| | |
| | | * ===================================================================================== |
| | | */ |
| | | #include "util.h" |
| | | #include <boost/date_time/posix_time/posix_time.hpp> |
| | | |
| | | using namespace boost::posix_time; |
| | | |
| | | BOOST_AUTO_TEST_CASE(SpeedTest) |
| | | { |
| | | const int mem_size = 1024 * 1024 * 50; |
| | | SharedMemory &shm = TestShm(); |
| | | MsgI::BindShm(shm); |
| | | GlobalInit(shm); |
| | | |
| | | MQId id = boost::uuids::random_generator()(); |
| | | MQId id = ShmMsgQueue::NewId(); |
| | | const int timeout = 1000; |
| | | const uint32_t data_size = 4000; |
| | | const std::string proc_id = "demo_proc"; |
| | |
| | | MsgRequestTopic body; |
| | | body.set_topic("topic"); |
| | | body.set_data(str); |
| | | auto head(InitMsgHead(GetType(body), proc_id)); |
| | | auto head(InitMsgHead(GetType(body), proc_id, mq.Id())); |
| | | msg.Make(head, body); |
| | | assert(msg.valid()); |
| | | DEFER1(msg.Release();); |
| | |
| | | }; |
| | | auto Reader = [&](int reader_id, std::atomic<bool> *run, bool isfork) { |
| | | ShmMsgQueue mq(id, shm, 1000); |
| | | auto now = []() { return steady_clock::now(); }; |
| | | auto tm = now(); |
| | | while (*run) { |
| | | MsgI msg; |
| | | BHMsgHead head; |
| | | if (mq.Recv(msg, timeout)) { |
| | | if (mq.TryRecv(msg)) { |
| | | DEFER1(msg.Release()); |
| | | // ok |
| | | tm = now(); |
| | | } else if (isfork) { |
| | | exit(0); // for forked quit after 1s. |
| | | if (now() > tm + 1s) { |
| | | exit(0); // for forked quit after 1s. |
| | | } |
| | | } |
| | | } |
| | | }; |
| | |
| | | } |
| | | }; |
| | | |
| | | int nwriters[] = {1, 2, 4}; |
| | | int nreaders[] = {1, 2}; |
| | | int nwriters[] = {1, 4, 16}; |
| | | int nreaders[] = {1, 4}; |
| | | |
| | | auto Test = [&](auto &www, auto &rrr, bool isfork) { |
| | | for (auto nreader : nreaders) { |
| | |
| | | const std::string server_proc_id = "server_proc"; |
| | | |
| | | SharedMemory &shm = TestShm(); |
| | | MsgI::BindShm(shm); |
| | | GlobalInit(shm); |
| | | |
| | | auto Avail = [&]() { return shm.get_free_memory(); }; |
| | | auto init_avail = Avail(); |
| | |
| | | MsgRequestTopic req_body; |
| | | req_body.set_topic("topic"); |
| | | req_body.set_data(msg_content); |
| | | auto req_head(InitMsgHead(GetType(req_body), client_proc_id)); |
| | | req_head.add_route()->set_mq_id(&cli.id(), cli.id().size()); |
| | | return cli.Send(&srv.id(), req_head, req_body); |
| | | auto req_head(InitMsgHead(GetType(req_body), client_proc_id, cli.id())); |
| | | req_head.add_route()->set_mq_id(cli.id()); |
| | | return cli.Send(srv.id(), req_head, req_body); |
| | | }; |
| | | |
| | | Req(); |
| | |
| | | DEFER1(req.Release()); |
| | | |
| | | if (req.ParseHead(req_head) && req_head.type() == kMsgTypeRequestTopic) { |
| | | auto &mqid = req_head.route()[0].mq_id(); |
| | | MQId src_id; |
| | | memcpy(&src_id, mqid.data(), sizeof(src_id)); |
| | | auto src_id = req_head.route()[0].mq_id(); |
| | | auto Reply = [&]() { |
| | | MsgRequestTopic reply_body; |
| | | reply_body.set_topic("topic"); |
| | | reply_body.set_data(msg_content); |
| | | auto reply_head(InitMsgHead(GetType(reply_body), server_proc_id, req_head.msg_id())); |
| | | return srv.Send(&src_id, reply_head, reply_body); |
| | | auto reply_head(InitMsgHead(GetType(reply_body), server_proc_id, srv.id(), req_head.msg_id())); |
| | | return srv.Send(src_id, reply_head, reply_body); |
| | | }; |
| | | Reply(); |
| | | } |