예제 #1
0
  private List<AgentStatusRepresentation> waitForAgentsToStart(List<String> instanceIds) {
    for (int loop = 0; true; loop++) {
      try {
        URI uri = uriBuilderFrom(coordinatorUri).replacePath("v1/admin/agent").build();
        Request request = RequestBuilder.prepareGet().setUri(uri).build();
        List<AgentStatusRepresentation> agents =
            client.execute(request, createJsonResponseHandler(AGENTS_CODEC));

        Map<String, AgentStatusRepresentation> runningAgents = newHashMap();
        for (AgentStatusRepresentation agent : agents) {
          if (agent.getState() == AgentLifecycleState.ONLINE) {
            runningAgents.put(agent.getInstanceId(), agent);
          }
        }
        if (runningAgents.keySet().containsAll(instanceIds)) {
          WaitUtils.clearWaitMessage();
          runningAgents.keySet().retainAll(instanceIds);
          return ImmutableList.copyOf(runningAgents.values());
        }
      } catch (Exception ignored) {
      }

      WaitUtils.wait(loop);
    }
  }
예제 #2
0
  @Override
  public List<AgentStatusRepresentation> provisionAgents(
      String agentConfig,
      int agentCount,
      String instanceType,
      String availabilityZone,
      String ami,
      String keyPair,
      String securityGroup,
      boolean waitForStartup) {
    URI uri = uriBuilderFrom(coordinatorUri).replacePath("v1/admin/agent").build();

    AgentProvisioningRepresentation agentProvisioning =
        new AgentProvisioningRepresentation(
            agentConfig, agentCount, instanceType, availabilityZone, ami, keyPair, securityGroup);

    Request request =
        RequestBuilder.preparePost()
            .setUri(uri)
            .setHeader("Content-Type", "application/json")
            .setBodyGenerator(jsonBodyGenerator(AGENT_PROVISIONING_CODEC, agentProvisioning))
            .build();

    List<AgentStatusRepresentation> agents =
        client.execute(request, createJsonResponseHandler(AGENTS_CODEC));
    if (waitForStartup) {
      List<String> instanceIds = newArrayList();
      for (AgentStatusRepresentation agent : agents) {
        instanceIds.add(agent.getInstanceId());
      }
      agents = waitForAgentsToStart(instanceIds);
    }
    return agents;
  }