使用 lambda 动态检索 SQS 队列 URL 通过事件或上下文

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《使用 lambda 动态检索 SQS 队列 URL 通过事件或上下文》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

我有一个 lambda 函数，它响应两个不同的 SQS 队列调用它（第二个是退避队列），因此我想动态确定消息来自哪个队列，以便我可以将其删除一次已处理完毕。

我能够创建 EventSourceMapping 来触发两个队列中的 lambda，但我看不到如何干净地检索执行 DeleteMessage 调用所需的 QueueURL。我正在使用 AWS Golang SDK。

事件消息正文中有 EventSourceArn，我似乎可以使用 GetQueueURL，但这将是一个额外的 API 调用。

避免这种情况的唯一方法（我能想到）是将两个队列 URL 作为环境变量传递，并使用 EventSourceArn 中的队列名称来查找它。

有更好的方法吗？上下文中是否隐藏着未记录的内容？

正确答案

如果您使用 lambda、sqs 和事件源映射，则无需手动从 sqs 中删除对象：When your function successfully processes a batch, Lambda deletes its messages from the queue

只需返回成功代码，该消息就会自动从相关 sqs 队列中删除

您可以从事件中的 arn 重建队列 url，如 eventsourcearn。

package main

import (
    "context"
    "fmt"

    "github.com/aws/aws-lambda-go/events"
    "github.com/aws/aws-lambda-go/lambda"
    "github.com/aws/aws-sdk-go/aws/arn"
    "github.com/aws/aws-sdk-go/aws/session"
    "github.com/aws/aws-sdk-go/service/sqs"
)

var client *sqs.SQS

func main() {
  client = sqs.New(session.Must(session.NewSessionWithOptions(session.Options{
        SharedConfigState: session.SharedConfigEnable,
    })))
    lambda.Start(Handle)
}

func Handle(ctx context.Context, event events.SQSEvent) {  
    for _, record := range event.Records {
        // Process message
        fmt.Printf(record.Body)

        // Rebuild Queue URL from ARN 
        queueArn, _ := arn.Parse(record.EventSourceARN)
        queueUrl := fmt.Sprintf("https://sqs.%v.amazonaws.com/%v/%v", queueArn.Region, queueArn.AccountID, queueArn.Resource)

        // Delete message from queue
        client.DeleteMessage(&sqs.DeleteMessageInput{
            QueueUrl:      &queueUrl,
            ReceiptHandle: &record.ReceiptHandle,
        })
    }
}

如果您使用非标准 aws 分区（例如中国），您还需要更改 url 的根域以进行匹配。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。