pinpoint-node-agent/lib/instrumentation/module/redis/redis-internal-send-command-interceptor.js

Pinpoint node agent provided by NAVER

/**
 * Pinpoint Node.js Agent
 * Copyright 2020-present NAVER Corp.
 * Apache License v2.0
 */

'use strict'

const serviceType = require('./redis-service-type')
const MethodDescriptorBuilder = require('../../../context/method-descriptor-builder')
const { addressStringOf } = require('../../../utils/convert-utils')
const InstrumentArrowFunction = require('../../instrument-arrow-function')

class RedisInternalSendCommandInterceptor {
    constructor(traceContext) {
        this.traceContext = traceContext
        this.methodDescriptorBuilder = MethodDescriptorBuilder.makeRuntimeDetectionMethodDescriptorBuilder()
            .setClassName('RedisClient')
        this.serviceType = serviceType
    }

    makeFunctionNameDetectedMethodDescriptorBuilder(thisArg, args) {
        if (!args || args.length < 1 || typeof args[0].command !== 'string') {
            return this.methodDescriptorBuilder
        }

        const command = args[0]
        return this.methodDescriptorBuilder.makeCloneOf(command.command)
    }

    doInBeforeTrace(recorder, target, args) {
        if (!args || args.length < 1 || typeof args[0].command !== 'string') {
            return
        }
        recorder.recordDestinationId('Redis')

        if (target.connection_options) {
            recorder.recordEndPoint(addressStringOf(target.connection_options.host, target.connection_options.port))
        }

        const command = args[0]
        InstrumentArrowFunction.make(command, 'callback', this.traceContext).addChildTraceInterceptor(recorder)
    }
}

module.exports = RedisInternalSendCommandInterceptor