dynamodb-helpers.ts

import {
  DeleteItemCommand,
  DeleteItemCommandOutput,
  DynamoDBClient,
  PutItemCommand,
  PutItemCommandOutput,
} from '@aws-sdk/client-dynamodb';
import { QueryCommand, QueryCommandOutput } from '@aws-sdk/lib-dynamodb';
import { marshall } from '@aws-sdk/util-dynamodb';
import { REGION, TTL_TABLE_NAME } from 'constants/backend-constants';
import { TtlDynamodbRecord } from 'utils';

const dynamoDbClient = new DynamoDBClient({ region: REGION });

export async function getTtl(messageUri: string) {
  const params = {
    TableName: TTL_TABLE_NAME,
    KeyConditionExpression: `PK = :messageUri`,
    ExpressionAttributeValues: {
      ':messageUri': messageUri,
    },
  };
  const request = new QueryCommand(params);
  const { Items }: QueryCommandOutput = await dynamoDbClient.send(request);

  return Items ?? [];
}

export async function putTtl(ttlItem: TtlDynamodbRecord) {
  const params = {
    TableName: TTL_TABLE_NAME,
    Item: marshall(ttlItem),
  };
  const request = new PutItemCommand(params);
  const output: PutItemCommandOutput = await dynamoDbClient.send(request);

  return output.$metadata.httpStatusCode;
}

export async function deleteTtl(messageUri: string) {
  const params = {
    TableName: TTL_TABLE_NAME,
    Key: {
      PK: {
        S: messageUri,
      },
      SK: {
        S: 'TTL',
      },
    },
  };
  const request = new DeleteItemCommand(params);
  const output: DeleteItemCommandOutput = await dynamoDbClient.send(request);

  return output.$metadata.httpStatusCode;
}