关于Linux 系统下 C++ 的多线程基类 Thread

最近在Linux 下用C++做的东西，原来是使用“多进程”方式实现，现在随着工程的增大，“多进程”的“变量共享”问题已经十分突出了，虽然可以使“内存共享”等方式实现，但大量的“内存共享”会导致代码混乱。。

所以决心将整个工程重新写，使用“多线程”方式实现，这样可以大大较少“内存共享”的使用次数。

下面是我写的基类，把代码保存在名为 Thread.h 的头文件中。

====================================传说中的分割线=================================

/*
* File:   Thread.h
* Author: Null
* Blog: http://hi.baidu.com/hetaoos * Created on 2008年7月30日, 上午10:13
*/

/*
* 在编译的时候记得加上参数：-lpthread
*
*/

#ifndef _THREAD_H
#define _THREAD_H

#include <pthread.h>
#include <unistd.h>

class Thread
{
private:
//当前线程的线程ID
pthread_t tid;
//线程的状态
int threadStatus;
//获取执行方法的指针
static void* run0(void* pVoid);
//内部执行方法
void* run1();
public:
//线程的状态－新建
static const int THREAD_STATUS_NEW = 0;
//线程的状态－正在运行
static const int THREAD_STATUS_RUNNING = 1;
//线程的状态－运行结束
static const int THREAD_STATUS_EXIT = -1;
//构造函数
Thread();
//线程的运行实体
virtual void run() = 0;
//开始执行线程
bool start();
//获取线程ID
pthread_t getThreadID();
//获取线程状态
int getState();
//等待线程直至退出
void join();
//等待线程退出或者超时
void join(unsigned long millisTime);
};

void* Thread::run0(void* pVoid)
{
Thread* p = (Thread*) pVoid;
p->run1();
return p;
}

void* Thread::run1()
{

threadStatus = THREAD_STATUS_RUNNING;
tid = pthread_self();
run();
threadStatus = THREAD_STATUS_EXIT;
tid = 0;
pthread_exit(NULL);
}

Thread::Thread()
{
tid = 0;
threadStatus = THREAD_STATUS_NEW;
}

bool Thread::start()
{
return pthread_create(&tid, NULL, run0, this) == 0;
}

pthread_t Thread::getThreadID()
{
return tid;
}

int Thread::getState()
{
return threadStatus;
}

void Thread::join()
{
if (tid > 0)
{
pthread_join(tid, NULL);
}
}

void Thread::join(unsigned long millisTime)
{

if (tid == 0)
{
return;
}
if (millisTime == 0)
{
join();
}else
{
unsigned long k = 0;
while (threadStatus != THREAD_STATUS_EXIT && k <= millisTime)
{
usleep(100);
k++;
}
}
}
#endif /* _THREAD_H */

====================================传说中的分割线=================================[/code]
用法简单类似于 Java 的 Thread，继承 Thread 类，然后重写 void run() 方法，然后用 bool start () 方法开始运行。这样完全屏蔽掉了线程的具体操作。使用简单方便。

具体看下面的示例代码。

====================================传说中的分割线=================================

/*
* File:   newmain.cc
* Author: Null
* Blog: http://hi.baidu.com/hetaoos * Created on 2008年7月30日, 上午11:49
*/

#include "Thread.h"
#include <iostream.h>

class MultiThread : public Thread
{
public:

void run()
{
int number = 0;
for (int i = 0; i < 10; i++)
{
cout << "Current number is " << number++;
cout << " PID is " << getpid() << " TID is " << getThreadID() << endl;
sleep(1);
}
}
};

int main(int argc, char** argv)
{
bool ret;
MultiThread *mt;
mt = new MultiThread();
ret = mt->start();
mt->join(6000);
return (EXIT_SUCCESS);
}

====================================传说中的分割线=================================

上面的代码，由于 join 设置超时为 6 秒，所以没能完全打印出10条记录就已经退出了。

上面的代码还有些不足：

1，���有线程属性的设置，这个当初也考虑过，但是有些麻烦，也很少用。如果哪位大侠实现了比较完成的 Thread 类，麻烦发个给我。

2，线程的控制有待完善，这个阿，暂时没有时间研究。



在原来的基础上，进行了加强，使之进一步接近 Java 中 Thread 和 Runnable 的用法。

下面是我写的基类，把代码保存在名为 Thread.h 的头文件中。

====================================传说中的分割线=================================

/*
* File:   Thread.h
* Author: Null
* Blog: http://hi.baidu.com/hetaoos * Created on 2008年7月30日, 上午10:13
*/

/*
* 在编译的时候记得加上参数：-lpthread
*
*/

#ifndef _THREAD_H
#define _THREAD_H

#include <pthread.h>
#include <unistd.h>

/*
* 线程运行实体类
*/
class Runnable
{
public:
//运行实体
virtual void run() = 0;
};

/*
* 线程类
*
*/
class Thread : public Runnable
{
private:
//线程初始化序号
static int threadInitNumber;
//当前线程初始化序号
int curThreadInitNumber;
//线程体
Runnable *target;
//当前线程的线程ID
pthread_t tid;
//线程的状态
int threadStatus;
//线程属性
pthread_attr_t attr;
//线程优先级
sched_param param;
//获取执行方法的指针
static void* run0(void* pVoid);
//内部执行方法
void* run1();
//获取一个线程序号
static int getNextThreadNum();

public:
//线程的状态－新建
static const int THREAD_STATUS_NEW = 0;
//线程的状态－正在运行
static const int THREAD_STATUS_RUNNING = 1;
//线程的状态－运行结束
static const int THREAD_STATUS_EXIT = -1;
//构造函数
Thread();
//构造函数
Thread(Runnable *iTarget);
//析构
~Thread();
//线程的运行实体
void run();
//开始执行线程
bool start();
//获取线程状态
int getState();
//等待线程直至退出
void join();
//等待线程退出或者超时
void join(unsigned long millisTime);
//比较两个线程时候相同,通过 curThreadInitNumber 判断
bool operator ==(const Thread *otherThread);
//获取This线程ID
pthread_t getThreadID();
//获取当前线程ID
static pthread_t getCurrentThreadID();
//当前线程是否和某个线程相等,通过 tid 判断
static bool isEquals(Thread *iTarget);
//设置线程的类型:绑定/非绑定
void setThreadScope(bool isSystem);
//获取线程的类型:绑定/非绑定
bool getThreadScope();
//设置线程的优先级,1-99,其中99为实时;意外的为普通
void setThreadPriority(int priority);
//获取线程的优先级
int getThreadPriority();
};

int Thread::threadInitNumber = 1;

inline int Thread::getNextThreadNum()
{
return threadInitNumber++;
}

void* Thread::run0(void* pVoid)
{
Thread* p = (Thread*) pVoid;
p->run1();
return p;
}

void* Thread::run1()
{

threadStatus = THREAD_STATUS_RUNNING;
tid = pthread_self();
run();
threadStatus = THREAD_STATUS_EXIT;
tid = 0;
pthread_exit(NULL);
}

void Thread::run()
{
if (target != NULL)
{
(*target).run();
}
}

Thread::Thread()
{
tid = 0;
threadStatus = THREAD_STATUS_NEW;
curThreadInitNumber = getNextThreadNum();
pthread_attr_init(&attr);
}

Thread::Thread(Runnable *iTarget)
{
target = iTarget;
tid = 0;
threadStatus = THREAD_STATUS_NEW;
curThreadInitNumber = getNextThreadNum();
pthread_attr_init(&attr);
}

Thread::~Thread()
{
pthread_attr_destroy(&attr);
}

bool Thread::start()
{
return pthread_create(&tid, &attr, run0, this) == 0;
}

inline pthread_t Thread::getCurrentThreadID()
{
return pthread_self();
}

inline pthread_t Thread::getThreadID()
{
return tid;
}

inline int Thread::getState()
{
return threadStatus;
}

void Thread::join()
{
if (tid > 0)
{
pthread_join(tid, NULL);
}
}

void Thread::join(unsigned long millisTime)
{

if (tid == 0)
{
return;
}
if (millisTime == 0)
{
join();
}else
{
unsigned long k = 0;
while (threadStatus != THREAD_STATUS_EXIT && k <= millisTime)
{
usleep(100);
k++;
}
}
}

bool Thread::operator ==(const Thread *otherThread)
{
if (otherThread == NULL)
{
return false;
}
if (curThreadInitNumber == (*otherThread).curThreadInitNumber)
{
return true;
}
return false;
}

bool Thread::isEquals(Thread *iTarget)
{
if (iTarget == NULL)
{
return false;
}
return pthread_self() == iTarget->tid;
}

void Thread::setThreadScope(bool isSystem)
{
if (isSystem)
{
pthread_attr_setscope(&attr, PTHREAD_SCOPE_SYSTEM);
}else
{
pthread_attr_setscope(&attr, PTHREAD_SCOPE_PROCESS);
}
}

bool Thread::getThreadScope()
{
int scopeType = 0;
pthread_attr_getscope(&attr, &scopeType);
return scopeType == PTHREAD_SCOPE_SYSTEM;
}

void Thread::setThreadPriority(int priority)
{
pthread_attr_getschedparam(&attr, ¶m);
param.__sched_priority = priority;
pthread_attr_setschedparam(&attr, ¶m);
}

int Thread::getThreadPriority()
{
pthread_attr_getschedparam(&attr, ¶m);
return param.__sched_priority;
}
#endif /* _THREAD_H */

====================================传说中的分割线=================================

具体看下面的示例代码。

呵呵，用法和 Java 的 Thread 很相似吧。

下面一共生成了3个线程，3个线程都共享 MultiThread::run()

在 run() 里面通过判断线程的实体来分离各个线程的工作。

====================================传说中的分割线=================================

/*
* File:   newmain.cc
* Author: Null
* Blog: http://hi.baidu.com/hetaoos * Created on 2008年7月30日, 上午12:49
*/

#include "Thread.h"
#include <iostream.h>

class MultiThread : public Thread
{
public:

Thread *th1;
Thread *th2;

void Test()
{
th1 = new Thread(this);
th1->setThreadPriority(99);
th2 = new Thread(this);
start();
th1->start();
th2->start();
th1->join();
th2->join();
}

void run()
{
//Thread->isEquals(th1)
if (Thread::isEquals(th1))
{
int number = 100;
for (int i = 0; i < 10; i++)
{
cout << "this is thread1 number is " << number++;
cout << " \tpid is " << getpid() << " tid is " << getCurrentThreadID() << "  Priority:" << th1->getThreadPriority() << endl;
sleep(1);
}
}else if (Thread::isEquals(th2))
{
int number = 200;
for (int i = 0; i < 10; i++)
{
cout << "this is thread2 number is " << number++;
cout << " \tpid is " << getpid() << " tid is " << getCurrentThreadID() << "  Priority:" << th2->getThreadPriority() << endl;
sleep(1);
}
}else if (Thread::isEquals(this))
{
int number = 300;
for (int i = 0; i < 10; i++)
{
cout << "this is thread0 number is " << number++;
cout << " \tpid is " << getpid() << " tid is " << getCurrentThreadID() << "  Priority:" << this->getThreadPriority() << endl;
sleep(1);
}
}

}
};

/*
*
*/
int main(int argc, char** argv)
{
bool ret;
MultiThread *mt;
mt = new MultiThread();
mt->Test();
return (EXIT_SUCCESS);
}

====================================传说中的分割线=================================

PS：

1，还没有进行深入的测试，使用之前请进行必要的测试，以确保它符合你的要求。

2，如果你发现了 BUG，请联系我。