AdminResource.java

package io.airlift.airship.coordinator;

import com.google.common.base.Predicate;
import com.google.inject.Inject;
import io.airlift.airship.shared.AgentStatus;
import io.airlift.airship.shared.CoordinatorStatus;
import io.airlift.airship.shared.Repository;
import io.airlift.airship.shared.SlotStatus;

import javax.ws.rs.DELETE;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.Status;
import javax.ws.rs.core.UriInfo;

import java.util.List;

import static com.google.common.collect.Lists.transform;
import static io.airlift.airship.shared.AgentStatus.idGetter;
import static io.airlift.airship.shared.AgentStatusRepresentation.fromAgentStatus;
import static io.airlift.airship.shared.CoordinatorStatusRepresentation.fromCoordinatorStatus;
import static io.airlift.airship.shared.VersionsUtil.AIRSHIP_AGENTS_VERSION_HEADER;
import static io.airlift.airship.shared.VersionsUtil.createAgentsVersion;

@Path("/v1/admin/")
public class AdminResource
{
    private final Coordinator coordinator;
    private final Repository repository;

    @Inject
    public AdminResource(Coordinator coordinator, Repository repository)
    {
        this.coordinator = coordinator;
        this.repository = repository;
    }

    @GET
    @Path("/coordinator")
    @Produces(MediaType.APPLICATION_JSON)
    public Response getAllCoordinators(@Context UriInfo uriInfo)
    {
        Predicate<CoordinatorStatus> coordinatorPredicate = CoordinatorFilterBuilder.build(uriInfo);
        List<CoordinatorStatus> coordinators = coordinator.getCoordinators(coordinatorPredicate);
        return Response.ok(transform(coordinators, fromCoordinatorStatus(coordinator.getCoordinators()))).build();
    }

    @POST
    @Path("/coordinator")
    @Produces(MediaType.APPLICATION_JSON)
    public Response provisionCoordinator(
            CoordinatorProvisioningRepresentation provisioning,
            @Context UriInfo uriInfo)
            throws Exception
    {
        List<CoordinatorStatus> coordinators = coordinator.provisionCoordinators(
                provisioning.getCoordinatorConfig(),
                provisioning.getCoordinatorCount(),
                provisioning.getInstanceType(),
                provisioning.getAvailabilityZone(),
                provisioning.getAmi(),
                provisioning.getKeyPair(),
                provisioning.getSecurityGroup(),
                provisioning.getProvisioningScriptsArtifact());

        return Response.ok(transform(coordinators, fromCoordinatorStatus(coordinator.getCoordinators()))).build();
    }

    @GET
    @Path("/agent")
    @Produces(MediaType.APPLICATION_JSON)
    public Response getAllAgents(@Context UriInfo uriInfo)
    {
        List<SlotStatus> allSlotStatus = coordinator.getAllSlotStatus();
        Predicate<AgentStatus> agentPredicate = AgentFilterBuilder.build(uriInfo,
                transform(coordinator.getAgents(), idGetter()),
                transform(allSlotStatus, SlotStatus.uuidGetter()),
                false,
                repository);

        List<AgentStatus> agents = coordinator.getAgents(agentPredicate);

        return Response.ok(transform(agents, fromAgentStatus(coordinator.getAgents(), repository)))
                .header(AIRSHIP_AGENTS_VERSION_HEADER, createAgentsVersion(agents))
                .build();
    }

    @POST
    @Path("/agent")
    @Produces(MediaType.APPLICATION_JSON)
    public Response provisionAgent(
            AgentProvisioningRepresentation provisioning,
            @Context UriInfo uriInfo)
            throws Exception
    {
        List<AgentStatus> agents = coordinator.provisionAgents(
                provisioning.getAgentConfig(),
                provisioning.getAgentCount(),
                provisioning.getInstanceType(),
                provisioning.getAvailabilityZone(),
                provisioning.getAmi(),
                provisioning.getKeyPair(),
                provisioning.getSecurityGroup(),
                provisioning.getProvisioningScriptsArtifact());

        return Response.ok(transform(agents, fromAgentStatus(coordinator.getAgents(), repository))).build();
    }

    @DELETE
    @Path("/agent/{agentId: [a-z0-9-]+}")
    public Response terminateAgent(String agentId, @Context UriInfo uriInfo)
    {
        if (coordinator.terminateAgent(agentId) != null) {
            return Response.ok().build();
        }
        return Response.status(Status.NOT_FOUND).build();
    }
}