ZeroMQ实践之路(2)

从网上找了一个ZMQ实现简单负载均衡的例子实验了一把:客户端同时连接两个启动的服务器,然后与服务器之间收发数据,你会发现程序会自动进行负载均衡操作–与两个服务器交替的通信.

• 客户端:

/*
 * 
 * g++ -o request request.cpp -lzmq
 *
 */

#include <zmq.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include<assert.h>

#define MAX_BUFFER_LEN 1024

//同时定义了两个服务器，这里可以更多，格式就这么写即可，下面只用一个套接字
const char* hosts[] = {
    "tcp://192.168.116.130:9000",
    "tcp://192.168.116.130:8000"

};

int main (void){    

    printf ("Connecting to hello world server…\n");
    void *context = zmq_ctx_new ();
    assert(context);
    void *requester = zmq_socket (context, ZMQ_REQ);

    for (int i = 0; i < sizeof(hosts) / sizeof(hosts[0]); i ++){
        if (0 != zmq_connect (requester, hosts[i])) {
            printf("connect server %s failed.\n", hosts[i]);
        }
    }
	//会轮流发送给连接的服务器，负载均衡，而且可以先启动客户端，真香
    for (int i = 0; i < 50; i ++){
        char buffer [MAX_BUFFER_LEN] = {0};
        sprintf(buffer, "hello_%d", i);
        printf ("Send %s to server.\n", buffer);
        zmq_send (requester, buffer, strlen(buffer), 0);
        memset(buffer, 0, MAX_BUFFER_LEN);
        zmq_recv (requester, buffer, 10, 0);
        printf ("Received %s.\n", buffer);
    }

    zmq_close (requester);
    zmq_ctx_destroy (context);
   
    return 0;
}

• 服务端:

/*
 * 
 *  g++ -o responser responser.cpp -lzmq
 */

#include <zmq.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <assert.h>

#define MAX_BUFFER_LEN 1024

int main (int argc, char * argv[])
{
    //  Socket to talk to clients
    void *context = zmq_ctx_new ();
    assert(context);
    void *responder = zmq_socket (context, ZMQ_REP);
    char strIpPort[256];
    //别忘了‘\0’
    sprintf(strIpPort, "tcp://%s:%s\0", argv[1], argv[2]);
    int rc = zmq_bind (responder, strIpPort);
    if (rc != 0)
    {
        printf ("Error occurred during zmq_bind(): %s\n", zmq_strerror (errno));
        return -1;
    }

    int count = 0;

    while (1) 
    {
        char buffer [MAX_BUFFER_LEN] = {0};
        zmq_recv (responder, buffer, 10, 0);
        printf ("%s.\n", buffer);
        sleep (1);
        zmq_send (responder, "World", 5, 0);
        printf("send world to client with count=%d.\n", count);
        count ++;
    }

    return 0;
}

参考文章:
zmq初次使用