AWS RDS PostgreSQL实例通过Java Lambda创建快照的实现问询

DevOps 小满

2026-5-15

当然可行！你完全可以在Java Lambda执行数据库更新逻辑之前，调用AWS SDK来创建RDS实例的快照。下面我会把实现步骤、代码示例和注意事项都给你讲清楚：

一、先搞定权限配置

Lambda要创建RDS快照，必须通过IAM角色获得对应的权限。你需要给Lambda的执行角色添加以下IAM策略：

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": [
                "rds:CreateDBInstanceSnapshot",
                "rds:DescribeDBInstanceSnapshots"
            ],
            "Resource": [
                "arn:aws:rds:你的区域:你的账号ID:db:你的RDS实例名",
                "arn:aws:rds:你的区域:你的账号ID:snapshot:*"
            ]
        }
    ]
}

解释下：CreateDBInstanceSnapshot是创建快照的核心权限，DescribeDBInstanceSnapshots用来后续查询快照状态（如果需要等待快照完成的话）。记得把占位符换成你自己的实际信息。

二、Java代码实现

首先，你的Lambda项目需要引入AWS RDS的SDK依赖。如果用Maven，在pom.xml里加这段：

<dependencies>
    <!-- AWS Lambda Core -->
    <dependency>
        <groupId>com.amazonaws</groupId>
        <artifactId>aws-lambda-java-core</artifactId>
        <version>1.2.21</version>
    </dependency>
    <!-- AWS RDS SDK v2（推荐使用的新版本） -->
    <dependency>
        <groupId>software.amazon.awssdk</groupId>
        <artifactId>rds</artifactId>
        <version>2.20.130</version>
    </dependency>
    <!-- 如果你用的是旧版SDK，可以用这个，但更推荐上面的v2版 -->
    <!--
    <dependency>
        <groupId>com.amazonaws</groupId>
        <artifactId>aws-java-sdk-rds</artifactId>
        <version>1.12.500</version>
    </dependency>
    -->
</dependencies>

接下来是Lambda的核心代码，我用AWS SDK v2来写（这是当前官方推荐的版本）：

import software.amazon.awssdk.regions.Region;
import software.amazon.awssdk.services.rds.RdsClient;
import software.amazon.awssdk.services.rds.model.CreateDbInstanceSnapshotRequest;
import software.amazon.awssdk.services.rds.model.CreateDbInstanceSnapshotResponse;
import software.amazon.awssdk.services.rds.model.DescribeDbInstanceSnapshotsRequest;
import software.amazon.awssdk.services.rds.model.DescribeDbInstanceSnapshotsResponse;
import software.amazon.awssdk.services.rds.model.DbInstanceSnapshot;
import com.amazonaws.services.lambda.runtime.Context;
import com.amazonaws.services.lambda.runtime.RequestHandler;
import java.time.Instant;
import java.util.concurrent.TimeUnit;

public class RdsSnapshotAndUpdateHandler implements RequestHandler<Object, String> {

    // 替换成你的RDS实例名和AWS区域
    private static final String RDS_INSTANCE_ID = "your-rds-instance-name";
    private static final Region AWS_REGION = Region.US_EAST_1;

    @Override
    public String handleRequest(Object input, Context context) {
        RdsClient rdsClient = RdsClient.builder().region(AWS_REGION).build();
        String snapshotId = null;

        try {
            // 1. 创建RDS快照：用时间戳生成唯一快照ID，避免重名
            snapshotId = "rds-snapshot-" + Instant.now().toEpochMilli();
            CreateDbInstanceSnapshotRequest snapshotRequest = CreateDbInstanceSnapshotRequest.builder()
                    .dbInstanceIdentifier(RDS_INSTANCE_ID)
                    .dbSnapshotIdentifier(snapshotId)
                    .build();

            CreateDbInstanceSnapshotResponse snapshotResponse = rdsClient.createDbInstanceSnapshot(snapshotRequest);
            context.getLogger().log("快照创建请求已提交，快照ID: " + snapshotId);

            // 2. 可选：等待快照创建完成（如果需要确保快照完成后再执行更新）
            waitForSnapshotCompletion(rdsClient, snapshotId, context);

            // 3. 执行你的数据库更新逻辑（这里替换成你原来的更新代码）
            executeDatabaseUpdate(context);

            return "快照创建完成，数据库更新执行成功";
        } catch (Exception e) {
            context.getLogger().log("操作失败: " + e.getMessage());
            throw new RuntimeException("执行快照和更新失败", e);
        } finally {
            if (rdsClient != null) {
                rdsClient.close();
            }
        }
    }

    // 轮询等待快照创建完成的方法
    private void waitForSnapshotCompletion(RdsClient rdsClient, String snapshotId, Context context) throws InterruptedException {
        boolean snapshotReady = false;
        while (!snapshotReady) {
            DescribeDbInstanceSnapshotsRequest describeRequest = DescribeDbInstanceSnapshotsRequest.builder()
                    .dbSnapshotIdentifier(snapshotId)
                    .build();

            DescribeDbInstanceSnapshotsResponse describeResponse = rdsClient.describeDbInstanceSnapshots(describeRequest);
            DbInstanceSnapshot snapshot = describeResponse.dbInstanceSnapshots().get(0);
            String status = snapshot.status();

            context.getLogger().log("当前快照状态: " + status);

            if ("available".equals(status)) {
                snapshotReady = true;
            } else if ("failed".equals(status)) {
                throw new RuntimeException("快照创建失败，状态为: failed");
            } else {
                // 每30秒轮询一次状态，避免过于频繁调用API
                TimeUnit.SECONDS.sleep(30);
            }
        }
    }

    // 你的数据库更新逻辑，替换成你原来的代码
    private void executeDatabaseUpdate(Context context) {
        context.getLogger().log("开始执行数据库更新操作...");
        // 这里写你原来的Java代码，比如用JDBC连接RDS PostgreSQL并执行更新
        // 示例伪代码：
        // try (Connection conn = DriverManager.getConnection(dbUrl, dbUser, dbPass)) {
        //     String sql = "UPDATE your_table SET ...";
        //     try (PreparedStatement stmt = conn.prepareStatement(sql)) {
        //         stmt.executeUpdate();
        //     }
        // } catch (SQLException e) {
        //     throw new RuntimeException("数据库更新失败", e);
        // }
        context.getLogger().log("数据库更新操作完成");
    }
}

三、几个关键注意事项

快照名称唯一性：我用时间戳生成快照ID，确保每次创建的快照名称不重复——RDS不允许同名快照存在，这一点一定要注意。
等待快照完成：如果你不需要严格等待快照完成再更新，可以去掉waitForSnapshotCompletion方法，直接提交快照请求后就执行更新。但如果需要数据一致性（比如确保更新前的所有数据都被快照捕获），建议等待快照状态变为available。
Lambda超时设置：快照创建时间取决于你的RDS实例大小，可能需要几分钟到几十分钟。一定要在Lambda控制台把超时时间设置得足够长（比如30分钟），避免Lambda提前终止。
成本问题：RDS快照会占用存储空间，记得定期清理不需要的旧快照，避免不必要的费用。
多AZ实例兼容性：如果你的RDS是多AZ部署，创建快照时会自动从备用AZ获取数据，不会影响主实例的性能，不用担心更新操作被阻塞。

内容的提问来源于stack exchange，提问作者ShaiNe Ram