feat: writeControl

Bisonai · Jul 29, 2024 · 4926a4a · 4926a4a
1 parent 73ba02b
commit 4926a4a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/node/pkg/dal/api/hub.go b/node/pkg/dal/api/hub.go
@@ -2,6 +2,7 @@ package api
 
 import (
 	"context"
+	"time"
 
 	"bisonai.com/orakl/node/pkg/common/types"
 	"bisonai.com/orakl/node/pkg/dal/collector"
@@ -42,8 +43,13 @@ func (c *Hub) Start(ctx context.Context, collector *collector.Collector) {
 			case conn := <-c.unregister:
 				c.mu.Lock()
 				delete(c.clients, conn)
-				conn.Close()
 				c.mu.Unlock()
+				conn.WriteControl(
+					websocket.CloseMessage,
+					websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""),
+					time.Now().Add(time.Second),
+				)
+				conn.Close()
 			}
 		}
 	}()