TechEmpower.FrameworkBenchmarks/frameworks/JavaScript/fastify/db/postgres.js

const postgres = require("postgres");

const clientOpts = {
  host: process.env.PG_HOST,
  user: process.env.PG_USER,
  password: process.env.PG_PSWD,
  database: process.env.PG_DBNAME,
};

const sql = postgres({ ...clientOpts, max: 1 });

async function allFortunes() {
  return sql`select id, message from fortune`;
}

async function getWorld(id) {
  return sql`select id, randomNumber from world where id = ${id}`.then(
    (arr) => arr[0]
  );
}

async function bulkUpdate(worlds) {
  const values = sql(
    worlds
      .map((world) => [world.id, world.randomnumber])
      .sort((a, b) => (a[0] < b[0] ? -1 : 1))
  );

  return sql`update world set randomNumber = (update_data.randomNumber)::int
    from (values ${values}) as update_data (id, randomNumber)
    where world.id = (update_data.id)::int`;
}

module.exports = {
  getWorld,
  bulkUpdate,
  allFortunes,
};