add lock on close

6 years ago · 503997660b
parent 3409e5e702
commit 503997660b
1 changed files with 3 additions and 0 deletions
--- a/conn.go
+++ b/conn.go
@ -317,6 +317,9 @@ func (c *ServerConn) serve(s *ServerStream) {

 // Close close the server connection
 func (c *ServerConn) Close() error {
+	c.lock.Lock()
+	defer c.lock.Unlock()
+
 	if c.closed {
 		return nil
 	}