// test1.cpp : 定义控制台应用程序的入口点。

//

#include "stdafx.h"

#include <zhelpers.hpp>

#include <thread>

void RouterFunc()

{

    zmq::context_t context();

    zmq::socket_t broker(context, ZMQ_ROUTER);

    broker.bind("tcp://*:5671");

    while ()

    {

        std::string identity = s_recv(broker);

        std::string recvStr = s_recv(broker);

        recvStr = s_recv(broker);

        std::cout << recvStr << std::endl;    

        s_sendmore(broker, identity);

        s_sendmore(broker, "");

        s_send(broker, "reply1");

        identity = s_recv(broker);

        recvStr = s_recv(broker);

        recvStr = s_recv(broker);

        std::cout << recvStr << std::endl;

        s_sendmore(broker, identity);

        s_sendmore(broker, "");

        s_send(broker, "reply2");

    }

}

void DealerFunc()

{

    zmq::context_t context();

    zmq::socket_t worker(context, ZMQ_DEALER);

    worker.connect("tcp://localhost:5671");

    while ()

    {

        s_sendmore(worker, "");

        s_send(worker, "hello world1");

        std::string recvStr = s_recv(worker);

        recvStr = s_recv(worker);

        std::cout << recvStr << std::endl;

        s_sendmore(worker, "");

        s_send(worker, "hello world2");

        recvStr = s_recv(worker);

        recvStr = s_recv(worker);

        std::cout << recvStr << std::endl;

    }

}

void TestRouterDealer()

{

    std::thread routerThread = std::thread(RouterFunc);

    std::thread dealerThread = std::thread(DealerFunc);

    routerThread.join();

    dealerThread.join();

    return;

}

void RouterFun()

{

    zmq::context_t context();

    zmq::socket_t broker(context, ZMQ_ROUTER);

    broker.bind("tcp://*:5671");

    while () {

        std::string identity = s_recv(broker);

        std::string recvStr = s_recv(broker);

        recvStr = s_recv(broker);

        std::cout << recvStr << std::endl;

        s_sendmore(broker, identity);

        s_sendmore(broker, "");

        s_send(broker, "Work harder1");

        identity = s_recv(broker);

        recvStr = s_recv(broker);

        recvStr = s_recv(broker);

        std::cout << recvStr << std::endl;

        s_sendmore(broker, identity);

        s_sendmore(broker, "");

        s_send(broker, "Work harder2");

    }

}

void ReqFun()

{

    zmq::context_t context();

    zmq::socket_t worker(context, ZMQ_REQ);

    s_set_id(worker, (intptr_t));

    worker.connect("tcp://localhost:5671");

    while ()

    {

        s_send(worker, "Hi Boss1");

        std::string recvStr = s_recv(worker);

        std::cout << recvStr << std::endl;

        s_send(worker, "Hi Boss2");

        recvStr = s_recv(worker);

        std::cout << recvStr << std::endl;

    }

}

void TestRouterReQ()

{

    std::thread routerThread = std::thread(RouterFun);

    std::thread reqThread = std::thread(ReqFun);

    routerThread.join();

    reqThread.join();

    return;

}

int main()

{

    //TestRouterDealer();

    TestRouterReQ();

    return ;

}