From ebf98f3a153d5526520e848ee14a0fa9335eadd7 Mon Sep 17 00:00:00 2001
From: hzy <h1260071766@126.com>
Date: Thu, 24 Jun 2021 22:03:42 +0800
Subject: [PATCH] Fix bugs

---
 TCPMode/demoWeb.go  | 3 ---
 UDPMode/listener.go | 4 ++--
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/TCPMode/demoWeb.go b/TCPMode/demoWeb.go
index 3e99213..0dbdb26 100644
--- a/TCPMode/demoWeb.go
+++ b/TCPMode/demoWeb.go
@@ -23,9 +23,6 @@ func (s *demoWeb) server_handle(conn net.Conn) {
 	s.worker.Add(+1)
 	defer s.worker.Done()
 
-	closing := make(chan int)
-	defer close(closing)
-
 	defer conn.Close()
 
 	log.Info("new connection", conn.RemoteAddr().String())
diff --git a/UDPMode/listener.go b/UDPMode/listener.go
index e44b8d3..5b861f3 100644
--- a/UDPMode/listener.go
+++ b/UDPMode/listener.go
@@ -57,11 +57,11 @@ type listenerConn struct {
 }
 
 func (s *listener) server_main() {
+	defer s.Close()
+
 	s.worker.Add(1)
 	defer s.worker.Done()
 
-	defer s.Close()
-
 	for {
 		buf := make([]byte, bufSize)
 		n, addr, err := s.server.ReadFrom(buf)