adapter.go

package algnhsa

import (
	"context"
	"encoding/json"
	"net/http"
	"net/http/httptest"

	"github.com/aws/aws-lambda-go/lambda"
)

var defaultOptions = &Options{}

type lambdaHandler struct {
	httpHandler http.Handler
	opts        *Options
}

func (handler lambdaHandler) Invoke(ctx context.Context, payload []byte) ([]byte, error) {
	resp, err := handler.handleEvent(ctx, payload)
	if err != nil {
		return nil, err
	}
	return json.Marshal(resp)
}

func (handler lambdaHandler) handleEvent(ctx context.Context, payload []byte) (lambdaResponse, error) {
	eventReq, err := newLambdaRequest(ctx, payload, handler.opts)
	if err != nil {
		return lambdaResponse{}, err
	}
	r, err := newHTTPRequest(eventReq)
	if err != nil {
		return lambdaResponse{}, err
	}
	w := httptest.NewRecorder()
	handler.httpHandler.ServeHTTP(w, r)
	return newLambdaResponse(w, handler.opts.binaryContentTypeMap)
}

// ListenAndServe starts the AWS Lambda runtime (aws-lambda-go lambda.Start) with a given handler.
func ListenAndServe(handler http.Handler, opts *Options) {
	if handler == nil {
		handler = http.DefaultServeMux
	}
	if opts == nil {
		opts = defaultOptions
	}
	opts.setBinaryContentTypeMap()
	lambda.StartHandler(lambdaHandler{httpHandler: handler, opts: opts})
}