ZeroMQ學習 （九）zmq_poll使用

11 zmq_poll使用

​ 每一個zmq的套接字也可以認爲是一路IO，那麼在多路IO讀寫的時候，我們可以使用zmq_poll進行IO複用，使用了zmq_poll()函數，當收到了消息的第一幀時，其它幀其實也已經收到了。

訂閱者代碼如下：

​
#include <zmq.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <signal.h>
​
static int interrupted = 0;
void signal_handler (int sig)
{
    (void)sig;
    interrupted = 1;
}
​
void catch_signals (void)
{
    struct sigaction action;
    action.sa_handler = signal_handler;
    action.sa_flags = 0;
    sigemptyset (&action.sa_mask);
    sigaction (SIGINT, &action, NULL);
}
​
void dump_msg(const void * data, int size)
{
    unsigned char * ptr = (unsigned char *)data;
    printf ("[%03d] ", size);
    int i = 0;
    for (i = 0; i < size; i++)
        printf ("%02X", ptr[i]);
    printf ("\n");
}
​
int main (void)
{
    catch_signals();
    void *context = zmq_ctx_new();
    void *subscriber = zmq_socket (context, ZMQ_SUB);
    /*訂閱任何內容*/
    zmq_setsockopt(subscriber, ZMQ_SUBSCRIBE, "", 0);
    zmq_connect (subscriber, "tcp://127.0.0.1:5555");
​
    int ret;
    zmq_pollitem_t item[] = {{subscriber, 0, ZMQ_POLLIN, 0}};
    while (!interrupted)
    {
        ret = zmq_poll(item, 1, -1);
        if(ret == -1)
            perror("zmq_poll");
        if(item[0].revents & ZMQ_POLLIN)
        {
            zmq_msg_t msg;
            zmq_msg_init(&msg);
            ret = zmq_msg_recv (&msg, subscriber, 0);
            dump_msg(zmq_msg_data(&msg), ret);
            zmq_msg_close(&msg);
        }
    }
    //  程序不會運行到這裏，以下只是演示我們應該如何結束
    zmq_close (subscriber);
    zmq_ctx_destroy(context);
    return 0;
}

​ 以上代碼使用zmq_poll來接受套接字信息，zmq_poll會被信號打斷，並且不能自動重啓，這裏我們打印了一下中斷錯誤信息，被中斷後變量interrupted設爲1, 退出循環。