消息队列的基本使用思路和接口介绍

最新推荐文章于 2024-06-23 19:02:10 发布

原创最新推荐文章于 2024-06-23 19:02:10 发布 · 1.6k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c 编程 #线程 #消息队列

c编程专栏收录该内容

17 篇文章

订阅专栏

本文档介绍了如何在C编程环境下使用消息队列。首先，通过`mk_id`函数生成消息队列的关键字，接着使用`msgget`创建消息队列。然后，展示了`Write_msg`函数用于写入消息，而`Recv_msg`函数则用于读取消息。在多进程环境中，两个进程并行读取消息队列，最后通过`Close_msg`函数删除消息队列。

#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <sys/types.h>

#define MSGLENGTH 256
#define MSG 0010
/*
 *消息队列，在不关闭的情况下，只要拿到id就可以读取内容，但是当执行过msgrcv后
 * 该队列将被删除
 **/

/*生成key用于创建消息队列*/
key_t mk_id(int id)
{
    static char path[128 +1];
    static int flag=0;
    key_t key ;

    if( flag )
        goto _not_first_;

    sprintf(path,"%s/%s",getenv("HOME"),".bashrc");
    flag = 1 ;

_not_first_:
    key = ftok(path,id);
    return ( key );
}

/*销毁消息队列*/
int Close_msg(int id)
{
    return msgctl(id,IPC_RMID,0);
}

/*读取消息队列的内容*/
int Recv_msg(int id,int p_no)
{
    struct s_msg{
            long type;
            char mtext[256];
    } msg;
    int msg_len=0;
    while(1)
    {
        msg_len=msgrcv(id,&msg,MSGLENGTH,0,IPC_NOWAIT|MSG_NOERROR);

        /*消息读完了就break*/
        if(errno == ENOMSG)
            break;
        /*读取消息出错*/
        if(msg_len==-1 )
        {
            perror("msgrcv");
            return -1;
        }
        printf("%d:%s\n",p_no,msg.mtext);
    }
    return 0;
}
int Write_msg(int id)
{
    struct s_msg{
            long type;
            char mtext[256];
    } msg;

    char buff   [256];
    char file_name   [256];
    FILE *in_file;
    int snd_len=0;

    msg.type=MSG;

    sprintf(file_name,"%s/%s",getenv("HOME"),"tmp/1.dat");

    if((in_file=fopen(file_name,"r"))==NULL)
    {
        perror("fopen");
        exit(-1);
    }

    while(fgets(buff,256,in_file) !=NULL)
    {
        memcpy((void *)msg.mtext,buff,MSGLENGTH);

        /*复制string到消息队列*/
        snd_len==msgsnd(id,&msg,MSGLENGTH,IPC_NOWAIT);

        /*出现EAGAIN好像是队列没释放，等会儿就能放进去了*/
        if(-1==snd_len && errno != EAGAIN)
        {
            perror("msgsnd");
            exit(-1);
        }

        memset(msg.mtext,0x00,sizeof(msg.mtext));
    }
    fclose(in_file);
    return 0;
}

/*创建消息队列*/
int Creat_msg(key_t key)
{
    int id=0;
    if((id=msgget(key,0666|IPC_CREAT)) == -1)
    {
        perror("msgget");
        exit(-1);
    }
    return id;
}
main()
{
    int id=0;
    int proc_num=2;
    pid_t pid=0;

    id=Creat_msg(mk_id(MSG));
    printf("msg id %d\n",id);
    Write_msg(id);

    /*2个线程同时读取消息队列*/
    while(proc_num--)
    {
        pid=fork();
        if(pid==0)
        {
            Recv_msg(id,proc_num);
            exit(-1);
        }
        else if (pid <0)
        {
            perror("fork");
            exit(-1);
        }
    }

    /*等大家都读取完成后销毁队列*/
    wait(NULL);
    Close_msg(id);
    exit(0);
}