Class AWS2DynamoDBPlugin

src/plugins/AWS2DynamoDBPlugin.ts:28–64

import { getAWS, execute } from '../aws/SDK2';

class AWS2DynamoDBPlugin implements SwPlugin {
  readonly module = 'aws-sdk';
  readonly versions = '2.*';

  install(installer: PluginInstaller): void {
    const AWS = getAWS(installer);
    const DocumentClient = AWS.DynamoDB.DocumentClient;

    function instrument(name: string): void {
      const _func = DocumentClient.prototype[name];

      DocumentClient.prototype[name] = function (params: any, callback?: any): any {
        const span = ContextManager.current.newExitSpan(`AWS/DynamoDB/${name}`, Component.AWS_DYNAMODB, Component.HTTP);

        span.component = Component.AWS_DYNAMODB;
        span.layer = SpanLayer.DATABASE;
        // span.peer = `${this.service.endpoint.host ?? '<unknown>'}:${this.service.endpoint.port ?? '<unknown>'}`;

        span.tag(Tag.dbType('DynamoDB'));
        span.tag(Tag.dbStatement(name));

        return execute(span, this, _func, params, callback);
      };
    }

    instrument('batchGet');
    instrument('batchWrite');
    instrument('delete');
    instrument('get');
    instrument('put');
    instrument('query');
    instrument('scan');
    instrument('update');
    instrument('transactGet');
    instrument('transactWrite');
  }
}

// noinspection JSUnusedGlobalSymbols
export default new AWS2DynamoDBPlugin();