diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/hotstuff_app.cpp | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/src/hotstuff_app.cpp b/src/hotstuff_app.cpp index f8f15f9..30d127b 100644 --- a/src/hotstuff_app.cpp +++ b/src/hotstuff_app.cpp @@ -110,6 +110,11 @@ class HotStuffApp: public HotStuff { */ } +#ifdef HOTSTUFF_MSG_STAT + std::unordered_set<conn_t> client_conns; + void print_stat() const; +#endif + public: HotStuffApp(uint32_t blk_size, double stat_period, @@ -313,6 +318,7 @@ void HotStuffApp::client_request_cmd_handler(MsgReqCmd &&msg, const conn_t &conn void HotStuffApp::start(const std::vector<std::pair<NetAddr, bytearray_t>> &reps) { ev_stat_timer = TimerEvent(ec, [this](TimerEvent &) { HotStuff::print_stat(); + HotStuffApp::print_stat(); //HotStuffCore::prune(100); ev_stat_timer.add(stat_period); }); @@ -327,6 +333,35 @@ void HotStuffApp::start(const std::vector<std::pair<NetAddr, bytearray_t>> &reps HOTSTUFF_LOG_INFO("conns = %lu", HotStuff::size()); HOTSTUFF_LOG_INFO("** starting the event loop..."); HotStuff::start(reps); + cn.reg_conn_handler([this](const salticidae::ConnPool::conn_t &_conn, bool connected) { + auto conn = salticidae::static_pointer_cast<conn_t::type>(_conn); + if (connected) + client_conns.insert(conn); + else + client_conns.erase(conn); + }); /* enter the event main loop */ ec.dispatch(); } + +void HotStuffApp::print_stat() const { +#ifdef HOTSTUFF_MSG_STAT + HOTSTUFF_LOG_INFO("--- client msg. (10s) ---"); + size_t _nsent = 0; + size_t _nrecv = 0; + for (const auto &conn: client_conns) + { + if (conn == nullptr) continue; + size_t ns = conn->get_nsent(); + size_t nr = conn->get_nrecv(); + size_t nsb = conn->get_nsentb(); + size_t nrb = conn->get_nrecvb(); + conn->clear_msgstat(); + HOTSTUFF_LOG_INFO("%s: %u(%u), %u(%u)", + std::string(conn->get_addr()).c_str(), ns, nsb, nr, nrb); + _nsent += ns; + _nrecv += nr; + } + HOTSTUFF_LOG_INFO("--- end client msg. ---"); +#endif +} |