fix handling of different event types

functions/ws/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 
 	"github.com/aws/aws-lambda-go/events"
@@ -13,16 +14,22 @@ func main() {
 	lambda.Start(handler)
 }
 
-func handler(ctx context.Context, req interface{}) (interface{}, error) {
+func handler(ctx context.Context, event interface{}) (interface{}, error) {
 	h, err := ws.NewHandler()
 	if err != nil {
 		return nil, err
 	}
-	switch req.(type) {
-	case events.APIGatewayProxyRequest:
-		return h.HandleApiGatewayRequest(ctx, req.(events.APIGatewayWebsocketProxyRequest))
-	case events.SQSEvent:
-		return nil, h.HandleSQSEvent(ctx, req.(events.SQSEvent))
+	data, err := json.Marshal(event)
+	if err != nil {
+		return nil, err
+	}
+	gwr := events.APIGatewayWebsocketProxyRequest{}
+	if err := json.Unmarshal(data, &gwr); err == nil {
+		return h.HandleApiGatewayRequest(ctx, gwr)
+	}
+	sqsr := events.SQSEvent{}
+	if err := json.Unmarshal(data, &sqsr); err == nil {
+		return nil, h.HandleSQSEvent(ctx, sqsr)
 	}
 	return nil, fmt.Errorf("unknown event type")
 }