@@ -40,6 +40,7 @@ func (filterLocalDC) String() string {
4040func PreferLocalDC (balancer * balancerConfig.Config ) * balancerConfig.Config {
4141 balancer .Filter = filterLocalDC {}
4242 balancer .DetectLocalDC = true
43+
4344 return balancer
4445}
4546
@@ -49,6 +50,7 @@ func PreferLocalDC(balancer *balancerConfig.Config) *balancerConfig.Config {
4950func PreferLocalDCWithFallBack (balancer * balancerConfig.Config ) * balancerConfig.Config {
5051 balancer = PreferLocalDC (balancer )
5152 balancer .AllowFallback = true
53+
5254 return balancer
5355}
5456
@@ -61,6 +63,7 @@ func (locations filterLocations) Allow(_ balancerConfig.Info, c conn.Conn) bool
6163 return true
6264 }
6365 }
66+
6467 return false
6568}
6669
@@ -91,6 +94,7 @@ func PreferLocations(balancer *balancerConfig.Config, locations ...string) *bala
9194 }
9295 sort .Strings (locations )
9396 balancer .Filter = filterLocations (locations )
97+
9498 return balancer
9599}
96100
@@ -100,6 +104,7 @@ func PreferLocations(balancer *balancerConfig.Config, locations ...string) *bala
100104func PreferLocationsWithFallback (balancer * balancerConfig.Config , locations ... string ) * balancerConfig.Config {
101105 balancer = PreferLocations (balancer , locations ... )
102106 balancer .AllowFallback = true
107+
103108 return balancer
104109}
105110
@@ -129,6 +134,7 @@ func Prefer(balancer *balancerConfig.Config, filter func(endpoint Endpoint) bool
129134 balancer .Filter = filterFunc (func (_ balancerConfig.Info , c conn.Conn ) bool {
130135 return filter (c .Endpoint ())
131136 })
137+
132138 return balancer
133139}
134140
@@ -138,6 +144,7 @@ func Prefer(balancer *balancerConfig.Config, filter func(endpoint Endpoint) bool
138144func PreferWithFallback (balancer * balancerConfig.Config , filter func (endpoint Endpoint ) bool ) * balancerConfig.Config {
139145 balancer = Prefer (balancer , filter )
140146 balancer .AllowFallback = true
147+
141148 return balancer
142149}
143150
0 commit comments