服务端主动发送 Ping 探测连接性

lanthora · Dec 2, 2023 · 12abca1 · 12abca1
1 parent aa8532f
commit 12abca1
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/src/core/server.cc b/src/core/server.cc
@@ -167,6 +167,7 @@ void Server::handleAuthMessage(WebSocketMessage &message) {
     }
 
     spdlog::info("client connected: {}", address.getIpStr());
+    this->ws.keepalive(message.conn);
     this->ipWsMap[address.getIp()] = message.conn;
     this->wsIpMap[message.conn] = address.getIp();
 }

diff --git a/src/websocket/server.cc b/src/websocket/server.cc
@@ -65,6 +65,15 @@ class WebSockeServerImpl {
         return 0;
     }
 
+    int keepalive(WebSocketConn conn) {
+        std::weak_ptr<ix::WebSocket> weakConn = std::any_cast<std::weak_ptr<ix::WebSocket>>(conn.conn);
+        auto ws = weakConn.lock();
+        if (ws) {
+            ws->setPingInterval(30);
+        }
+        return 0;
+    }
+
     int close(WebSocketConn conn) {
         std::weak_ptr<ix::WebSocket> weakConn = std::any_cast<std::weak_ptr<ix::WebSocket>>(conn.conn);
         auto ws = weakConn.lock();
@@ -169,4 +178,10 @@ int WebSocketServer::close(WebSocketConn conn) {
     return server->close(conn);
 }
 
+int WebSocketServer::keepalive(WebSocketConn conn) {
+    std::shared_ptr<WebSockeServerImpl> server;
+    server = std::any_cast<std::shared_ptr<WebSockeServerImpl>>(this->impl);
+    return server->keepalive(conn);
+}
+
 } // namespace Candy
diff --git a/src/websocket/server.h b/src/websocket/server.h
@@ -27,6 +27,9 @@ class WebSocketServer {
     // 关闭单个客户端连接
     int close(WebSocketConn conn);
 
+    // 服务端主动探测客户端连接状态
+    int keepalive(WebSocketConn conn);
+
 private:
     std::any impl;
 };