fix: reconnect

yomorun · Nov 26, 2023 · 96d07ec · 96d07ec
1 parent e913b0b
commit 96d07ec
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/core/client.go b/core/client.go
@@ -221,13 +221,9 @@ func (c *Client) handleConn(ctx context.Context, conn frame.Conn) error {
 	for {
 		select {
 		case <-ctx.Done():
-			se := errors.New("yomo: parent context done")
-			conn.CloseWithError(se.Error())
-			return se
+			conn.CloseWithError("yomo: parent context done")
 		case <-c.ctx.Done():
-			se := context.Cause(c.ctx)
-			conn.CloseWithError(se.Error())
-			return se
+			conn.CloseWithError(context.Cause(c.ctx).Error())
 		case f := <-c.wrCh:
 			if err := conn.WriteFrame(f); err != nil {
 				return err

diff --git a/core/server.go b/core/server.go
@@ -223,7 +223,7 @@ func (s *Server) handleFrameConn(fconn frame.Conn, logger *slog.Logger) {
 		return
 	}
 
-	s.connHandler(conn, route) // s.handleConn(conn) with middlewares
+	s.connHandler(conn, route) // s.handleConnRoute(conn, route) with middlewares
 
 	if conn.ClientType() == ClientTypeStreamFunction {
 		_ = route.Remove(conn.ID())