@Override
public ImmutableList<ServiceDescriptor> getServiceInventory(Iterable<SlotStatus> allSlotStatus) {
ImmutableList.Builder<ServiceDescriptor> newDescriptors = ImmutableList.builder();
for (SlotStatus slotStatus : allSlotStatus) {
// if the self reference is null, the slot is totally offline so skip for now
if (slotStatus.getSelf() == null) {
continue;
}
List<ServiceDescriptor> serviceDescriptors = getServiceInventory(slotStatus);
if (serviceDescriptors == null) {
continue;
}
for (ServiceDescriptor serviceDescriptor : serviceDescriptors) {
newDescriptors.add(
new ServiceDescriptor(
null,
slotStatus.getId().toString(),
serviceDescriptor.getType(),
serviceDescriptor.getPool(),
slotStatus.getLocation(),
slotStatus.getState() == SlotLifecycleState.RUNNING
? ServiceState.RUNNING
: ServiceState.STOPPED,
interpolateProperties(serviceDescriptor.getProperties(), slotStatus)));
}
}
return newDescriptors.build();
}
@BeforeMethod
public void setup() throws Exception {
NodeInfo nodeInfo = new NodeInfo("testing");
MockProvisioner provisioner = new MockProvisioner();
coordinator =
new Coordinator(
nodeInfo,
new HttpServerInfo(new HttpServerConfig(), nodeInfo),
new CoordinatorConfig().setStatusExpiration(new Duration(1, TimeUnit.DAYS)),
provisioner.getCoordinatorFactory(),
provisioner.getAgentFactory(),
MOCK_REPO,
provisioner,
new InMemoryStateManager(),
new MockServiceInventory());
resource = new CoordinatorLifecycleResource(coordinator, MOCK_REPO);
apple1SlotId = UUID.randomUUID();
SlotStatus appleSlotStatus1 =
createSlotStatus(
apple1SlotId,
URI.create("fake://foo/v1/agent/slot/apple1"),
URI.create("fake://foo/v1/agent/slot/apple1"),
"instance",
"/location",
STOPPED,
APPLE_ASSIGNMENT,
"/apple1",
ImmutableMap.<String, Integer>of());
apple2SlotId = UUID.randomUUID();
SlotStatus appleSlotStatus2 =
createSlotStatus(
apple2SlotId,
URI.create("fake://foo/v1/agent/slot/apple1"),
URI.create("fake://foo/v1/agent/slot/apple1"),
"instance",
"/location",
STOPPED,
APPLE_ASSIGNMENT,
"/apple2",
ImmutableMap.<String, Integer>of());
bananaSlotId = UUID.randomUUID();
SlotStatus bananaSlotStatus =
createSlotStatus(
bananaSlotId,
URI.create("fake://foo/v1/agent/slot/banana"),
URI.create("fake://foo/v1/agent/slot/banana"),
"instance",
"/location",
STOPPED,
BANANA_ASSIGNMENT,
"/banana",
ImmutableMap.<String, Integer>of());
agentId = UUID.randomUUID().toString();
AgentStatus agentStatus =
new AgentStatus(
agentId,
ONLINE,
"instance-id",
URI.create("fake://foo/"),
URI.create("fake://foo/"),
"/unknown/location",
"instance.type",
ImmutableList.of(appleSlotStatus1, appleSlotStatus2, bananaSlotStatus),
ImmutableMap.of("cpu", 8, "memory", 1024));
prefixSize =
shortestUniquePrefix(
asList(
appleSlotStatus1.getId().toString(),
appleSlotStatus2.getId().toString(),
bananaSlotStatus.getId().toString()),
MIN_PREFIX_SIZE);
provisioner.addAgents(agentStatus);
coordinator.updateAllAgents();
}
