fix: serialize NetworkTransport data race on ServerAddr()

hashicorp · Apr 3, 2024 · c6584b3 · c6584b3
1 parent cc2bb08
commit c6584b3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/net_transport.go b/net_transport.go
@@ -91,6 +91,7 @@ type NetworkTransport struct {
 	maxPool     int
 	maxInFlight int
 
+	serverAddressLock     sync.RWMutex
 	serverAddressProvider ServerAddressProvider
 
 	shutdown     bool
@@ -384,6 +385,8 @@ func (n *NetworkTransport) getConnFromAddressProvider(id ServerID, target Server
 }
 
 func (n *NetworkTransport) getProviderAddressOrFallback(id ServerID, target ServerAddress) ServerAddress {
+	n.serverAddressLock.RLock()
+	defer n.serverAddressLock.RUnlock()
 	if n.serverAddressProvider != nil {
 		serverAddressOverride, err := n.serverAddressProvider.ServerAddr(id)
 		if err != nil {