fix typo

gansidui · Oct 9, 2015 · bc20b14 · ronalfei · Oct 30, 2015 · ronalfei
1 parent 4745093
commit bc20b14
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/conn.go b/conn.go
@@ -121,17 +121,18 @@ func (c *Conn) AsyncWritePacket(p Packet, timeout time.Duration) (err error) {
 
 // Do it
 func (c *Conn) Do() {
+	defer c.srv.waitGroup.Done()
 	if !c.srv.callback.OnConnect(c) {
 		return
 	}
 
+	c.srv.waitGroup.Add(3)
 	go c.handleLoop()
 	go c.readLoop()
 	go c.writeLoop()
 }
 
 func (c *Conn) readLoop() {
-	c.srv.waitGroup.Add(1)
 	defer func() {
 		recover()
 		c.Close()
@@ -159,7 +160,6 @@ func (c *Conn) readLoop() {
 }
 
 func (c *Conn) writeLoop() {
-	c.srv.waitGroup.Add(1)
 	defer func() {
 		recover()
 		c.Close()
@@ -186,7 +186,6 @@ func (c *Conn) writeLoop() {
 }
 
 func (c *Conn) handleLoop() {
-	c.srv.waitGroup.Add(1)
 	defer func() {
 		recover()
 		c.Close()

diff --git a/server.go b/server.go
@@ -53,6 +53,7 @@ func (s *Server) Start(listener *net.TCPListener, acceptTimeout time.Duration) {
 			continue
 		}
 
+		s.waitGroup.Add(1)
 		go newConn(conn, s).Do()
 	}
 }