/** * Handles message from client. * @param message message to process */ private void handleMessage(final ControlMessage.ClientToDriverMessage message) { final ControlMessage.ClientToDriverMessageType type = message.getType(); final EventHandler<ControlMessage.ClientToDriverMessage> handler = handlers.get(type); if (handler == null) { throw new RuntimeException(String.format("Handler for message type %s not registered", type)); } else { handler.onNext(message); } }
serializedDAG = Base64.getEncoder().encodeToString(SerializationUtils.serialize(dag)); jobDoneLatch = new CountDownLatch(1); driverRPCServer.send(ControlMessage.ClientToDriverMessage.newBuilder() .setType(ControlMessage.ClientToDriverMessageType.LaunchDAG) .setLaunchDAG(ControlMessage.LaunchDAGMessage.newBuilder()
public org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage buildPartial() { org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage result = new org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage(this); int from_bitField0_ = bitField0_; int to_bitField0_ = 0; if (((from_bitField0_ & 0x00000001) == 0x00000001)) { to_bitField0_ |= 0x00000001; } result.type_ = type_; if (((from_bitField0_ & 0x00000002) == 0x00000002)) { to_bitField0_ |= 0x00000002; } if (launchDAGBuilder_ == null) { result.launchDAG_ = launchDAG_; } else { result.launchDAG_ = launchDAGBuilder_.build(); } result.bitField0_ = to_bitField0_; onBuilt(); return result; }
com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException { initFields(); int mutable_bitField0_ = 0; com.google.protobuf.UnknownFieldSet.Builder unknownFields = break; default: { if (!parseUnknownField(input, unknownFields, extensionRegistry, tag)) { done = true; } finally { this.unknownFields = unknownFields.build(); makeExtensionsImmutable();
com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException { initFields(); int mutable_bitField0_ = 0; com.google.protobuf.UnknownFieldSet.Builder unknownFields = break; default: { if (!parseUnknownField(input, unknownFields, extensionRegistry, tag)) { done = true; } finally { this.unknownFields = unknownFields.build(); makeExtensionsImmutable();
startSchedulingUserDAG(message.getLaunchDAG().getDag()); final Map<Serializable, Object> broadcastVars = SerializationUtils.deserialize(message.getLaunchDAG().getBroadcastVars().toByteArray()); BroadcastManagerMaster.registerBroadcastVariablesFromClient(broadcastVars); });
startSchedulingUserDAG(message.getLaunchDAG().getDag()); final Map<Serializable, Object> broadcastVars = SerializationUtils.deserialize(message.getLaunchDAG().getBroadcastVars().toByteArray()); BroadcastManagerMaster.registerBroadcastVariablesFromClient(broadcastVars); });
driverRPCServer.send(ControlMessage.ClientToDriverMessage.newBuilder() .setType(ControlMessage.ClientToDriverMessageType.DriverShutdown).build());
public org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage buildPartial() { org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage result = new org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage(this); int from_bitField0_ = bitField0_; int to_bitField0_ = 0; if (((from_bitField0_ & 0x00000001) == 0x00000001)) { to_bitField0_ |= 0x00000001; } result.type_ = type_; if (((from_bitField0_ & 0x00000002) == 0x00000002)) { to_bitField0_ |= 0x00000002; } if (launchDAGBuilder_ == null) { result.launchDAG_ = launchDAG_; } else { result.launchDAG_ = launchDAGBuilder_.build(); } result.bitField0_ = to_bitField0_; onBuilt(); return result; }
public int getSerializedSize() { int size = memoizedSerializedSize; if (size != -1) return size; size = 0; if (((bitField0_ & 0x00000001) == 0x00000001)) { size += com.google.protobuf.CodedOutputStream .computeEnumSize(1, type_.getNumber()); } if (((bitField0_ & 0x00000002) == 0x00000002)) { size += com.google.protobuf.CodedOutputStream .computeMessageSize(2, launchDAG_); } size += getUnknownFields().getSerializedSize(); memoizedSerializedSize = size; return size; }
public Builder toBuilder() { return newBuilder(this); }
public Builder toBuilder() { return newBuilder(this); }
public final boolean isInitialized() { byte isInitialized = memoizedIsInitialized; if (isInitialized != -1) return isInitialized == 1; if (!hasType()) { memoizedIsInitialized = 0; return false; } if (hasLaunchDAG()) { if (!getLaunchDAG().isInitialized()) { memoizedIsInitialized = 0; return false; } } memoizedIsInitialized = 1; return true; }
public Builder newBuilderForType() { return newBuilder(); } public static Builder newBuilder(org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage prototype) {
public int getSerializedSize() { int size = memoizedSerializedSize; if (size != -1) return size; size = 0; if (((bitField0_ & 0x00000001) == 0x00000001)) { size += com.google.protobuf.CodedOutputStream .computeEnumSize(1, type_.getNumber()); } if (((bitField0_ & 0x00000002) == 0x00000002)) { size += com.google.protobuf.CodedOutputStream .computeMessageSize(2, launchDAG_); } size += getUnknownFields().getSerializedSize(); memoizedSerializedSize = size; return size; }
public final boolean isInitialized() { byte isInitialized = memoizedIsInitialized; if (isInitialized != -1) return isInitialized == 1; if (!hasType()) { memoizedIsInitialized = 0; return false; } if (hasLaunchDAG()) { if (!getLaunchDAG().isInitialized()) { memoizedIsInitialized = 0; return false; } } memoizedIsInitialized = 1; return true; }
public Builder newBuilderForType() { return newBuilder(); } public static Builder newBuilder(org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage prototype) {
public org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage getDefaultInstanceForType() { return org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage.getDefaultInstance(); }
public static Builder newBuilder(org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage prototype) { return newBuilder().mergeFrom(prototype); } public Builder toBuilder() { return newBuilder(this); }
public org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage getDefaultInstanceForType() { return org.apache.nemo.runtime.common.comm.ControlMessage.ClientToDriverMessage.getDefaultInstance(); }