endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_A).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_B).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_C).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_D).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_E).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_F).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_G).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_H).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_I).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_J).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_K).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_L).setControlPort(1234).build()); endpoints.add(DrillbitEndpoint.newBuilder().setAddress(HOST_M).setControlPort(1234).build());
.setAddress(ipAddress) .setUserPort(Integer.parseInt(port)) .build();
md = md.toBuilder().setHttpPort(httpPort).build();
return; case 1: builder.setAddress(input.readString()); break; case 2: builder.setUserPort(input.readInt32()); break; case 3: builder.setControlPort(input.readInt32()); break; case 4: builder.setDataPort(input.readInt32()); break; case 5: builder.setRoles(input.mergeObject(org.apache.drill.exec.proto.CoordinationProtos.Roles.newBuilder(), org.apache.drill.exec.proto.SchemaCoordinationProtos.Roles.MERGE)); builder.setVersion(input.readString()); break; case 7: builder.setState(org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.State.valueOf(input.readEnum())); break; case 8: builder.setHttpPort(input.readInt32()); break; default:
bitField0_ |= 0x00000001; address_ = other.address_; onChanged(); setUserPort(other.getUserPort()); setControlPort(other.getControlPort()); setDataPort(other.getDataPort()); mergeRoles(other.getRoles()); onChanged(); setState(other.getState()); setHttpPort(other.getHttpPort()); this.mergeUnknownFields(other.getUnknownFields()); return this;
/** * Update drillbit endpoint state. Drillbit advertises its * state in Zookeeper when a shutdown request of drillbit is * triggered. State information is used during planning and * initial client connection phases. */ public RegistrationHandle update(RegistrationHandle handle, State state) { ZKRegistrationHandle h = (ZKRegistrationHandle) handle; try { endpoint = h.endpoint.toBuilder().setState(state).build(); ServiceInstance<DrillbitEndpoint> serviceInstance = ServiceInstance.<DrillbitEndpoint>builder() .name(serviceName) .id(h.id) .payload(endpoint).build(); discovery.updateService(serviceInstance); } catch (Exception e) { Throwables.throwIfUnchecked(e); throw new RuntimeException(e); } return handle; }
/** * <code>optional .exec.DrillbitEndpoint foreman = 7;</code> */ public Builder setForeman( org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.Builder builderForValue) { if (foremanBuilder_ == null) { foreman_ = builderForValue.build(); onChanged(); } else { foremanBuilder_.setMessage(builderForValue.build()); } bitField0_ |= 0x00000040; return this; } /**
/** * <code>optional .exec.DrillbitEndpoint foreman = 11;</code> */ public Builder setForeman( org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.Builder builderForValue) { if (foremanBuilder_ == null) { foreman_ = builderForValue.build(); onChanged(); } else { foremanBuilder_.setMessage(builderForValue.build()); } bitField0_ |= 0x00000100; return this; } /**
/** * <code>optional .exec.DrillbitEndpoint foreman = 5;</code> */ public Builder setForeman( org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.Builder builderForValue) { if (foremanBuilder_ == null) { foreman_ = builderForValue.build(); onChanged(); } else { foremanBuilder_.setMessage(builderForValue.build()); } bitField0_ |= 0x00000010; return this; } /**
/** * <code>optional .exec.DrillbitEndpoint endpoint = 3;</code> */ public Builder setEndpoint( org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.Builder builderForValue) { if (endpointBuilder_ == null) { endpoint_ = builderForValue.build(); onChanged(); } else { endpointBuilder_.setMessage(builderForValue.build()); } bitField0_ |= 0x00000004; return this; } /**
/** * <code>optional .exec.DrillbitEndpoint assignment = 10;</code> */ public Builder setAssignment( org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.Builder builderForValue) { if (assignmentBuilder_ == null) { assignment_ = builderForValue.build(); onChanged(); } else { assignmentBuilder_.setMessage(builderForValue.build()); } bitField0_ |= 0x00000080; return this; } /**
/** * <code>optional .exec.DrillbitEndpoint endpoint = 3;</code> */ public Builder setEndpoint( org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.Builder builderForValue) { if (endpointBuilder_ == null) { endpoint_ = builderForValue.build(); onChanged(); } else { endpointBuilder_.setMessage(builderForValue.build()); } bitField0_ |= 0x00000004; return this; } /**
/** * <code>optional .exec.DrillbitEndpoint endpoint = 9;</code> */ public Builder setEndpoint( org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.Builder builderForValue) { if (endpointBuilder_ == null) { endpoint_ = builderForValue.build(); onChanged(); } else { endpointBuilder_.setMessage(builderForValue.build()); } bitField0_ |= 0x00000100; return this; } /**
/** * <code>optional .exec.DrillbitEndpoint endpoint = 1;</code> */ public Builder setEndpoint( org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.Builder builderForValue) { if (endpointBuilder_ == null) { endpoint_ = builderForValue.build(); onChanged(); } else { endpointBuilder_.setMessage(builderForValue.build()); } bitField0_ |= 0x00000001; return this; } /**
/** * <code>optional .exec.DrillbitEndpoint endpoint = 2;</code> */ public Builder setEndpoint( org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint.Builder builderForValue) { if (endpointBuilder_ == null) { endpoint_ = builderForValue.build(); onChanged(); } else { endpointBuilder_.setMessage(builderForValue.build()); } bitField0_ |= 0x00000002; return this; } /**
public DrillbitEndpoint start() throws DrillbitStartupException, UnknownHostException { // loopback address check if (isDistributedMode && InetAddress.getByName(hostName).isLoopbackAddress()) { throw new DrillbitStartupException("Drillbit is disallowed to bind to loopback address in distributed mode."); } userPort = userServer.bind(intialUserPort, allowPortHunting); DrillbitEndpoint partialEndpoint = DrillbitEndpoint.newBuilder() .setAddress(hostName) .setUserPort(userPort) .setVersion(DrillVersionInfo.getVersion()) .setState(State.STARTUP) .build(); partialEndpoint = controller.start(partialEndpoint, allowPortHunting); return dataPool.start(partialEndpoint, allowPortHunting); }
/** * Update drillbit endpoint state. Drillbit advertises its * state. State information is used during planning and initial * client connection phases. */ @Override public RegistrationHandle update(RegistrationHandle handle, State state) { DrillbitEndpoint endpoint = handle.getEndPoint(); endpoint = endpoint.toBuilder().setState(state).build(); handle.setEndPoint(endpoint); endpoints.put(handle,endpoint); return handle; }
@Override public RegistrationHandle register(DrillbitEndpoint data) { try { data = data.toBuilder().setState(State.ONLINE).build(); ServiceInstance<DrillbitEndpoint> serviceInstance = newServiceInstance(data); discovery.registerService(serviceInstance); return new ZKRegistrationHandle(serviceInstance.getId(),data); } catch (Exception e) { Throwables.throwIfUnchecked(e); throw new RuntimeException(e); } }
@Override public DrillbitEndpoint start(DrillbitEndpoint partialEndpoint, final boolean allowPortHunting) { server = new ControlServer(config, connectionRegistry); int port = config.getBootstrapContext().getConfig().getInt(ExecConstants.INITIAL_BIT_PORT); port = server.bind(port, allowPortHunting); DrillbitEndpoint completeEndpoint = partialEndpoint.toBuilder().setControlPort(port).build(); connectionRegistry.setLocalEndpoint(completeEndpoint); handlerRegistry.setEndpoint(completeEndpoint); return completeEndpoint; }