Search in sources :

Example 1 with ECSCluster

use of com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster 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());
}
Also used : ClusterProfileProperties(com.thoughtworks.gocd.elasticagent.ecs.domain.ClusterProfileProperties) HashMap(java.util.HashMap) ContainerInstance(com.amazonaws.services.ecs.model.ContainerInstance) Instance(com.amazonaws.services.ec2.model.Instance) ArrayList(java.util.ArrayList) ECSCluster(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster) Cluster(com.amazonaws.services.ecs.model.Cluster) JsonObject(com.google.gson.JsonObject) ECSCluster(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster) TemplateException(freemarker.template.TemplateException) IOException(java.io.IOException) Template(freemarker.template.Template) ContainerInstance(com.amazonaws.services.ecs.model.ContainerInstance) ECSContainer(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSContainer) JsonObject(com.google.gson.JsonObject)

Example 2 with ECSCluster

use of com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster 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);
}
Also used : ContainerInstance(com.amazonaws.services.ecs.model.ContainerInstance) ECSContainerInstance(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSContainerInstance) ECSContainer(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSContainer) ContainerInstance(com.amazonaws.services.ecs.model.ContainerInstance) ContainerInstanceMother.containerInstance(com.thoughtworks.gocd.elasticagent.ecs.aws.ContainerInstanceMother.containerInstance) Instance(com.amazonaws.services.ec2.model.Instance) ECSContainerInstance(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSContainerInstance) ECSCluster(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster) Cluster(com.amazonaws.services.ecs.model.Cluster) JobIdentifier(com.thoughtworks.gocd.elasticagent.ecs.domain.JobIdentifier) ECSCluster(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster) Template(freemarker.template.Template) Test(org.junit.jupiter.api.Test)

Example 3 with ECSCluster

use of com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster in project gocd-ecs-elastic-agent by gocd.

the class PluginStatusReportViewTest method shouldBuildViewWithoutContainerInstanceAndPrintNoRunningContainerInstanceMessage.

@Test
void shouldBuildViewWithoutContainerInstanceAndPrintNoRunningContainerInstanceMessage() throws Exception {
    final Cluster cluster = clusterWith("GoCD", 0, 0, 0);
    final ECSCluster ecsCluster = new ECSCluster(cluster, emptyList(), emptyList(), emptyList(), 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);
    assertView(view, ecsCluster);
}
Also used : ECSCluster(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster) Cluster(com.amazonaws.services.ecs.model.Cluster) ECSCluster(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster) Template(freemarker.template.Template) Test(org.junit.jupiter.api.Test)

Example 4 with ECSCluster

use of com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster in project gocd-ecs-elastic-agent by gocd.

the class PluginStatusReportViewTest method shouldBuildStatusReportViewWithDefaultValues.

@Test
void shouldBuildStatusReportViewWithDefaultValues() throws Exception {
    final Cluster cluster = clusterWith("GoCD", 0, 0, 0);
    final ContainerInstance containerInstance = containerInstance("instance-id", "arn/container-instance-1", "ACTIVE", 0, 0, 0, 0);
    final Instance instance = instance("instance-id", C3Large, "ami-23456", toDate("13/05/2017 12:50:20"));
    final ECSContainer alpineContainer = containerWith("arn/container-instance-1", "container-name", "alpine", 100, 200, "13/05/2017 12:55:20", null);
    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("maxAllowedInstances", 5);
    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);
    assertView(view, ecsCluster);
}
Also used : ContainerInstance(com.amazonaws.services.ecs.model.ContainerInstance) ECSContainerInstance(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSContainerInstance) ECSContainer(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSContainer) ContainerInstance(com.amazonaws.services.ecs.model.ContainerInstance) ContainerInstanceMother.containerInstance(com.thoughtworks.gocd.elasticagent.ecs.aws.ContainerInstanceMother.containerInstance) Instance(com.amazonaws.services.ec2.model.Instance) ECSContainerInstance(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSContainerInstance) ECSCluster(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster) Cluster(com.amazonaws.services.ecs.model.Cluster) ECSCluster(com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster) Template(freemarker.template.Template) Test(org.junit.jupiter.api.Test)

Aggregations

Cluster (com.amazonaws.services.ecs.model.Cluster)4 ECSCluster (com.thoughtworks.gocd.elasticagent.ecs.domain.ECSCluster)4 Template (freemarker.template.Template)4 Instance (com.amazonaws.services.ec2.model.Instance)3 ContainerInstance (com.amazonaws.services.ecs.model.ContainerInstance)3 ECSContainer (com.thoughtworks.gocd.elasticagent.ecs.domain.ECSContainer)3 Test (org.junit.jupiter.api.Test)3 ContainerInstanceMother.containerInstance (com.thoughtworks.gocd.elasticagent.ecs.aws.ContainerInstanceMother.containerInstance)2 ECSContainerInstance (com.thoughtworks.gocd.elasticagent.ecs.domain.ECSContainerInstance)2 JsonObject (com.google.gson.JsonObject)1 ClusterProfileProperties (com.thoughtworks.gocd.elasticagent.ecs.domain.ClusterProfileProperties)1 JobIdentifier (com.thoughtworks.gocd.elasticagent.ecs.domain.JobIdentifier)1 TemplateException (freemarker.template.TemplateException)1 IOException (java.io.IOException)1 ArrayList (java.util.ArrayList)1 HashMap (java.util.HashMap)1