Skip to content

Commit 5da2731

Browse files
authored
balancer/weightedtarget: stop forwarding UpdateSubConnState calls (#6525)
1 parent 182b0ad commit 5da2731

File tree

2 files changed

+13
-12
lines changed

2 files changed

+13
-12
lines changed

balancer/weightedtarget/weightedtarget.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ func (b *weightedTargetBalancer) ResolverError(err error) {
163163
}
164164

165165
func (b *weightedTargetBalancer) UpdateSubConnState(sc balancer.SubConn, state balancer.SubConnState) {
166-
b.bg.UpdateSubConnState(sc, state)
166+
b.logger.Errorf("UpdateSubConnState(%v, %+v) called unexpectedly", sc, state)
167167
}
168168

169169
func (b *weightedTargetBalancer) Close() {

balancer/weightedtarget/weightedtarget_test.go

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1211,23 +1211,24 @@ var errTestInitIdle = fmt.Errorf("init Idle balancer error 0")
12111211
func init() {
12121212
stub.Register(initIdleBalancerName, stub.BalancerFuncs{
12131213
UpdateClientConnState: func(bd *stub.BalancerData, opts balancer.ClientConnState) error {
1214-
sc, err := bd.ClientConn.NewSubConn(opts.ResolverState.Addresses, balancer.NewSubConnOptions{})
1214+
sc, err := bd.ClientConn.NewSubConn(opts.ResolverState.Addresses, balancer.NewSubConnOptions{
1215+
StateListener: func(state balancer.SubConnState) {
1216+
err := fmt.Errorf("wrong picker error")
1217+
if state.ConnectivityState == connectivity.Idle {
1218+
err = errTestInitIdle
1219+
}
1220+
bd.ClientConn.UpdateState(balancer.State{
1221+
ConnectivityState: state.ConnectivityState,
1222+
Picker: &testutils.TestConstPicker{Err: err},
1223+
})
1224+
},
1225+
})
12151226
if err != nil {
12161227
return err
12171228
}
12181229
sc.Connect()
12191230
return nil
12201231
},
1221-
UpdateSubConnState: func(bd *stub.BalancerData, sc balancer.SubConn, state balancer.SubConnState) {
1222-
err := fmt.Errorf("wrong picker error")
1223-
if state.ConnectivityState == connectivity.Idle {
1224-
err = errTestInitIdle
1225-
}
1226-
bd.ClientConn.UpdateState(balancer.State{
1227-
ConnectivityState: state.ConnectivityState,
1228-
Picker: &testutils.TestConstPicker{Err: err},
1229-
})
1230-
},
12311232
})
12321233
}
12331234

0 commit comments

Comments
 (0)