express-slonik

import express from 'express'; import { createPool, sql, NotFoundError } from 'slonik'; import createMiddleware from 'express-slonik'; import { z } from 'zod'; import { Server } from 'http'; const userSchema = z.object({ id: z.number().int(), name: z.string(), email: z.string().email(), }); export const createExpressApp = ({ app, pool }) => { const transaction = createMiddleware(pool); app.get( '/user/:id', // Starts the transaction. transaction.begin(), async (req, res, next) => { try { const userId = parseInt(req.params.id, 10); if (isNaN(userId)) { return res.status(400).json({ message: 'Invalid user ID' }); } const user = await req.transaction.one( sql.type(userSchema)`SELECT * FROM users WHERE users.id = ${userId}` ); res.json(user); } catch (error) { if (error instanceof NotFoundError) { res.status(404).json({ name: error.name, message: `User with given id (${req.params.id}) not found.`, }); return; } next(error); } }, // Optional: Explicitly commits the transaction. If omitted, it auto-commits on response send or rolls back on error. transaction.end() ); return app; }; /** * Gracefully attempt to shut down the server. */ async function shutdownHandler(server: Server, pool) { return async function () { server.close(); await pool.end(); // Ensure the Slonik pool is also terminated }; } (async function () { const app = express(); // Ensure DATABASE_URL is set in your environment variables, e.g., 'postgres://user:pass@host:port/database' const pool = await createPool(process.env.DATABASE_URL ?? 'postgres://user:password@localhost:5432/mydb'); const server = createExpressApp({ app, pool }).listen(8080, () => { console.log('Server listening on http://localhost:8080'); }); process .on('SIGTERM', shutdownHandler(server, pool)) .on('SIGINT', shutdownHandler(server, pool)); })();