使用select实现消息队列超时

Zoelov

1. #include <stdio.h>
   
2. #include <string.h>
   
3. #include <stdlib.h>
   
4. #include <unistd.h>
   
5. #include <sys/ipc.h>
   
6. #include <sys/stat.h>
   
7. #include <sys/msg.h>
   
8. #include <errno.h>
   
9. 
   
10. #define BUFFER 255
    
11. #define PERM S_IRUSR | S_IWUSR
    
12. 
    
13. struct msgtype
    
14. {
    
15.     long mtype;
    
16.     char buffer[BUFFER+1];
    
17. };
    
18. 
    
19. void recv(int msgid)
    
20. {
    
21.     while(1)
    
22.     {
    
23.     int fd[2];
    
24.     pid_t pid;
    
25.     if(pipe(fd) < 0)
    
26.     {
    
27.         fprintf(stderr,"pipe error!\n");
    
28.         exit(-1);
    
29.     }
    
30. 
    
31.     if((pid = fork()) == 0)
    
32.     {
    
33.         close(fd[0]);
    
34.         int rc;
    
35.         struct msgtype msg;
    
36.         if((rc = msgrcv(msgid,&msg,sizeof(struct msgtype),0,0)) == -1)
    
37.         {
    
38.             fprintf(stderr,"msgrcv error!\n");
    
39.             exit(-1);
    
40.                            }
    
41. 
    
42. 
    
43.         rc = write(fd[1],msg.buffer,strlen(msg.buffer));
    
44.         if(rc < 0)
    
45.         {
    
46.             fprintf(stderr,"write error!\n");
    
47.             exit(-1);
    
48.         }
    
49.         exit(0);
    
50.     }
    
51.     else if(pid > 0)
    
52.     {
    
53.         char buf[256] = {0};
    
54.         fd_set rfds;
    
55.         struct timeval tv;
    
56.         int rc;
    
57.         FD_ZERO(&rfds);
    
58.         FD_SET(fd[0],&rfds);
    
59.         tv.tv_sec = 3;
    
60.         tv.tv_usec = 0;
    
61. 
    
62.         rc = select(fd[0]+1,&rfds,NULL,NULL,&tv);
    
63.         if(-1 == rc)
    
64.         {
    
65.             fprintf(stderr,"select error!\n");
    
66.             exit(-1);
    
67.         }
    
68.         else if(rc)
    
69.         {
    
70.             if(FD_ISSET(fd[0],&rfds))
    
71.             {
    
72.                 rc = read(fd[0],buf,256);
    
73.                 buf[rc] = 0;
    
74.                 if(rc == -1)
    
75.                 {
    
76.                     fprintf(stderr,"read error!\n");
    
77.                     exit(-1);
    
78.                 }
    
79.                       printf("I have receive a message : %s\n",buf);
    
80.             }
    
81.         }
    
82.         else
    
83.         {
    
84.             printf("no data timeout!\n");
    
85.             return ;
    
86.         }
    
87.     }
    
88.     }
    
89. 
    
90. }
    
91. 
    
92. 
    
93. int main(int argc, char **argv)
    
94. {
    
95.     if(argc != 2)
    
96.     {
    
97.         printf("usage ./test string\n");
    
98.         exit(-1);
    
99.     }
    
100. 
     
101.     int msgid = msgget(IPC_PRIVATE, PERM | IPC_CREAT | IPC_EXCL);
     
102.     if(msgid == -1)
     
103.     {
     
104.         fprintf(stderr,"msgget error!\n");
     
105.         exit(-1);
     
106.     }
     
107. 
     
108.     pid_t pid;
     
109.     int i;
     
110.     for(i = 0; i < 4; i++)
     
111.     {
     
112.         pid = fork();
     
113. if(pid == 0 || pid == -1)
     
114.             break;
     
115.     }
     
116. 
     
117.     if(pid == 0)
     
118.     {
     
119. 
     
120.         struct msgtype msg;
     
121.         msg.mtype = 1;
     
122.         strcpy(msg.buffer,argv[1]);
     
123. 
     
124.         int rc = msgsnd(msgid,&msg,sizeof(struct msgtype),0);
     
125.         if(rc == -1)
     
126.         {
     
127.             fprintf(stderr,"msgsnd error!\n");
     
128.             exit(-1);
     
129.         }
     
130.         else
     
131.         {
     
132.             printf("msgsnd success!\n");
     
133.         }
     
134.     }
     
135.     else if(pid > 0)
     
136.     {
     
137.         pthread_t tid,tid2,tid3,tid4,tid5;
     
138. 
     
139.             pthread_create(&tid,NULL,(void*)recv,(void*)msgid);
     
140.             pthread_join(tid,NULL);
     
141. wait(NULL);
     
142.     }
     
143.     else
     
144.     {
     
145.         fprintf(stderr,"fork error!\n");
     
146.         exit(-1);
     
147.     }
     
148. 
     
149.     return 0;
     
150. 
     
151. }           

复制代码

以上是我的代码，使用select与管道来实现消息队列超时退出。其中是在子进程中调用msgrcv函数，在这里可能会产生阻塞，现在有个问题，如果msgrcv阻塞了，父进程可以监控到超时，那么父进程中就退出了，那么子进程会不会退出呢？我不了解select是让谁退出了呢？

井蛙夏虫

回复 1# Zoelov
父子进程是独立的，父进程退出不会导致子进程退出，子进程退出也不会导致父进程退出


   

Zoelov

那是不是子进程一直阻塞，根本就无法退出？回复 2# 井蛙夏虫


   

井蛙夏虫

回复 3# Zoelov
应当是的


   

天赐之道

slecet（）一般就是用来避免使用多进程或者多进程的，select是通过轮询的方式来监听（暂且这么说吧），所以可以取代多进程，一般用于单进程中，进程结束时，

它也就自然结束了。再说，你调用select时的第一个参数，应当是进程中，最大的文件描述符+1，你确定你的fd【0】是当前进程中最大的描述符么？

调用select（）时随着需要监听的描述符越多，它的效率就会下降（因为轮询监听），这时如果还是不想使用多进程，就可以用epoll系列函数实现了。

天赐之道

slecet（）一般就是用来避免使用多进程或者多进程的，select是通过轮询的方式来监听（暂且这么说吧），所以可以取代多进程，一般用于单进程中，进程结束时，

它也就自然结束了。再说，你调用select时的第一个参数，应当是进程中，最大的文件描述符+1，你确定你的fd【0】是当前进程中最大的描述符么？

调用select（）时随着需要监听的描述符越多，它的效率就会下降（因为轮询监听），这时如果还是不想使用多进程，就可以用epoll系列函数实现了。