com.amazonaws.serverless.proxy.spring.SpringBootProxyHandlerBuilder Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
 * with the License. A copy of the License is located at
 *
 * http://aws.amazon.com/apache2.0/
 *
 * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES
 * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions
 * and limitations under the License.
 */
package com.amazonaws.serverless.proxy.spring;

import com.amazonaws.serverless.exceptions.ContainerInitializationException;
import com.amazonaws.serverless.proxy.ExceptionHandler;
import com.amazonaws.serverless.proxy.internal.servlet.ServletLambdaContainerHandlerBuilder;
import com.amazonaws.serverless.proxy.model.AwsProxyResponse;
import org.springframework.boot.WebApplicationType;

import jakarta.servlet.http.HttpServletRequest;

public final class SpringBootProxyHandlerBuilder extends ServletLambdaContainerHandlerBuilder<
        RequestType,
            AwsProxyResponse,
            HttpServletRequest,
            SpringBootLambdaContainerHandler,
            SpringBootProxyHandlerBuilder> {
    private Class springBootInitializer;
    private String[] profiles;
    private WebApplicationType applicationType = WebApplicationType.REACTIVE;

    @Override
    protected SpringBootProxyHandlerBuilder self() {
        return this;
    }


    public SpringBootProxyHandlerBuilder springBootApplication(Class app) {
        springBootInitializer = app;
        return self();
    }

    public SpringBootProxyHandlerBuilder profiles(String... profiles) {
        this.profiles = profiles;
        return self();
    }

    public SpringBootProxyHandlerBuilder servletApplication() {
        this.applicationType = WebApplicationType.SERVLET;
        return self();
    }

    @Override
    public SpringBootLambdaContainerHandler build() throws ContainerInitializationException {
        validate();
        if (springBootInitializer == null) {
            throw new ContainerInitializationException("Missing spring boot application class in builder", null);
        }
        SpringBootLambdaContainerHandler handler =  new SpringBootLambdaContainerHandler(
                requestTypeClass,
                responseTypeClass,
                requestReader,
                responseWriter,
                securityContextWriter,
                exceptionHandler,
                springBootInitializer,
                initializationWrapper,
                applicationType
        );
        if (profiles != null) {
            handler.activateSpringProfiles(profiles);
        }
        return handler;
    }

    @Override
    public SpringBootLambdaContainerHandler buildAndInitialize() throws ContainerInitializationException {
        SpringBootLambdaContainerHandler handler = build();
        initializationWrapper.start(handler);
        return handler;
    }

    @Override
    protected ExceptionHandler defaultExceptionHandler() {
        return new SpringBootAwsProxyExceptionHandler();
    }
}