pg-node-pg.js

import { Pool } from "pg";
import { Stats } from './lib/bench.js'

const pool_size = 3

const sql = new Pool({
  host: 'tfb-database',
  user: 'benchmarkdbuser',
  password: 'benchmarkdbpass',
  database: 'hello_world',
  port: 5432,
  max: pool_size, // Maximum connections in pool
  idle_timeout: 30, // Close idle connections after 30s
  max_lifetime: 0, // Connection lifetime in seconds (0 = forever)
  connect_timeout: 30, // Timeout when establishing new connections
  ssl: false,
});

const stats = new Stats()

const query = { text: `SELECT id FROM Fortune`, name: 'foo' }

async function connection () {
  let values = await sql.query(query)
  stats.rps++
  stats.conn++
  while (values.rows.length === 12) {
    values = await sql.query(query)
    stats.rps++
  }
}

const connections = []

for (let i = 0; i < pool_size; i++) {
  connections.push(connection())
}

setInterval(() => stats.log(), 1000)

await Promise.all(connections)