discoverd/client: lock it up

flynn · Jan 6, 2014 · a090611 · a090611
1 parent 91480cc
commit a090611
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/client.go b/client.go
@@ -430,6 +430,7 @@ func (c *Client) UnregisterAll() error {
 }
 
 var defaultClient *Client
+var defaultEnsureLock = &sync.Mutex{}
 
 func Connect(addr string) (err error) {
 	if addr == "" {
@@ -441,6 +442,8 @@ func Connect(addr string) (err error) {
 }
 
 func ensureDefaultConnected() error {
+	defaultEnsureLock.Lock()
+	defer defaultEnsureLock.Unlock()
 	if defaultClient == nil {
 		return Connect("")
 	}