帮我用C语言编写两个线程,然后两个线程间的通信用消息队列
时间: 2024-02-16 13:03:28 浏览: 72
好的,下面是一个简单的例子,其中包含两个线程,一个生产者线程和一个消费者线程,它们之间通过消息队列进行通信。在此之前,需要先安装 `librt` 库,它提供了 POSIX 消息队列的支持。
```c
#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <mqueue.h>
#include <string.h>
#include <errno.h>
#define MAX_MSG_SIZE 1024
#define MSG_QUEUE_NAME "/test_msg_queue"
void* producer_thread(void* arg);
void* consumer_thread(void* arg);
int main() {
mqd_t mq;
pthread_t producer_tid, consumer_tid;
int ret;
/* 定义消息队列属性 */
struct mq_attr attr = {
.mq_flags = 0, /* 未阻塞 */
.mq_maxmsg = 10, /* 最多10个消息 */
.mq_msgsize = MAX_MSG_SIZE, /* 每个消息大小不超过1024字节 */
.mq_curmsgs = 0 /* 当前消息个数为0 */
};
/* 创建消息队列 */
mq = mq_open(MSG_QUEUE_NAME, O_CREAT | O_RDWR, 0666, &attr);
if (mq == (mqd_t)-1) {
perror("mq_open failed");
exit(EXIT_FAILURE);
}
/* 创建生产者线程 */
ret = pthread_create(&producer_tid, NULL, producer_thread, &mq);
if (ret != 0) {
perror("pthread_create failed");
exit(EXIT_FAILURE);
}
/* 创建消费者线程 */
ret = pthread_create(&consumer_tid, NULL, consumer_thread, &mq);
if (ret != 0) {
perror("pthread_create failed");
exit(EXIT_FAILURE);
}
/* 等待线程结束 */
pthread_join(producer_tid, NULL);
pthread_join(consumer_tid, NULL);
/* 删除消息队列 */
mq_close(mq);
mq_unlink(MSG_QUEUE_NAME);
return 0;
}
void* producer_thread(void* arg) {
mqd_t mq = *(mqd_t*)arg;
char buffer[MAX_MSG_SIZE];
int i, ret;
for (i = 0; i < 10; ++i) {
snprintf(buffer, MAX_MSG_SIZE, "Message %d from producer.", i);
/* 发送消息到消息队列 */
ret = mq_send(mq, buffer, strlen(buffer) + 1, 0);
if (ret == -1) {
perror("mq_send failed");
exit(EXIT_FAILURE);
}
printf("Producer sent: %s\n", buffer);
sleep(1);
}
/* 发送结束标志 */
ret = mq_send(mq, "exit", strlen("exit") + 1, 0);
if (ret == -1) {
perror("mq_send failed");
exit(EXIT_FAILURE);
}
pthread_exit(NULL);
}
void* consumer_thread(void* arg) {
mqd_t mq = *(mqd_t*)arg;
char buffer[MAX_MSG_SIZE];
int ret;
while (1) {
/* 从消息队列接收消息 */
ret = mq_receive(mq, buffer, MAX_MSG_SIZE, NULL);
if (ret == -1) {
if (errno == EINTR) {
continue;
} else {
perror("mq_receive failed");
exit(EXIT_FAILURE);
}
}
printf("Consumer received: %s\n", buffer);
/* 判断是否接收到了结束标志 */
if (strcmp(buffer, "exit") == 0) {
break;
}
}
pthread_exit(NULL);
}
```
在上面的代码中,我们先创建了一个消息队列,然后创建了一个生产者线程和一个消费者线程。生产者线程每隔1秒向消息队列中发送一条消息,消费者线程不断从消息队列中接收消息并输出。当生产者线程发送完10条消息后,它会向消息队列中发送一个特殊的消息 "exit",表示生产者线程已经结束。消费者线程会接收到这个消息后退出。
需要注意的是,在实际使用中,要根据具体需求对消息队列的属性进行调整,并对消息的格式和内容进行设计。
阅读全文