g3keymess: ignore more early connection errors

bytedance · Aug 23, 2023 · 891fe75 · 891fe75
1 parent e104e4c
commit 891fe75
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/g3keymess/src/serve/task/multiplex.rs b/g3keymess/src/serve/task/multiplex.rs
@@ -109,7 +109,13 @@ impl KeylessTask {
                             msg_count += 1;
                         }
                         Ok(false) => return Ok(()),
-                        Err(e) => return Err(ServerTaskError::ReadFailed(e)),
+                        Err(e) => {
+                            return if msg_count == 0 {
+                                Err(ServerTaskError::ConnectionClosedEarly)
+                            } else {
+                                Err(ServerTaskError::ReadFailed(e))
+                            };
+                        }
                     }
                 }
                 r = self.ctx.reload_notifier.recv() => {

diff --git a/g3keymess/src/serve/task/simplex.rs b/g3keymess/src/serve/task/simplex.rs
@@ -49,7 +49,11 @@ impl KeylessTask {
                         }
                         Ok(false) => break,
                         Err(e) => {
-                            self.log_task_err(ServerTaskError::ReadFailed(e));
+                            if msg_count == 0 {
+                                self.log_task_err(ServerTaskError::ConnectionClosedEarly);
+                            } else {
+                                self.log_task_err(ServerTaskError::ReadFailed(e));
+                            }
                             break;
                         }
                     }