@uyu423/pinpoint-node-agent/lib/instrumentation/inst-manager.js

Pinpoint node agent provided by NAVER (Personalized version)

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

'use strict'

const fs = require('fs')
const path = require('path')
const findNodeModules = require('find-node-modules');

const log = require('../utils/logger')
const Hook = require('require-in-the-middle')
const nodeModules = findNodeModules({relative: false })
const MODULES = [
  'express',
  'http',
  'https',
  // 'koa',
  'koa-router',
  'redis',
  'ioredis',
  'mongodb-core',
  'elasticsearch',
  'request',
  // 'bluebird',
]

function init(agent) {

  try {
    this.hook = Hook(MODULES, function (exports, name) {
      loadModule(name, agent, exports);
      return exports
    })
  } catch (e) {
    log.error('error occurred', e)
  }
}

function loadModule(name, agent, m) {
  if (MODULES.includes(name) && agent.includedModules(name)) {
    try {
      const version = _getModuleVersion(name);
      require('./module/' + name)(agent, version, m);
      agent.setModules(name);
      log.info('loader ==> %s (%s)', name, version);
    }
    catch (e) {
      log.error('fail to load:', e);
    }
  }
}

function _getModuleVersion(name) {
  let version
  nodeModules.some((nodePath) => {
    const pkg = path.join(`${nodePath}/${name}/`, 'package.json')
    try {
      if (fs.existsSync(pkg)) {
        const json = JSON.parse(fs.readFileSync(pkg))
        version = json.version
      }
    } catch (e) {
      log.error('error occurred', e)
    }
  })
  if (!version) {
    version = process.versions.node
  }
  return version
}

module.exports = {
  init,
}