消息队列模板类.docVIP

消息队列模板类 template class TData class TMessageQueue { public: vectorTData queue; // 存储消息的队列 pthread_mutex_t mutex; sem_t sem; public: TMessageQueue() { pthread_mutex_init( mutex, NULL); sem_init(sem, 0, 0); } ~TMessageQueue() { pthread_mutex_destroy(mutex); sem_destroy(sem); FreeQueue(); } /*将记录插入消息队列的尾部，正常情况下，放在队列尾部，从头部取*/ void AddElement(TData pData) { Lock(); queue.push_back(pData); SetQueueEvent(); Unlock(); } /* 将记录放在队列的头部， 当外部请求取消、断开、异常退出时， 接入层将消息放入队列的头部，分发时，将还没有处理的相应消 息从队列中删除，将取消的消息放入业务线程的接收队列，业务 线程进行释放处理 */ void AddElementAtFront(TData pData) { Lock(); queue.insert(queue.begin());//push_back(pData); SetQueueEvent(); Unlock(); } int count() { return queue.size(); } /*从队列头部取一条记录*/ TData GetElement() { WaitQueueEvent(); TData p;// = NULL; //vectorTData::iterator it; memset(p, 0, sizeof(TData)); Lock(); if (false == queue.empty()) { p = *queue.begin(); queue.erase(queue.begin()); } Unlock(); return p; } TData GetElement(int dd) { if (dd == 1) WaitQueueEvent(); TData p;// = NULL; //vectorTData::iterator it; memset(p, 0, sizeof(TData)); Lock(); if (false == queue.empty()) { p = *queue.begin(); queue.erase(queue.begin()); } Unlock(); return p; } private: /* 操作之前加锁处理 */ void Lock() { pthread_mutex_lock( mutex); } /* 操作之后解锁处理 */ void Unlock() { pthread_mutex_unlock( mutex); } /* 设置事件，触发线程中的处理 */ void SetQueueEvent() { sem_post(sem); }