feat: 长保持连接，控制端可重启

python/omnisocket/_omnisocket.c

@@ -67,7 +67,7 @@ static PyObject *build_recv_meta_dict(
 
 static PyObject *build_stats_dict(const omnisocket_session_stats_t *stats) {
     return Py_BuildValue(
-        "{s:K,s:K,s:K,s:K,s:K,s:K,s:K,s:i}",
+        "{s:K,s:K,s:K,s:K,s:K,s:K,s:K,s:i,s:i,s:s}",
         "send_calls",
         (unsigned long long) stats->send_calls,
         "send_bytes",
@@ -83,7 +83,11 @@ static PyObject *build_stats_dict(const omnisocket_session_stats_t *stats) {
         "recv_errors",
         (unsigned long long) stats->recv_errors,
         "connected",
-        stats->connected
+        stats->connected,
+        "registered",
+        stats->registered,
+        "last_server_error",
+        stats->last_server_error
     );
 }
 

python/omnisocket/omnisocket_client.c

@@ -1,5 +1,33 @@
 #include "omnisocket_client.h"
 
+static void omnisocket_session_sync_client_state_locked(omnisocket_session_t *session, kcp_client_t *client) {
+    kcp_client_state_t client_state;
+
+    if (session == NULL) {
+        return;
+    }
+    memset(&client_state, 0, sizeof(client_state));
+    if (client != NULL) {
+        kcp_client_state_snapshot(client, &client_state);
+    }
+    session->stats.connected = client_state.connected;
+    session->stats.registered = client_state.registered;
+    snprintf(
+        session->stats.last_server_error,
+        sizeof(session->stats.last_server_error),
+        "%s",
+        client_state.last_server_error
+    );
+}
+
+static void omnisocket_session_mark_disconnected_locked(omnisocket_session_t *session) {
+    if (session == NULL) {
+        return;
+    }
+    session->stats.connected = 0;
+    session->stats.registered = 0;
+}
+
 int omnisocket_session_init(omnisocket_session_t *session) {
     int rc;
 
@@ -97,7 +125,7 @@ int omnisocket_session_connect(
         return -1;
     }
     session->client = client;
-    session->stats.connected = 1;
+    omnisocket_session_sync_client_state_locked(session, client);
     pthread_mutex_unlock(&session->mutex);
     return 0;
 }
@@ -119,7 +147,7 @@ int omnisocket_session_close(omnisocket_session_t *session) {
         session->closing = 1;
         session->client = NULL;
     }
-    session->stats.connected = 0;
+    omnisocket_session_mark_disconnected_locked(session);
     pthread_mutex_unlock(&session->mutex);
 
     if (client != NULL) {
@@ -158,6 +186,7 @@ int omnisocket_session_send(omnisocket_session_t *session, const char *to, const
     } else {
         session->stats.send_errors += 1;
     }
+    omnisocket_session_sync_client_state_locked(session, client);
     if (session->active_ops > 0) {
         session->active_ops -= 1;
     }
@@ -190,6 +219,7 @@ int omnisocket_session_recv(omnisocket_session_t *session, message_t *out_msg, i
     } else {
         session->stats.recv_errors += 1;
     }
+    omnisocket_session_sync_client_state_locked(session, client);
     if (session->active_ops > 0) {
         session->active_ops -= 1;
     }
@@ -228,6 +258,7 @@ int omnisocket_session_recv_into(
     } else {
         session->stats.recv_errors += 1;
     }
+    omnisocket_session_sync_client_state_locked(session, client);
     if (session->active_ops > 0) {
         session->active_ops -= 1;
     }
@@ -376,6 +407,8 @@ int omnisocket_udp_session_connect(
     }
     session->client = client;
     session->stats.connected = 1;
+    session->stats.registered = 1;
+    session->stats.last_server_error[0] = '\0';
     pthread_mutex_unlock(&session->mutex);
     return 0;
 }
@@ -398,6 +431,7 @@ int omnisocket_udp_session_close(omnisocket_udp_session_t *session) {
         session->client = NULL;
     }
     session->stats.connected = 0;
+    session->stats.registered = 0;
     pthread_mutex_unlock(&session->mutex);
 
     if (client != NULL) {

python/omnisocket/omnisocket_client.h

@@ -13,6 +13,8 @@ typedef struct omnisocket_session_stats {
     uint64_t recv_timeouts;
     uint64_t recv_errors;
     int connected;
+    int registered;
+    char last_server_error[256];
 } omnisocket_session_stats_t;
 
 typedef struct omnisocket_session_kcp_stats {