use defer for wg.Done to ensure its called

2013-10-06 12:19:25 -07:00 · 2013-10-06 12:19:25 -07:00 · f5295e5b81
commit f5295e5b81
parent ae3f6c71a7
1 changed files with 1 additions and 1 deletions
--- a/grace.go
+++ b/grace.go
@ -148,11 +148,11 @@ func Wait(listeners []Listener) (err error) {
 			wg.Add(len(listeners))
 			for _, l := range listeners {
 				go func(l Listener) {
+					defer wg.Done()
 					cErr := l.Close()
 					if cErr != nil {
 						err = cErr
 					}
-					wg.Done()
 				}(l)
 			}
 			wg.Wait()