fix: 节点断开后列表还显示

opentdp · Nov 22, 2023 · c3047fb · c3047fb
1 parent 512b1f2
commit c3047fb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/module/workhub/main.go b/module/workhub/main.go
@@ -57,11 +57,11 @@ func Connect(ws *websocket.Conn, rq *ConnectParam) error {
 
 func Receiver(worker *Worker) error {
 
+	defer DeleteWorker(worker)
+
 	recv := &RecvPod{worker}
 	resp := &RespPod{worker}
 
-	defer delete(workerPool, worker.WorkerId)
-
 	for {
 		var rq *SocketData
 

diff --git a/module/workhub/poolman.go b/module/workhub/poolman.go
@@ -2,13 +2,17 @@ package workhub
 
 import (
 	"time"
+
+	"github.com/opentdp/go-helper/logman"
 )
 
 var workerResp = map[uint]any{}
 var workerPool = map[string]*Worker{}
 
 func DeleteWorker(Worker *Worker) {
 
+	logman.Info("Worker disconnect", "Id", Worker.WorkerId)
+
 	if Worker.WorkerId != "" {
 		delete(workerPool, Worker.WorkerId)
 	}