new SegmentMessageHandlerFactory(fetcherAndLoader, _serverInstance.getInstanceDataManager(), serverMetrics); _helixManager.getMessagingService() .registerMessageHandlerFactory(Message.MessageType.USER_DEFINE_MSG.toString(), messageHandlerFactory);
@Override public List<String> getMessageTypes() { return ImmutableList.of(Message.MessageType.PARTICIPANT_ERROR_REPORT.name()); }
@Override public HelixTaskResult handleMessage() throws InterruptedException { String type = _message.getMsgType(); HelixTaskResult result = new HelixTaskResult(); if (!type.equals(MessageType.CONTROLLER_MSG.name())) { throw new HelixException("Unexpected msg type for message " + _message.getMsgId() + " type:" + _message.getMsgType()); } result.getTaskResultMap().put("ControllerResult", "msg " + _message.getMsgId() + " from " + _message.getMsgSrc() + " processed"); result.setSuccess(true); return result; }
@Override public List<String> getMessageTypes() { return ImmutableList.of(MessageType.CONTROLLER_MSG.name()); }
@Override public String getMessageType() { return Message.MessageType.USER_DEFINE_MSG.toString(); }
@Override public HelixTaskResult handleMessage() throws InterruptedException { String type = _message.getMsgType(); HelixTaskResult result = new HelixTaskResult(); if (!type.equals(MessageType.CONTROLLER_MSG.name())) { throw new HelixException("Unexpected msg type for message " + _message.getMsgId() + " type:" + _message.getMsgType()); } result.getTaskResultMap().put("ControllerResult", "msg " + _message.getMsgId() + " from " + _message.getMsgSrc() + " processed"); result.setSuccess(true); return result; }
@Override public String getMessageType() { return Message.MessageType.PARTICIPANT_ERROR_REPORT.name(); }
@Override public String getMessageType() { return MessageType.STATE_TRANSITION.name(); }
@Override public HelixTaskResult handleMessage() throws InterruptedException { String messageSubType = this._message.getMsgSubType(); if (messageSubType.equalsIgnoreCase(HelixMessageSubTypes.TOKEN_FILE_UPDATED.toString())) { LOGGER.info("Handling message " + HelixMessageSubTypes.TOKEN_FILE_UPDATED.toString()); eventBus.post(new DelegationTokenUpdatedEvent()); HelixTaskResult helixTaskResult = new HelixTaskResult(); helixTaskResult.setSuccess(true); return helixTaskResult; } throw new IllegalArgumentException(String.format("Unknown %s message subtype: %s", Message.MessageType.USER_DEFINE_MSG.toString(), messageSubType)); }
@Override public String getMessageType() { return Message.MessageType.USER_DEFINE_MSG.toString(); }
@Override public List<String> getMessageTypes() { return ImmutableList.of(MessageType.CONTROLLER_MSG.name()); }
@Override public List<String> getMessageTypes() { return ImmutableList.of(MessageType.TASK_REPLY.name()); }
@Override public String getMessageType() { return MessageType.CONTROLLER_MSG.name(); }
@Test public void testMultipleMessageTypeRegisteration() { HelixManager manager = new MockManager(); MockDefaultMessagingService svc = new MockDefaultMessagingService(manager); TestStateTransitionHandlerFactory factory = new TestStateTransitionHandlerFactory(); svc.registerMessageHandlerFactory(factory.getMessageTypes(), factory); Assert.assertTrue( svc.getMessageHandlerFactoryMap().containsKey(Message.MessageType.STATE_TRANSITION.name())); Assert.assertTrue(svc.getMessageHandlerFactoryMap() .containsKey(Message.MessageType.STATE_TRANSITION_CANCELLATION.name())); Assert.assertTrue( svc.getMessageHandlerFactoryMap().containsKey(Message.MessageType.CONTROLLER_MSG.name())); } }
@Override public HelixTaskResult handleMessage() throws InterruptedException { String messageSubType = this._message.getMsgSubType(); if (messageSubType.equalsIgnoreCase(org.apache.gobblin.cluster.HelixMessageSubTypes.TOKEN_FILE_UPDATED.toString())) { LOGGER.info("Handling message " + org.apache.gobblin.cluster.HelixMessageSubTypes.TOKEN_FILE_UPDATED.toString()); eventBus.post(new DelegationTokenUpdatedEvent()); HelixTaskResult helixTaskResult = new HelixTaskResult(); helixTaskResult.setSuccess(true); return helixTaskResult; } throw new IllegalArgumentException(String .format("Unknown %s message subtype: %s", Message.MessageType.USER_DEFINE_MSG.toString(), messageSubType)); }
@Override public String getMessageType() { return Message.MessageType.USER_DEFINE_MSG.toString(); }
@Override public String getMessageType() { return Message.MessageType.USER_DEFINE_MSG.toString(); }
@Override public List<String> getMessageTypes() { return ImmutableList.of(Message.MessageType.PARTICIPANT_ERROR_REPORT.name()); }
@Override public List<String> getMessageTypes() { return ImmutableList.of(MessageType.USER_DEFINE_MSG.name()); }