Utility method to create an AWS Timestream Database

timestream-db.ts

import { RemovalPolicy } from 'aws-cdk-lib';
import { CfnDatabase, CfnTable } from 'aws-cdk-lib/aws-timestream';
import { Construct } from 'constructs';

export interface TimeStreamConstructProps {
  databaseName: string;
  tableName: string;
}

export class TimeStreamConstruct extends Construct {
  public readonly database: CfnDatabase;
  public readonly table: CfnTable;

  constructor(scope: Construct, id: string, props: TimeStreamConstructProps) {
    super(scope, id);
    this.database = new CfnDatabase(this, props.databaseName+'Cfn', {
      databaseName: props.databaseName,
    });
    this.database.applyRemovalPolicy(RemovalPolicy.DESTROY);

    this.table = new CfnTable(this, props.tableName+'Cfn', {
      tableName: props.tableName,
      databaseName: props.databaseName,
      retentionProperties: {
        memoryStoreRetentionPeriodInHours: (24 * 60).toString(10),
        magneticStoreRetentionPeriodInDays: (60 * 10).toString(10),
      },
    });
    this.table.node.addDependency(this.database);
    this.table.applyRemovalPolicy(RemovalPolicy.DESTROY);
  }
}

export const addTimeStreamDB = (
    scope: Construct,
    id: string,
    databaseName: string,
    tableName:string) => {
  return new TimeStreamConstruct(scope, id, {
    databaseName: databaseName,
    tableName: tableName,
  });
};