Everyrealm-AWS-CDK / lib /dynamodb-stack.ts

create aws cdk

2d78d32 over 1 year ago

3.8 kB

import * as cdk from 'aws-cdk-lib';
import * as dynamodb from 'aws-cdk-lib/aws-dynamodb';
import * as cr from 'aws-cdk-lib/custom-resources';
import * as iam from 'aws-cdk-lib/aws-iam';
import { Construct } from 'constructs';
import { AppConfig } from './config';
import { generateSeedData } from './seed-data';

export interface DynamoDBStackProps extends cdk.StackProps {
  pointInTimeRecovery?: boolean;
  removalPolicy?: cdk.RemovalPolicy;
  deploymentEnvironment?: string;
  useStaticTableName?: boolean;
  config: AppConfig;
}

export class DynamoDBStack extends cdk.Stack {
  public readonly bonusClaimsTable: dynamodb.Table;

  constructor(scope: Construct, id: string, props: DynamoDBStackProps) {
    super(scope, id, props);

    // Create DynamoDB table with conditional naming
    const tableProps: Omit<dynamodb.TableProps, 'tableName'> = {
      partitionKey: { name: 'PK', type: dynamodb.AttributeType.STRING },
      sortKey: { name: 'SK', type: dynamodb.AttributeType.STRING },
      billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
      removalPolicy: props.removalPolicy || cdk.RemovalPolicy.DESTROY,
      pointInTimeRecoverySpecification: {
        pointInTimeRecoveryEnabled:
          props.pointInTimeRecovery !== undefined ? props.pointInTimeRecovery : true,
      },
    };

    // Create table with or without static name
    this.bonusClaimsTable = props.useStaticTableName
      ? new dynamodb.Table(this, 'BonusClaimsTable', {
          ...tableProps,
          tableName: `bonus-claims-${props.deploymentEnvironment || props.config.environment}`,
        })
      : new dynamodb.Table(this, 'BonusClaimsTable', tableProps);

    // Add GSI for querying by timestamp
    this.bonusClaimsTable.addGlobalSecondaryIndex({
      indexName: 'TimestampIndex',
      partitionKey: { name: 'PK', type: dynamodb.AttributeType.STRING },
      sortKey: { name: 'timestamp', type: dynamodb.AttributeType.STRING },
    });

    // Seed data using custom resource
    const seedDataRole = new iam.Role(this, 'SeedDataRole', {
      assumedBy: new iam.ServicePrincipal('lambda.amazonaws.com'),
    });

    seedDataRole.addManagedPolicy(
      iam.ManagedPolicy.fromAwsManagedPolicyName('service-role/AWSLambdaBasicExecutionRole')
    );

    seedDataRole.addToPolicy(
      new iam.PolicyStatement({
        actions: ['dynamodb:BatchWriteItem', 'dynamodb:PutItem', 'dynamodb:DescribeTable'],
        resources: [this.bonusClaimsTable.tableArn],
      })
    );

    // Get seed data from the separate module
    const seedData = generateSeedData();

    // Create custom resource to seed data
    new cr.AwsCustomResource(this, 'SeedBonusClaimsData', {
      onCreate: {
        service: 'DynamoDB',
        action: 'batchWriteItem',
        parameters: {
          RequestItems: {
            [this.bonusClaimsTable.tableName]: seedData,
          },
        },
        physicalResourceId: cr.PhysicalResourceId.of(Date.now().toString()), // Use timestamp to force update
      },
      policy: cr.AwsCustomResourcePolicy.fromStatements([
        new iam.PolicyStatement({
          actions: ['dynamodb:BatchWriteItem', 'dynamodb:PutItem', 'dynamodb:DescribeTable'],
          resources: [this.bonusClaimsTable.tableArn],
        }),
      ]),
      role: seedDataRole,
    });

    // Add outputs
    new cdk.CfnOutput(this, 'BonusClaimsTableName', {
      value: this.bonusClaimsTable.tableName,
      description: 'The name of the DynamoDB table for bonus claims',
      exportName: `${this.stackName}-BonusClaimsTableName`,
    });

    new cdk.CfnOutput(this, 'BonusClaimsTableArn', {
      value: this.bonusClaimsTable.tableArn,
      description: 'The ARN of the DynamoDB table for bonus claims',
      exportName: `${this.stackName}-BonusClaimsTableArn`,
    });
  }
}

	import * as cdk from 'aws-cdk-lib';
	import * as dynamodb from 'aws-cdk-lib/aws-dynamodb';
	import * as cr from 'aws-cdk-lib/custom-resources';
	import * as iam from 'aws-cdk-lib/aws-iam';
	import { Construct } from 'constructs';
	import { AppConfig } from './config';
	import { generateSeedData } from './seed-data';

	export interface DynamoDBStackProps extends cdk.StackProps {
	pointInTimeRecovery?: boolean;
	removalPolicy?: cdk.RemovalPolicy;
	deploymentEnvironment?: string;
	useStaticTableName?: boolean;
	config: AppConfig;
	}

	export class DynamoDBStack extends cdk.Stack {
	public readonly bonusClaimsTable: dynamodb.Table;

	constructor(scope: Construct, id: string, props: DynamoDBStackProps) {
	super(scope, id, props);

	// Create DynamoDB table with conditional naming
	const tableProps: Omit<dynamodb.TableProps, 'tableName'> = {
	partitionKey: { name: 'PK', type: dynamodb.AttributeType.STRING },
	sortKey: { name: 'SK', type: dynamodb.AttributeType.STRING },
	billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,
	removalPolicy: props.removalPolicy \|\| cdk.RemovalPolicy.DESTROY,
	pointInTimeRecoverySpecification: {
	pointInTimeRecoveryEnabled:
	props.pointInTimeRecovery !== undefined ? props.pointInTimeRecovery : true,
	},
	};

	// Create table with or without static name
	this.bonusClaimsTable = props.useStaticTableName
	? new dynamodb.Table(this, 'BonusClaimsTable', {
	...tableProps,
	tableName: `bonus-claims-${props.deploymentEnvironment \|\| props.config.environment}`,
	})
	: new dynamodb.Table(this, 'BonusClaimsTable', tableProps);

	// Add GSI for querying by timestamp
	this.bonusClaimsTable.addGlobalSecondaryIndex({
	indexName: 'TimestampIndex',
	partitionKey: { name: 'PK', type: dynamodb.AttributeType.STRING },
	sortKey: { name: 'timestamp', type: dynamodb.AttributeType.STRING },
	});

	// Seed data using custom resource
	const seedDataRole = new iam.Role(this, 'SeedDataRole', {
	assumedBy: new iam.ServicePrincipal('lambda.amazonaws.com'),
	});

	seedDataRole.addManagedPolicy(
	iam.ManagedPolicy.fromAwsManagedPolicyName('service-role/AWSLambdaBasicExecutionRole')
	);

	seedDataRole.addToPolicy(
	new iam.PolicyStatement({
	actions: ['dynamodb:BatchWriteItem', 'dynamodb:PutItem', 'dynamodb:DescribeTable'],
	resources: [this.bonusClaimsTable.tableArn],
	})
	);

	// Get seed data from the separate module
	const seedData = generateSeedData();

	// Create custom resource to seed data
	new cr.AwsCustomResource(this, 'SeedBonusClaimsData', {
	onCreate: {
	service: 'DynamoDB',
	action: 'batchWriteItem',
	parameters: {
	RequestItems: {
	[this.bonusClaimsTable.tableName]: seedData,
	},
	},
	physicalResourceId: cr.PhysicalResourceId.of(Date.now().toString()), // Use timestamp to force update
	},
	policy: cr.AwsCustomResourcePolicy.fromStatements([
	new iam.PolicyStatement({
	actions: ['dynamodb:BatchWriteItem', 'dynamodb:PutItem', 'dynamodb:DescribeTable'],
	resources: [this.bonusClaimsTable.tableArn],
	}),
	]),
	role: seedDataRole,
	});

	// Add outputs
	new cdk.CfnOutput(this, 'BonusClaimsTableName', {
	value: this.bonusClaimsTable.tableName,
	description: 'The name of the DynamoDB table for bonus claims',
	exportName: `${this.stackName}-BonusClaimsTableName`,
	});

	new cdk.CfnOutput(this, 'BonusClaimsTableArn', {
	value: this.bonusClaimsTable.tableArn,
	description: 'The ARN of the DynamoDB table for bonus claims',
	exportName: `${this.stackName}-BonusClaimsTableArn`,
	});
	}
	}