使用AWS Lambda将记录推送到AWS DynamoDB

Question

我正在尝试将一条记录从AWS Lambda函数推送到AWS DynamoDB。我可以使用AWSAPI成功地从我的pc上推送一条记录，但是当我将代码部署到AWS时，我就卡住了。

下面是可以从我的电脑成功推送记录的示例代码：

const AWS = require("aws-sdk");
AWS.config.update({
  region: "eu-west-1",
  endpoint: "https://dynamodb.eu-west-1.amazonaws.com"
});
const docClient = new AWS.DynamoDB.DocumentClient();

const params = {
  TableName: "hello-world-bot",
  Item: {
    timestamp: Date.now(),
    chat_id: 123,
    text: 'text',
  },
};

console.log("Adding a new item...");
docClient.put(params, function(err, data) {
    if (err) {
        console.error("Unable to add item. Error JSON:", JSON.stringify(err, null, 2));
    } else {
        console.log("Added item:", JSON.stringify(data, null, 2));
    }
});

以及DynamoDB中的结果：来自DynamoDB控制台的屏幕截图

下面是我试图部署到Lambda的代码：

const AWS = require("aws-sdk");
AWS.config.update({
  region: "eu-west-1",
  endpoint: "https://dynamodb.eu-west-1.amazonaws.com",
});
const docClient = new AWS.DynamoDB.DocumentClient();

module.exports.helloworld = (event) => {
  const params = {
    TableName: "hello-world-bot",
    Item: {
      timestamp: Date.now(),
      chat_id: 123,
      text: "text",
    },
  };

  console.log("Adding a new item...");
  docClient.put(params, function (err, data) {
    if (err) {
      console.error(
        "Unable to add item. Error JSON:",
        JSON.stringify(err, null, 2)
      );
    } else {
      console.log("Added item:", JSON.stringify(data, null, 2));
    }
  });
  return { statusCode: 200 };
};

问题是，当Lambda函数运行时，DynamoDB中没有任何记录。以下是我拉取Lambda外部API并尝试运行函数时CloudWatch的日志：

START RequestId: d9e171c2-a763-4df1-9ec3-06d79eb68a1e Version: $LATEST
2021-08-13T06:03:01.974Z    d9e171c2-a763-4df1-9ec3-06d79eb68a1e    INFO    Adding a new item...
END RequestId: d9e171c2-a763-4df1-9ec3-06d79eb68a1e
REPORT RequestId: d9e171c2-a763-4df1-9ec3-06d79eb68a1e  Duration: 45.04 ms  Billed Duration: 46 ms  Memory Size: 1024 MB    Max Memory Used: 88 MB  Init Duration: 524.54 ms

正如您所看到的，没有任何关于权限或其他方面的错误。以下是用户权限，其凭据用于Lambda中的此函数：

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": [
                "dynamodb:BatchGetItem",
                "dynamodb:GetItem",
                "dynamodb:Query",
                "dynamodb:Scan",
                "dynamodb:BatchWriteItem",
                "dynamodb:PutItem",
                "dynamodb:UpdateItem"
            ],
            "Resource": "arn:aws:dynamodb:eu-west-1:120400915578:table/hello-world-bot"
        }
    ]
}

Answer 1

您的返回{statusCode:200}结束lambda调用，但不会等待put请求完成。事实上，它甚至可能不会等到它开始。

从put回调返回主函数，或者使用promise。promise的例子：

module.exports.helloworld = async (event) => { // <- use async
  const params = {
    TableName: "hello-world-bot",
    Item: {
      timestamp: Date.now(),
      chat_id: 123,
      text: "text",
    },
  }

  console.log("Adding a new item...")
  // Create a promise and return it
  return docClient
    .put(params)
    .promise()
    .then(() => {
      console.log("success")
      return {statusCode: 200}
    })
    .catch(e => {
      console.log("error: ", e)
      return {statusCode: 500}
    })
}

此外，正如@jens在评论中指出的，不需要明确地设置dynamodbendpoint

Answer 2

正如我在另一个答案的评论中指出的，他们的回报声明似乎并不完全正确。我更喜欢对lambda和AWS SDK开箱即用支持的promise使用异步等待语法。

这会将您的代码更改为如下所示：

const AWS = require("aws-sdk");
const docClient = new AWS.DynamoDB.DocumentClient();

module.exports.helloworld = async (event) => {
  const params = {
    TableName: "hello-world-bot",
    Item: {
      timestamp: Date.now(),
      chat_id: 123,
      text: "text",
    },
  };

  console.log("Adding a new item...");
  try {
    await docClient.put(params).promise();
    return { statusCode: 200 };
  } catch (err) {
    console.error("Unable to add item. Error JSON:",
        JSON.stringify(err, null, 2));
    return { statusCode: 500 };
  }
};

PS：正如评论中所指出的，如果您使用的是默认endpoint，则不需要为DynamoDB设置endpoint。由于您以Lambda的形式运行函数，AWS_REGION环境变量将自动设置，因此您也无需自行配置（除非您希望访问另一个区域中的DynamoDB表）。