如何在AWS Amplify创建Networkoperator表时插入JSON初始化数据？

核心思路 #

通过CDK定义目标数据库表（以DynamoDB为例，其他数据库可适配调整），再创建一个Lambda函数，利用CDK的自定义资源机制，让Lambda仅在表首次创建时触发，读取你准备好的JSON文件批量写入数据。

具体步骤 #

1. 准备JSON数据文件 #

把你的network-operator-data.json数据文件放在CDK项目的资源目录（比如src/assets/），确保打包时能被Lambda包含。

2. 编写Lambda初始化代码 #

Lambda的核心逻辑是：只响应创建事件、读取JSON数据、批量写入表、返回执行状态给CDK。以下是Node.js示例：

const AWS = require('aws-sdk');
const fs = require('fs');
const path = require('path');

const dynamodb = new AWS.DynamoDB.DocumentClient();

exports.handler = async (event) => {
  // 只在资源创建时执行，更新/删除操作直接跳过
  if (event.RequestType !== 'Create') {
    return sendResponse(event, 'SUCCESS', { Message: '跳过非创建事件' });
  }

  try {
    // 读取打包进来的JSON数据
    const dataPath = path.join(__dirname, 'network-operator-data.json');
    const operators = JSON.parse(fs.readFileSync(dataPath, 'utf8'));

    // 批量写入DynamoDB
    const putRequests = operators.map(operator => ({
      PutRequest: { Item: operator }
    }));

    await dynamodb.batchWrite({
      RequestItems: {
        'Networkoperator': putRequests
      }
    }).promise();

    return sendResponse(event, 'SUCCESS', { Message: '数据初始化完成' });
  } catch (error) {
    console.error('初始化失败:', error);
    return sendResponse(event, 'FAILED', { Error: error.message });
  }
};

// 向CloudFormation发送执行结果的辅助函数
function sendResponse(event, status, data) {
  const responseBody = JSON.stringify({
    Status: status,
    Reason: `查看CloudWatch日志流: ${process.env.AWS_LAMBDA_LOG_STREAM_NAME}`,
    PhysicalResourceId: event.PhysicalResourceId || `network-operator-init-${Date.now()}`,
    StackId: event.StackId,
    RequestId: event.RequestId,
    LogicalResourceId: event.LogicalResourceId,
    Data: data
  });

  const https = require('https');
  const url = new URL(event.ResponseURL);
  const options = {
    hostname: url.hostname,
    port: 443,
    path: url.pathname + url.search,
    method: 'PUT',
    headers: {
      'Content-Type': '',
      'Content-Length': responseBody.length
    }
  };

  return new Promise((resolve, reject) => {
    const req = https.request(options, (res) => resolve());
    req.on('error', (err) => reject(err));
    req.write(responseBody);
    req.end();
  });
}

3. 在CDK栈中配置资源 #

用CDK定义表、Lambda，并通过自定义资源关联，确保Lambda仅在表创建时触发。以下是TypeScript示例：

import * as cdk from 'aws-cdk-lib';
import { Construct } from 'constructs';
import * as lambda from 'aws-cdk-lib/aws-lambda';
import * as dynamodb from 'aws-cdk-lib/aws-dynamodb';
import * as cr from 'aws-cdk-lib/custom-resources';

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

    // 1. 创建Networkoperator表
    const networkOperatorTable = new dynamodb.Table(this, 'NetworkoperatorTable', {
      tableName: 'Networkoperator',
      partitionKey: { name: 'id', type: dynamodb.AttributeType.STRING }, // 按你的表结构调整主键
      removalPolicy: cdk.RemovalPolicy.RETAIN, // 根据需求调整删除策略
    });

    // 2. 创建初始化数据的Lambda函数
    const initDataLambda = new lambda.Function(this, 'InitNetworkOperatorData', {
      runtime: lambda.Runtime.NODEJS_18_X,
      handler: 'index.handler',
      code: lambda.Code.fromAsset('src/lambdas/init-data', {
        bundling: {
          image: lambda.Runtime.NODEJS_18_X.bundlingImage,
          command: [
            'bash', '-c',
            'npm install && cp ../assets/network-operator-data.json . && zip -r /asset-output/index.zip .'
          ],
        },
      }),
      environment: {
        TABLE_NAME: networkOperatorTable.tableName,
      },
    });

    // 给Lambda授予表的写入权限
    networkOperatorTable.grantWriteData(initDataLambda);

    // 3. 创建自定义资源，触发Lambda仅在创建阶段执行
    new cr.AwsCustomResource(this, 'InitNetworkOperatorResource', {
      onCreate: {
        service: 'Lambda',
        action: 'invoke',
        parameters: {
          FunctionName: initDataLambda.functionName,
          InvocationType: 'RequestResponse',
        },
        physicalResourceId: cr.PhysicalResourceId.of(`init-network-operator-${Date.now()}`),
      },
      policy: cr.AwsCustomResourcePolicy.fromSdkCalls({
        resources: cr.AwsCustomResourcePolicy.ANY_RESOURCE,
      }),
    });
  }
}

4. 部署验证 #

• 运行cdk deploy部署栈，CloudFormation会先创建表，再触发Lambda执行数据插入
• 查看CloudWatch日志确认Lambda执行成功，再检查Networkoperator表数据是否已填充

注意事项 #

• 如果是RDS等其他数据库，需把Lambda代码替换为对应数据库的SDK（比如PostgreSQL用pg），并配置Lambda的VPC、安全组以访问数据库
• 确保JSON数据的字段和表结构完全匹配，避免写入失败
• 自定义资源的onCreate仅在栈首次创建时触发，后续更新栈不会重复执行，可避免数据重复插入

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