defer the Done() call

2013-11-03 13:47:47 -05:00 · 2013-11-03 13:47:47 -05:00 · 43ccb346c3
commit 43ccb346c3
parent c846d177f2
1 changed files with 1 additions and 1 deletions
--- a/grace.go
+++ b/grace.go
@ -64,7 +64,7 @@ type conn struct {
 }

 func (c *conn) Close() error {
-	c.once.Do(func() { c.wg.Done() })
+	c.once.Do(func() { defer c.wg.Done() })
 	return c.Conn.Close()
 }