@ -1358,6 +1358,47 @@ ServerAddressList ExtractBalancerAddresses(const ChannelArgs& args) {
ChannelArgs BuildBalancerChannelArgs (
FakeResolverResponseGenerator * response_generator ,
const ChannelArgs & args ) {
ChannelArgs grpclb_channel_args ;
const grpc_channel_args * lb_channel_specific_args =
args . GetPointer < grpc_channel_args > (
GRPC_ARG_EXPERIMENTAL_GRPCLB_CHANNEL_ARGS ) ;
if ( lb_channel_specific_args ! = nullptr ) {
grpclb_channel_args = ChannelArgs : : FromC ( lb_channel_specific_args ) ;
} else {
// Set grpclb_channel_args based on the parent channel's channel args.
grpclb_channel_args =
args
// LB policy name, since we want to use the default (pick_first) in
// the LB channel.
. Remove ( GRPC_ARG_LB_POLICY_NAME )
// Strip out the service config, since we don't want the LB policy
// config specified for the parent channel to affect the LB channel.
. Remove ( GRPC_ARG_SERVICE_CONFIG )
// The channel arg for the server URI, since that will be different
// for the LB channel than for the parent channel. The client
// channel factory will re-add this arg with the right value.
. Remove ( GRPC_ARG_SERVER_URI )
// The fake resolver response generator, because we are replacing it
// with the one from the grpclb policy, used to propagate updates to
// the LB channel.
. Remove ( GRPC_ARG_FAKE_RESOLVER_RESPONSE_GENERATOR )
// The LB channel should use the authority indicated by the target
// authority table (see \a ModifyGrpclbBalancerChannelArgs),
// as opposed to the authority from the parent channel.
. Remove ( GRPC_ARG_DEFAULT_AUTHORITY )
// Just as for \a GRPC_ARG_DEFAULT_AUTHORITY, the LB channel should
// be treated as a stand-alone channel and not inherit this argument
// from the args of the parent channel.
. Remove ( GRPC_SSL_TARGET_NAME_OVERRIDE_ARG )
// Don't want to pass down channelz node from parent; the balancer
// channel will get its own.
. Remove ( GRPC_ARG_CHANNELZ_CHANNEL_NODE )
// Remove the channel args for channel credentials and replace it
// with a version that does not contain call credentials. The
// loadbalancer is not necessarily trusted to handle bearer token
// credentials.
. Remove ( GRPC_ARG_CHANNEL_CREDENTIALS ) ;
}
// Create channel args for channel credentials that does not contain bearer
// token credentials.
auto * channel_credentials = args . GetObject < grpc_channel_credentials > ( ) ;
@ -1365,36 +1406,7 @@ ChannelArgs BuildBalancerChannelArgs(
RefCountedPtr < grpc_channel_credentials > creds_sans_call_creds =
channel_credentials - > duplicate_without_call_credentials ( ) ;
GPR_ASSERT ( creds_sans_call_creds ! = nullptr ) ;
return args
// LB policy name, since we want to use the default (pick_first) in
// the LB channel.
. Remove ( GRPC_ARG_LB_POLICY_NAME )
// Strip out the service config, since we don't want the LB policy
// config specified for the parent channel to affect the LB channel.
. Remove ( GRPC_ARG_SERVICE_CONFIG )
// The channel arg for the server URI, since that will be different for
// the LB channel than for the parent channel. The client channel
// factory will re-add this arg with the right value.
. Remove ( GRPC_ARG_SERVER_URI )
// The fake resolver response generator, because we are replacing it
// with the one from the grpclb policy, used to propagate updates to
// the LB channel.
. Remove ( GRPC_ARG_FAKE_RESOLVER_RESPONSE_GENERATOR )
// The LB channel should use the authority indicated by the target
// authority table (see \a ModifyGrpclbBalancerChannelArgs),
// as opposed to the authority from the parent channel.
. Remove ( GRPC_ARG_DEFAULT_AUTHORITY )
// Just as for \a GRPC_ARG_DEFAULT_AUTHORITY, the LB channel should be
// treated as a stand-alone channel and not inherit this argument from the
// args of the parent channel.
. Remove ( GRPC_SSL_TARGET_NAME_OVERRIDE_ARG )
// Don't want to pass down channelz node from parent; the balancer
// channel will get its own.
. Remove ( GRPC_ARG_CHANNELZ_CHANNEL_NODE )
// Remove the channel args for channel credentials and replace it
// with a version that does not contain call credentials. The loadbalancer
// is not necessarily trusted to handle bearer token credentials.
. Remove ( GRPC_ARG_CHANNEL_CREDENTIALS )
return grpclb_channel_args
// A channel arg indicating the target is a grpclb load balancer.
. Set ( GRPC_ARG_ADDRESS_IS_GRPCLB_LOAD_BALANCER , 1 )
// Tells channelz that this is an internal channel.
Vignesh Babu
3 years ago
committed by
GitHub