@Test
public void testGetExecutionSummaryWithRange() throws Exception {
storeEntity(EntityType.CLUSTER, "testCluster");
storeEntity(EntityType.FEED, "clicksFeed");
storeEntity(EntityType.FEED, "clicksSummary");
long instance1Time = System.currentTimeMillis() - 180000;
long instance2Time = System.currentTimeMillis();
EntityState entityState = getEntityState(EntityType.PROCESS, "clicksProcess");
ExecutionInstance processExecutionInstance1 =
BeanMapperUtil.getExecutionInstance(
entityState.getEntity().getEntityType(),
entityState.getEntity(),
instance1Time,
"cluster1",
instance1Time);
InstanceState instanceState1 = new InstanceState(processExecutionInstance1);
instanceState1.setCurrentState(InstanceState.STATE.RUNNING);
ExecutionInstance processExecutionInstance2 =
BeanMapperUtil.getExecutionInstance(
entityState.getEntity().getEntityType(),
entityState.getEntity(),
instance2Time,
"cluster1",
instance2Time);
InstanceState instanceState2 = new InstanceState(processExecutionInstance2);
instanceState2.setCurrentState(InstanceState.STATE.SUCCEEDED);
stateStore.putExecutionInstance(instanceState1);
stateStore.putExecutionInstance(instanceState2);
Map<InstanceState.STATE, Long> summary =
stateStore.getExecutionInstanceSummary(
entityState.getEntity(),
"cluster1",
new DateTime(instance1Time),
new DateTime(instance1Time + 60000));
Assert.assertEquals(summary.size(), 1);
Assert.assertEquals(summary.get(InstanceState.STATE.RUNNING).longValue(), 1L);
summary =
stateStore.getExecutionInstanceSummary(
entityState.getEntity(),
"cluster1",
new DateTime(instance2Time),
new DateTime(instance2Time + 60000));
Assert.assertEquals(summary.size(), 1);
Assert.assertEquals(summary.get(InstanceState.STATE.SUCCEEDED).longValue(), 1L);
}
@Override
public InstancesSummaryResult getSummary(
Entity entity, Date start, Date end, List<LifeCycle> lifeCycles) throws FalconException {
Set<String> clusters = EntityUtil.getClustersDefinedInColos(entity);
List<InstancesSummaryResult.InstanceSummary> instanceSummaries = new ArrayList<>();
// Iterate over entity clusters
for (String cluster : clusters) {
LOG.debug("Retrieving summary of instances for cluster : {}", cluster);
Map<InstanceState.STATE, Long> summaries =
STATE_STORE.getExecutionInstanceSummary(
entity, cluster, new DateTime(start), new DateTime(end));
Map<String, Long> summaryMap = new HashMap<>();
// Iterate over the map and convert STATE to String
for (Map.Entry<InstanceState.STATE, Long> summary : summaries.entrySet()) {
summaryMap.put(summary.getKey().name(), summary.getValue());
}
instanceSummaries.add(new InstancesSummaryResult.InstanceSummary(cluster, summaryMap));
}
InstancesSummaryResult instancesSummaryResult =
new InstancesSummaryResult(APIResult.Status.SUCCEEDED, JobAction.SUMMARY.name());
instancesSummaryResult.setInstancesSummary(
instanceSummaries.toArray(
new InstancesSummaryResult.InstanceSummary[instanceSummaries.size()]));
return instancesSummaryResult;
}