Linux多線程──讀者寫者問題

讀者寫者問題

這也是一個非常經典的多線程題目，題目大意如下：有一個寫者很多讀者，多個讀者可以同時讀文件，但寫者在寫文件時不允許有讀者在讀文件，同樣有讀者讀時寫者也不能寫。

程序：

1. // reader_writer.cpp
2. //////////////////////////////////////////////////////////////////////
3. // 讀者寫者問題
4. // 有一個寫者很多讀者，多個讀者可以同時讀文件，但寫者在寫文件時不允許有讀者在讀文件，
5. // 同樣有讀者讀時寫者也不能寫。
6. //////////////////////////////////////////////////////////////////////
8. #include <pthread.h>
9. #include <stdio.h>
10. #include <unistd.h>
12. // 定義數據類
13. class data
14. {
15. public:
16. data(int i, float f):
17. I(i), F(f)
18. {}
20. int I;
21. float F;
22. };
24. // 讀者寫者讀寫的內容
25. data *p_data = NULL;
27. pthread_rwlock_t lock;
29. // 寫者數目
30. const int WRITER_NUMBER = 2;
32. void *reader(void *arg);
33. void *writer(void *arg);
35. int main(int argc, char **argv)
36. {
37. pthread_t reader_tid;
38. pthread_t writer_tid[WRITER_NUMBER];
40. pthread_create(&reader_tid, NULL, reader, NULL);
41. for (int i = 0; i < WRITER_NUMBER; ++i)
42. {
43. pthread_create(&writer_tid[i], NULL, writer, (void *)i);
44. }
46. sleep(1);
48. return 0;
49. }
51. void *reader(void *arg)
52. {
53. int id = (int)arg;
55. pthread_detach(pthread_self());
57. while (true)
58. {
59. pthread_rwlock_rdlock(&lock);
60. printf("reader %d is reading the data; ", id);
61. if (p_data == NULL)
62. {
63. printf("the data is NULL\n");
64. }
65. else
66. {
67. printf("the data is (%d, %f)\n", p_data->I, p_data->F);
68. }
69. pthread_rwlock_unlock(&lock);
70. }
71. return (void *)0;
72. }
74. void *writer(void *arg)
75. {
76. pthread_detach(pthread_self());
78. while (true)
79. {
80. pthread_rwlock_wrlock(&lock);
81. printf("writer is writing the data; ");
82. if (p_data == NULL)
83. {
84. p_data = new data(1, 1.1f);
85. printf("writer create the data (%d, %f)\n", p_data->I, p_data->F);
86. }
87. else
88. {
89. delete p_data;
90. p_data = NULL;
91. printf("writer free the data\n");
92. }
93. pthread_rwlock_unlock(&lock);
94. }
95. return (void *)0;
96. }