.setRunningTasks(stage.getTotalScheduledObjectsCount() - stage.getCompletedTaskCount()) .addAllCandidateNodes(candidateWorkers) .setQueue(context.getMasterContext().getQueryContext().get("queue", "default")); //TODO set queue
.setRunningTasks(stage.getTotalScheduledObjectsCount() - stage.getCompletedTaskCount()) .addAllCandidateNodes(candidateWorkers) .setQueue(context.getMasterContext().getQueryContext().get("queue", "default")); //TODO set queue
private NodeResourceRequest createResourceRequest(QueryId queryId, int containerNum, List<Integer> candidateWorkers) { NodeResourceRequest.Builder request = NodeResourceRequest.newBuilder(); request.setCapacity(scheduler.getMinimumResourceCapability().getProto()) .setNumContainers(containerNum) .setPriority(1) .setQueryId(queryId.getProto()) .setType(ResourceType.LEAF) .setUserId("test user") .setRunningTasks(0) .addAllCandidateNodes(candidateWorkers) .setQueue("default"); return request.build(); }