队列、信号量、多线程同步综合使用

被封禁言

1. #include <stdio.h>
   
2. #include <unistd.h>
   
3. #include <stdlib.h>
   
4. #include <string.h>
   
5. #include <pthread.h>
   
6. #include <semaphore.h>
   
7. #include "queue.h"
   
8. 
   
9. void *thread_function(void *arg);
   
10. 
    
11. #define WORK_SIZE 1024
    
12. #define THREAD_NUM 10
    
13. 
    
14. //互斥量work_mutex 将time_to_exit和strQueue管理在其中
    
15. pthread_mutex_t work_mutex;
    
16. queue_t* strQueue;
    
17. int time_to_exit = 0;
    
18. 
    
19. //信号量的作用是不通过轮询让多线程阻塞或者取消阻塞
    
20. sem_t bin_sem;
    
21. 
    
22. int main()
    
23. {
    
24.         char *local_work_area;
    
25.         int i, res;
    
26.         int thread_id[THREAD_NUM];
    
27.         pthread_t thread[THREAD_NUM];
    
28.         void *thread_result;
    
29. 
    
30.         strQueue = init_queue();
    
31.         res = pthread_mutex_init(&work_mutex, NULL);
    
32.         if (res != 0)
    
33.         {
    
34.                 perror("Mutex initialization failed");
    
35.                 exit(EXIT_FAILURE);
    
36.         }
    
37.         res = sem_init(&bin_sem, 0, 0);
    
38.         if (res != 0)
    
39.         {
    
40.                 perror("Semaphore initialization failed");
    
41.                 exit(EXIT_FAILURE);
    
42.         }
    
43.         for (i = 0; i < THREAD_NUM; i++)
    
44.         {
    
45.                 thread_id[i] = i;
    
46.                 res = pthread_create(&thread[i], NULL, thread_function, &thread_id[i]);
    
47.                 if (res != 0)
    
48.                 {
    
49.                         perror("Thread creation failed");
    
50.                         exit(EXIT_FAILURE);
    
51.                 }
    
52.         }
    
53. 
    
54.         printf("Input some text. Enter 'end' to finish\n");
    
55.         while (1)
    
56.         {
    
57.                 local_work_area = (char*) malloc(WORK_SIZE);
    
58.                 printf(">");
    
59.                 fgets(local_work_area, WORK_SIZE, stdin); //gets(local_work_area);
    
60.                 if (strcmp(local_work_area, "end\n") == 0)
    
61.                 {
    
62.                         printf("program ready to exit.\n");
    
63.                         pthread_mutex_lock(&work_mutex);
    
64.                         time_to_exit = 1;
    
65.                         pthread_mutex_unlock(&work_mutex);
    
66.                         break;
    
67.                 }
    
68.                 else if (strcmp(local_work_area, "\n") == 0)
    
69.                         free(local_work_area);
    
70.                 else
    
71.                 {
    
72.                         //begin input queue
    
73.                         pthread_mutex_lock(&work_mutex);
    
74.                         //将字符串的地址 加入到队列中
    
75.                         enqueue(strQueue, local_work_area);
    
76.                         pthread_mutex_unlock(&work_mutex);
    
77. 
    
78.                         //通知其他线程，有字符串加入到队列中了
    
79.                         sem_post(&bin_sem);
    
80.                 }
    
81.         }
    
82. 
    
83.         printf("\nWaiting for thread to finish...\n");
    
84.         for (i = 0; i < THREAD_NUM; i++)
    
85.         {
    
86.                 sem_post(&bin_sem); //激活正在阻塞的线程
    
87.         }
    
88. 
    
89.         //上,下两段for语句，不能组合在一个for中。
    
90.         //因为thread[i]完全有可能不把收到sem_post()的信号，就会一无限等待
    
91.         for (i = 0; i < THREAD_NUM; i++)
    
92.         {
    
93.                 res = pthread_join(thread[i], &thread_result);
    
94.                 if (res != 0)
    
95.                 {
    
96.                         perror("Thread join failed");
    
97.                         exit(EXIT_FAILURE);
    
98.                 }
    
99.         }
    
100.         pthread_mutex_destroy(&work_mutex);
     
101.         sem_destroy(&bin_sem);
     
102.         exit(EXIT_SUCCESS);
     
103. }
     
104. 
     
105. /**
     
106. * src转换为大写到dst
     
107. */
     
108. void CopyAndUpper(char *dst, char *src)
     
109. {
     
110.         while (((*src) != '\n') && ((*src) != '\0'))
     
111.         {
     
112.                 *dst = toupper(*src);
     
113.                 src++;
     
114.                 dst++;
     
115.         }
     
116.         *dst = '\0';
     
117. }
     
118. 
     
119. void *thread_function(void *arg)
     
120. {
     
121.         char strlocal[WORK_SIZE], strtemp[WORK_SIZE];
     
122.         int thread_id = *(int*) (arg);
     
123.         char *p;
     
124.         while (1)
     
125.         {
     
126.                 //排队
     
127.                 sem_wait(&bin_sem);
     
128.                 pthread_mutex_lock(&work_mutex);
     
129.                 if (time_to_exit == 1)
     
130.                 {
     
131.                         pthread_mutex_unlock(&work_mutex);
     
132.                         break;
     
133.                 }
     
134.                 //出队，获取到地址后，strcpy拷贝到strtemp
     
135.                 dequeue(strQueue, &p);
     
136.                 strcpy(strtemp, p);
     
137.                 pthread_mutex_unlock(&work_mutex);        //互斥量时间越少越好
     
138. 
     
139.                 CopyAndUpper(strlocal, strtemp);
     
140.                 if (strlen(strlocal) > 3)
     
141.                         sleep(strlen(strlocal));        //模拟长时间运算，每字符一秒
     
142.                 printf("thread-%d: %s len=%d\n", thread_id, strlocal, strlen(strlocal));
     
143.                 free(p);
     
144.         }
     
145.         pthread_exit(0);
     
146. }

复制代码

yuzhong

111111111111111111111111111111111111111111111

nangu66

qtcpl

有e源码就好了

wxc6688

好高深啊  看不懂啊

fml224

沙发终结者，