implement apex gateway ListenAndServe internally (#101)

* implement apex gateway ListenAndServe internally

* implement apex gateway ListenAndServe internally for api gateway v2

go.mod

@@ -14,4 +14,6 @@ require (
 	github.com/pkg/errors v0.9.1
 	github.com/shirou/gopsutil v2.20.8+incompatible
 	github.com/stretchr/testify v1.6.1
+	github.com/apex/gateway v1.1.2
+	github.com/apex/gateway/v2 v2.0.0
 )

wrappers/apexgateway/thundra_apex_gateway.go

@@ -0,0 +1,34 @@
+package apexgateway
+
+import (
+	"context"
+	"github.com/apex/gateway"
+	"github.com/thundra-io/thundra-lambda-agent-go/v2/thundra"
+	"net/http"
+
+	"github.com/aws/aws-lambda-go/events"
+	"github.com/aws/aws-lambda-go/lambda"
+)
+
+func ListenAndServe(h http.Handler) error {
+	if h == nil {
+		h = http.DefaultServeMux
+	}
+
+	lambda.Start(thundra.Wrap(wrapper(h)))
+
+	return nil
+}
+
+func wrapper(h http.Handler) func(ctx context.Context, e events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
+	return func(ctx context.Context, e events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
+		r, err := gateway.NewRequest(ctx, e)
+		if err != nil {
+			return events.APIGatewayProxyResponse{}, err
+		}
+
+		w := gateway.NewResponse()
+		h.ServeHTTP(w, r)
+		return w.End(), nil
+	}
+}

wrappers/apexgatewayv2/thundra_apex_gateway_v2.go

@@ -0,0 +1,35 @@
+package apexgatewayv2
+
+import (
+	"context"
+	"github.com/apex/gateway/v2"
+	"github.com/thundra-io/thundra-lambda-agent-go/v2/thundra"
+	"net/http"
+
+	"github.com/aws/aws-lambda-go/events"
+	"github.com/aws/aws-lambda-go/lambda"
+)
+
+func ListenAndServe(h http.Handler) error {
+	if h == nil {
+		h = http.DefaultServeMux
+	}
+
+	lambda.Start(thundra.Wrap(wrapper(h)))
+
+	return nil
+}
+
+func wrapper(h http.Handler) func(ctx context.Context, e events.APIGatewayV2HTTPRequest) (events.APIGatewayV2HTTPResponse, error) {
+	return func(ctx context.Context, e events.APIGatewayV2HTTPRequest) (events.APIGatewayV2HTTPResponse, error) {
+		r, err := gateway.NewRequest(ctx, e)
+		if err != nil {
+			return events.APIGatewayV2HTTPResponse{}, err
+		}
+
+		w := gateway.NewResponse()
+		h.ServeHTTP(w, r)
+
+		return w.End(), nil
+	}
+}