使用 golang 调用另一个 Lambda 函数的方法

在 Go 语言中调用其他 Lambda 函数时，可以通过 Amazon Web Services (AWS) Lambda 函数服务来实现。该服务提供了调用 Lambda 函数并接收其响应的机制。本文档将指导你如何使用 Go 语言中的 Lambda SDK 来调用另一个 Lambda 函数，包括设置请求有效负载、处理响应以及解决常见问题。

我正在尝试在另一个 lambda 函数中调用一个 lambda 函数。我可以调用 lambda 函数，但是，我似乎无法让使用的 lambda 函数从发送的 lambda 函数接收有效负载/主体。

有关调用 lambda 函数的 lambda go 文档

这是我的发送/调用 lambda 函数

type response events.apigatewayproxyresponse

func handler(ctx context.context) (response, error) {
    region := os.getenv("aws_region")
    session, err := session.newsession(&aws.config{ // use aws sdk to connect to dynamodb
        region: &region,
    })
    svc := invoke.new(session)

    payload, err := json.marshal(map[string]interface{}{
        "message": "message to other lambda func",
    })

    if err != nil {
        fmt.println("json marshalling error")
    }
    input := &invoke.invokeinput{
        functionname:   aws.string("invokeconsume"),
        invocationtype: aws.string("requestresponse"),
        logtype:        aws.string("tail"),
        payload:        payload,
    }
    result, err := svc.invoke(input)
    if err != nil {
        fmt.println("error")
        fmt.println(err.error())
    }
    var m map[string]interface{}
    json.unmarshal(result.payload, &m)
    fmt.println(m["body"])

    body, err := json.marshal(m["body"])
    resp := response{
        statuscode:      200,
        isbase64encoded: false,
        headers: map[string]string{
            "content-type": "application/json",
        },
        body: string(body),
    }
    fmt.println(resp)

    return resp, nil
}
func main() {
    lambda.start(handler)
}

我从调用的 lambda 中得到的响应...

{200 map[content-type:application/json] "{\"message\":\"something\"}" false}

我使用的 lambda 函数

type response events.apigatewayproxyresponse

func handler(ctx context.context, req events.apigatewayproxyrequest) (response, error) {
    fmt.println(req)

    var m map[string]interface{}
    err := json.unmarshal([]byte(req.body), &m)
    if err != nil {
        fmt.println("json unmarshalling error")
        fmt.println(err.error())
    }
    fmt.println(m)

    body, _ := json.marshal(map[string]interface{}{
        "message": "something",
    })
    resp := response{
        statuscode:      200,
        isbase64encoded: false,
        headers: map[string]string{
            "content-type": "application/json",
        },
        body: string(body),
    }
    return resp, nil
}
func main() {
    lambda.start(handler)
}

使用 lambda 函数的日志

{ map[] map[] map[] map[] { { } map[] } false}
json unmarshalling error
unexpected end of json input
map[]

似乎使用 lambda 函数没有接收任何 events.apigatewayproxyrequest，但我不确定为什么。

编辑： 我的解决方案 - 我还必须在有效负载中包含 json 主体对象。这是我解决的方法

body, err := json.Marshal(map[string]interface{}{
    "name": "Jimmy",
})
type Payload struct {
    Body string `json:"body"`
}
p := Payload{
    Body: string(body),
}
payload, err := json.Marshal(p) // This should give you {"body":"{\"name\":\"Jimmy\"}"} if you print it out which is the required format for the lambda request body.

解决方案

感谢 @yorodm 的输入，我也必须将 body 对象包含到有效负载中。

这是使用 golang 在另一个 lambda 函数中调用 lambda 函数的完整解决方案

region := os.Getenv("AWS_REGION")
session, err := session.NewSession(&aws.Config{ // Use aws sdk to connect to dynamoDB
    Region: &region,
})
svc := invoke.New(session)

body, err := json.Marshal(map[string]interface{}{
    "name": "Jimmy",
})

type Payload struct {
    // You can also include more objects in the structure like below, 
    // but for my purposes body was all that was required
    // Method string `json:"httpMethod"`
    Body string `json:"body"`
}
p := Payload{
    // Method: "POST",
    Body: string(body),
}
payload, err := json.Marshal(p) 
 // Result should be: {"body":"{\"name\":\"Jimmy\"}"} 
 // This is the required format for the lambda request body.

if err != nil {
    fmt.Println("Json Marshalling error")
}
fmt.Println(string(payload))

input := &invoke.InvokeInput{
    FunctionName:   aws.String("invokeConsume"),
    InvocationType: aws.String("RequestResponse"),
    LogType:        aws.String("Tail"),
    Payload:        payload,
}
result, err := svc.Invoke(input)
if err != nil {
    fmt.Println("error")
    fmt.Println(err.Error())
}
var m map[string]interface{}
json.Unmarshal(result.Payload, &m)
fmt.Println(m["body"])

invokeReponse, err := json.Marshal(m["body"])
resp := Response{
    StatusCode:      200,
    IsBase64Encoded: false,
    Headers: map[string]string{
        "Content-Type": "application/json",
    },
    Body: string(invokeReponse),
}
fmt.Println(resp)

return resp, nil

终于介绍完啦！小伙伴们，这篇关于《使用 golang 调用另一个 Lambda 函数的方法》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！