support auto client command for evaluation

1 files changed, 10 insertions, 0 deletions

diff --git a/src/hotstuff_app.cpp b/src/hotstuff_app.cpp
index 30d127b..be9d3cd 100644
--- a/src/hotstuff_app.cpp
+++ b/src/hotstuff_app.cpp
@@ -110,6 +110,16 @@ class HotStuffApp: public HotStuff {
         */
     }
 
+#ifdef HOTSTUFF_AUTOCLI
+    void do_demand_commands(size_t blk_size) override {
+        size_t ncli = client_conns.size();
+        size_t bsize = (blk_size + ncli - 1) / ncli;
+        hotstuff::MsgDemandCmd mdc{bsize};
+        for(const auto &conn: client_conns)
+            cn.send_msg(mdc, conn);
+    }
+#endif
+
 #ifdef HOTSTUFF_MSG_STAT
     std::unordered_set<conn_t> client_conns;
     void print_stat() const;