nicktrn
2c91c492df
* Add notification system * Graceful worker shutdown on migration notification * Add notification catalog * Rename pgListen call to on
28 lines
806 B
TypeScript
28 lines
806 B
TypeScript
import { z } from "zod";
|
|
import { PrismaClient, prisma } from "~/db.server";
|
|
import { logger } from "~/services/logger.server";
|
|
import { NotificationCatalog, NotificationChannel } from "./types";
|
|
|
|
export class PgNotifyService {
|
|
#prismaClient: PrismaClient;
|
|
|
|
constructor(prismaClient: PrismaClient = prisma) {
|
|
this.#prismaClient = prismaClient;
|
|
}
|
|
|
|
public async call<TChannel extends NotificationChannel>(
|
|
channelName: TChannel,
|
|
payload: z.infer<NotificationCatalog[TChannel]>
|
|
) {
|
|
this.#logDebug("Sending notification", { channelName, notifyPayload: payload });
|
|
|
|
await this.#prismaClient.$executeRaw`
|
|
SELECT pg_notify(${channelName}, ${JSON.stringify(payload)})
|
|
`;
|
|
}
|
|
|
|
#logDebug(message: string, args?: any) {
|
|
logger.debug(`[pgNotify] ${message}`, args);
|
|
}
|
|
}
|