client close nicely

src/agent/client.go

@@ -49,6 +49,7 @@ type client struct {
 	wantResponseId uint64
 	wantResponse   chan []byte
 	lock           sync.Mutex
+	once           sync.Once
 }
 
 func newClient(conn *websocket.Conn, a *Agent) *client {
@@ -142,9 +143,12 @@ func (c *client) writePump() {
 }
 
 func (c *client) Close() {
+	once.Do(func(){
 		c.fnCancel()
 		c.agent.removeClient(c)
 		c.conn.Close()
+		log.Info("client close", "remote", c.conn.RemoteAddr().String(), "connected", c.agent.clients.Size())
+	})
 }
 
 func (c *client) WriteJson(data interface{}, block bool) error {

src/workflow/task.go

@@ -47,8 +47,8 @@ func (task *Task) SendExecFail(errMsg string) {
 
 func (task *Task) do() {
 	defer func() {
-		task.pool.remove(task.client)
 		<-task.pool.sem
+		task.pool.remove(task.client)
 		if err := recover(); err != nil {
 			log.Error("[Task] do panic", "err", err)
 		}