一个多线程的控制问题请教

protossYSZ

小弟最近初学pthread，现在有一个场景，
在一个4核CPU，每个核上启动2个线程，线程和核的映射有相应函数处理，
假设线程是A1A2B1B2C1C2D1D2，A1输出一个A，并且向一个缓冲区里写一个字母A，B1C1D1输出BCD，A2
从缓冲区内消耗一个字母A，并输出E,B2C2D2输出FGH字母，
逻辑是A1B1C1D1一起执行，容器满了换A2B2C2D2执行，空了再切换
单核上线程切换用条件变量会做，请问如何根据缓冲区是否为空，来控制多个线程的切换呢？求各位指导

cokeboL

目测信号量比较合适，随便signal什么的都能搞啊

cokeboL

以A1 A2为例

1. #include <pthread.h>
   
2. #include <semaphore.h>
   
3. #include <sys/types.h>
   
4. #include <stdio.h>
   
5. #include <unistd.h>
   
6. 
   
7. #define BUF_LEN_A 5
   
8. 
   
9. sem_t sem_id_A1;
   
10. sem_t sem_id_A2;
    
11. int buf_len_A = 0;
    
12. char buf_A[BUF_LEN_A] = {0};
    
13. 
    
14. void* thread_A1(void *arg)
    
15. {
    
16.         while(1)
    
17.         {
    
18.                 //example
    
19.                 printf("\n");
    
20.                 for(int i=0; i < BUF_LEN_A; i++)
    
21.                 {
    
22.                         sleep(1);
    
23.                         buf_A[buf_len_A] = 'A';
    
24.                         printf("thread_A1 set %d: %c\n", i, buf_A[i]);
    
25.                         buf_len_A++;
    
26.                 }
    
27.                
    
28.                 if(BUF_LEN_A == buf_len_A)
    
29.                 {
    
30.                         buf_len_A--;
    
31.                         sem_post(&sem_id_A2);
    
32.                         sem_wait(&sem_id_A1);
    
33.                 }
    
34.         }
    
35. }
    
36. 
    
37. void* thread_A2(void *arg)
    
38. {
    
39.         while(1)
    
40.         {
    
41.                 sem_wait(&sem_id_A2);
    
42. 
    
43.                 //example
    
44.                 printf("\n");
    
45.                 for(int i=buf_len_A; i >= 0; i--)
    
46.                 {
    
47.                         printf("thread_A2 get %d: %c\n", i, buf_A[i]);
    
48.                         buf_len_A--;
    
49.                         sleep(1);
    
50.                 }
    
51. 
    
52.                 if(buf_len_A < 0)
    
53.                 {
    
54.                         buf_len_A++;
    
55.                         sem_post(&sem_id_A1);
    
56.                 }
    
57.         }
    
58. }
    
59. 
    
60. int main(int argc, char *argv[])
    
61. {
    
62.         pthread_t tid_A1, tid_A2;
    
63.         sem_init(&sem_id_A1, 0, 0);
    
64.         sem_init(&sem_id_A2, 0, 0);
    
65.         pthread_create(&tid_A1,NULL,thread_A1, NULL);
    
66.         pthread_create(&tid_A2,NULL,thread_A2, NULL);
    
67. 
    
68.         getchar();
    
69. 
    
70.         return 0;
    
71. }
    
72. 
    

复制代码

hanxin83

骚年, 我想你需要的不是多线程, 而是"协程".....

cokeboL

回复 4# hanxin83


    一个循环里套两个循环就行，协程都不需要了。。但是不知道楼主具体需求是啥

lxyscls

回复 1# protossYSZ


    挺奇怪的需求，只要buffer里面有数据，A2B2...这种就可以取数据，为什么非要等到buffer满？
    你这种模型，可以算是pipeline，无非是一级pipeline有多个线程，并且互为pipeline

    A1B1...

1.     pthread_mutex_lock(mutex)
   
2.     while (buf == FULL)
   
3.         pthread_cond_wait(not_full_cond, mutex)
   
4. 
   
5.     insert && ++i1
   
6. 
   
7.     if (i1 == FULL) {
   
8.         buf += i1
   
9.         i1 = 0
   
10.         pthread_cond_broadcast(full_cond)
    
11.     }
    
12.     pthread_mutex_unlock(mutex)

复制代码

A2B2...

1.     pthread_mutex_lock(mutex)
   
2.     while (buf != FULL)
   
3.         pthread_cond_wait(full_cond, mutex)
   
4.    
   
5.     consume && ++i2
   
6. 
   
7.     if (i2 == FULL) {
   
8.         buf -=i2
   
9.         i2 = 0
   
10.         pthread_cond_broadcast(not_full_cond)
    
11.     }
    
12.     pthread_mutex_unlock(mutex)

复制代码

lxyscls

回复 1# protossYSZ


   

protossYSZ

回复 3# cokeboL


    谢谢~我详细看一下，刚开始学

protossYSZ

回复 5# cokeboL


    我开始想用条件变量，但是没有看到过多个线程成一组用条件变量控制组内线程切换的例子，就奇怪是不是需要多个条件变量来控制

protossYSZ

回复 6# lxyscls


    请问如果4个核，那么需要4个锁和4个条件变量嘛，开始没想出来的地方是，线程A1A2根据缓冲区为空用条件变量切换，可是B1C1D1如何跟随A1A2进行切换