Xray

xray-trace.js

import * as AWSXRay from "aws-xray-sdk";
import { logger } from "./logger";

export function trace(name: string) {
  return (target: any, key: string, descriptor: PropertyDescriptor) => {
    const originalMethod = descriptor.value;
    descriptor.value = async function() {
      let segment: AWSXRay.Subsegment | null = null;
      try {
        try {
          segment = AWSXRay.getSegment()!.addNewSubsegment(`${name}.${key}`);
        } catch (err) {
          // ignores this (only happens locally)
          logError(err);
        }
        return await originalMethod.apply(this, arguments);
      } catch (err) {
        try {
          segment!.addError(err);
        } catch (err) {
          // ignores this (only happens locally)
          logError(err);
        }
        throw err;
      } finally {
        try {
          segment!.close();
        } catch (err) {
          // ignores this (only happens locally)
          logError(err);
        }
      }
    };
    return descriptor;
  };
}

function logError(err: any) {
  logger.warn("AWSXRay error", err);
}