From ff3daaed0f4bed27bae650214274c30359029336 Mon Sep 17 00:00:00 2001
From: esrrhs <esrrhs@163.com>
Date: Sat, 26 Oct 2019 09:39:36 +0800
Subject: [PATCH] add

---
 server.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/server.go b/server.go
index 4681464..e6a3b2e 100644
--- a/server.go
+++ b/server.go
@@ -285,7 +285,12 @@ func (p *Server) Recv(conn *ServerConn, id string, src *net.IPAddr) {
 
 func (p *Server) Close(conn *ServerConn) {
 	if p.localConnMap[conn.id] != nil {
-		conn.conn.Close()
+		if conn.conn != nil {
+			conn.conn.Close()
+		}
+		if conn.tcpconn != nil {
+			conn.tcpconn.Close()
+		}
 		delete(p.localConnMap, conn.id)
 	}
 }