Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/openstatusHQ/openstatus. 🫖 Status page with uptime monitoring & API monitoring as code 🫖 openstatus.dev
bun drizzle-orm monitoring monitoring-as-code nextjs observability on-call open-source shadcn-ui status-page statuspage synthetic-monitoring tinybird turso uptime uptime-checker uptime-monitor
Something went wrong. Try again.
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292import { CloudTasksClient, type protos } from "@google-cloud/tasks";import { and, db, eq, gte, isNotNull, lte, notInArray } from "@openstatus/db";import type { MonitorStatus } from "@openstatus/db/src/schema";import { maintenance, monitor, monitorStatusTable, selectMonitorSchema, selectMonitorStatusSchema,} from "@openstatus/db/src/schema";import type { Region } from "@openstatus/db/src/schema/constants";import { maintenancesToPageComponents, pageComponent,} from "@openstatus/db/src/schema/page_components";import { regionDict } from "@openstatus/regions";import { type httpPayloadSchema, type tpcPayloadSchema, transformHeaders,} from "@openstatus/utils";import type { NextRequest } from "next/server";import { z } from "zod";
import { env } from "../../../../env";
const periodicityAvailable = selectMonitorSchema.pick({ periodicity: true });
// FIXME: do coerce in zod instead
const DEFAULT_URL = process.env.VERCEL_URL ? `https://${process.env.VERCEL_URL}` : "http://localhost:3000";
// We can't secure cron endpoint by vercel thus we should make sure they are called by the generated urlexport const isAuthorizedDomain = (url: string) => { return url.includes(DEFAULT_URL);};
export const cron = async ({ periodicity, // oxlint-disable-next-line eslint/no-unused-vars req,}: z.infer<typeof periodicityAvailable> & { req: NextRequest }) => { const client = new CloudTasksClient({ projectId: env.GCP_PROJECT_ID, credentials: { client_email: process.env.GCP_CLIENT_EMAIL, private_key: env.GCP_PRIVATE_KEY.replaceAll("\\n", "\n"), }, }); const parent = client.queuePath( env.GCP_PROJECT_ID, env.GCP_LOCATION, periodicity, );
const timestamp = Date.now();
const currentMaintenance = db .select({ id: maintenance.id }) .from(maintenance) .where( and(lte(maintenance.from, new Date()), gte(maintenance.to, new Date())), ) .as("currentMaintenance");
const currentMaintenanceMonitors = db .select({ id: pageComponent.monitorId }) .from(maintenancesToPageComponents) .innerJoin( currentMaintenance, eq(maintenancesToPageComponents.maintenanceId, currentMaintenance.id), ) .innerJoin( pageComponent, eq(maintenancesToPageComponents.pageComponentId, pageComponent.id), ) .where(isNotNull(pageComponent.monitorId));
const result = await db .select() .from(monitor) .where( and( eq(monitor.periodicity, periodicity), eq(monitor.active, true), notInArray(monitor.id, currentMaintenanceMonitors), ), ) .all();
console.log(`Start cron for ${periodicity}`);
const monitors = z.array(selectMonitorSchema).safeParse(result); const allResult = []; if (!monitors.success) { console.error( `Error while fetching the monitors ${monitors.error.issues.map((issue) => issue.message).join(", ")}`, ); throw new Error("Error while fetching the monitors"); }
for (const row of monitors.data) { // const selectedRegions = row.regions.length > 0 ? row.regions : ["ams"];
const result = await db .select() .from(monitorStatusTable) .where(eq(monitorStatusTable.monitorId, row.id)) .all(); const monitorStatus = z.array(selectMonitorStatusSchema).safeParse(result); if (!monitorStatus.success) { console.error( `Error while fetching the monitor status ${monitorStatus.error.issues.map((issue) => issue.message).join(", ")}`, ); continue; }
for (const region of row.regions) { const status = monitorStatus.data.find((m) => region === m.region)?.status || "active";
const r = regionDict[region as keyof typeof regionDict];
if (!r) { console.error(`Invalid region ${region}`); continue; } if (r.deprecated) { // Let's uncomment this when we are ready to remove deprecated regions // We should not use deprecated regions anymore console.error(`Deprecated region ${region}`); continue; } const response = createCronTask({ row, timestamp, client, parent, status, region, }); allResult.push(response); if (periodicity === "30s") { // we schedule another task in 30s const scheduledAt = timestamp + 30 * 1000; const response = createCronTask({ row, timestamp: scheduledAt, client, parent, status, region, }); allResult.push(response); } } }
const allRequests = await Promise.allSettled(allResult);
const success = allRequests.filter((r) => r.status === "fulfilled").length; const failed = allRequests.filter((r) => r.status === "rejected").length;
console.log( `End cron for ${periodicity} with ${allResult.length} jobs with ${success} success and ${failed} failed`, );};// timestamp needs to be in msconst createCronTask = async ({ row, timestamp, client, parent, status, region,}: { row: z.infer<typeof selectMonitorSchema>; timestamp: number; client: CloudTasksClient; parent: string; status: MonitorStatus; region: Region;}) => { let payload: | z.infer<typeof httpPayloadSchema> | z.infer<typeof tpcPayloadSchema> | null = null;
// if (row.jobType === "http") { payload = { workspaceId: String(row.workspaceId), monitorId: String(row.id), url: row.url, method: row.method || "GET", cronTimestamp: timestamp, body: row.body, headers: row.headers, status: status, assertions: row.assertions ? JSON.parse(row.assertions) : null, degradedAfter: row.degradedAfter, timeout: row.timeout, trigger: "cron", otelConfig: row.otelEndpoint ? { endpoint: row.otelEndpoint, headers: transformHeaders(row.otelHeaders), } : undefined, retry: row.retry || 3, followRedirects: row.followRedirects || true, }; } if (row.jobType === "tcp") { payload = { workspaceId: String(row.workspaceId), monitorId: String(row.id), uri: row.url, status: status, assertions: row.assertions ? JSON.parse(row.assertions) : null, cronTimestamp: timestamp, degradedAfter: row.degradedAfter, timeout: row.timeout, trigger: "cron", retry: row.retry || 3, otelConfig: row.otelEndpoint ? { endpoint: row.otelEndpoint, headers: transformHeaders(row.otelHeaders), } : undefined, }; }
if (!payload) { throw new Error("Invalid jobType"); } const regionInfo = regionDict[region]; let regionHeader = {}; if (regionInfo.provider === "fly") { regionHeader = { "fly-prefer-region": region }; } if (regionInfo.provider === "koyeb") { regionHeader = { "X-KOYEB-REGION-OVERRIDE": region.replace("koyeb_", "") }; } if (regionInfo.provider === "railway") { regionHeader = { "railway-region": region.replace("railway_", "") }; } const newTask: protos.google.cloud.tasks.v2beta3.ITask = { httpRequest: { headers: { "Content-Type": "application/json", // Set content type to ensure compatibility your application's request parsing ...regionHeader, Authorization: `Basic ${env.CRON_SECRET}`, }, httpMethod: "POST", url: generateUrl({ row, region }), body: Buffer.from(JSON.stringify(payload)).toString("base64"), }, scheduleTime: { seconds: timestamp / 1000, }, };
const request = { parent: parent, task: newTask }; return client.createTask(request);};
function generateUrl({ row, region,}: { row: z.infer<typeof selectMonitorSchema>; region: Region;}) { const regionInfo = regionDict[region];
switch (regionInfo.provider) { case "fly": return `https://openstatus-checker.fly.dev/checker/${row.jobType}?monitor_id=${row.id}`; case "koyeb": return `https://openstatus-checker.koyeb.app/checker/${row.jobType}?monitor_id=${row.id}`; case "railway": return `https://railway-proxy-production-9cb1.up.railway.app/checker/${row.jobType}?monitor_id=${row.id}`;
default: throw new Error("Invalid jobType"); }}