tweak ping pong

This commit is contained in:
Florian Lüke 2024-11-21 23:00:13 +01:00
parent 78cab52b91
commit 9e95897283

View file

@ -1,14 +1,25 @@
#include <mesytec-mvlc/mesytec-mvlc.h>
#include "proto/service.pb.h"
#include "mesy_nng.h"
using namespace mesytec;
using namespace std::literals;
void requester(nng_socket socket, unsigned id)
{
mvlc::util::Stopwatch sw;
std::uint32_t seq = 0;
mnode::Ping ping;
mnode::Pong pong;
while (true)
{
mnode::Ping ping;
if (sw.get_elapsed() >= 1s)
{
spdlog::info("Requester {} sent {} requests", id, seq);
break;
}
ping.set_peer_id(id);
ping.set_sequence_number(seq++);
auto reqMsg = nng::make_message(ping.SerializeAsString());
@ -19,7 +30,7 @@ void requester(nng_socket socket, unsigned id)
return;
continue;
}
spdlog::info("Sent ping request: {}", ping.ShortDebugString());
spdlog::debug("Sent ping request: {}", ping.ShortDebugString());
auto [repMsg, rc] = nng::receive_message(socket);
if (rc)
{
@ -28,16 +39,25 @@ void requester(nng_socket socket, unsigned id)
return;
continue;
}
mnode::Pong pong;
pong.ParseFromArray(nng_msg_body(repMsg.get()), nng_msg_len(repMsg.get()));
spdlog::info("Received pong response: {}", pong.ShortDebugString());
spdlog::debug("Received pong response: {}", pong.ShortDebugString());
}
}
void responder(nng_socket socket, unsigned id)
{
mvlc::util::Stopwatch sw;
mnode::Ping ping;
mnode::Pong pong;
while (true)
{
if (sw.get_elapsed() >= 1s)
{
spdlog::info("Responder {} answered {} requests", id, ping.sequence_number());
break;
}
auto [reqMsg, rc] = nng::receive_message(socket);
if (rc)
{
@ -46,11 +66,9 @@ void responder(nng_socket socket, unsigned id)
return;
continue;
}
mnode::Ping ping;
ping.ParseFromArray(nng_msg_body(reqMsg.get()), nng_msg_len(reqMsg.get()));
spdlog::info("Received ping request: {}", ping.ShortDebugString());
spdlog::debug("Received ping request: {}", ping.ShortDebugString());
mnode::Pong pong;
pong.set_peer_id(id);
pong.set_sequence_number(ping.sequence_number());
auto repMsg = nng::make_message(pong.SerializeAsString());
@ -61,22 +79,50 @@ void responder(nng_socket socket, unsigned id)
return;
continue;
}
spdlog::info("Sent pong response: {}", pong.ShortDebugString());
spdlog::debug("Sent pong response: {}", pong.ShortDebugString());
}
}
void client_aio(nng_socket socket, unsigned id)
{
}
void server_aio(nng_socket socket, unsigned id)
{
}
int main()
{
spdlog::set_level(spdlog::level::info);
unsigned peerId = 0;
std::vector<std::thread> threads;
#if 1
{
auto reqSocket = nng::make_req_socket();
auto repSocket = nng::make_rep_socket();
if (int rc = nng::marry_listen_dial(reqSocket, repSocket, "inproc://test"))
if (int rc = nng::marry_listen_dial(reqSocket, repSocket, fmt::format("inproc://test{}", peerId).c_str()))
return rc;
std::vector<std::thread> threads;
threads.emplace_back(std::thread(requester, reqSocket, 0));
threads.emplace_back(std::thread(responder, repSocket, 1));
threads.emplace_back(std::thread(requester, reqSocket, peerId++));
threads.emplace_back(std::thread(responder, repSocket, peerId++));
}
#endif
#if 1
{
auto reqSocket = nng::make_req_socket();
auto repSocket = nng::make_rep_socket();
if (int rc = nng::marry_listen_dial(reqSocket, repSocket, fmt::format("inproc://test{}", peerId).c_str()))
return rc;
threads.emplace_back(std::thread(client_aio, reqSocket, peerId++));
threads.emplace_back(std::thread(server_aio, repSocket, peerId++));
}
#endif
for (auto &t: threads)
if (t.joinable()) t.join();