use defer in conn.Close

2013-10-06 10:54:16 -07:00 · 2013-10-06 10:54:16 -07:00 · 220ebb1f4f
commit 220ebb1f4f
parent e3a666a3f1
1 changed files with 2 additions and 3 deletions
--- a/grace.go
+++ b/grace.go
@ -58,9 +58,8 @@ type conn struct {
 }

 func (c conn) Close() error {
-	err := c.Conn.Close()
-	c.wg.Done()
-	return err
+	defer c.wg.Done()
+	return c.Conn.Close()
 }

 // Wraps an existing File listener to provide a graceful Close() process.