@Override public void onError(final Throwable e) { LOG.warn("RPC request call failed with msg.id={}, msg.listenerId={}, msg.type={}, e.cause={}, e.message={}", message.getId(), message.getListenerId(), message.getType(), e.getCause(), e.getMessage()); completableFuture.completeExceptionally(e); }
@Override public void onError(final Throwable e) { LOG.warn("RPC request call failed with msg.id={}, msg.listenerId={}, msg.type={}, e.cause={}, e.message={}", message.getId(), message.getListenerId(), message.getType(), e.getCause(), e.getMessage()); completableFuture.completeExceptionally(e); }
@Override public <U> void reply(final U replyMessage) { final ControlMessage.Message message = (ControlMessage.Message) replyMessage; LOG.debug("[REQUEST] response msg.id={}, msg.listenerId={}, msg.type={}", message.getId(), message.getListenerId(), message.getType()); responseObserver.onNext((ControlMessage.Message) replyMessage); responseObserver.onCompleted(); } }
@Override public void onMessage(final ControlMessage.Message message) { switch (message.getType()) { case ScheduleTask: final ControlMessage.ScheduleTaskMsg scheduleTaskMsg = message.getScheduleTaskMsg(); final Task task = SerializationUtils.deserialize(scheduleTaskMsg.getTask().toByteArray()); onTaskReceived(task); break; case RequestMetricFlush: metricMessageSender.flush(); break; default: throw new IllegalMessageException( new Exception("This message should not be received by an executor :" + message.getType())); } }
@Override public void onMessage(final ControlMessage.Message message) { switch (message.getType()) { case ScheduleTask: final ControlMessage.ScheduleTaskMsg scheduleTaskMsg = message.getScheduleTaskMsg(); final Task task = SerializationUtils.deserialize(scheduleTaskMsg.getTask().toByteArray()); onTaskReceived(task); break; case RequestMetricFlush: metricMessageSender.flush(); break; default: throw new IllegalMessageException( new Exception("This message should not be received by an executor :" + message.getType())); } }
/** * Issue {@link org.apache.nemo.runtime.common.message.MessageSender#send(Object)} rpc call. * * @param message a message to send */ void send(final ControlMessage.Message message) { LOG.debug("[SEND] request msg.id={}, msg.listenerId={}, msg.type={}", message.getId(), message.getListenerId(), message.getType()); try { blockingStub.send(message); } catch (final StatusRuntimeException e) { LOG.warn("RPC send call failed with msg.id={}, msg.listenerId={}, msg.type={}, e.cause={}, e.message={}", message.getId(), message.getListenerId(), message.getType(), e.getCause(), e.getMessage()); } }
/** * Issue {@link org.apache.nemo.runtime.common.message.MessageSender#send(Object)} rpc call. * * @param message a message to send */ void send(final ControlMessage.Message message) { LOG.debug("[SEND] request msg.id={}, msg.listenerId={}, msg.type={}", message.getId(), message.getListenerId(), message.getType()); try { blockingStub.send(message); } catch (final StatusRuntimeException e) { LOG.warn("RPC send call failed with msg.id={}, msg.listenerId={}, msg.type={}, e.cause={}, e.message={}", message.getId(), message.getListenerId(), message.getType(), e.getCause(), e.getMessage()); } }
@Override public void onMessageWithContext(final ControlMessage.Message message, final MessageContext messageContext) { switch (message.getType()) { case RequestBlockLocation: registerLocationRequest(message, messageContext); break; default: throw new IllegalMessageException( new Exception("This message should not be received by " + BlockManagerMaster.class.getName() + ":" + message.getType())); } } }
@Override public void onNext(final ControlMessage.Message responseMessage) { LOG.debug("[REQUEST] response msg.id={}, msg.listenerId={}, msg.type={}", responseMessage.getId(), responseMessage.getListenerId(), responseMessage.getType()); completableFuture.complete(responseMessage); }
private long getRequestId(final ControlMessage.Message controlMessage) { switch (controlMessage.getType()) { case BlockLocationInfo: return controlMessage.getBlockLocationInfoMsg().getRequestId(); case InMasterBroadcastVariable: return controlMessage.getBroadcastVariableMsg().getRequestId(); case PipeLocInfo: return controlMessage.getPipeLocInfoMsg().getRequestId(); default: throw new IllegalArgumentException(controlMessage.toString()); } } }
@Override public <U> void reply(final U replyMessage) { final ControlMessage.Message message = (ControlMessage.Message) replyMessage; LOG.debug("[REQUEST] response msg.id={}, msg.listenerId={}, msg.type={}", message.getId(), message.getListenerId(), message.getType()); responseObserver.onNext((ControlMessage.Message) replyMessage); responseObserver.onCompleted(); } }
@Override public void onMessageWithContext(final ControlMessage.Message message, final MessageContext messageContext) { switch (message.getType()) { case RequestBlockLocation: registerLocationRequest(message, messageContext); break; default: throw new IllegalMessageException( new Exception("This message should not be received by " + BlockManagerMaster.class.getName() + ":" + message.getType())); } } }
private String getExecutorId(final ControlMessage.Message controlMessage) { switch (controlMessage.getType()) { case RequestBlockLocation: return controlMessage.getRequestBlockLocationMsg().getExecutorId(); case RequestBroadcastVariable: return controlMessage.getRequestbroadcastVariableMsg().getExecutorId(); case RequestPipeLoc: return controlMessage.getRequestPipeLocMsg().getExecutorId(); default: throw new IllegalArgumentException(controlMessage.toString()); } }
private long getRequestId(final ControlMessage.Message controlMessage) { switch (controlMessage.getType()) { case BlockLocationInfo: return controlMessage.getBlockLocationInfoMsg().getRequestId(); case InMasterBroadcastVariable: return controlMessage.getBroadcastVariableMsg().getRequestId(); case PipeLocInfo: return controlMessage.getPipeLocInfoMsg().getRequestId(); default: throw new IllegalArgumentException(controlMessage.toString()); } } }
@Override public void onCompleted() { LOG.debug("[REQUEST] completed. msg.id={}, msg.listenerId={}, msg.type={}", message.getId(), message.getListenerId(), message.getType()); } });
private String getExecutorId(final ControlMessage.Message controlMessage) { switch (controlMessage.getType()) { case RequestBlockLocation: return controlMessage.getRequestBlockLocationMsg().getExecutorId(); case RequestBroadcastVariable: return controlMessage.getRequestbroadcastVariableMsg().getExecutorId(); case RequestPipeLoc: return controlMessage.getRequestPipeLocMsg().getExecutorId(); default: throw new IllegalArgumentException(controlMessage.toString()); } }
@Override public void onCompleted() { LOG.debug("[REQUEST] completed. msg.id={}, msg.listenerId={}, msg.type={}", message.getId(), message.getListenerId(), message.getType()); } });
@Override public void onNext(final ControlMessage.Message responseMessage) { LOG.debug("[REQUEST] response msg.id={}, msg.listenerId={}, msg.type={}", responseMessage.getId(), responseMessage.getListenerId(), responseMessage.getType()); completableFuture.complete(responseMessage); }