def _get_channel(self,
config: metadata_store_pb2.MetadataStoreClientConfig,
target: Text):
"""Configures the channel, which could be secure or insecure.
It returns a channel that can be specified to be secure or insecure,
depending on whether ssl_config is specified in the config.
Args:
config: metadata_store_pb2.MetadataStoreClientConfig.
target: target host with port.
Returns:
an initialized gRPC channel.
"""
if not config.HasField('ssl_config'):
return grpc.insecure_channel(target)
root_certificates = None
private_key = None
certificate_chain = None
if config.ssl_config.HasField('custom_ca'):
root_certificates = config.ssl_config.custom_ca
if config.ssl_config.HasField('client_key'):
private_key = config.ssl_config.client_key
if config.ssl_config.HasField('server_cert'):
certificate_chain = config.ssl_config.server_cert
credentials = grpc.ssl_channel_credentials(root_certificates,
private_key,
certificate_chain)
return grpc.secure_channel(target, credentials)
def _get_channel(self,
config: metadata_store_pb2.MetadataStoreClientConfig):
"""Configures the channel, which could be secure or insecure.
It returns a channel that can be specified to be secure or insecure,
depending on whether ssl_config is specified in the config.
Args:
config: metadata_store_pb2.MetadataStoreClientConfig.
Returns:
an initialized gRPC channel.
"""
target = ':'.join([config.host, str(config.port)])
if not config.HasField('ssl_config'):
return grpc.insecure_channel(target)
root_certificates = None
private_key = None
certificate_chain = None
if config.ssl_config.HasField('custom_ca'):
root_certificates = bytes(
str(config.ssl_config.custom_ca).encode('ascii'))
if config.ssl_config.HasField('client_key'):
private_key = bytes(
str(config.ssl_config.client_key).encode('ascii'))
if config.ssl_config.HasField('server_cert'):
certificate_chain = bytes(
str(config.ssl_config.server_cert).encode('ascii'))
credentials = grpc.ssl_channel_credentials(root_certificates,
private_key,
certificate_chain)
return grpc.secure_channel(target, credentials)
def _get_channel(self,
config: metadata_store_pb2.MetadataStoreClientConfig):
"""Configures the channel, which could be secure or insecure.
It returns a channel that can be specified to be secure or insecure,
depending on whether ssl_config is specified in the config.
Args:
config: metadata_store_pb2.MetadataStoreClientConfig.
Returns:
an initialized gRPC channel.
"""
target = ':'.join([config.host, str(config.port)])
if config.HasField('client_timeout_sec'):
self._grpc_timeout_sec = config.client_timeout_sec
options = None
if (config.HasField('channel_arguments')
and config.channel_arguments.HasField(
'max_receive_message_length')):
options = [('grpc.max_receive_message_length',
config.channel_arguments.max_receive_message_length)]
if not config.HasField('ssl_config'):
return grpc.insecure_channel(target, options=options)
root_certificates = None
private_key = None
certificate_chain = None
if config.ssl_config.HasField('custom_ca'):
root_certificates = bytes(
str(config.ssl_config.custom_ca).encode('ascii'))
if config.ssl_config.HasField('client_key'):
private_key = bytes(
str(config.ssl_config.client_key).encode('ascii'))
if config.ssl_config.HasField('server_cert'):
certificate_chain = bytes(
str(config.ssl_config.server_cert).encode('ascii'))
credentials = grpc.ssl_channel_credentials(root_certificates,
private_key,
certificate_chain)
return grpc.secure_channel(target, credentials, options=options)