com.boozallen.aissemble.pipeline.invocation.service.endpoint.HttpEndpoint Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of extensions-pipeline-invocation-service Show documentation

Contains an invocation service for submitting spark applications

There is a newer version: 1.10.0-rc2

Show newest version

package com.boozallen.aissemble.pipeline.invocation.service.endpoint;

/*-
 * #%L
 * aiSSEMBLE::Extensions::Pipeline Invocation Service
 * %%
 * Copyright (C) 2021 Booz Allen
 * %%
 * This software package is licensed under the Booz Allen Public License. All Rights Reserved.
 * #L%
 */

import com.boozallen.aissemble.pipeline.invocation.service.PipelineInvocationAgent;
import com.boozallen.aissemble.pipeline.invocation.service.serialization.PipelineInvocationRequest;
import org.jboss.logging.Logger;

import javax.enterprise.context.ApplicationScoped;
import javax.inject.Inject;
import javax.ws.rs.Consumes;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import java.io.IOException;

/**
 * Entrypoint for HTTP-based requests for SparkApplication submission.
 */
@Path("/invoke-pipeline")
@ApplicationScoped
public class HttpEndpoint {
    private static final Logger logger = Logger.getLogger(MessageEndpoint.class);

    @Inject
    private PipelineInvocationAgent pipelineInvocationAgent;

    @GET
    @Path("/healthcheck")
    public Response healthCheck() throws IOException {
        return Response.ok().entity("Service is available").build();
    }

    /**
     * Receives a message requesting the submission of a SparkApplication to the cluster for processing
     * @param pipelineInvocationRequest object representing the required parameters for submitting the desired SparkApplication.
     */
    @POST
    @Path("/start-spark-operator-job")
    @Consumes(MediaType.APPLICATION_JSON)
    public Response startSparkOperatorJob(PipelineInvocationRequest pipelineInvocationRequest) {
        try {
            logger.info("Received HTTP request to submit " + pipelineInvocationRequest.getApplicationName() + ".");
            pipelineInvocationAgent.submitSparkApplication(pipelineInvocationRequest);
            logger.info("Submitted " + pipelineInvocationRequest.getApplicationName() + " for processing.");
        } catch (Exception ex) {
            return Response.serverError().entity("Failed to submit pipeline, see service pod logs for additional details.").build();
        }
        return Response.ok().entity("Submitted " + pipelineInvocationRequest.getApplicationName()).build();
    }

}