public void clearPartitions() { for(Stage eachStage : getStages()) { eachStage.clearPartitions(); } }
private List<StageHistory> makeStageHistories() { List<StageHistory> stageHistories = new ArrayList<StageHistory>(); for(Stage eachStage : getStages()) { stageHistories.add(eachStage.getStageHistory()); } return stageHistories; }
private List<StageHistory> makeStageHistories() { List<StageHistory> stageHistories = new ArrayList<>(); for(Stage eachStage : getStages()) { stageHistories.add(eachStage.getStageHistory()); } return stageHistories; }
public void clearPartitions() { for(Stage eachStage : getStages()) { eachStage.clearPartitions(); } }
public List<PartitionDescProto> getPartitions() { Set<PartitionDescProto> partitions = new HashSet<>(); for(Stage eachStage : getStages()) { partitions.addAll(eachStage.getPartitions()); } return Lists.newArrayList(partitions); }
public List<PartitionDescProto> getPartitions() { Set<PartitionDescProto> partitions = TUtil.newHashSet(); for(Stage eachStage : getStages()) { partitions.addAll(eachStage.getPartitions()); } return Lists.newArrayList(partitions); }
private void cleanupQuery(final QueryId queryId) { if (getQuery() != null) { Set<InetSocketAddress> workers = Sets.newHashSet(); for (Stage stage : getQuery().getStages()) { workers.addAll(stage.getAssignedWorkerMap().values()); } LOG.info("Cleanup resources of all workers. Query: " + queryId + ", workers: " + workers.size()); for (final InetSocketAddress worker : workers) { queryMasterContext.getEventExecutor().submit(new Runnable() { @Override public void run() { try { AsyncRpcClient rpc = RpcClientManager.getInstance().getClient(worker, TajoWorkerProtocol.class, true, rpcParams); TajoWorkerProtocol.TajoWorkerProtocolService tajoWorkerProtocolService = rpc.getStub(); tajoWorkerProtocolService.stopQuery(null, queryId.getProto(), NullCallback.get()); } catch (Throwable e) { LOG.error(e.getMessage(), e); } } }); } } }
private void cleanupQuery(final QueryId queryId) { if (getQuery() != null) { Set<InetSocketAddress> workers = Sets.newHashSet(); for (Stage stage : getQuery().getStages()) { workers.addAll(stage.getAssignedWorkerMap().values()); } LOG.info("Cleanup resources of all workers. Query: " + queryId + ", workers: " + workers.size()); for (final InetSocketAddress worker : workers) { queryMasterContext.getEventExecutor().submit(new Runnable() { @Override public void run() { try { AsyncRpcClient rpc = RpcClientManager.getInstance().getClient(worker, TajoWorkerProtocol.class, true, rpcParams); TajoWorkerProtocol.TajoWorkerProtocolService tajoWorkerProtocolService = rpc.getStub(); tajoWorkerProtocolService.stopQuery(null, queryId.getProto(), NullCallback.get()); } catch (Throwable e) { LOG.error(e.getMessage(), e); } } }); } } }
Stage stage = queryMasterTask.getQuery().getStages().iterator().next(); assertNotNull(stage);