bool has_more(void* socket)
{
int more = 0;
size_t more_size = sizeof(more);
int rc = zmq_getsockopt(socket, ZMQ_RCVMORE, &more, &more_size);
if (rc != 0)
return false;
return more != 0;
}
#include <string>
#include <map>
#include <vector>
#include <set>
bool bExitClient = false;
bool bExitServer = false;
//ZmqStreamServerProc - 服务端线程函数
DWORD WINAPI ZmqStreamServerProc(LPVOID lpThreadParameter) {
int rc;
std::set<std::string> peer_ids;
DWORD prev_tick = GetTickCount();
//创建context
void *ctx = zmq_ctx_new();
assert(ctx);
//创建socket并设置属性
void *server = zmq_socket(ctx, ZMQ_STREAM);
assert(server);
int enabled = 1;
rc = zmq_setsockopt(server, ZMQ_STREAM_NOTIFY, &enabled, sizeof(enabled));
assert(rc == 0);
//绑定端口
rc = zmq_bind(server, "tcp://127.0.0.1:9070");
assert(rc == 0);
while (!bExitServer) {
//对socket进行poll操作
zmq_pollitem_t items[] = {
{ server, 0, ZMQ_POLLIN, 0 },
};
rc = zmq_poll(items, 1, 100);
if (rc == 1) {
if (items[0].revents & ZMQ_POLLIN) {
//有数据可读,分为两种情况,一种是STREAM NOTIFY,一种是真实的客户端发出的数据
zmq_msg_t peer_frame;
//接收1st 对端ID帧
rc = zmq_msg_init(&peer_frame);
assert(rc == 0);
rc = zmq_msg_recv(&peer_frame, server, 0);
assert(rc != -1);
assert(zmq_msg_size(&peer_frame) > 0);
assert(has_more(server));
char * p_peer_frame_data = (char*)zmq_msg_data(&peer_frame);
//TRACE("peer_id:%s\n", p_peer_frame_data);
std::string str_peer_id(p_peer_frame_data, zmq_msg_size(&peer_frame));
zmq_msg_close(&peer_frame);
//接收2nd 对端数据帧
zmq_msg_t data_frame;
rc = zmq_msg_init(&data_frame);
assert(rc == 0);
rc = zmq_msg_recv(&data_frame, server, 0);
assert(rc != -1);
if (zmq_msg_size(&data_frame) == 0) {
if (peer_ids.find(str_peer_id) == peer_ids.end()) {
TRACE("S :a new conn incoming\n");
peer_ids.insert(str_peer_id);
}
else {
TRACE("S :an exsit conn disconnected\n");
peer_ids.erase(str_peer_id);
}
}
else {
char * p_data_frame_data = (char*)zmq_msg_data(&data_frame);
TRACE("S :recv [%s]\n", p_data_frame_data);
}
zmq_msg_close(&data_frame);
}
/*if (items[0].revents & ZMQ_POLLERR) {
TRACE("S :ZMQ_POLLERR\n");
}*/
}
else if (rc == 0) {
//0 表示poll没有相应事件,可认为socket出于空闲状态。
//TRACE("S :zmq_poll() return 0\n");
//assert(false);
//模拟每3秒钟给所有客户端发出请求
if (peer_ids.size() > 0 && (GetTickCount() - prev_tick) > 3000) {
for (auto iter = peer_ids.begin(); iter != peer_ids.end(); ++iter) {
//服务器端先发出请求
//发出1st ID帧
rc = zmq_send(server, iter->c_str(), iter->size(), ZMQ_SNDMORE);
//TRACE("S :send identity res=%d errno=%d\n", rc, errno);
//assert(rc != -1);
//发出2nd 数据帧
TRACE("S :send:[%s]\n", "hello");
rc = zmq_send(server, "hello", 6, ZMQ_SNDMORE);
//assert(rc != -1);
}
prev_tick = GetTickCount();
}
}
else {
TRACE("S :zmq_poll() return error\n");
assert(false);
}
}
rc = zmq_close(server);
assert(rc == 0);
rc = zmq_ctx_term(ctx);
assert(rc == 0);
return 0;
}
//ZmqStreamClientProc - 客户端线程函数
DWORD WINAPI ZmqStreamClientProc(LPVOID lpThreadParameter) {
int rc;
// Set-up our context and sockets
void *ctx = zmq_ctx_new();
assert(ctx);
int enabled = 1;
void *client = zmq_socket(ctx, ZMQ_STREAM);
assert(client);
rc = zmq_setsockopt(client, ZMQ_STREAM_NOTIFY, &enabled, sizeof(enabled));
assert(rc == 0);
rc = zmq_connect(client, "tcp://localhost:9070");
uint8_t id_opt[256];
size_t id_size_opt;
uint8_t id_c[256];
size_t id_size_c = 256;
uint8_t buffer[256];
rc = zmq_getsockopt(client, ZMQ_IDENTITY, id_opt, &id_size_opt);
assert(rc == 0);
Client: First frame is identity, second frame is zero
//id_size_c = zmq_recv(client, id_c, 256, 0);
///*TRACE("id_size = zmq_recv (client, id, 256, 0) -> id_size_c=%d\r\n", id_size_c);
//for (size_t i = 0; i < id_size_c; ++i) {
//TRACE("0x%02x \n", id_c[i]);
//}
//TRACE("\r\n");*/
//assert(id_size_c > 0);
//rc = zmq_recv(client, buffer, 256, 0);
//assert(rc == 0);
while (!bExitClient) {
//对socket进行poll操作
zmq_pollitem_t items[] = {
{ client, 0, ZMQ_POLLIN, 0 },
};
rc = zmq_poll(items, 1, 100);
if (rc == 1) {
if (items[0].revents & ZMQ_POLLIN) {
//有数据可读,分为两种情况,一种是STREAM NOTIFY,一种是真实的客户端发出的数据
zmq_msg_t peer_frame;
//接收1st 对端ID帧
rc = zmq_msg_init(&peer_frame);
assert(rc == 0);
rc = zmq_msg_recv(&peer_frame, client, 0);
assert(rc != -1);
assert(zmq_msg_size(&peer_frame) > 0);
assert(has_more(client));
char * p_peer_frame_data = (char*)zmq_msg_data(&peer_frame);
//TRACE("peer_id:%s\n", p_peer_frame_data);
std::string str_peer_id(p_peer_frame_data, zmq_msg_size(&peer_frame));
zmq_msg_close(&peer_frame);
//接收2nd 对端数据帧
zmq_msg_t data_frame;
rc = zmq_msg_init(&data_frame);
assert(rc == 0);
rc = zmq_msg_recv(&data_frame, client, 0);
assert(rc != -1);
if (zmq_msg_size(&data_frame) == 0) {
}
else {
char * p_data_frame_data = (char*)zmq_msg_data(&data_frame);
TRACE("C :recv [%s]\n", p_data_frame_data);
assert(memcmp(p_data_frame_data, "hello", 6) == 0);
rc = zmq_send(client, str_peer_id.c_str(), str_peer_id.size(), ZMQ_SNDMORE);
assert(rc != -1);
TRACE("C :send:[world!]\n");
rc = zmq_send(client, "world!", 7, 0);
assert(rc != -1);
}
zmq_msg_close(&data_frame);
}
if (items[0].revents & ZMQ_POLLOUT) {
}
if (items[0].revents & ZMQ_POLLERR) {
TRACE("C :ZMQ_POLLERR");
}
}
else if (rc == 0) {
//TRACE("C :zmq_poll() return 0\n");
//assert(false);
}
else {
TRACE("C :zmq_poll() return error\n");
//assert(false);
}
}
rc = zmq_close(client);
assert(rc == 0);
rc = zmq_ctx_term(ctx);
assert(rc == 0);
return 0;
}
void CZmqStreamCommDemoDlg::OnBnClickedButton1()
{
DWORD tid;
bExitServer = false;
HANDLE hServerThread = CreateThread(NULL, 0, ZmqStreamServerProc, NULL, 0, &tid);
CloseHandle(hServerThread);
}
void CZmqStreamCommDemoDlg::OnBnClickedButton2()
{
bExitClient = false;
DWORD tid;
HANDLE hThread = CreateThread(NULL, 0, ZmqStreamClientProc, NULL, 0, &tid);
CloseHandle(hThread);
}
void CZmqStreamCommDemoDlg::OnBnClickedButton3()
{
bExitClient = true;
}
void CZmqStreamCommDemoDlg::OnBnClickedButton4()
{
bExitServer = true;
}
2016-06-08 在实际代码中测试发现接收方收到的数据虽然是以zmq_msg_t 为边界,但有可能是由发送方的多包数据组成的,所以接收方还应严格按照自定义的协议格式来分解出多个包。