index.js

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

'use strict'

// This is the file where we will use our instrumented datastore.

const SimpleDatastore = require('./simple-datastore')
const fastify = require('fastify')({ logger: true })

const { PORT: port = 3000, HOST: host = '127.0.0.1' } = process.env
const datastore = new SimpleDatastore(host, port, "DUMMY_DATABASE")

// We will be using the fastify web framework to host the
// datastore, but any host (Docker, web framework, etc.) can be used
fastify.post('/query', async (request, reply) => {
  // Execute a single query
  const query = 'SELECT * FROM users'
  const result = await datastore.execute(query)
  return reply.send({ result })
})

fastify.post('/batch', async (request, reply) => {
  // Execute a batch of queries.
  const queries = ['SELECT * FROM orders', 'SELECT * FROM products']
  const result = await datastore.batch(queries)
  return reply.send({ result })
})

fastify.addHook('onClose', async (instance) => {
  // Close the connection to the datastore
  datastore.close()
})

// Start the server where the database will be hosted
const start = async () => {
  try {
    datastore.connect()
    fastify.listen({ port, host })
  } catch (err) {
    fastify.log.error(err)
    process.exit(1)
  }
}

start()