/**
* Send all the remote fragments belonging to a single target drillbit in one request.
*
* @param assignment the drillbit assigned to these fragments
* @param fragments the set of fragments
* @param latch the countdown latch used to track the requests to all endpoints
* @param fragmentSubmitFailures the submission failure counter used to track the requests to all
* endpoints
*/
private void sendRemoteFragments(
final DrillbitEndpoint assignment,
final Collection<PlanFragment> fragments,
final CountDownLatch latch,
final FragmentSubmitFailures fragmentSubmitFailures) {
@SuppressWarnings("resource")
final Controller controller = drillbitContext.getController();
final InitializeFragments.Builder fb = InitializeFragments.newBuilder();
for (final PlanFragment planFragment : fragments) {
fb.addFragment(planFragment);
}
final InitializeFragments initFrags = fb.build();
logger.debug("Sending remote fragments to \nNode:\n{} \n\nData:\n{}", assignment, initFrags);
final FragmentSubmitListener listener =
new FragmentSubmitListener(assignment, initFrags, latch, fragmentSubmitFailures);
controller.getTunnel(assignment).sendFragments(listener, initFrags);
}
public DrillbitEndpoint start() throws DrillbitStartupException, UnknownHostException {
int userPort =
userServer.bind(config.getInt(ExecConstants.INITIAL_USER_PORT), allowPortHunting);
String address =
useIP
? InetAddress.getLocalHost().getHostAddress()
: InetAddress.getLocalHost().getCanonicalHostName();
DrillbitEndpoint partialEndpoint =
DrillbitEndpoint.newBuilder()
.setAddress(address)
// .setAddress("localhost")
.setUserPort(userPort)
.build();
partialEndpoint = controller.start(partialEndpoint);
return dataPool.start(partialEndpoint);
}
