Class IORedisPlugin

src/plugins/IORedisPlugin.ts:27–66

import ContextManager from '../trace/context/ContextManager';

class IORedisPlugin implements SwPlugin {
  readonly module = 'ioredis';
  readonly versions = '*';

  install(installer: PluginInstaller): void {
    const Redis = installer.require?.('ioredis') ?? require('ioredis');

    this.interceptOperation(Redis, 'sendCommand');
  }

  interceptOperation(Cls: any, operation: string): void {
    const _original = Cls.prototype[operation];

    if (!_original) return;

    Cls.prototype[operation] = function (...args: any[]) {
      const command = args[0];
      const host = `${this.options.host}:${this.options.port}`;
      const span = ContextManager.current.newExitSpan(`redis/${command?.name}`, Component.REDIS);

      span.start();
      span.component = Component.REDIS;
      span.layer = SpanLayer.CACHE;
      span.peer = host;
      span.tag(Tag.dbType('Redis'));
      span.tag(Tag.dbInstance(`${this.condition?.select ?? host}`));

      try {
        const ret = wrapPromise(span, _original.apply(this, args));
        span.async();
        return ret;
      } catch (err) {
        span.error(err);
        span.stop();

        throw err;
      }
    };
  }
}

export default new IORedisPlugin();