-
-
Save ilovelili/e60e4ce430e75259df907528467de32e to your computer and use it in GitHub Desktop.
Wrap GoLang Echo server in Lambda handler for API Gateway
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| package main | |
| import( | |
| "github.com/aws/aws-lambda-go/events" | |
| "github.com/aws/aws-lambda-go/lambda" | |
| "github.com/labstack/echo" | |
| "github.com/labstack/echo/middleware" | |
| ) | |
| func main() { | |
| e := echo.New() | |
| e.Use(middleware.Logger()) | |
| e.GET("/hello", func(c echo.Context) error { | |
| return c.JSON(200, "HELLO") | |
| }) | |
| server := utils.wrapRouter(e) | |
| lambda.Start(server) | |
| } | |
| // WrapEchoServer wraps echo server into Lambda Handler | |
| func wrapRouter(e *echo.Echo) func(ctx context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) { | |
| return func(ctx context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) { | |
| body := strings.NewReader(request.Body) | |
| req := httptest.NewRequest(request.HTTPMethod, request.Path, body) | |
| for k, v := range request.Headers { | |
| req.Header.Add(k, v) | |
| } | |
| q := req.URL.Query() | |
| for k, v := range request.QueryStringParameters { | |
| q.Add(k, v) | |
| } | |
| req.URL.RawQuery = q.Encode() | |
| rec := httptest.NewRecorder() | |
| e.ServeHTTP(rec, req) | |
| res := rec.Result() | |
| responseBody, err := ioutil.ReadAll(res.Body) | |
| if err != nil { | |
| return formatAPIErrorResponse(http.StatusInternalServerError, res.Header, err.Error()) | |
| } | |
| return formatAPIResponse(res.StatusCode, res.Header, string(responseBody)) | |
| } | |
| } | |
| func formatAPIResponse(statusCode int, headers http.Header, responseData string) (events.APIGatewayProxyResponse, error) { | |
| responseHeaders := make(map[string]string) | |
| responseHeaders["Content-Type"] = "application/json" | |
| for key, value := range headers { | |
| responseHeaders[key] = "" | |
| if len(value) > 0 { | |
| responseHeaders[key] = value[0] | |
| } | |
| } | |
| responseHeaders["Access-Control-Allow-Origin"] = "*" | |
| responseHeaders["Access-Control-Allow-Headers"] = "origin,Accept,Authorization,Content-Type" | |
| return events.APIGatewayProxyResponse{ | |
| Body: responseData, | |
| Headers: responseHeaders, | |
| StatusCode: statusCode, | |
| }, nil | |
| } | |
| func formatAPIErrorResponse(statusCode int, headers http.Header, err string) (events.APIGatewayProxyResponse, error) { | |
| responseHeaders := make(map[string]string) | |
| responseHeaders["Content-Type"] = "application/json" | |
| for key, value := range headers { | |
| responseHeaders[key] = "" | |
| if len(value) > 0 { | |
| responseHeaders[key] = value[0] | |
| } | |
| } | |
| responseHeaders["Access-Control-Allow-Origin"] = "*" | |
| responseHeaders["Access-Control-Allow-Headers"] = "origin,Accept,Authorization,Content-Type" | |
| return events.APIGatewayProxyResponse{ | |
| Body: err, | |
| Headers: responseHeaders, | |
| StatusCode: statusCode, | |
| }, nil | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment