thread.h

一套linux下的C++开发库

/***************************************************************************
                          thread.h  -  description
                             -------------------
    begin                : Fri Jul 20 2001
    copyright            : (C) 2001 by Mark
    email                : [e-mail]
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/


#ifndef THREAD_H
#define THREAD_H

#include <pthread.h>
#include <semaphore.h>

#include <tinycxx/exception.h>


class CThCond;
class CThMutex
{
	friend class CThCond;

protected:
	pthread_mutex_t m_tMutex;

public:
	CThMutex();
	virtual ~CThMutex();

	bool Lock();
	bool TryLock();
	bool Unlock();
};


class CThSem
{
protected:
	sem_t m_tSem;
	
public:
	CThSem(int iValue = 0);
	virtual ~CThSem();
	
	bool Wait();
	bool TryWait();
	bool Post();
	int GetValue();
};


class CThCond
{
protected:
	pthread_cond_t m_tCond;
	
public:
	CThCond();
	virtual ~CThCond();
	
	bool Signal();
	bool Broadcast();
	bool Wait(CThMutex& stThMutex);
	bool TimedWait(CThMutex& stThMutex, const struct timespec* pstAbsTime);
};


class CThRWLock
{
public:
	CThRWLock(bool bWriterFirst = false, int iMaxReader = 0);	//0 means no limit
	~CThRWLock();
	
	bool ReadLock();
	bool WriteLock();
	bool Unlock();
	
protected:
	const bool m_bWriterFirst;
	const int m_iMaxReader;
	
	CThMutex m_stMutex;
	CThCond m_stReaderCond;
	CThCond m_stWriterCond;
	int m_iWaitingReader;
	int m_iWaitingWriter;
	int m_iHoldingLock;	//-1 means writer holds the lock, >0 means reader count hloding the lock
};


class CThread
{
	friend void* ThreadRun(CThread* pstThread);

protected:
	pthread_attr_t m_tThreadAttr;
	pthread_t m_tThread;
	bool m_bDetach;
	bool m_bStart;

public:
	CThread(bool bDetach = true);
	virtual ~CThread();
	
	void Start();
	
protected:
	virtual void Run() = 0;
	virtual void OnEnd() {}	//perhaps you will delete THIS here.
	virtual void OnDelete();
	virtual void OnException(CException& e) {}
};

#endif