Subscription.ts

import { attribute, hashKey, rangeKey } from '@aws/dynamodb-data-mapper-annotations';
import { APIGatewayEventRequestContext } from 'aws-lambda';

@table('SUBSCRIPTION_TABLE')
export class Subscription {
  /*
   * Combination of connectionId and subscriptionId
   * `${connectionId}|${subscriptionId}`
   */
  @hashKey({ type: 'String' })
  id: string;

  @rangeKey({
    type: 'String',
    indexKeyConfigurations: { TopicIndex: 'HASH' },
  })
  topic: string;

  @attribute({
    type: 'String',
    indexKeyConfigurations: { ConnectionIndex: 'HASH' },
  })
  connectionId: string;
  
  @attribute({ defaultProvider: () => new Date() })
  createdAt: Date;
  
  @attribute()
  requestContext: APIGatewayEventRequestContext;
  
  /* Reference to connection_init payload */
  @attribute()
  connectionRef: object;

  @attribute()
  subscription: {
    id: string;
    query: string;
    /** Actual value of variables for given field */
    variables?: any;
    /** Value of variables for user provided subscription */
    variableValues?: any;
  };
}