Examples with Cluster com.amazonaws.services.ecs.model.Cluster used on opensource projects

Example 1 with Cluster

use of com.amazonaws.services.ecs.model.Cluster in project gocd-ecs-elastic-agent by gocd.

the class ClusterStatusReportExecutor method execute.

@Override
public GoPluginApiResponse execute() throws IOException, TemplateException {
    final Map<String, Object> dataModel = new HashMap<>();
    try {
        final ClusterProfileProperties clusterProfileProperties = request.clusterProfileProperties();
        agentInstances.refreshAll(clusterProfileProperties);
        if (clusterProfileProperties == null || !clusterProfileProperties.isConfigured()) {
            throw new RuntimeException("Please configure the cluster profile properly before using the plugin.");
        }
        final Cluster cluster = helper.getCluster(clusterProfileProperties);
        final List<ContainerInstance> containerInstances = new ArrayList<>();
        final List<Instance> instances = new ArrayList<>();
        final List<ECSContainer> ecsContainers = new ArrayList<>();
        if (cluster.getRegisteredContainerInstancesCount() != 0) {
            containerInstances.addAll(helper.getContainerInstances(clusterProfileProperties));
            instances.addAll(helper.ec2InstancesFromContainerInstances(clusterProfileProperties, containerInstances));
            ecsContainers.addAll(taskHelper.allRunningContainers(clusterProfileProperties));
        }
        dataModel.put("region", clusterProfileProperties.getRegion());
        final ECSCluster ecsCluster = new ECSCluster(cluster, containerInstances, instances, ecsContainers, clusterProfileProperties.getMaxLinuxInstancesAllowed(), clusterProfileProperties.getMaxWindowsInstancesAllowed(), clusterProfileProperties.getMaxLinuxSpotInstanceAllowed(), clusterProfileProperties.getMaxWindowsSpotInstanceAllowed());
        dataModel.put("cluster", ecsCluster);
        eventStream.remove(EventFingerprint.forStatusReport());
    } catch (Exception e) {
        LOG.error("[status report] Error accessing ECS cluster details", e);
        final String errorMessage = e.getCause() != null ? e.getCause().getMessage() : e.getMessage();
        eventStream.update(Event.errorEvent(EventFingerprint.forStatusReport(), "Error accessing ECS cluster details", errorMessage));
    }
    addPluginVersionInformation(dataModel);
    dataModel.put("errors", eventStream.allErrors());
    dataModel.put("warnings", eventStream.allWarnings());
    final Template template = pluginStatusReportViewBuilder.getTemplate("status-report.template.ftlh");
    final String view = pluginStatusReportViewBuilder.build(template, dataModel);
    JsonObject jsonObject = new JsonObject();
    jsonObject.addProperty("view", view);
    return DefaultGoPluginApiResponse.success(jsonObject.toString());
}

Example 2 with Cluster

use of com.amazonaws.services.ecs.model.Cluster in project gocd-ecs-elastic-agent by gocd.

the class PluginStatusReportViewTest method shouldBuildStatusReportView.

@Test
void shouldBuildStatusReportView() throws Exception {
    final Cluster cluster = clusterWith("GoCD", 5, 10, 0);
    final ContainerInstance containerInstance = containerInstance("instance-id-1", "arn/container-instance-1", "ACTIVE", 8, 4096, 4, 1024);
    final Instance instance = instance("instance-id-1", C3Large, "ami-2dad3da", toDate("13/05/2017 12:50:20"));
    final ECSContainer alpineContainer = containerWith("arn/container-instance-1", "alpine-container", "alpine", 100, 200, "13/05/2017 12:55:00", "13/05/2017 12:56:30", new JobIdentifier("up42", 1L, "foo", "up42_stage", "2", "up42_job", 25632868237L));
    final ECSCluster ecsCluster = new ECSCluster(cluster, singletonList(containerInstance), singletonList(instance), singletonList(alpineContainer), 2, 3, 0, 0);
    Map<String, Object> dataModel = new HashMap<>();
    dataModel.put("cluster", ecsCluster);
    dataModel.put("errors", Collections.emptyList());
    dataModel.put("region", "us-east-2");
    final PluginStatusReportViewBuilder statusReportViewBuilder = PluginStatusReportViewBuilder.instance();
    final Template template = statusReportViewBuilder.getTemplate("status-report.template.ftlh");
    final String view = statusReportViewBuilder.build(template, dataModel);
    assertThat(view).contains("/go/admin/status_reports/com.thoughtworks.gocd.elastic-agent.ecs/agent/alpine-container?job_id=25632868237");
    assertView(view, ecsCluster);
}

Example 3 with Cluster

use of com.amazonaws.services.ecs.model.Cluster in project gocd-ecs-elastic-agent by gocd.

the class ClusterStatusReportExecutorTest method shouldNotFetchClusterInfoIfClusterHasNoContainerInstances.

@Test
void shouldNotFetchClusterInfoIfClusterHasNoContainerInstances() throws Exception {
    final Cluster cluster = mock(Cluster.class);
    when(cluster.getRegisteredContainerInstancesCount()).thenReturn(0);
    when(request.clusterProfileProperties()).thenReturn(clusterProfileProperties);
    when(containerInstanceHelper.getCluster(clusterProfileProperties)).thenReturn(cluster);
    when(pluginStatusReportViewBuilder.build(any(Template.class), anyMap())).thenReturn("plugin_health_html");
    clusterStatusReportExecutor.execute();
    verify(containerInstanceHelper).getCluster(clusterProfileProperties);
    verify(containerInstanceHelper, times(0)).getContainerInstances(clusterProfileProperties);
    verify(containerInstanceHelper, times(0)).ec2InstancesFromContainerInstances(eq(clusterProfileProperties), anyList());
    verifyNoInteractions(taskHelper);
}

Example 4 with Cluster

use of com.amazonaws.services.ecs.model.Cluster in project gocd-ecs-elastic-agent by gocd.

the class ClusterStatusReportExecutorTest method shouldGenerateHealthView.

@Test
void shouldGenerateHealthView() throws Exception {
    final List<ContainerInstance> containerInstances = Collections.emptyList();
    final List<Instance> ec2Instances = Collections.emptyList();
    final List<ECSContainer> ecsContainers = Collections.emptyList();
    final Cluster cluster = mock(Cluster.class);
    when(request.clusterProfileProperties()).thenReturn(clusterProfileProperties);
    when(containerInstanceHelper.getCluster(clusterProfileProperties)).thenReturn(cluster);
    when(containerInstanceHelper.getContainerInstances(clusterProfileProperties)).thenReturn(containerInstances);
    when(containerInstanceHelper.ec2InstancesFromContainerInstances(clusterProfileProperties, containerInstances)).thenReturn(ec2Instances);
    when(taskHelper.allRunningContainers(clusterProfileProperties)).thenReturn(ecsContainers);
    when(pluginStatusReportViewBuilder.build(any(), anyMap())).thenReturn("plugin_health_html");
    final GoPluginApiResponse response = clusterStatusReportExecutor.execute();
    final String expectedJSON = "{\n" + "\"view\" : \"plugin_health_html\"\n" + "}";
    assertThat(response.responseCode()).isEqualTo(200);
    JSONAssert.assertEquals(expectedJSON, response.responseBody(), true);
}

Example 5 with Cluster

use of com.amazonaws.services.ecs.model.Cluster in project gocd-ecs-elastic-agent by gocd.

the class ClusterStatusReportExecutorTest method shouldFetchClusterInformation.

@Test
void shouldFetchClusterInformation() throws Exception {
    final List<ContainerInstance> containerInstances = Collections.emptyList();
    final List<Instance> ec2Instances = Collections.emptyList();
    final List<ECSContainer> ecsContainers = Collections.emptyList();
    final Cluster cluster = mock(Cluster.class);
    when(cluster.getRegisteredContainerInstancesCount()).thenReturn(2);
    when(request.clusterProfileProperties()).thenReturn(clusterProfileProperties);
    when(containerInstanceHelper.getCluster(clusterProfileProperties)).thenReturn(cluster);
    when(containerInstanceHelper.getContainerInstances(clusterProfileProperties)).thenReturn(containerInstances);
    when(containerInstanceHelper.ec2InstancesFromContainerInstances(clusterProfileProperties, containerInstances)).thenReturn(ec2Instances);
    when(taskHelper.allRunningContainers(clusterProfileProperties)).thenReturn(ecsContainers);
    when(pluginStatusReportViewBuilder.build(any(Template.class), anyMap())).thenReturn("plugin_health_html");
    clusterStatusReportExecutor.execute();
    verify(ecsTasks).refreshAll(clusterProfileProperties);
    verify(containerInstanceHelper).getCluster(clusterProfileProperties);
    verify(containerInstanceHelper).getContainerInstances(clusterProfileProperties);
    verify(containerInstanceHelper).ec2InstancesFromContainerInstances(eq(clusterProfileProperties), anyList());
    verify(taskHelper).allRunningContainers(clusterProfileProperties);
}