@VisibleForTesting void connectHelixManager() { try { this.jobHelixManager.connect(); this.jobHelixManager.getMessagingService() .registerMessageHandlerFactory(GobblinHelixConstants.SHUTDOWN_MESSAGE_TYPE, new ParticipantShutdownMessageHandlerFactory()); this.jobHelixManager.getMessagingService() .registerMessageHandlerFactory(Message.MessageType.USER_DEFINE_MSG.toString(), getUserDefinedMessageHandlerFactory()); if (this.taskDriverHelixManager.isPresent()) { this.taskDriverHelixManager.get().connect(); } } catch (Exception e) { logger.error("HelixManager failed to connect", e); throw Throwables.propagate(e); } }
@VisibleForTesting void connectHelixManager() { try { this.helixManager.connect(); this.helixManager.getMessagingService() .registerMessageHandlerFactory(GobblinHelixConstants.SHUTDOWN_MESSAGE_TYPE, new ParticipantShutdownMessageHandlerFactory()); this.helixManager.getMessagingService() .registerMessageHandlerFactory(Message.MessageType.USER_DEFINE_MSG.toString(), getUserDefinedMessageHandlerFactory()); } catch (Exception e) { logger.error("HelixManager failed to connect", e); throw Throwables.propagate(e); } }