关闭

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

标签: c 编程线程消息队列
626人阅读 评论(0) 收藏 举报
分类:
#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);
}


0
0
猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:37992次
    • 积分:624
    • 等级:
    • 排名:千里之外
    • 原创:25篇
    • 转载:5篇
    • 译文:0篇
    • 评论:11条
    文章分类
    最新评论