1
0
mirror of https://github.com/astaxie/beego.git synced 2025-07-11 22:01:01 +00:00

fix panic: sync: negative WaitGroup counter

This commit is contained in:
huwenbo
2017-06-13 15:19:51 +08:00
parent cab8458c1c
commit 55e6c15073
2 changed files with 13 additions and 2 deletions

View File

@ -3,11 +3,14 @@ package grace
import (
"errors"
"net"
"sync"
)
type graceConn struct {
net.Conn
server *Server
m sync.Mutex
closed bool
}
func (c graceConn) Close() (err error) {
@ -23,6 +26,14 @@ func (c graceConn) Close() (err error) {
}
}
}()
c.m.Lock()
if c.closed {
c.m.Unlock()
return
}
c.server.wg.Done()
c.closed = true
c.m.Unlock()
return c.Conn.Close()
}
}