break;
}
}
if (supervisorId == null) {
throw new TException("No supervisor of " + host);
}
Map<String, Assignment> assignments = new HashMap<String, Assignment>();
// get all active topology's StormBase
Map<String, StormBase> bases = Cluster
.topology_bases(stormClusterState);
for (Entry<String, StormBase> entry : bases.entrySet()) {
String topologyId = entry.getKey();
StormBase base = entry.getValue();
Assignment assignment = stormClusterState.assignment_info(
topologyId, null);
if (assignment == null) {
LOG.error("Failed to get assignment of " + topologyId);
continue;
}
assignments.put(topologyId, assignment);
}
Map<Integer, WorkerSummary> portWorkerSummarys = new TreeMap<Integer, WorkerSummary>();
Map<String, Integer> supervisorToUsedSlotNum = new HashMap<String, Integer>();
for (Entry<String, Assignment> entry : assignments.entrySet()) {
String topologyId = entry.getKey();
Assignment assignment = entry.getValue();
Map<Integer, String> taskToComponent = Cluster
.topology_task_info(stormClusterState, topologyId);
Map<Integer, String> taskToComponentType = Cluster
.topology_task_compType(stormClusterState, topologyId);
Set<ResourceWorkerSlot> workers = assignment.getWorkers();
for (ResourceWorkerSlot worker : workers) {
if (supervisorId.equals(worker.getNodeId()) == false) {
continue;
}
Integer slotNum = supervisorToUsedSlotNum
.get(supervisorId);
if (slotNum == null) {
slotNum = 0;
supervisorToUsedSlotNum.put(supervisorId, slotNum);
}
supervisorToUsedSlotNum.put(supervisorId, ++slotNum);
Integer port = worker.getPort();
WorkerSummary workerSummary = portWorkerSummarys
.get(port);
if (workerSummary == null) {
workerSummary = new WorkerSummary();
workerSummary.set_port(port);
workerSummary.set_topology(topologyId);
workerSummary
.set_tasks(new ArrayList<TaskSummary>());
portWorkerSummarys.put(port, workerSummary);
}
for (Integer taskId : worker.getTasks()) {
String componentName = taskToComponent.get(taskId);
String componentType = taskToComponentType.get(taskId);
int uptime = TimeUtils.time_delta(assignment
.getTaskStartTimeSecs().get(taskId));
List<TaskSummary> tasks = workerSummary.get_tasks();
TaskSummary taskSummary = NimbusUtils
.mkSimpleTaskSummary(worker, taskId,
componentName, componentType, host, uptime);
tasks.add(taskSummary);
}
}
}
List<WorkerSummary> wokersList = new ArrayList<WorkerSummary>();
wokersList.addAll(portWorkerSummarys.values());
SupervisorSummary supervisorSummary = NimbusUtils
.mkSupervisorSummary(supervisorInfo, supervisorId,
supervisorToUsedSlotNum);
return new SupervisorWorkers(supervisorSummary, wokersList);
} catch (TException e) {
LOG.info("Failed to get ClusterSummary ", e);
throw e;
} catch (Exception e) {
LOG.info("Failed to get ClusterSummary ", e);
throw new TException(e);
}
}