Go语言AWS Lambda兼容SQS/API Gateway触发类型识别失败求助

问题描述 #

我正在开发一个可由API Gateway和SQS触发的AWS Lambda（基于Go语言），核心代码如下：

package main

import (
    "context"
    "fmt"
    "log"

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

    regcode "company/gateway/lambda/regcode/lib"
)

func main() {

    // Start the Lambda function with both SQS and Lambda URL handlers
    lambda.Start(func(ctx context.Context, event interface{}) error {
        fmt.Println("Event type: ", event)
        switch event.(type) {
        case events.SQSEvent:
            return regcode.HandlerSqs(ctx, event.(events.SQSEvent))
        case events.APIGatewayProxyRequest:
            response, err := regcode.HandlerLambdaURL(ctx, event.(events.APIGatewayProxyRequest))
            if err != nil {
                log.Println(err)
                return err
            }
            log.Printf("API Gateway response: %v", response)
            return nil
        default:
            return fmt.Errorf("unknown event type: %T", event)
        }
    })
}

但代码中的类型断言无法识别SQS或API Gateway事件，总是进入default分支。测试SQS事件：

{
  "Records": [
    {
      "messageId": "19dd0b57-b21e-4ac1-bd88-01bbb068cb78",
      "receiptHandle": "MessageReceiptHandle",
      "body": "Hello from SQS!",
      "attributes": {
        "ApproximateReceiveCount": "1",
        "SentTimestamp": "1523232000000",
        "SenderId": "123456789012",
        "ApproximateFirstReceiveTimestamp": "1523232000001"
      },
      "messageAttributes": {},
      "md5OfBody": "{{{md5_of_body}}}",
      "md5OfMessageAttributes": "test",
      "eventSource": "aws:sqs",
      "eventSourceARN": "arn:aws:sqs:us-east-1:123456789012:MyQueue",
      "awsRegion": "us-east-1"
    }
  ]
}

会报错：

unknown event type: map[string]interface {}: errorString

尝试直接转换事件类型时，转换后sqsEvent为空且出现类型转换错误：

func main() {

    // Start the Lambda function with both SQS and Lambda URL handlers
    lambda.Start(func(ctx context.Context, event interface{}) error {
        sqsEvent, _ := event.(SQSEvent)
        // ...
    })
}

问题原因 #

AWS Lambda的Go运行时在处理事件时，若handler的参数为interface{}类型，会将JSON格式的事件自动反序列化为map[string]interface{}，而不是直接转换为events.SQSEvent或events.APIGatewayProxyRequest结构体。因此直接对interface{}进行类型断言会失败，因为实际类型是map而非目标结构体。

解决方案 #

需要手动将map[string]interface{}转换为对应的事件结构体，步骤如下：

1. 先将事件断言为map[string]interface{}类型
2. 通过检查map中的特征字段判断事件类型（比如SQS事件包含Records字段，API Gateway事件包含resource字段）
3. 将map序列化为JSON字节流，再反序列化为目标事件结构体

修改后的核心代码如下：

package main

import (
    "context"
    "encoding/json"
    "fmt"
    "log"

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

    regcode "company/gateway/lambda/regcode/lib"
)

func main() {
    lambda.Start(func(ctx context.Context, event interface{}) error {
        fmt.Println("Event type: ", event)
        
        // 先断言为map类型
        eventMap, ok := event.(map[string]interface{})
        if !ok {
            return fmt.Errorf("unsupported event type: %T", event)
        }

        // 判断是否为SQS事件
        if _, hasRecords := eventMap["Records"]; hasRecords {
            // 将map转为JSON字节流
            jsonBytes, err := json.Marshal(eventMap)
            if err != nil {
                return fmt.Errorf("failed to marshal event map: %v", err)
            }
            // 反序列化为SQSEvent
            var sqsEvent events.SQSEvent
            if err := json.Unmarshal(jsonBytes, &sqsEvent); err != nil {
                return fmt.Errorf("failed to unmarshal to SQSEvent: %v", err)
            }
            return regcode.HandlerSqs(ctx, sqsEvent)
        }

        // 判断是否为API Gateway事件
        if _, hasResource := eventMap["resource"]; hasResource {
            jsonBytes, err := json.Marshal(eventMap)
            if err != nil {
                return fmt.Errorf("failed to marshal event map: %v", err)
            }
            var apiEvent events.APIGatewayProxyRequest
            if err := json.Unmarshal(jsonBytes, &apiEvent); err != nil {
                return fmt.Errorf("failed to unmarshal to APIGatewayProxyRequest: %v", err)
            }
            response, err := regcode.HandlerLambdaURL(ctx, apiEvent)
            if err != nil {
                log.Println(err)
                return err
            }
            log.Printf("API Gateway response: %v", response)
            return nil
        }

        return fmt.Errorf("unknown event type: %T", event)
    })
}

额外说明 #

• 不要忽略类型断言或反序列化的错误，这些错误能帮你快速定位转换问题
• 如果需要处理更多类型的事件，可以通过添加对应的特征字段检查和结构体转换来扩展
• 若只需要处理单一类型事件，直接将handler的参数定义为对应结构体即可（比如func handler(ctx context.Context, event events.SQSEvent) error），Lambda运行时会自动完成反序列化

内容的提问来源于stack exchange，提问作者gmilleo