public UUID getId() { return _workItem.getId(); }
@Nullable @Override public QueuedWork getWork(UUID workItemId) { for (QueuedWork work : this) { if (work.getWorkItem().getId().equals(workItemId)) { return work; } } return null; } }
public void handleSigInt() { _logger.info("Got SIGINT\n"); WorkItem wItem = _polledWorkItem; if (wItem != null) { _logger.outputf("Killing %s\n", wItem.getId()); boolean result = _workHelper.killWork(_polledWorkItem.getId()); _logger.outputf("Result of killing %s: %s\n", wItem.getId(), result); } else { _logger.output("No work being polled\n"); } }
private boolean execute(WorkItem wItem, @Nullable FileWriter outWriter) { _logger.infof("work-id is %s\n", wItem.getId()); ActiveSpan activeSpan = GlobalTracer.get().activeSpan(); if (activeSpan != null) { activeSpan.setTag("work-id", wItem.getId().toString()); } wItem.addRequestParam(BfConsts.ARG_LOG_LEVEL, _settings.getBatfishLogLevel()); for (String option : _additionalBatfishOptions.keySet()) { wItem.addRequestParam(option, _additionalBatfishOptions.get(option)); } boolean queueWorkResult = _workHelper.queueWork(wItem); _logger.infof("Queuing result: %s\n", queueWorkResult); if (!queueWorkResult) { return queueWorkResult; } boolean result = true; if (!_backgroundExecution) { _polledWorkItem = wItem; result = pollWorkAndGetAnswer(wItem, outWriter); _polledWorkItem = null; } return result; }
return new WorkItem(workItem.getId(), networkId.getId(), snapshotId.getId(), params);
@Test public void testGetWorkStatusPresent() throws IOException { String network = "network1"; String snapshot = "snapshot1"; UUID workId = UUID.randomUUID(); Main.getWorkMgr().initNetwork(network, null); WorkMgrTestUtils.initSnapshotWithTopology(network, snapshot, ImmutableSet.of()); WorkItem workItem = new WorkItem(workId, network, snapshot, new HashMap<>()); Main.getWorkMgr().queueWork(workItem); Response response = getWorkItemTarget(network, workId.toString()).get(); // work item should exist assertThat(response.getStatus(), equalTo(OK.getStatusCode())); WorkStatus workStatus = response.readEntity(WorkStatus.class); // work ID should match assertThat(workStatus.getWorkItem().getId(), equalTo(workId)); } }
private boolean pollWorkAndGetAnswer(WorkItem wItem, @Nullable FileWriter outWriter) { boolean pollResult = pollWork(wItem.getId()); if (!pollResult) { return false; String ansFileName = wItem.getId() + BfConsts.SUFFIX_ANSWER_JSON_FILE; String downloadedAnsFile = _workHelper.getObject(wItem.getContainerName(), wItem.getTestrigName(), ansFileName); String logFileName = wItem.getId() + BfConsts.SUFFIX_LOG_FILE; String downloadedFileStr = _workHelper.getObject(wItem.getContainerName(), wItem.getTestrigName(), logFileName);
CoordConsts.DEFAULT_API_KEY, Version.getVersion(), workItem.getId().toString());
GlobalTracer.get().activeSpan().setTag("work-id", workItem.getId().toString());
String taskStr = BatfishObjectMapper.writePrettyString(work.getLastTaskCheckResult()); response .put(CoordConsts.SVC_KEY_WORKID, work.getWorkItem().getId()) .put(CoordConsts.SVC_KEY_WORKSTATUS, work.getStatus().toString()) .put(CoordConsts.SVC_KEY_TASKSTATUS, taskStr);
String taskStr = BatfishObjectMapper.writePrettyString(work.getLastTaskCheckResult()); response .put(CoordConsts.SVC_KEY_WORKID, work.getWorkItem().getId()) .put(CoordConsts.SVC_KEY_WORKSTATUS, work.getStatus().toString()) .put(CoordConsts.SVC_KEY_TASKSTATUS, taskStr);
String taskStr = BatfishObjectMapper.writePrettyString(work.getLastTaskCheckResult()); response .put(CoordConsts.SVC_KEY_WORKID, work.getWorkItem().getId()) .put(CoordConsts.SVC_KEY_WORKSTATUS, work.getStatus().toString()) .put(CoordConsts.SVC_KEY_TASKSTATUS, taskStr);
String taskStr = BatfishObjectMapper.writePrettyString(work.getLastTaskCheckResult()); response .put(CoordConsts.SVC_KEY_WORKID, work.getWorkItem().getId()) .put(CoordConsts.SVC_KEY_WORKSTATUS, work.getStatus().toString()) .put(CoordConsts.SVC_KEY_TASKSTATUS, taskStr);
return successResponse( new JSONObject() .put(CoordConsts.SVC_KEY_WORKID, work.getWorkItem().getId()) .put(CoordConsts.SVC_KEY_WORKSTATUS, work.getStatus().toString()) .put(CoordConsts.SVC_KEY_TASKSTATUS, taskStr));
String taskStr = BatfishObjectMapper.writePrettyString(work.getLastTaskCheckResult()); response .put(CoordConsts.SVC_KEY_WORKID, work.getWorkItem().getId()) .put(CoordConsts.SVC_KEY_WORKSTATUS, work.getStatus().toString()) .put(CoordConsts.SVC_KEY_TASKSTATUS, taskStr);
String taskStr = BatfishObjectMapper.writePrettyString(work.getLastTaskCheckResult()); response .put(CoordConsts.SVC_KEY_WORKID, work.getWorkItem().getId()) .put(CoordConsts.SVC_KEY_WORKSTATUS, work.getStatus().toString()) .put(CoordConsts.SVC_KEY_TASKSTATUS, taskStr);
String taskStr = BatfishObjectMapper.writePrettyString(work.getLastTaskCheckResult()); response .put(CoordConsts.SVC_KEY_WORKID, work.getWorkItem().getId()) .put(CoordConsts.SVC_KEY_WORKSTATUS, work.getStatus().toString()) .put(CoordConsts.SVC_KEY_TASKSTATUS, taskStr);