From bf21b9f280571c01ac7f2c84ed30a7ed40647e48 Mon Sep 17 00:00:00 2001
From: hdt3213 <hdt3213@outlook.com>
Date: Sat, 12 Dec 2020 22:08:04 +0800
Subject: [PATCH] tiny bug fix

---
 src/tcp/echo.go   | 2 +-
 src/tcp/server.go | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/tcp/echo.go b/src/tcp/echo.go
index 586dcb5..4b581e9 100644
--- a/src/tcp/echo.go
+++ b/src/tcp/echo.go
@@ -55,7 +55,7 @@ func (h *EchoHandler)Handle(ctx context.Context, conn net.Conn) {
         if err != nil {
             if err == io.EOF {
                 logger.Info("connection close")
-                h.activeConn.Delete(conn)
+                h.activeConn.Delete(client)
             } else {
                 logger.Warn(err)
             }
diff --git a/src/tcp/server.go b/src/tcp/server.go
index 0b6f7b8..0f3dab6 100644
--- a/src/tcp/server.go
+++ b/src/tcp/server.go
@@ -68,11 +68,11 @@ func ListenAndServe(cfg *Config, handler tcp.Handler) {
         }
         // handle
         logger.Info("accept link")
+        waitDone.Add(1)
         go func() {
             defer func() {
                 waitDone.Done()
             }()
-            waitDone.Add(1)
             handler.Handle(ctx, conn)
         }()
     }