From e0b84f5481d49d743af1aa45d26e37fff6c8515a Mon Sep 17 00:00:00 2001 From: Max Mayfield Date: Mon, 2 Mar 2026 03:59:26 +0000 Subject: [PATCH] Fix drift SET LOCAL: use string interpolation with UUID validation (SET doesn't support params) --- products/02-iac-drift-detection/saas/src/data/db.ts | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/products/02-iac-drift-detection/saas/src/data/db.ts b/products/02-iac-drift-detection/saas/src/data/db.ts index c5852d7..6753403 100644 --- a/products/02-iac-drift-detection/saas/src/data/db.ts +++ b/products/02-iac-drift-detection/saas/src/data/db.ts @@ -15,7 +15,9 @@ export function createPool(connectionString: string): pg.Pool { * MUST be cleared when returning the connection to the pool. */ export async function setTenantContext(client: pg.PoolClient, tenantId: string): Promise { - await client.query('SET LOCAL app.tenant_id = $1', [tenantId]); + // SET doesn't support parameterized queries — validate UUID format then interpolate + if (!/^[0-9a-f-]{36}$/i.test(tenantId)) throw new Error('Invalid tenant ID'); + await client.query(`SET LOCAL app.tenant_id = '${tenantId}'`); } /**