feat: add @apollo/server@4 support (#3203)

Closes: #2980
Co-authored-by: Trent Mick <[email protected]>

Author: jmesimak

.ci/tav.json

@@ -2,6 +2,7 @@
   "// todo": "We want versions=['20','19','18','16','14','12','10','8'], but versions*modules needs to be <256 for the GH Actions jobs limit",
   "versions": [ "20", "18", "16", "14", "12", "10", "8" ],
   "modules": [
+    "@apollo/server",
     "@aws-sdk/client-s3",
     "@elastic/elasticsearch",
     "@elastic/elasticsearch-canary",

.tav.yml

@@ -1,6 +1,12 @@
 # This file is used by tool TAV - Test All Versions.
 # See: https://github.com/watson/test-all-versions
 
+'@apollo/server':
+  versions: ^4.0.0
+  peerDependencies: graphql@^16.6.0
+  node: '>=14.16.0'
+  commands: node test/instrumentation/modules/@apollo/server.test.js
+
 generic-pool:
   versions: ^2.0.0 || ^3.1.0
   commands: node test/instrumentation/modules/generic-pool.test.js

docs/supported-technologies.asciidoc

@@ -127,6 +127,7 @@ These modules override that behavior to give better insights into specialized HT
 |Module |Version |Note
 |https://www.npmjs.com/package/express-graphql[express-graphql] |>=0.6.1 <0.13.0 |Will name all transactions by the GraphQL query name. There is a https://github.com/elastic/apm-agent-nodejs/issues/2516[known issue with node <10.4]. This module is deprecated and is no longer tested.
 |https://www.npmjs.com/package/apollo-server-express[apollo-server-express] |>=2.0.4 <4|Will name all transactions by the GraphQL query name. Versions before 2.9.6 are no longer tested.
+|https://www.npmjs.com/package/@apollo/server[@apollo/server] |>=4.0.0|Will name all transactions by the GraphQL query name
 |=======================================================================
 
 [float]

lib/instrumentation/index.js

@@ -31,6 +31,7 @@ const nodeSupportsAsyncLocalStorage = semver.satisfies(process.versions.node, '>
 const nodeHasInstrumentableFetch = typeof (global.fetch) === 'function'
 
 var MODULES = [
+  '@apollo/server',
   '@aws-sdk/smithy-client', // Instrument the base client which all AWS-SDK v3 clients extends
   ['@elastic/elasticsearch', '@elastic/elasticsearch-canary'],
   '@node-redis/client/dist/lib/client',

lib/instrumentation/modules/@apollo/server.js

@@ -0,0 +1,26 @@
+/*
+ * 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
+}