Eric Allam
a5aed0d139
Also adding a /metrics endpoint with a few custom metrics and the built-in prom-client ones
45 lines
1.1 KiB
TypeScript
45 lines
1.1 KiB
TypeScript
import { z } from "zod";
|
|
import { singleton } from "~/utils/singleton";
|
|
import { ZodPubSub, ZodSubscriber } from "../utils/zodPubSub.server";
|
|
import { env } from "~/env.server";
|
|
import { Gauge } from "prom-client";
|
|
import { metricsRegister } from "~/metrics.server";
|
|
|
|
const messageCatalog = {
|
|
CANCEL_ATTEMPT: z.object({
|
|
version: z.literal("v1").default("v1"),
|
|
backgroundWorkerId: z.string(),
|
|
attemptId: z.string(),
|
|
taskRunId: z.string(),
|
|
}),
|
|
};
|
|
|
|
export type DevSubscriber = ZodSubscriber<typeof messageCatalog>;
|
|
|
|
export const devPubSub = singleton("devPubSub", initializeDevPubSub);
|
|
|
|
function initializeDevPubSub() {
|
|
const pubSub = new ZodPubSub({
|
|
redis: {
|
|
port: env.REDIS_PORT,
|
|
host: env.REDIS_HOST,
|
|
username: env.REDIS_USERNAME,
|
|
password: env.REDIS_PASSWORD,
|
|
enableAutoPipelining: true,
|
|
...(env.REDIS_TLS_DISABLED === "true" ? {} : { tls: {} }),
|
|
},
|
|
schema: messageCatalog,
|
|
});
|
|
|
|
new Gauge({
|
|
name: "dev_pub_sub_subscribers",
|
|
help: "Number of dev pub sub subscribers",
|
|
collect() {
|
|
this.set(pubSub.subscriberCount);
|
|
},
|
|
registers: [metricsRegister],
|
|
});
|
|
|
|
return pubSub;
|
|
}
|