#ifndef THREAD_EVENT_H_
#define THREAD_EVENT_H_
// #include <standard C library headers>
#include <time.h>
// #include <standard C++ library headers>
// #include <other library headers>
#ifdef _WIN32
#include <Windows.h>
#else
#include <pthread.h>
#include <sys/time.h>
#endif
// #include "custom headers"
namespace thread
{
class Event
{
public:
Event();
/*virtual*/ ~Event();
private:
Event(const Event& rhs);
Event& operator=(const Event& rhs);
public:
void NotifyAll();
void Wait();
bool Wait(int ms);
protected:
private:
#ifdef _WIN32
HANDLE event;
#else
pthread_cond_t cond;
pthread_mutex_t mutex;
#endif
};
#ifdef _WIN32
inline Event::Event()
{
event = ::CreateEvent(NULL, TRUE, FALSE, NULL);
}
inline Event::~Event()
{
::CloseHandle(event);
}
inline void Event::NotifyAll()
{
::PulseEvent(event);
}
inline void Event::Wait()
{
::WaitForSingleObject(event, INFINITE);
}
inline bool Event::Wait(int ms)
{
return (::WaitForSingleObject(event, ms) == WAIT_OBJECT_0);
}
#else
inline Event::Event()
{
pthread_cond_init(&cond, NULL);
pthread_mutex_init(&mutex, NULL);
}
inline Event::~Event()
{
pthread_cond_destroy(&cond);
pthread_mutex_destroy(&mutex);
}
inline void Event::NotifyAll()
{
pthread_cond_broadcast(&cond);
}
inline void Event::Wait()
{
pthread_mutex_lock(&mutex);
pthread_cond_wait(&cond, &mutex);
pthread_mutex_unlock(&mutex);
}
inline bool Event::Wait(int ms)
{
struct timespec ts;
struct timeval tv;
gettimeofday(&tv, NULL);
ts.tv_sec = tv.tv_sec + (ms / 1000);
ts.tv_nsec = tv.tv_usec * 1000 + ((ms % 1000) * 1000 * 1000);
// clock_gettime(CLOCK_REALTIME, &ts);
// ts.tv_sec += (ms / 1000);
// ts.tv_nsec += ((ms % 1000) * 1000 * 1000);
pthread_mutex_lock(&mutex);
bool ret = (pthread_cond_timedwait(&cond, &mutex, &ts) == 0);
pthread_mutex_unlock(&mutex);
return ret;
}
#endif // _WIN32
} // namespace thread
#endif // EVENT_H_
线程类封装(4)
最新推荐文章于 2012-03-26 10:07:27 发布