lock closing new shards in case ring was closed in the meantime

use the internal logger as it is used elsewhere

Signed-off-by: Sandor Szücs <sandor.szuecs@zalando.de>

Author: szuecs

ring.go

@@ -5,7 +5,6 @@ import (
 	"crypto/tls"
 	"errors"
 	"fmt"
-	"log"
 	"net"
 	"strconv"
 	"sync"
@@ -245,7 +244,7 @@ func (c *ringShards) SetAddrs(addrs map[string]string) {
 			defer func() {
 				err := shard.Client.Close()
 				if err != nil {
-					log.Printf("Failed to close ring shard client %s: %v", shard.addr, err)
+					internal.Logger.Printf(context.Background(), "Failed to close ring shard client %s %s: %v", k, shard.addr, err)
 				}
 			}()
 		}
@@ -266,14 +265,12 @@ func (c *ringShards) SetAddrs(addrs map[string]string) {
 	if !c.closed {
 		c.shards = shards
 		c.list = list
-	}
-	c.mu.Unlock()
-
-	if c.closed {
+	} else {
 		for _, shard := range shards {
 			shard.Client.Close()
 		}
 	}
+	c.mu.Unlock()
 
 	c.rebalance()
 }