如何使用TypeScript版CDK在已创建的Postgres RDS中创建表？

嘿，我懂你现在的困扰——CDK本身并没有直接提供创建RDS表的原生能力，因为它的核心是管理基础设施资源，而表属于数据库内部的Schema层内容。不过别担心，有几种实用的方法能让你在CDK的部署流程里自动完成表的创建，下面给你详细拆解：

RDS支持在实例首次启动时执行指定的SQL脚本，你可以把创建表的逻辑写在SQL文件里，通过CDK将脚本上传到S3，再关联到RDS实例上。

步骤1：编写初始化SQL脚本 #

创建一个init.sql文件，写入建表逻辑：

-- 创建用户表（如果不存在的话）
CREATE TABLE IF NOT EXISTS users (
    id SERIAL PRIMARY KEY,
    name VARCHAR(50) NOT NULL,
    email VARCHAR(100) UNIQUE NOT NULL,
    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);

步骤2：CDK代码实现 #

import * as cdk from 'aws-cdk-lib';
import * as rds from 'aws-cdk-lib/aws-rds';
import * as s3assets from 'aws-cdk-lib/aws-s3-assets';
import * as ec2 from 'aws-cdk-lib/aws-ec2';
import { Construct } from 'constructs';

export class RdsTableDemoStack extends cdk.Stack {
  constructor(scope: Construct, id: string, props?: cdk.StackProps) {
    super(scope, id, props);

    // 假设你已经有了VPC配置，这里替换成你自己的VPC
    const vpc = ec2.Vpc.fromLookup(this, 'ExistingVpc', {
      vpcId: 'vpc-xxxxxx',
    });

    // 将SQL脚本打包成S3资产
    const initScript = new s3assets.Asset(this, 'DbInitScript', {
      path: './init.sql', // 脚本的本地路径
    });

    // 创建Postgres RDS实例
    const dbInstance = new rds.DatabaseInstance(this, 'PostgresDb', {
      engine: rds.DatabaseInstanceEngine.postgres({
        version: rds.PostgresEngineVersion.VER_15_4,
      }),
      instanceType: ec2.InstanceType.of(ec2.InstanceClass.T3, ec2.InstanceSize.MICRO),
      credentials: rds.Credentials.fromGeneratedSecret('postgres'),
      vpc: vpc,
      vpcSubnets: {
        subnetType: ec2.SubnetType.PRIVATE_ISOLATED,
      },
      // 指定初始化脚本
      initScript: initScript,
    });

    // 给RDS实例权限读取S3里的初始化脚本
    initScript.grantRead(dbInstance);
  }
}

⚠️ 注意：这个脚本只会在RDS实例第一次创建时执行，如果后续需要修改表结构，得用其他方式（比如数据库迁移工具）。

如果需要更灵活的控制（比如在实例更新时也能执行表结构变更），可以用CDK的Custom Resource结合Lambda函数，在RDS部署完成后自动触发SQL执行。

步骤1：编写Lambda函数（用于执行SQL） #

创建create-table-lambda/index.ts文件：

import { Client } from 'pg';
import { CloudFormationCustomResourceEvent } from 'aws-lambda';

export async function handler(event: CloudFormationCustomResourceEvent) {
  // 从环境变量获取数据库连接信息
  const dbConfig = {
    host: process.env.DB_HOST!,
    port: parseInt(process.env.DB_PORT!),
    user: process.env.DB_USER!,
    password: process.env.DB_PASSWORD!,
    database: 'postgres',
  };

  const client = new Client(dbConfig);

  try {
    await client.connect();

    // 根据CloudFormation事件类型处理逻辑（这里只处理Create/Update）
    if (event.RequestType === 'Create' || event.RequestType === 'Update') {
      const createTableQuery = `
        CREATE TABLE IF NOT EXISTS users (
          id SERIAL PRIMARY KEY,
          name VARCHAR(50) NOT NULL,
          email VARCHAR(100) UNIQUE NOT NULL,
          created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        );
        -- 可以在这里添加更多表结构变更逻辑
      `;
      await client.query(createTableQuery);
      console.log('表创建/更新成功');
    }

    return { Status: 'SUCCESS' };
  } catch (error) {
    console.error('执行SQL出错:', error);
    return { Status: 'FAILED', Reason: (error as Error).message };
  } finally {
    await client.end();
  }
}

记得给Lambda安装依赖：在create-table-lambda目录下运行npm init -y && npm install pg。

步骤2：CDK代码关联Lambda和Custom Resource #

import * as cdk from 'aws-cdk-lib';
import * as rds from 'aws-cdk-lib/aws-rds';
import * as lambda from 'aws-cdk-lib/aws-lambda';
import * as customResources from 'aws-cdk-lib/custom-resources';
import * as ec2 from 'aws-cdk-lib/aws-ec2';
import { Construct } from 'constructs';

export class RdsTableDemoStack extends cdk.Stack {
  constructor(scope: Construct, id: string, props?: cdk.StackProps) {
    super(scope, id, props);

    const vpc = ec2.Vpc.fromLookup(this, 'ExistingVpc', {
      vpcId: 'vpc-xxxxxx',
    });

    // 创建RDS实例
    const dbInstance = new rds.DatabaseInstance(this, 'PostgresDb', {
      engine: rds.DatabaseInstanceEngine.postgres({
        version: rds.PostgresEngineVersion.VER_15_4,
      }),
      instanceType: ec2.InstanceType.of(ec2.InstanceClass.T3, ec2.InstanceSize.MICRO),
      credentials: rds.Credentials.fromGeneratedSecret('postgres'),
      vpc: vpc,
      vpcSubnets: {
        subnetType: ec2.SubnetType.PRIVATE_ISOLATED,
      },
    });

    // 创建执行SQL的Lambda函数
    const createTableLambda = new lambda.Function(this, 'CreateTableLambda', {
      runtime: lambda.Runtime.NODEJS_18_X,
      handler: 'index.handler',
      code: lambda.Code.fromAsset('create-table-lambda'),
      environment: {
        DB_HOST: dbInstance.dbInstanceEndpointAddress,
        DB_PORT: dbInstance.dbInstanceEndpointPort,
        DB_USER: 'postgres',
        DB_PASSWORD: dbInstance.secret?.secretValueFromJson('password').toString()!,
      },
      vpc: vpc,
      vpcSubnets: {
        subnetType: ec2.SubnetType.PRIVATE_ISOLATED,
      },
    });

    // 允许Lambda访问RDS的5432端口
    dbInstance.connections.allowFrom(createTableLambda, ec2.Port.tcp(5432));

    // 创建Custom Resource，触发Lambda执行
    new customResources.CustomResource(this, 'CreateTableResource', {
      serviceToken: new customResources.Provider(this, 'CreateTableProvider', {
        onEventHandler: createTableLambda,
      }).serviceToken,
    });
  }
}

如果你的表结构变更比较频繁，推荐用专业的数据库迁移工具（Flyway或Liquibase）。你可以在CDK部署完RDS后，用Lambda触发迁移工具执行脚本，或者在CI/CD流程里添加迁移步骤。这种方式更适合复杂的Schema管理场景。

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