def __init__(self, project_config, receiver_config): self._project = project_config self._receiver = receiver_config self._auth_info = service_pb2.ProjAuthInfo( project_name=self._project.name, source_domain=self._project.domain_name, target_domain=self._receiver.domain_name, auth_token=self._project.token) self._client = service_pb2_grpc.WebConsoleV2ServiceStub( _build_channel(self._receiver.grpc_spec.egress_url, self._receiver.grpc_spec.authority))
def check_connection(self): msg = service_pb2.CheckConnectionRequest( auth_info=service_pb2.ProjAuthInfo( project_name=self._project.project_name, sender_name=self._project.self_name, receiver_name=self._receiver.name, auth_token=self._receiver.sender_auth_token)) try: response = self._client.CheckConnection( request=msg, metadata=self._get_metadata()) return response.status except Exception as e: return common_pb2.Status(code=common_pb2.STATUS_UNKNOWN_ERROR, msg=repr(e))
def __init__(self, project_config, receiver_config): self._project = project_config self._receiver = receiver_config self._auth_info = service_pb2.ProjAuthInfo( project_name=self._project.name, target_domain=self._receiver.domain_name, auth_token=self._project.token) egress_url = 'fedlearner-stack-ingress-nginx-controller.default.svc:80' for variable in self._project.variables: if variable.name == 'EGRESS_URL': egress_url = variable.value break self._client = service_pb2_grpc.WebConsoleV2ServiceStub( _build_channel(egress_url, self._receiver.grpc_spec.authority))