Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异，请参阅 中国的 Amazon Web Services 服务入门 (PDF)。

教程：使用 Lambda 函数，以便在 Amazon VPC 中访问 Amazon RDS

在本教程中，您使用 Lambda 函数将数据写入 Amazon Relational Database Service（Amazon RDS）数据库。您的 Lambda 函数在 Amazon Simple Queue Service（Amazon SQS）队列中读取记录，每当添加消息时，都将新的项目写入您数据库的表中。在此示例中，您使用 Amazon Web Services Management Console 手动向队列添加消息。下图显示了您用于完成教程的 Amazon 资源。

借助 Amazon RDS，您可以使用 Microsoft SQL Server、MySQL 和 PostgreSQL 等常见数据库产品在云中运行托管关系数据库。通过使用 Lambda 访问您的数据库，您可以读取和写入数据以响应事件，例如在您的网站上注册的新客户。您的函数和数据库实例还可以自动扩展以满足高需求时期。

要完成本教程，请执行以下任务：

通过完成这些步骤，您将学习：

您可以使用 Amazon Web Services Management Console 或 Amazon Command Line Interface (Amazon CLI) 完成此教程。

先决条件

创建 Amazon RDS 数据库实例

Amazon RDS 数据库实例是在 Amazon Web Services 云 中运行的独立数据库环境。实例可以包含一个或多个由用户创建的数据库。除非您另行指定，否则 Amazon RDS 会在您 Amazon Web Services 账户 包含的默认 Amazon VPC 中创建新的数据库实例。有关 Amazon VPC 的更多信息，请参阅 Amazon Virtual Private Cloud 用户指南。

在本教程中，您在您 Amazon Web Services 账户 的默认 VPC 中创建一个新实例，并在该实例中创建一个名为 ExampleDB 的数据库。您可以使用 Amazon Web Services Management Console 或 Amazon CLI 创建 Amazon RDS 实例和数据库。

Amazon Web Services Management Console

创建数据库实例和数据库（控制台）

1. 打开 Amazon RDS 控制台的数据库页面，然后选择创建数据库。

2. 保持标准创建选项处于选中状态，然后在引擎选项中选择 MySQL。

3. 在模板部分中，选择免费套餐。

4. 在设置中，为 DB 实例标识符输入 MySQLForLambda。

5. 请通过执行以下操作设置用户名和密码：

   1. 在凭证设置中，将主用户名设置为 admin

   2. 对于主密码，输入并确认密码以访问您的数据库。

6. 通过执行以下操作指定数据库名称：

   1. 将所有剩余的默认选项保持选中状态，然后向下滚动到其他配置窗格。

   2. 展开此窗格并输入 ExampleDB 作为初始数据库名称。

7. 保持所有其余默认选项处于选中状态，然后选择创建队列。

Amazon CLI

创建数据库实例和数据库（CLI）

• 要使用 Amazon CLI 创建数据库实例和数据库，请选择自己的密码并运行以下命令。Lambda 使用此密码来访问数据库。

  aws rds create-db-instance --db-name ExampleDB --engine MySQL \
  --db-instance-identifier MySQLForLambda \
  --db-instance-class db.t2.micro --allocated-storage 5 --no-publicly-accessible \
  --master-username admin --master-user-password password

要让 Lambda 连接到您的数据库，您需要知道其主机地址（端点）和 VPC 配置。

创建函数执行角色

在创建 Lambda 函数之前，您需要创建一个执行角色来为您的函数提供必要的权限。在本教程中，Lambda 需要权限来管理与包含您的数据库实例的 Amazon VPC 的网络连接，以及轮询来自 Amazon SQS 队列的消息。

为了向您的 Lambda 函数提供其所需的权限，本教程使用 IAM 托管策略。这些策略可授予许多常见使用案例的权限，可在您的 Amazon Web Services 账户 中使用。有关使用托管策略的更多信息，请参阅 策略最佳实践。

在本教程的后面部分，您需要提供刚刚创建的执行角色的 Amazon 资源名称（ARN）。

创建 Lambda 部署包

以下示例 Python 代码使用 PyMySQL 包打开与您的数据库的连接。首次调用函数时，它还会创建一个名为 Customer 的新表。该表使用以下架构，其中 CustID 是主键：

Customer(CustID, Name)

该函数还使用 PyMySQL 向该表添加记录。该函数使用您将添加到 Amazon SQS 队列的消息中指定的客户 ID 和名称来添加记录。

请注意，该代码在处理程序函数之外创建了与数据库的连接。在初始化代码中创建连接，使后续调用能够重用该连接并提高性能。在生产应用程序中，您还可以使用预置并发来初始化所请求数量的数据库连接。调用函数后，这些连接即可用。

import sys
import logging
import pymysql
import json

# rds settings
rds_host  = "mysqlforlambda.cdipnbm2csku.cn-north-1.rds.amazonaws.com.cn"
user_name = "admin"
password = "password"
db_name = "ExampleDB"

logger = logging.getLogger()
logger.setLevel(logging.INFO)

# create the database connection outside of the handler to allow connections to be
# re-used by subsequent function invocations.
try:
    conn = pymysql.connect(host=rds_host, user=user_name, passwd=password, db=db_name, connect_timeout=5)
except pymysql.MySQLError as e:
    logger.error("ERROR: Unexpected error: Could not connect to MySQL instance.")
    logger.error(e)
    sys.exit()

logger.info("SUCCESS: Connection to RDS MySQL instance succeeded")

def lambda_handler(event, context):
    """
    This function creates a new RDS database table and writes records to it
    """
    message = event['Records'][0]['body']
    data = json.loads(message)
    CustID = data['CustID']
    Name = data['Name']

    item_count = 0
    sql_string = f"insert into Customer (CustID, Name) values({CustID}, '{Name}')"

    with conn.cursor() as cur:
        cur.execute("create table if not exists Customer ( CustID  int NOT NULL, Name varchar(255) NOT NULL, PRIMARY KEY (CustID))")
        cur.execute(sql_string)
        conn.commit()
        cur.execute("select * from Customer")
        logger.info("The following items have been added to the database:")
        for row in cur:
            item_count += 1
            logger.info(row)
    conn.commit()

    return "Added %d items to RDS MySQL table" %(item_count)
    

要在函数代码中包含 PyMySQL 依赖关系，请创建一个 .zip 部署包。

创建 Lambda 函数

现在，您可以使用刚刚创建的 .zip 程序包，使用 Amazon CLI 或 Lambda 控制台创建 Lambda 函数。

Amazon Web Services Management Console

创建函数（控制台）

1. 打开 Lambda 控制台的函数页面，然后选择创建函数。

2. 选择从头开始编写，然后在基本信息中输入 LambdaFunctionWithRDS 函数名称。

3. 选择 Python 3.9 作为运行时。

4. 选择 Create function（创建函数）。

5. 在代码窗格中，选择上传自，然后选择 .zip 文件。

6. 选择您在前一阶段创建的 lambda_function.zip 文件，然后选择保存。

现在，使用您之前创建的执行角色和 VPC 设置来配置该函数。这会授予该函数访问您的数据库实例和轮询 Amazon SQS 队列所需的权限。

配置函数

1. 在 Lambda 控制台的函数页面中，选择配置选项卡，然后选择权限。

2. 在执行角色窗格中，选择编辑。

3. 在执行角色中，从下拉列表中选择执行角色 (lambda-rds-sqs-role)。

4. 选择 Save（保存）。

5. 通过执行以下操作配置 VPC 设置：

   1. 在配置选项卡中，选择 VPC，然后选择编辑。

   2. 在 VPC 中，从下拉列表中选择您 Amazon Web Services 账户 的默认 VPC。

   3. 在子网中，选中您之前记录的 VPC 子网的复选框。

   4. 在安全组中，选中您之前记录的 VPC 安全组的复选框，然后选择保存。

Amazon CLI

创建函数（CLI）

• 使用您自己的角色 ARN、VPC 子网 ID 和 VPC 安全组值运行以下命令。

  aws lambda create-function --function-name LambdaFunctionWithRDS --runtime python3.9 \
  --zip-file fileb://lambda_function.zip --handler lambda_function.lambda_handler \
  --role arn:aws:iam::111122223333:role/lambda-vpc-sqs-role \
  --vpc-config SubnetIds=subnet-1234567890abcdef0,subnet-abcdef01234567890,\
  subnet-021345abcdef6789,subnet-1234abcdef567890,SecurityGroupIds=sg-1234567890abcdef0

在控制台中测试 Lambda 函数。

您现在可以使用 Lambda 控制台测试您的函数。您可以创建一个测试事件，该事件模仿您在教程的最后阶段使用 Amazon SQS 调用函数时将收到的数据。您的测试事件包含一个 JSON 对象，该对象指定要添加到您的函数创建的 Customer 表中的客户 ID 和客户名称。

在执行结果选项卡中，您应该看到与函数日志中显示的以下内容类似的结果：

[INFO] 2023-02-14T19:31:35.149Z bdd06682-00c7-4d6f-9abb-89f4bbb4a27f The following items have been added to the database:
[INFO] 2023-02-14T19:31:35.149Z bdd06682-00c7-4d6f-9abb-89f4bbb4a27f (1021, 'Martha Rivera')

创建 Amazon SQS 队列

您已成功测试了您的 Lambda 函数和 Amazon RDS 数据库实例的集成。现在，您创建了 Amazon SQS 队列，用于在本教程的最后阶段调用 Lambda 函数。

Amazon Web Services Management Console

创建 Amazon SQS 队列（控制台）。

1. 打开 Amazon SQS 控制台的队列页面，然后选择创建队列。

2. 将类型保留为标准，然后为您的队列名称输入 LambdaRDSQueue。

3. 保持所有默认选项处于选中状态，然后选择创建队列。

Amazon CLI

创建 Amazon SQS 队列（CLI）

• 运行以下命令并记录 Amazon CLI 返回的 URL：

  aws sqs create-queue --queue-name LambdaRDSQueue

在本教程的后面部分，您将需要刚刚创建的队列的 Amazon 资源名称（ARN）。

查找队列 ARN

• 使用您在上一步中记录的 URL，运行以下命令并记录 Amazon CLI 返回的 ARN：

  aws sqs get-queue-attributes \
  --queue-url https://sqs.cn-north-1.amazonaws.com.cn/111122223333/LambdaRDSQueue \
  --attribute-names QueueArn

创建一个事件源映射以调用您的 Lambda 函数

事件源映射是 Lambda 中的一种资源，它从流或队列中读取项目并调用 Lambda 函数。配置事件源映射时，可以指定批处理大小，以便可以将来自您的流或队列的记录批处理成单个负载。在此示例中，您将批处理大小设置为 1，这样每次向队列发送消息时都会调用 Lambda 函数。您可以使用 Amazon CLI 或 Lambda 控制台配置事件源映射。

Amazon Web Services Management Console

创建事件源映射（控制台）

1. 打开 Lambda 控制台的函数页面，然后选择一个函数 (LambdaFunctionWithRDS)。

2. 在函数概述窗格中，选择添加触发器。

3. 对于源，选择 Amazon SQS，然后选择您的队列名称 (LambdaRDSQueue)。

4. 对于批处理大小，输入 1。

5. 将所有其他选项设置为默认值，然后选择添加。

Amazon CLI

创建事件源映射（CLI）

• 要使用 Amazon CLI 创建事件源映射，请使用您自己的 Amazon SQS 队列的 ARN 运行以下命令：

  aws lambda create-event-source-mapping --function-name LambdaFunctionWithRDS --batch-size 1 \
  --event-source-arn arn:aws:sqs:us-west-2:111122223333:LambdaRDSQueue

现在，您可以通过向 Amazon SQS 队列添加一条消息来测试您的完整设置。

测试和监控您的设置

要测试您的完整设置，请使用控制台向 Amazon SQS 队列添加消息。然后，您可以使用 CloudWatch Logs 确认您的 Lambda 函数正在按预期将记录写入数据库。

清除资源

除非您想要保留为本教程创建的资源，否则可立即将其删除。通过删除您不再使用的 Amazon 资源，可防止您的 Amazon 账户产生不必要的费用。