AWS SDK Go V2 Lambda本地调试：如何自定义Context Deadline值？

awslambdarpc默认不会为本地调试的Lambda上下文设置有效截止时间，所以会返回零值时间（1970-01-01 00:00:00 +0000 UTC）。你可以通过以下两种方式自定义这个值：

方法1：通过awslambdarpc调用参数指定 #

awslambdarpc支持通过--deadline参数直接设置截止时间，支持两种格式：

• 相对时间：比如30s（当前时间加30秒）、5m（当前时间加5分钟）
• 绝对时间：RFC3339格式，比如2024-05-20T14:30:00Z

示例调用命令：

awslambdarpc invoke --function-name your-lambda-function --deadline 30s

这样调用后，上下文的截止时间就会和AWS线上环境逻辑一致，为当前时间加上指定时长。

方法2：在代码中针对本地调试场景手动覆盖 #

如果不想每次调用都添加命令行参数，可以在代码中通过环境变量判断本地调试场景，手动生成带截止时间的上下文：

import (
    "context"
    "os"
    "time"
    "github.com/aws/aws-lambda-go/events"
)

func handleLambdaEvent(ctxOrg context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
    // 本地调试时手动设置截止时间
    if os.Getenv("LOCAL_LAMBDA_DEBUG") == "true" {
        // 设置为当前时间加30秒，可根据需求调整时长
        deadline := time.Now().Add(30 * time.Second)
        var cancel context.CancelFunc
        ctxOrg, cancel = context.WithDeadline(ctxOrg, deadline)
        defer cancel()
    }

    orgDL, _ := ctxOrg.Deadline()
    fmt.Println(orgDL)
    // 后续业务逻辑...

    return events.APIGatewayProxyResponse{StatusCode: 200}, nil
}

启动本地调试时，设置环境变量LOCAL_LAMBDA_DEBUG=true即可触发自定义截止时间逻辑，线上部署时无需修改代码。

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