aboutsummaryrefslogtreecommitdiff
path: root/include/salticidae/network.h
diff options
context:
space:
mode:
Diffstat (limited to 'include/salticidae/network.h')
-rw-r--r--include/salticidae/network.h22
1 files changed, 14 insertions, 8 deletions
diff --git a/include/salticidae/network.h b/include/salticidae/network.h
index 1b871f5..11f720a 100644
--- a/include/salticidae/network.h
+++ b/include/salticidae/network.h
@@ -79,12 +79,14 @@ class MsgNetwork: public ConnPool {
#ifdef SALTICIDAE_MSG_STAT
mutable std::atomic<size_t> nsent;
mutable std::atomic<size_t> nrecv;
+ mutable std::atomic<size_t> nsentb;
+ mutable std::atomic<size_t> nrecvb;
#endif
public:
Conn(): msg_state(HEADER)
#ifdef SALTICIDAE_MSG_STAT
- , nsent(0), nrecv(0)
+ , nsent(0), nrecv(0), nsentb(0), nrecvb(0)
#endif
{}
@@ -95,8 +97,14 @@ class MsgNetwork: public ConnPool {
#ifdef SALTICIDAE_MSG_STAT
size_t get_nsent() const { return nsent; }
size_t get_nrecv() const { return nrecv; }
- void clear_nsent() const { nsent = 0; }
- void clear_nrecv() const { nrecv = 0; }
+ size_t get_nsentb() const { return nsentb; }
+ size_t get_nrecvb() const { return nrecvb; }
+ void clear_msgstat() const {
+ nsent.store(0, std::memory_order_relaxed);
+ nrecv.store(0, std::memory_order_relaxed);
+ nsentb.store(0, std::memory_order_relaxed);
+ nrecvb.store(0, std::memory_order_relaxed);
+ }
#endif
protected:
@@ -116,8 +124,6 @@ class MsgNetwork: public ConnPool {
queue_t incoming_msgs;
protected:
-#ifdef SALTICIDAE_MSG_STAT
-#endif
ConnPool::Conn *create_conn() override { return new Conn(); }
@@ -157,10 +163,11 @@ class MsgNetwork: public ConnPool {
SALTICIDAE_LOG_DEBUG("got message %s from %s",
std::string(msg).c_str(),
std::string(*conn).c_str());
- it->second(msg, conn);
#ifdef SALTICIDAE_MSG_STAT
conn->nrecv++;
+ conn->nrecvb += msg.get_length();
#endif
+ it->second(msg, conn);
}
if (++cnt == burst_size) return true;
}
@@ -182,8 +189,6 @@ class MsgNetwork: public ConnPool {
template<typename MsgType>
bool send_msg(MsgType &&msg, const conn_t &conn);
using ConnPool::listen;
-#ifdef SALTICIDAE_MSG_STAT
-#endif
conn_t connect(const NetAddr &addr) {
return static_pointer_cast<Conn>(ConnPool::connect(addr));
}
@@ -467,6 +472,7 @@ bool MsgNetwork<OpcodeType>::send_msg(MsgType &&_msg, const conn_t &conn) {
std::string(*conn).c_str());
#ifdef SALTICIDAE_MSG_STAT
conn->nsent++;
+ conn->nsentb += msg.get_length();
#endif
return conn->write(std::move(msg_data));
}