pinpoint-node-agent/lib/instrumentation/module/mysql/mysql-connection-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('./mysql-service-type')
const MethodDescriptorBuilder = require('../../../context/method-descriptor-builder')
class MySQLConnectionConnectInterceptor {
    constructor() {
        this.methodDescriptorBuilder = new MethodDescriptorBuilder('connect')
        this.methodDescriptorBuilder.setClassName('Connection')
        this.methodDescriptorBuilder.setParameterVariableNames('connectionUri')
    }

    doInBeforeTrace(recorder, target, args) {
        recorder.recordServiceType(serviceType)
    }

    doInBeforeAsyncTrace(asyncEventRecorder, target, args) {
        asyncEventRecorder.recordServiceType(serviceType)
    }

    callbackIndexOf(args) {
        if (args.length > 1 && typeof args[1] === 'function') {
            return 1
        }

        if (typeof args[0] !== 'function') {
            return
        }

        return 0
    }
}

module.exports = MySQLConnectionConnectInterceptor