def _restart_service(service): print('Restarting %s...' % service) if service == 'magmad': _restart_all() return chan = ServiceRegistry.get_rpc_channel('magmad', ServiceRegistry.LOCAL) client = MagmadStub(chan) client.RestartServices(RestartServicesRequest(services=[service]))
class MagmadServiceGrpc(MagmadServiceClient): """ Handle magmad actions by making service calls over gRPC. """ def __init__(self): self._magmad_stub = MagmadStub(get_rpc_channel("magmad")) def restart_services(self, services): request = RestartServicesRequest() for s in services: request.services.append(s) try: self._magmad_stub.RestartServices(request) except grpc.RpcError as error: err_code = error.exception().code() if err_code == grpc.StatusCode.FAILED_PRECONDITION: logging.info("Ignoring FAILED_PRECONDITION exception") else: raise