New Relic GraphQL extension.

new-relic-extension.js

const { GraphQLExtension } = require("graphql-extensions");
const newrelic = require("newrelic");

function traceSummary(trace) {
    if (!trace) {
        return "No trace data.";
    }

    try {
        const { duration, execution: {resolvers}} = trace;

        const summary = [];
        summary.push(`Total Duration (ms): ${duration / 1000000}`);

        const rootResolverTimings = resolvers
        .filter(r => (r.parentType === "RootQuery"))
        .map(r => (`Field: ${r.fieldName} - ReturnType: ${r.returnType} - Duration: ${ r.duration / 1000000 }`));

        summary.push(...rootResolverTimings);
        return summary.join(" | ");
    }
    catch(err) {
        return "Error getting trace summary.";
    }
}

class NewRelicExtension extends GraphQLExtension {
    constructor() {
        super();
    }

    requestDidStart({queryString, operationName, variables, persistedQueryHit}) {
        newrelic.setTransactionName(`graphql (${operationName})`);
        newrelic.addCustomAttribute("gqlQuery", queryString);
        newrelic.addCustomAttribute("gqlVars", variables);
        newrelic.addCustomAttribute("persistedQueryHit", persistedQueryHit);
        return;
    }

    willSendResponse(payload) {
        newrelic.addCustomAttribute("traceSummary", traceSummary(payload.graphqlResponse
            && payload.graphqlResponse.extensions
            && payload.graphqlResponse.extensions.tracing));
    }
}



module.exports = NewRelicExtension;

This looks perfect, thanks!

I repurposed this a bit and published it as an NPM package for anyone interested https://github.com/localshred/apollo-newrelic-extension