tech.rsqn.useful.things.springboot.lambda.SpringBootLambdaWrapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lambda-utilities Show documentation

A collection of common utilities used by rsqn.tech opensource projects.

There is a newer version: 1.0.78

package tech.rsqn.useful.things.springboot.lambda;


import com.amazonaws.services.lambda.runtime.*;
import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyRequestEvent;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.apache.commons.codec.binary.Hex;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import tech.rsqn.useful.things.lambda.model.ApiGatewayResponse;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

public abstract class SpringBootLambdaWrapper {

    private static final Logger LOG = LoggerFactory.getLogger(SpringBootLambdaWrapper.class);

    protected  ResponseEntity handleRequest(HttpServletRequest req, HttpServletResponse resp, T model, RequestHandler lambdaHandler) {
        return handleRequest(req, resp, model, lambdaHandler, new HashMap<>());
    }

    protected  ResponseEntity handleRequest(HttpServletRequest req, HttpServletResponse resp, T model, RequestHandler lambdaHandler, Map pathVariables) {
        ObjectMapper objectMapper = new ObjectMapper();
        APIGatewayProxyRequestEvent lambdaEvent = new APIGatewayProxyRequestEvent();

        // headers
        HashMap hdrs = new HashMap();
        Enumeration hdrEn = req.getHeaderNames();
        while (hdrEn.hasMoreElements()) {
            String n = hdrEn.nextElement();
            hdrs.put(n, req.getHeader(n));
        }

        // query string
        HashMap qs = new HashMap();
        Enumeration qsEn = req.getParameterNames();
        while (qsEn.hasMoreElements()) {
            String n = qsEn.nextElement();
            qs.put(n, req.getParameter(n));
        }

        // body
        if (model != null) {
            try {
                lambdaEvent.setBody(objectMapper.writeValueAsString(model));
            } catch (JsonProcessingException e) {
                throw new RuntimeException(e);
            }
        }
        lambdaEvent.setHeaders(hdrs);
        lambdaEvent.setQueryStringParameters(qs);
        lambdaEvent.setHttpMethod(req.getMethod());
        lambdaEvent.setPath(req.getRequestURI());
        lambdaEvent.setPathParameters(pathVariables);

        ApiGatewayResponse ret = (ApiGatewayResponse) lambdaHandler.handleRequest(lambdaEvent, makeContext());

        if (ret.getStatusCode() == 302) {
            try {
                resp.sendRedirect(ret.getHeaders().get("Location"));
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            return null;
        } else {

            HttpHeaders h = new HttpHeaders();
            for (String n : ret.getHeaders().keySet()) {
                h.add(n, hdrs.get(n));
            }
            return new ResponseEntity