zeroMQ学习记录-请求应答模式(zmq/zmp和router/dealer)

//为了方便测试，没有另外启动一个工程，直接在一个工程中将客户端和服务端实现。tcp协议。
一对一形式。

#include <iostream>
#include <string>
#include <vector>
#include <thread>
#include "zmq.hpp"

using namespace std;

void reqClient()
{
    zmq::context_t context(1);
    zmq::socket_t socket(context, zmq::socket_type::req);
    socket.connect("tcp://localhost:5555");
    
    while (true)
    {
        std::string request="hello";
        zmq::message_t sendmessage(request.size());

        memcpy(sendmessage.data(),request.c_str(),request.size());
        socket.send(sendmessage, zmq::send_flags::none);

        //请求应答模式必须请求了要应答
        
        zmq::message_t receive_mes;

        socket.recv(receive_mes, zmq::recv_flags::none);
        std::string receive(static_cast<char*>(receive_mes.data()), receive_mes.size());
        std::cout << "client:  " << receive << std::endl;

    }
}

int main()
{
    zmq::context_t context(1);
    zmq::socket_t socket(context, zmq::socket_type::rep);
    socket.bind("tcp://*:5555");
    std::thread t1(reqClient);
    while (true)
    {
        //请求响应模式，发送和接收必须配套使用
        zmq::message_t request;
        socket.recv(request, zmq::recv_flags::none);
        std::string received_message(static_cast<char*>(request.data()), request.size());

        std::cout << "server:  " << received_message << std::endl;

        //发送响应
        std::string reply_message="world";
        zmq::message_t reply(reply_message.size());
        memcpy(reply.data(), reply_message.c_str(), reply_message.size());

        socket.send(reply,zmq::send_flags::none);
    }
    t1.join();
    return 0;

}

//对客户端对多服务端：主要用于负载均衡和任务分配，通过代理可以将客户端的请求分配到对应的服务端，进行处理后通过代理唯一标识发送到客户端

使用多对多通信，也可以尝试使用req和rep,通过连接代理：dealer(后端)和router(前端)连接
当然最好直接使用router和dealer套接字

//代理代码：
void agency()
{
    zmq::context_t context(1);
    zmq::socket_t frontend(context, zmq::socket_type::router);
    zmq::socket_t backend(context, zmq::socket_type::dealer);
    
    frontend.bind("tcp://*:5555");
    backend.bind("tcp://*:5556");

    zmq::proxy(frontend, backend);
}

//客户端代码：
void reqClient(int id) {
    zmq::context_t context(1);
    zmq::socket_t socket(context, zmq::socket_type::dealer);
    std::string identity = "client_" + std::to_string(id);
    socket.set(zmq::sockopt::routing_id, identity);

    socket.connect("tcp://localhost:5555");

    while (true) {
        std::string request = "hello from client: " + std::to_string(id);
        zmq::message_t sendmessage(request.size());
        memcpy(sendmessage.data(), request.c_str(), request.size());
        socket.send(sendmessage, zmq::send_flags::none);

        zmq::message_t reply;
        socket.recv(reply, zmq::recv_flags::none);

        std::string reply_message(static_cast<char*>(reply.data()), reply.size());
        std::cout << "Client " << id << " received: " << reply_message << std::endl;

        std::this_thread::sleep_for(std::chrono::seconds(1));
    }
}

//服务端代码：这里有个坑：注意router套接字接收到的是三部分消息：
//identity来源的生成码,
//一个空帧(一个空帧，用于分隔身份标识符和实际消息内容)，
//实际的消息。所以发出去也是要三部分，前两部分就直接返回就行。
void zmqServer(int id) {
    zmq::context_t context(1);
    zmq::socket_t socket(context, zmq::socket_type::router);
    socket.connect("tcp://localhost:5556");

    while (true) {
        zmq::message_t identity;
        zmq::message_t request;

        // Receive identity
        socket.recv(identity, zmq::recv_flags::none);
        std::string identity_str(static_cast<char*>(identity.data()), identity.size());
        std::cout << "Server " << id << " received identity: " << identity_str << std::endl;

        zmq::message_t delimaily;
        socket.recv(delimaily,zmq::recv_flags::none);
        // Receive request
        socket.recv(request, zmq::recv_flags::none);
        std::string received_message(static_cast<char*>(request.data()), request.size());
        std::cout << "Server " << id << " received message: " << received_message << std::endl;


        // Send reply
        std::string reply_message = "world from server " + std::to_string(id);
        zmq::message_t reply(reply_message.size());
        memcpy(reply.data(), reply_message.c_str(), reply_message.size());

        socket.send(identity, zmq::send_flags::sndmore);
        socket.send(delimaily, zmq::send_flags::sndmore);
        socket.send(reply, zmq::send_flags::none);

        std::this_thread::sleep_for(std::chrono::milliseconds(10));
    }
}