多线程环境下安全的消息队列存取---利用信号量

来源:互联网 发布:快照优化 编辑:程序博客网 时间:2024/06/06 21:34

       前面几篇博文中, 我们一直在讨论异质链表消息队列问题, 在本文中, 我们继续来探讨这一问题, 在多线程环境下, 考虑线程同步问题, 代码如下(我用Sleep(20);来故意捣乱):

#include <windows.h>#include <iostream>using namespace std;#define Rectangle MyRectangle  // 避免Rectangle与Windows中的Rectangle冲突HANDLE g_hSemp = NULL;  // 信号量handle// 对象的id值typedef enum{ErrorId = -1,IntegerId = 1,PointId = 2,RectangeId = 3,}ObjectID;// 基类struct Basic{ObjectID id;virtual Basic *copy() = 0; // 纯虚函数};// 整数类struct Integer : public Basic{int a;Basic *copy(){Integer *p = new Integer;p->a = ((Integer*)this)->a;p->id = ((Integer*)this)->id;return p;}};// 点类struct Point : public Basic{int x;int y;Basic *copy(){Point *p = new Point;p->x = ((Point*)this)->x;p->y = ((Point*)this)->y;p->id = ((Point*)this)->id;return p;}};// 矩形类struct Rectangle : public Basic{Point point;int width;int height;Basic *copy(){Rectangle *p = new Rectangle;p->point.x = ((Rectangle*)this)->point.x;p->point.y = ((Rectangle*)this)->point.y;p->width = ((Rectangle*)this)->width;p->height = ((Rectangle*)this)->height;p->id = ((Rectangle*)this)->id;return p;}};// 抽象对象的共同点, 构造成新的结点, 便于链接typedef struct node{node *next;Basic *pBasic;}Node;Node *head = NULL;  // 指向第一结点(采用不带头结点的链表)// 往链式消息队列中塞消息Node *addToMsgQueue(Basic* pb){WaitForSingleObject(g_hSemp, INFINITE);printf("add\n");Node *pn = new Node;Node *qn = NULL;Basic *p = pb->copy(); // 多态性if(NULL == head){Sleep(20);  head = pn;}else{qn = head;while(NULL != qn->next){Sleep(20);  qn = qn->next;}Sleep(20);  qn->next = pn;}pn->pBasic = p;   // 千万别忘记啊pn->next = NULL;  // 千万别忘记啊ReleaseSemaphore(g_hSemp, 1, NULL);  return head;}// 从链式消息队列中取出消息(结点)Node *getMsgFromQueue(){WaitForSingleObject(g_hSemp, INFINITE);  printf("get\n");if(NULL == head){Sleep(20);  ReleaseSemaphore(g_hSemp, 1, NULL);  return NULL;}Node *pn = head;head = head->next;Sleep(20);  ReleaseSemaphore(g_hSemp, 1, NULL);  return pn;} // 线程函数DWORD WINAPI ThreadFun(LPVOID pM){Node *p = NULL;// 从消息队列中取出消息while(1){p = getMsgFromQueue();if(NULL == p){Sleep(100);continue;}// 对指针进行还原switch(p->pBasic->id) {case IntegerId:{cout << ((Integer*)(p->pBasic))->a << endl;break;}case PointId:{cout << ((Point *)(p->pBasic))->x << endl;cout << ((Point *)(p->pBasic))->y << endl;break;}case RectangeId:{cout << ((Rectangle *)(p->pBasic))->point.x << endl;cout << ((Rectangle *)(p->pBasic))->point.y << endl;cout << ((Rectangle *)(p->pBasic))->width << endl;cout << ((Rectangle *)(p->pBasic))->height << endl;break;}default:{break;}}}return 0;}// 主线程int main(){HANDLE handle = CreateThread(NULL, 0, ThreadFun, NULL, 0, NULL);    CloseHandle(handle);g_hSemp = CreateSemaphore(NULL, 1, 10, NULL);  // 定义三个对象并赋值Integer i;Point po;Rectangle rect;i.id = IntegerId;po.id = PointId;rect.id = RectangeId;i.a = 11;po.x = 22;po.y = 33;rect.point.x = 44;rect.point.y = 55;rect.width = 66;rect.height = 77;// 塞入消息队列while(1){addToMsgQueue(&i);addToMsgQueue(&po);addToMsgQueue(&rect);Sleep(2000);}CloseHandle(g_hSemp);return 0;}
       测试过十几次, 没发现程序有什么问题。 如果大家有不同意见, 可以提出。 

       貌似该睡觉了啊。


0 0
原创粉丝点击