Add Idle

resetius · Nov 25, 2023 · ed59147 · ed59147
1 parent f750b9f
commit ed59147
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 3 deletions.
diff --git a/src/server.cpp b/src/server.cpp
@@ -1,3 +1,4 @@
+#include <chrono>
 #include <coroutine>
 #include <exception>
 #include <iostream>
@@ -66,7 +67,12 @@ NNet::TSimpleTask TRaftServer::InboundCounnection(NNet::TSocket socket) {
     co_return;
 }
 
-NNet::TSimpleTask TRaftServer::Serve() {
+void TRaftServer::Serve() {
+    Idle();
+    InboundServe();
+}
+
+NNet::TSimpleTask TRaftServer::InboundServe() {
     Socket.Bind();
     Socket.Listen();
     while (true) {
@@ -75,3 +81,18 @@ NNet::TSimpleTask TRaftServer::Serve() {
     }
     co_return;
 }
+
+NNet::TSimpleTask TRaftServer::Idle() {
+    auto t0 = TimeSource->Now();
+    auto dt = std::chrono::milliseconds(2000);
+    auto sleep = std::chrono::milliseconds(10);
+    while (true) {
+        Raft->Process(NewTimeout());
+        auto t1 = TimeSource->Now();
+        if (t1 > t0 + dt) {
+            t0 = t1;
+        }
+        Poller.Sleep(sleep);
+    }
+    co_return;
+}
diff --git a/src/server.h b/src/server.h
@@ -4,6 +4,8 @@
 #include <coroutine>
 
 #include <all.hpp>
+
+#include "timesource.h"
 #include "messages.h"
 #include "raft.h"
 #include "socket.hpp"
@@ -106,21 +108,31 @@ class TWriter {
 
 class TRaftServer {
 public:
-    TRaftServer(NNet::TPoll& poller, NNet::TAddress address, const std::shared_ptr<TRaft>& raft, const TNodeDict& nodes)
+    TRaftServer(
+        NNet::TPoll& poller,
+        NNet::TAddress address,
+        const std::shared_ptr<TRaft>& raft,
+        const TNodeDict& nodes,
+        const std::shared_ptr<ITimeSource>& ts)
         : Poller(poller)
         , Socket(std::move(address), Poller)
         , Raft(raft)
         , Nodes(nodes)
+        , TimeSource(ts)
     { }
 
-    NNet::TSimpleTask Serve();
+    void Serve();
 
 private:
+    NNet::TSimpleTask InboundServe();
+    NNet::TSimpleTask Idle();
+
     NNet::TSimpleTask InboundCounnection(NNet::TSocket socket);
     NNet::TTestTask Connector(std::shared_ptr<INode> node);
 
     NNet::TPoll& Poller;
     NNet::TPoll::TSocket Socket;
     std::shared_ptr<TRaft> Raft;
     TNodeDict Nodes;
+    std::shared_ptr<ITimeSource> TimeSource;
 };