newrelic/lib/subscribers/cassandra-driver/client-shutdown.js

New Relic agent

/*
 * Copyright 2025 New Relic Corporation. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

'use strict'

const DbOperationSubscriber = require('../db-operation')

/**
 * Subscribes to the `shutdown` event in `cassandra-driver`.
 */
class CassandraShutdownSubscriber extends DbOperationSubscriber {
  constructor({ agent, logger, channelName = 'nr_shutdown' }) {
    super({ agent, logger, channelName, packageName: 'cassandra-driver', system: 'Cassandra' })
    this.events = ['asyncStart', 'asyncEnd']
    this.operation = 'shutdown'
  }

  handler(data, ctx) {
    const { self } = data
    this.setParameters(self)
    return super.handler(data, ctx)
  }

  setParameters() {
    this.parameters = {}
    this.parameters.product = this.system
  }
}

module.exports = CassandraShutdownSubscriber