atatus-nodejs/lib/instrumentation/modules/@apollo/server.js

Atatus APM agent for Node.js

/*
 * Copyright Elasticsearch B.V. and other contributors where applicable.
 * Licensed under the BSD 2-Clause License; you may not use this file except in
 * compliance with the BSD 2-Clause License.
 */

'use strict';

const shimmer = require('../../shimmer');

module.exports = function (apolloServer, agent, { enabled }) {
  if (!enabled) {
    return apolloServer;
  }

  function wrapExecuteHTTPGraphQLRequest(orig) {
    return function wrappedExecuteHTTPGraphQLRequest() {
      var trans = agent._instrumentation.currTransaction();
      if (trans) trans._graphqlRoute = true;
      return orig.apply(this, arguments);
    };
  }

  shimmer.wrap(
    apolloServer.ApolloServer.prototype,
    'executeHTTPGraphQLRequest',
    wrapExecuteHTTPGraphQLRequest,
  );
  return apolloServer;
};