Compare commits
6 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 40bd6dcfea | |||
| 4884961c8e | |||
| 93be4d8f72 | |||
| f67b96ddfe | |||
| d1a68d93de | |||
| e9f94a2bd7 |
@@ -166,6 +166,7 @@ Expected: one row, `role = 'groomer'`. If zero rows return, the request hit the
|
|||||||
| TC-API-3.26 | Verify 25-35% medicalAlerts distribution | GET /api/pets (first 30 pets), count how many have non-empty medicalAlerts | Ratio is 25-35% (seed uses rand() < 0.3 for ~30% distribution) |
|
| TC-API-3.26 | Verify 25-35% medicalAlerts distribution | GET /api/pets (first 30 pets), count how many have non-empty medicalAlerts | Ratio is 25-35% (seed uses rand() < 0.3 for ~30% distribution) |
|
||||||
| TC-API-3.27 | Verify coat_type enum has all seed values | After UAT seed completes, inspect the coat_type enum on the UAT DB — it must contain: short, medium, long, double, wire, silky, curly, hairless | UAT seed jobs (`reset-demo-data`, `seed-test-data`) complete 1/1 with no `enum_in` error; coat_type includes all 8 values used by seed.ts `coatTypePool` |
|
| TC-API-3.27 | Verify coat_type enum has all seed values | After UAT seed completes, inspect the coat_type enum on the UAT DB — it must contain: short, medium, long, double, wire, silky, curly, hairless | UAT seed jobs (`reset-demo-data`, `seed-test-data`) complete 1/1 with no `enum_in` error; coat_type includes all 8 values used by seed.ts `coatTypePool` |
|
||||||
| TC-API-3.28 | Verify pet_size_category enum has all seed values | After UAT seed completes, inspect the pet_size_category enum on the UAT DB — it must contain: small, medium, large, extra_large | UAT seed jobs (`reset-demo-data`, `seed-test-data`) complete 1/1 with no `enum_in` error; pet_size_category includes all 4 values used by seed.ts `petSizeCategoryPool` (regression for GRO-1999, mirrors TC-API-3.27) |
|
| TC-API-3.28 | Verify pet_size_category enum has all seed values | After UAT seed completes, inspect the pet_size_category enum on the UAT DB — it must contain: small, medium, large, extra_large | UAT seed jobs (`reset-demo-data`, `seed-test-data`) complete 1/1 with no `enum_in` error; pet_size_category includes all 4 values used by seed.ts `petSizeCategoryPool` (regression for GRO-1999, mirrors TC-API-3.27) |
|
||||||
|
| TC-API-3.29 | Verify `reset-demo-data` CronJob does not fail with FK 23503 on `invoice_tip_splits` (GRO-2123) | Trigger the CronJob manually: `kubectl create job --from=cronjob/reset-demo-data verify-gro2123 -n groombook-uat`. Wait for pod to terminate. Inspect logs: `kubectl logs -n groombook-uat -l job-name=verify-gro2123` | Pod reaches `Completed` state; logs show `✓ Acquired seed advisory lock` and `✓ Released seed advisory lock` from `seed.ts`; no `PostgresError: … violates foreign key constraint "invoice_tip_splits_invoice_id_invoices_id_fk"` (code 23503); final counts unchanged (500 clients, ~4000 invoices) |
|
||||||
|
|
||||||
### 4.4 Appointment Scheduling
|
### 4.4 Appointment Scheduling
|
||||||
|
|
||||||
|
|||||||
@@ -12,8 +12,8 @@
|
|||||||
"test": "vitest run",
|
"test": "vitest run",
|
||||||
"db:generate": "drizzle-kit generate",
|
"db:generate": "drizzle-kit generate",
|
||||||
"db:migrate": "drizzle-kit migrate",
|
"db:migrate": "drizzle-kit migrate",
|
||||||
"db:seed": "tsx src/db/seed.ts",
|
"db:seed": "pnpm --filter @groombook/db seed",
|
||||||
"db:reset": "tsx src/db/reset.ts && drizzle-kit migrate && tsx src/db/seed.ts",
|
"db:reset": "pnpm --filter @groombook/db reset",
|
||||||
"db:studio": "drizzle-kit studio"
|
"db:studio": "drizzle-kit studio"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,66 @@
|
|||||||
|
-- Migration: 0041_route_optimization.sql
|
||||||
|
-- Route optimization schema: geocoding columns on clients, groomerRoutes +
|
||||||
|
-- routeStops tables, and route settings on business_settings.
|
||||||
|
-- Written idempotently so it is safe to re-run.
|
||||||
|
|
||||||
|
-- ─── Enums ────────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
DO $$ BEGIN
|
||||||
|
CREATE TYPE "route_status" AS ENUM ('draft', 'optimized', 'in_progress', 'completed');
|
||||||
|
EXCEPTION WHEN duplicate_object THEN NULL;
|
||||||
|
END $$;
|
||||||
|
|
||||||
|
-- ─── Clients: geocoding columns ───────────────────────────────────────────────
|
||||||
|
|
||||||
|
ALTER TABLE "clients" ADD COLUMN IF NOT EXISTS "latitude" double precision;
|
||||||
|
ALTER TABLE "clients" ADD COLUMN IF NOT EXISTS "longitude" double precision;
|
||||||
|
ALTER TABLE "clients" ADD COLUMN IF NOT EXISTS "geocoded_at" timestamp;
|
||||||
|
|
||||||
|
-- ─── Business settings: route optimization config ─────────────────────────────
|
||||||
|
|
||||||
|
ALTER TABLE "business_settings"
|
||||||
|
ADD COLUMN IF NOT EXISTS "default_travel_buffer_mins" integer NOT NULL DEFAULT 15;
|
||||||
|
ALTER TABLE "business_settings"
|
||||||
|
ADD COLUMN IF NOT EXISTS "route_optimization_provider" text DEFAULT 'nominatim';
|
||||||
|
-- Encrypted at rest at the application layer (AES-256-GCM).
|
||||||
|
ALTER TABLE "business_settings"
|
||||||
|
ADD COLUMN IF NOT EXISTS "google_maps_api_key" text;
|
||||||
|
|
||||||
|
-- ─── Groomer routes table ─────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS "groomer_routes" (
|
||||||
|
"id" uuid PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
"staff_id" uuid NOT NULL REFERENCES "staff"("id") ON DELETE CASCADE,
|
||||||
|
"route_date" date NOT NULL,
|
||||||
|
"status" "route_status" NOT NULL DEFAULT 'draft',
|
||||||
|
"total_travel_mins" integer,
|
||||||
|
"total_distance_km" numeric(8, 2),
|
||||||
|
"optimized_at" timestamp,
|
||||||
|
"created_at" timestamp NOT NULL DEFAULT now(),
|
||||||
|
"updated_at" timestamp NOT NULL DEFAULT now(),
|
||||||
|
CONSTRAINT "uq_groomer_routes_staff_date" UNIQUE ("staff_id", "route_date")
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS "idx_groomer_routes_staff_id"
|
||||||
|
ON "groomer_routes"("staff_id");
|
||||||
|
|
||||||
|
-- ─── Route stops table ────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS "route_stops" (
|
||||||
|
"id" uuid PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
"route_id" uuid NOT NULL REFERENCES "groomer_routes"("id") ON DELETE CASCADE,
|
||||||
|
"appointment_id" uuid NOT NULL REFERENCES "appointments"("id") ON DELETE CASCADE,
|
||||||
|
"stop_order" integer NOT NULL,
|
||||||
|
"latitude" double precision NOT NULL,
|
||||||
|
"longitude" double precision NOT NULL,
|
||||||
|
"travel_mins_from_prev" integer,
|
||||||
|
"travel_distance_km_from_prev" numeric(8, 2),
|
||||||
|
"buffer_mins" integer NOT NULL DEFAULT 15,
|
||||||
|
"created_at" timestamp NOT NULL DEFAULT now(),
|
||||||
|
"updated_at" timestamp NOT NULL DEFAULT now(),
|
||||||
|
CONSTRAINT "uq_route_stops_route_appointment" UNIQUE ("route_id", "appointment_id"),
|
||||||
|
CONSTRAINT "uq_route_stops_route_order" UNIQUE ("route_id", "stop_order")
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS "idx_route_stops_route_id"
|
||||||
|
ON "route_stops"("route_id");
|
||||||
@@ -281,6 +281,13 @@
|
|||||||
"when": 1780000000002,
|
"when": 1780000000002,
|
||||||
"tag": "0040_register_missing_coat_type_values",
|
"tag": "0040_register_missing_coat_type_values",
|
||||||
"breakpoints": true
|
"breakpoints": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"idx": 41,
|
||||||
|
"version": "7",
|
||||||
|
"when": 1780000000003,
|
||||||
|
"tag": "0041_route_optimization",
|
||||||
|
"breakpoints": true
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
@@ -78,6 +78,9 @@ export function buildClient(overrides: Partial<ClientRow> = {}): ClientRow {
|
|||||||
stripeCustomerId: null,
|
stripeCustomerId: null,
|
||||||
status: "active",
|
status: "active",
|
||||||
disabledAt: null,
|
disabledAt: null,
|
||||||
|
latitude: null,
|
||||||
|
longitude: null,
|
||||||
|
geocodedAt: null,
|
||||||
createdAt: new Date("2025-01-01T00:00:00Z"),
|
createdAt: new Date("2025-01-01T00:00:00Z"),
|
||||||
updatedAt: new Date("2025-01-01T00:00:00Z"),
|
updatedAt: new Date("2025-01-01T00:00:00Z"),
|
||||||
...overrides,
|
...overrides,
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
import {
|
import {
|
||||||
boolean,
|
boolean,
|
||||||
|
date,
|
||||||
|
doublePrecision,
|
||||||
index,
|
index,
|
||||||
integer,
|
integer,
|
||||||
jsonb,
|
jsonb,
|
||||||
@@ -140,6 +142,10 @@ export const clients = pgTable(
|
|||||||
stripeCustomerId: text("stripe_customer_id"),
|
stripeCustomerId: text("stripe_customer_id"),
|
||||||
status: clientStatusEnum("status").notNull().default("active"),
|
status: clientStatusEnum("status").notNull().default("active"),
|
||||||
disabledAt: timestamp("disabled_at"),
|
disabledAt: timestamp("disabled_at"),
|
||||||
|
// Geocoded coordinates for route optimization; null until geocoded.
|
||||||
|
latitude: doublePrecision("latitude"),
|
||||||
|
longitude: doublePrecision("longitude"),
|
||||||
|
geocodedAt: timestamp("geocoded_at"),
|
||||||
createdAt: timestamp("created_at").notNull().defaultNow(),
|
createdAt: timestamp("created_at").notNull().defaultNow(),
|
||||||
updatedAt: timestamp("updated_at").notNull().defaultNow(),
|
updatedAt: timestamp("updated_at").notNull().defaultNow(),
|
||||||
},
|
},
|
||||||
@@ -555,6 +561,16 @@ export const businessSettings = pgTable("business_settings", {
|
|||||||
accentColor: text("accent_color").notNull().default("#8b7355"),
|
accentColor: text("accent_color").notNull().default("#8b7355"),
|
||||||
messagingPhoneNumber: text("messaging_phone_number"),
|
messagingPhoneNumber: text("messaging_phone_number"),
|
||||||
telnyxMessagingProfileId: text("telnyx_messaging_profile_id"),
|
telnyxMessagingProfileId: text("telnyx_messaging_profile_id"),
|
||||||
|
// Route optimization settings.
|
||||||
|
defaultTravelBufferMins: integer("default_travel_buffer_mins")
|
||||||
|
.notNull()
|
||||||
|
.default(15),
|
||||||
|
routeOptimizationProvider: text("route_optimization_provider").default(
|
||||||
|
"nominatim"
|
||||||
|
),
|
||||||
|
// Encrypted at rest at the application layer (AES-256-GCM), mirroring
|
||||||
|
// the handling of authProviderConfigs.clientSecret.
|
||||||
|
googleMapsApiKey: text("google_maps_api_key"),
|
||||||
createdAt: timestamp("created_at").notNull().defaultNow(),
|
createdAt: timestamp("created_at").notNull().defaultNow(),
|
||||||
updatedAt: timestamp("updated_at").notNull().defaultNow(),
|
updatedAt: timestamp("updated_at").notNull().defaultNow(),
|
||||||
});
|
});
|
||||||
@@ -658,3 +674,69 @@ export const bufferRules = pgTable(
|
|||||||
index("idx_buffer_rules_service_id").on(t.serviceId),
|
index("idx_buffer_rules_service_id").on(t.serviceId),
|
||||||
]
|
]
|
||||||
);
|
);
|
||||||
|
|
||||||
|
// ─── Route Optimization ───────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
export const routeStatusEnum = pgEnum("route_status", [
|
||||||
|
"draft",
|
||||||
|
"optimized",
|
||||||
|
"in_progress",
|
||||||
|
"completed",
|
||||||
|
]);
|
||||||
|
|
||||||
|
// A groomer's optimized route for a single day. One row per (staff, date).
|
||||||
|
export const groomerRoutes = pgTable(
|
||||||
|
"groomer_routes",
|
||||||
|
{
|
||||||
|
id: uuid("id").primaryKey().defaultRandom(),
|
||||||
|
staffId: uuid("staff_id")
|
||||||
|
.notNull()
|
||||||
|
.references(() => staff.id, { onDelete: "cascade" }),
|
||||||
|
routeDate: date("route_date", { mode: "string" }).notNull(),
|
||||||
|
status: routeStatusEnum("status").notNull().default("draft"),
|
||||||
|
// Populated once the route is optimized.
|
||||||
|
totalTravelMins: integer("total_travel_mins"),
|
||||||
|
totalDistanceKm: numeric("total_distance_km", { precision: 8, scale: 2 }),
|
||||||
|
optimizedAt: timestamp("optimized_at"),
|
||||||
|
createdAt: timestamp("created_at").notNull().defaultNow(),
|
||||||
|
updatedAt: timestamp("updated_at").notNull().defaultNow(),
|
||||||
|
},
|
||||||
|
(t) => [
|
||||||
|
// One route per groomer per day.
|
||||||
|
unique("uq_groomer_routes_staff_date").on(t.staffId, t.routeDate),
|
||||||
|
index("idx_groomer_routes_staff_id").on(t.staffId),
|
||||||
|
]
|
||||||
|
);
|
||||||
|
|
||||||
|
// An ordered stop within a groomer's route, tied to an appointment.
|
||||||
|
export const routeStops = pgTable(
|
||||||
|
"route_stops",
|
||||||
|
{
|
||||||
|
id: uuid("id").primaryKey().defaultRandom(),
|
||||||
|
routeId: uuid("route_id")
|
||||||
|
.notNull()
|
||||||
|
.references(() => groomerRoutes.id, { onDelete: "cascade" }),
|
||||||
|
appointmentId: uuid("appointment_id")
|
||||||
|
.notNull()
|
||||||
|
.references(() => appointments.id, { onDelete: "cascade" }),
|
||||||
|
stopOrder: integer("stop_order").notNull(),
|
||||||
|
latitude: doublePrecision("latitude").notNull(),
|
||||||
|
longitude: doublePrecision("longitude").notNull(),
|
||||||
|
// Null for the first stop in the route.
|
||||||
|
travelMinsFromPrev: integer("travel_mins_from_prev"),
|
||||||
|
travelDistanceKmFromPrev: numeric("travel_distance_km_from_prev", {
|
||||||
|
precision: 8,
|
||||||
|
scale: 2,
|
||||||
|
}),
|
||||||
|
bufferMins: integer("buffer_mins").notNull().default(15),
|
||||||
|
createdAt: timestamp("created_at").notNull().defaultNow(),
|
||||||
|
updatedAt: timestamp("updated_at").notNull().defaultNow(),
|
||||||
|
},
|
||||||
|
(t) => [
|
||||||
|
// An appointment appears at most once per route.
|
||||||
|
unique("uq_route_stops_route_appointment").on(t.routeId, t.appointmentId),
|
||||||
|
// Stop order is unique within a route.
|
||||||
|
unique("uq_route_stops_route_order").on(t.routeId, t.stopOrder),
|
||||||
|
index("idx_route_stops_route_id").on(t.routeId),
|
||||||
|
]
|
||||||
|
);
|
||||||
|
|||||||
+116
-7
@@ -401,7 +401,9 @@ const servicesDef = [
|
|||||||
*
|
*
|
||||||
* In seedKnownUsers() this replaces the inline UAT-staff block.
|
* In seedKnownUsers() this replaces the inline UAT-staff block.
|
||||||
*/
|
*/
|
||||||
async function seedUatStaffAccounts(db: ReturnType<typeof drizzle>) {
|
async function seedUatStaffAccounts(
|
||||||
|
db: ReturnType<typeof drizzle>,
|
||||||
|
): Promise<string | null> {
|
||||||
// ── Staff: UAT Super User (oidcSub from SEED_UAT_SUPER_OIDC_SUB env var) ──
|
// ── Staff: UAT Super User (oidcSub from SEED_UAT_SUPER_OIDC_SUB env var) ──
|
||||||
const uatSuperOidcSub = process.env.SEED_UAT_SUPER_OIDC_SUB;
|
const uatSuperOidcSub = process.env.SEED_UAT_SUPER_OIDC_SUB;
|
||||||
if (uatSuperOidcSub) {
|
if (uatSuperOidcSub) {
|
||||||
@@ -677,7 +679,12 @@ async function seedUatStaffAccounts(db: ReturnType<typeof drizzle>) {
|
|||||||
// We deterministically link the UAT groomer to the UAT customer's first pet
|
// We deterministically link the UAT groomer to the UAT customer's first pet
|
||||||
// ("UAT Pup Alpha") and leave the second pet ("UAT Pup Beta") UNLINKED so
|
// ("UAT Pup Alpha") and leave the second pet ("UAT Pup Beta") UNLINKED so
|
||||||
// TC-UAT-2 (200) and TC-UAT-3 (403) can both hardcode the stable petIds.
|
// TC-UAT-2 (200) and TC-UAT-3 (403) can both hardcode the stable petIds.
|
||||||
await seedUatGroomerLinkage(db, uatCustomerClientId);
|
//
|
||||||
|
// The linkage call itself is performed by the caller AFTER the `services`
|
||||||
|
// catalogue has been seeded (this helper runs before services exist,
|
||||||
|
// which previously caused the linkage to be silently skipped on every
|
||||||
|
// reset). GRO-2100 follow-up.
|
||||||
|
return uatCustomerClientId;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -692,12 +699,18 @@ async function seedUatStaffAccounts(db: ReturnType<typeof drizzle>) {
|
|||||||
*/
|
*/
|
||||||
async function seedUatGroomerLinkage(
|
async function seedUatGroomerLinkage(
|
||||||
db: ReturnType<typeof drizzle>,
|
db: ReturnType<typeof drizzle>,
|
||||||
customerClientId: string,
|
customerClientId: string | null,
|
||||||
): Promise<void> {
|
): Promise<void> {
|
||||||
const uatGroomerEmail = "uat-groomer@groombook.dev";
|
const uatGroomerEmail = "uat-groomer@groombook.dev";
|
||||||
const LINKED_PET_ID = "c0000001-0000-0000-0000-000000000002"; // UAT Pup Alpha
|
const LINKED_PET_ID = "c0000001-0000-0000-0000-000000000002"; // UAT Pup Alpha
|
||||||
const APPT_ID = "a0000001-0000-0000-0000-000000000001";
|
const APPT_ID = "a0000001-0000-0000-0000-000000000001";
|
||||||
|
|
||||||
|
// Skip silently if the UAT Customer client wasn't created (non-UAT seed
|
||||||
|
// profile, e.g. seedKnownUsers() in an env without the UAT personas).
|
||||||
|
if (!customerClientId) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
// Only run if the UAT groomer staff record actually exists — dev/test seeds
|
// Only run if the UAT groomer staff record actually exists — dev/test seeds
|
||||||
// that don't set SEED_UAT_STAFF_OIDC_SUB should not crash.
|
// that don't set SEED_UAT_STAFF_OIDC_SUB should not crash.
|
||||||
const [uatGroomerStaff] = await db
|
const [uatGroomerStaff] = await db
|
||||||
@@ -720,6 +733,19 @@ async function seedUatGroomerLinkage(
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Skip if the linked pet hasn't been seeded yet (defensive: caller should
|
||||||
|
// ensure pets exist; if the helper is re-ordered later we don't want to
|
||||||
|
// crash here).
|
||||||
|
const [linkedPet] = await db
|
||||||
|
.select({ id: schema.pets.id })
|
||||||
|
.from(schema.pets)
|
||||||
|
.where(eq(schema.pets.id, LINKED_PET_ID))
|
||||||
|
.limit(1);
|
||||||
|
if (!linkedPet) {
|
||||||
|
console.warn(`⚠ GRO-2100: UAT Pup Alpha (${LINKED_PET_ID}) not found — skipping uat-groomer linkage`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
// The "Bath & Brush" service id is stable across the reset; falls back to
|
// The "Bath & Brush" service id is stable across the reset; falls back to
|
||||||
// any active service if it has not been seeded yet (e.g. seedKnownUsers
|
// any active service if it has not been seeded yet (e.g. seedKnownUsers
|
||||||
// runs in isolation).
|
// runs in isolation).
|
||||||
@@ -847,7 +873,7 @@ async function seedKnownUsers() {
|
|||||||
// ── UAT staff accounts + Better Auth credentials (shared impl) ──────────────
|
// ── UAT staff accounts + Better Auth credentials (shared impl) ──────────────
|
||||||
// Extracted into seedUatStaffAccounts() so it runs in both seedKnownUsers()
|
// Extracted into seedUatStaffAccounts() so it runs in both seedKnownUsers()
|
||||||
// and the full seed() UAT branch.
|
// and the full seed() UAT branch.
|
||||||
await seedUatStaffAccounts(db);
|
const uatCustomerClientId = await seedUatStaffAccounts(db);
|
||||||
|
|
||||||
// ── Services: idempotent upsert keyed on `id` ─────────────────────────────
|
// ── Services: idempotent upsert keyed on `id` ─────────────────────────────
|
||||||
// GRO-2064: previously keyed on `services.name` while writing a
|
// GRO-2064: previously keyed on `services.name` while writing a
|
||||||
@@ -875,6 +901,12 @@ async function seedKnownUsers() {
|
|||||||
}
|
}
|
||||||
console.log(`✓ Seeded ${demoSvcs.length} services`);
|
console.log(`✓ Seeded ${demoSvcs.length} services`);
|
||||||
|
|
||||||
|
// GRO-2100: deterministic uat-groomer ↔ UAT Pup Alpha linkage. Must run
|
||||||
|
// AFTER services are seeded (this helper looks up an active service id
|
||||||
|
// to attach to the appointment; on a fresh reset there are none yet at
|
||||||
|
// the time seedUatStaffAccounts() returns).
|
||||||
|
await seedUatGroomerLinkage(db, uatCustomerClientId);
|
||||||
|
|
||||||
// ── Client: Demo Client ──
|
// ── Client: Demo Client ──
|
||||||
const [existingClient] = await db
|
const [existingClient] = await db
|
||||||
.select()
|
.select()
|
||||||
@@ -944,6 +976,63 @@ async function seedKnownUsers() {
|
|||||||
|
|
||||||
// ── Main seed ────────────────────────────────────────────────────────────────
|
// ── Main seed ────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
// ── GRO-2123: serialize reset+seed with a Postgres advisory lock ────────
|
||||||
|
// The reset-demo-data CronJob runs on an hourly schedule. With
|
||||||
|
// concurrencyPolicy=Replace, a new pod can start while the previous one
|
||||||
|
// is still mid-seed; the new pod's TRUNCATE then deletes rows the old pod
|
||||||
|
// is still inserting, producing FK 23503 errors non-deterministically
|
||||||
|
// (see GRO-2123: invoice_tip_splits → invoices).
|
||||||
|
//
|
||||||
|
// We hold a session-level advisory lock for the full duration of the
|
||||||
|
// seed so that overlapping invocations block then proceed in order —
|
||||||
|
// not skip. The key is a stable 32-bit constant so it can be referenced
|
||||||
|
// from runbooks without ambiguity and binds to the single-argument
|
||||||
|
// `pg_advisory_lock(int)` form, which postgres-js serializes as a plain
|
||||||
|
// number (no bigint type plumbing required).
|
||||||
|
const SEED_ADVISORY_LOCK_KEY = 0x47524f4f; // "GROO" in ASCII — arbitrary, stable
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Reserve a dedicated connection from `pool`, take the seed advisory lock
|
||||||
|
* on it, run `fn`, and release the lock + connection in a try/finally.
|
||||||
|
*
|
||||||
|
* CRITICAL: with postgres-js connection pooling, a session-level
|
||||||
|
* `pg_advisory_lock(KEY)` acquired on one pooled connection and released
|
||||||
|
* on a *different* one is a no-op (the lock is bound to the session /
|
||||||
|
* pg-backend that took it). We therefore reserve a dedicated connection
|
||||||
|
* for the lock and release it from the same reserved connection. The
|
||||||
|
* seed work itself still runs on the pooled connections.
|
||||||
|
*/
|
||||||
|
async function withSeedAdvisoryLock<T>(
|
||||||
|
pool: ReturnType<typeof postgres>,
|
||||||
|
fn: () => Promise<T>,
|
||||||
|
): Promise<T> {
|
||||||
|
const lockConnection = await pool.reserve();
|
||||||
|
let lockHeld = false;
|
||||||
|
try {
|
||||||
|
await lockConnection`SELECT pg_advisory_lock(${SEED_ADVISORY_LOCK_KEY})`;
|
||||||
|
lockHeld = true;
|
||||||
|
console.log(`✓ Acquired seed advisory lock (key=${SEED_ADVISORY_LOCK_KEY})`);
|
||||||
|
const result = await fn();
|
||||||
|
await lockConnection`SELECT pg_advisory_unlock(${SEED_ADVISORY_LOCK_KEY})`;
|
||||||
|
lockHeld = false;
|
||||||
|
console.log(`✓ Released seed advisory lock`);
|
||||||
|
return result;
|
||||||
|
} finally {
|
||||||
|
if (lockHeld) {
|
||||||
|
try {
|
||||||
|
await lockConnection`SELECT pg_advisory_unlock(${SEED_ADVISORY_LOCK_KEY})`;
|
||||||
|
} catch (err) {
|
||||||
|
console.error("Failed to release seed advisory lock during cleanup:", err);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
lockConnection.release();
|
||||||
|
} catch (err) {
|
||||||
|
console.error("Failed to release reserved lock connection:", err);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
async function seed() {
|
async function seed() {
|
||||||
const url = process.env.DATABASE_URL;
|
const url = process.env.DATABASE_URL;
|
||||||
if (!url) {
|
if (!url) {
|
||||||
@@ -961,6 +1050,22 @@ async function seed() {
|
|||||||
const client = postgres(url, { max: 5 });
|
const client = postgres(url, { max: 5 });
|
||||||
const db = drizzle(client, { schema });
|
const db = drizzle(client, { schema });
|
||||||
|
|
||||||
|
// GRO-2123: hold the seed advisory lock for the full body of runSeedBody.
|
||||||
|
// See the withSeedAdvisoryLock comment for why a reserved connection is
|
||||||
|
// required (postgres-js pooling would silently drop the lock otherwise).
|
||||||
|
await withSeedAdvisoryLock(client, async () => {
|
||||||
|
return await runSeedBody(client, db, profile, cfg);
|
||||||
|
});
|
||||||
|
|
||||||
|
await client.end();
|
||||||
|
}
|
||||||
|
|
||||||
|
async function runSeedBody(
|
||||||
|
client: ReturnType<typeof postgres>,
|
||||||
|
db: ReturnType<typeof drizzle>,
|
||||||
|
profile: SeedProfile,
|
||||||
|
cfg: ProfileConfig,
|
||||||
|
): Promise<void> {
|
||||||
console.log(`Seeding Groom Book database (profile: ${profile})...\n`);
|
console.log(`Seeding Groom Book database (profile: ${profile})...\n`);
|
||||||
|
|
||||||
// ── Staff ──
|
// ── Staff ──
|
||||||
@@ -1031,7 +1136,7 @@ async function seed() {
|
|||||||
// ── UAT staff accounts + Better Auth credentials (shared impl) ──────────────
|
// ── UAT staff accounts + Better Auth credentials (shared impl) ──────────────
|
||||||
// Seeds deterministic UAT staff with numeric OIDC subs and Better Auth credentials.
|
// Seeds deterministic UAT staff with numeric OIDC subs and Better Auth credentials.
|
||||||
// Must run AFTER random staff are created so upserts land correctly.
|
// Must run AFTER random staff are created so upserts land correctly.
|
||||||
await seedUatStaffAccounts(db);
|
const uatCustomerClientId = await seedUatStaffAccounts(db);
|
||||||
|
|
||||||
// ── Services ──
|
// ── Services ──
|
||||||
// GRO-2064: key the upsert on `services.id` (not `name`) so deterministic
|
// GRO-2064: key the upsert on `services.id` (not `name`) so deterministic
|
||||||
@@ -1058,6 +1163,12 @@ async function seed() {
|
|||||||
}
|
}
|
||||||
console.log(`✓ Created ${servicesDef.length} services`);
|
console.log(`✓ Created ${servicesDef.length} services`);
|
||||||
|
|
||||||
|
// GRO-2100: deterministic uat-groomer ↔ UAT Pup Alpha linkage. Must run
|
||||||
|
// AFTER services are seeded (this helper looks up an active service id
|
||||||
|
// to attach to the appointment; on a fresh reset there are none yet at
|
||||||
|
// the time seedUatStaffAccounts() returns).
|
||||||
|
await seedUatGroomerLinkage(db, uatCustomerClientId);
|
||||||
|
|
||||||
// ── Clients & Pets ──
|
// ── Clients & Pets ──
|
||||||
const now = new Date();
|
const now = new Date();
|
||||||
const appointmentsBackDate = new Date(now);
|
const appointmentsBackDate = new Date(now);
|
||||||
@@ -1576,8 +1687,6 @@ async function seed() {
|
|||||||
}
|
}
|
||||||
console.log(`✓ Created ${visitLogCount} grooming visit logs`);
|
console.log(`✓ Created ${visitLogCount} grooming visit logs`);
|
||||||
console.log("\nSeed complete!");
|
console.log("\nSeed complete!");
|
||||||
|
|
||||||
await client.end();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
seed().catch((err) => {
|
seed().catch((err) => {
|
||||||
|
|||||||
Reference in New Issue
Block a user