从 Aurora PostgreSQL 数据库集群中调用 AWS Lambda 函数

配置 IAM 权限以将集群与 Lambda 搭配使用

1. 使用 create-policy AWS CLI 命令创建允许 Aurora PostgreSQL 数据库集群调用指定 Lambda 函数的 IAM 策略。（语句 ID (Sid) 是策略语句的可选描述，对使用没有影响。） 此策略授予 Aurora 数据库集群调用指定 Lambda 函数所需的最低权限。

   aws iam create-policy  --policy-name rds-lambda-policy --policy-document '{
       "Version": "2012-10-17",
       "Statement": [
           {
           "Sid": "AllowAccessToExampleFunction",
           "Effect": "Allow",
           "Action": "lambda:InvokeFunction",
           "Resource": "arn:aws:lambda:aws-region:444455556666:function:my-function"
           }
       ]
   }'

   您也可以使用允许您调用任何 Lambda 函数的预定义 AWSLambdaRole 策略。有关更多信息，请参阅 Lambda 的基于身份的 IAM 策略

2. 使用 create-role AWS CLI 命令创建该策略可在运行时担任的 IAM 角色。

   aws iam create-role  --role-name rds-lambda-role --assume-role-policy-document '{
       "Version": "2012-10-17",
       "Statement": [
           {
           "Effect": "Allow",
           "Principal": {
               "Service": "rds.amazonaws.com"
           },
           "Action": "sts:AssumeRole"
           }
       ]
   }'

3. 使用 attach-role-policy AWS CLI 命令将策略应用于角色。

   aws iam attach-role-policy \
       --policy-arn arn:aws:iam::444455556666:policy/rds-lambda-policy \
       --role-name rds-lambda-role --region aws-region
   

4. 使用 add-role-to-db-cluster AWS CLI 命令，将该角色应用于 Aurora PostgreSQL 数据库集群。最后一步允许数据库集群的数据库用户调用 Lambda 函数。

   aws rds add-role-to-db-cluster \
          --db-cluster-identifier my-cluster-name \
          --feature-name Lambda \
          --role-arn  arn:aws:iam::444455556666:role/rds-lambda-role   \
          --region aws-region

为 Aurora PostgreSQL 数据库集群安装 aws_lambda 扩展

使用 PostgreSQL psql 命令行或 pgAdmin 工具连接到 Aurora PostgreSQL 数据库集群。

1. 以具有 rds_superuser 权限的用户身份，连接到 Aurora PostgreSQL 数据库集群。默认 postgres 用户如示例所示。

   psql -h cluster-instance.444455556666.aws-region.rds.amazonaws.com -U postgres -p 5432

2. 安装 aws_lambda 扩展。aws_commons 扩展也是必要项。其为 PostgreSQL 的 aws_lambda 和许多其他 Aurora 扩展提供帮助程序函数。如果其尚未安装到 Aurora PostgreSQL 数据库集群上，则会按如下所示一并安装该扩展和 aws_lambda。

   CREATE EXTENSION IF NOT EXISTS aws_lambda CASCADE;
   NOTICE:  installing required extension "aws_commons"
   CREATE EXTENSION

调用示例函数

1. 使用 psql 或 pgAdmin 连接到主数据库实例。

   psql -h cluster.444455556666.aws-region.rds.amazonaws.com -U postgres -p 5432

2. 使用函数的 ARN 调用该函数。

   SELECT * from aws_lambda.invoke(aws_commons.create_lambda_function_arn('arn:aws:lambda:aws-region:444455556666:function:simple', 'us-west-1'), '{"body": "Hello from Postgres!"}'::json );

   响应如下所示。

   status_code |                        payload                        | executed_version | log_result
   -------------+-------------------------------------------------------+------------------+------------
            200 | {"statusCode": 200, "body": "\"Hello from Lambda!\""} | $LATEST          |
   (1 row)

要授予调用 Lambda 函数的其他权限，请执行以下操作：

1. 使用 psql 或 pgAdmin 连接到主数据库实例。

   psql -h cluster.444455556666.aws-region.rds.amazonaws.com -U postgres -p 5432

2. 运行以下 SQL 命令：

   postgres=>  GRANT USAGE ON SCHEMA aws_lambda TO db_username;
   GRANT EXECUTE ON ALL FUNCTIONS IN SCHEMA aws_lambda TO db_username;