Merge branch 'main' of https://github.com/camargo2019/cmr-cache

camargo2019 · Sep 10, 2024 · cb11884 · cb11884
2 parents a36c576 + 03d7533
commit cb11884
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/core/socket/socket.cpp b/core/socket/socket.cpp
@@ -23,8 +23,9 @@
 
 CoreSocket::CoreSocket(boost::asio::io_context& io_context, std::string ip, short port, Cache& cache_)
     : acceptor_(io_context, boost::asio::ip::tcp::endpoint(boost::asio::ip::make_address(ip), port)),
-      cache_(cache_), client_(0) {
+      cache_(cache_), client_(0), socket_(io_context) {
         accept();
+        ping();
 }
 
 void CoreSocket::accept() {
@@ -47,3 +48,14 @@ void CoreSocket::accept() {
     );
 }
 
+void CoreSocket::ping() {
+    std::thread([this](){
+        while (true) {
+            if (socket_.is_open()) {
+                boost::asio::write(socket_, boost::asio::buffer(std::string("PING\r\n")));
+            }
+
+            std::this_thread::sleep_for(std::chrono::seconds(5));
+        }
+    }).detach();
+}
diff --git a/core/socket/socket.h b/core/socket/socket.h
@@ -37,10 +37,12 @@
 
         private:
             void accept();
+            void ping();
 
-            boost::asio::ip::tcp::acceptor acceptor_;
             Cache& cache_;
             std::atomic<int> client_;
+            boost::asio::ip::tcp::socket socket_;
+            boost::asio::ip::tcp::acceptor acceptor_;
             ConfigConnect ConfigConn = Config::getConfigConnect();
     };