使用nanomsg库实现多线程之间的通信（三）

概述：

   这次使用nanomsg库实现多线程一对多的双向通信，其实类似一个服务器，其他均为客户端，且它们之间是长连接，所以服务器可以主动和客户端通信，客户端也可以和服务器通信，这样的一个方式。

多线程一对多双向通信Demo

#include <stdio.h>
#include <pthread.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <nanomsg/survey.h>
#include <nanomsg/nn.h>

/*
  此程序为nanomsg多线程一对多双向通信demo,服务器可以发送消息，所
  有客户端都可接收并回应数据。
*/

//inproc 标识用于多线程通信
char *url = "inproc://sky_test";

//服务器socket初始化
int server_sock_init(int *sock)
{
  *sock = nn_socket (AF_SP, NN_SURVEYOR);
  if (*sock < 0) {
    printf("create server data sock failed\r\n");
    return 1;
  }
  if (nn_bind(*sock, url) < 0) {
    printf("bind server data sock failed\r\n");
    return 1;
  }
  printf("server data socket init success...\r\n");
  return 0;
}

//客户端socket初始化
int client_sock_init(int *sock)
{
  *sock = nn_socket (AF_SP, NN_RESPONDENT);
  if (*sock < 0) {
    printf("create client data sock failed\r\n");
    return 1;
  }
  if (nn_connect(*sock, url) < 0) {
    printf("connect client data sock failed\r\n");
    return 1;
  }
  printf("client data socket init success...\r\n");
  return 0;
}

//线程1测试
void *thread_test(void *arg)
{
  int c_sock;
  if (0 != client_sock_init(&c_sock)) {
    return;
  }
  while (1) {
    char *txmsg = "Hi Main Pro, I`m Thread 1";
    char *rx_msg = NULL;
    int result = nn_recv(c_sock, &rx_msg, NN_MSG, NN_DONTWAIT);
    if (result > 0) {
      printf("Thread 1 Recieve: %s\r\n\r\n", rx_msg);
      nn_freemsg (rx_msg);
      //回应数据
      nn_send(c_sock, txmsg, strlen(txmsg)+1, 0);
    }
    sleep(1);
  }
}

//线程2测试
void *thread_test2(void *arg)
{
  int c_sock;
  if (0 != client_sock_init(&c_sock)) {
    return;
  }
  while (1) {
    char *rx_msg = NULL;
    char *txmsg = "Hi Main Pro, I`m Thread 2";
    int result = nn_recv(c_sock, &rx_msg, NN_MSG, NN_DONTWAIT);
    if (result > 0) {
      printf("Thread 2 Recieve: %s\r\n\r\n", rx_msg);
      nn_freemsg (rx_msg);
      //回应数据
      nn_send(c_sock, txmsg, strlen(txmsg)+1, 0);
    }
    sleep(1);
  }
}

//发送数据
int send_data(int sock, char *data)
{
  if (data == NULL) {
    return 1;
  }
  if (nn_send(sock, data, strlen(data)+1, 0) < 0) {
    return 1;
  }
  printf("Main Server Send:%s\r\n\r\n", data);
  return 0;
}

int main()
{
  int s_sock, ret, i = 0;
  pthread_t ps, ps2;
  char *tx_msg = "Hello Every Thread";
  if (0 != server_sock_init(&s_sock)) {
    return 1;
  }
  //创建子线程，接收信息
  pthread_create(&ps, NULL, thread_test, NULL);
  pthread_create(&ps2, NULL, thread_test2, NULL);
  sleep(1);
  while (1) {
    //发送信息
    ret = send_data(s_sock, tx_msg);
    sleep(1);
    while (1) {
      //接收客户端信息
      char *rx_msg = NULL;
      int result = nn_recv(s_sock, &rx_msg, NN_MSG, NN_DONTWAIT);
      if (result > 0) {
        printf("Main Recieve: %s\r\n\r\n", rx_msg);
        nn_freemsg (rx_msg);
      }
    }
  }
  return 0;
}

编译
gcc -o nanomsg_survey nanomsg_survey.c -lnanomsg -lpthread

运行结果：

sky@ubuntu:~/Study/nanomsg/code_test/inproc/survey$ ./nanomsg_survey
server data socket init success...
client data socket init success...
client data socket init success...
Main Server Send:Hello Every Thread

Thread 2 Recieve: Hello Every Thread

Thread 1 Recieve: Hello Every Thread

Main Recieve: Hi Main Pro, I`m Thread 2

Main Recieve: Hi Main Pro, I`m Thread 1
...

通过结果可以看到主线程可以发送数据到子线程，子线程也可以发送数据到主线程。