jobqueue and stauts service

Author: ajhollid

server/src/init/services.ts

@@ -36,6 +36,8 @@ import { MeService } from "@/services/index.js";
 import { MongoMonitorRepository } from "@/repositories/index.js";
 
 export const initServices = async () => {
+  const monitorRepository = new MongoMonitorRepository();
+
   const checkService = new CheckService();
   const inviteService = new InviteService();
   const maintenanceService = new MaintenanceService();
@@ -46,7 +48,7 @@ export const initServices = async () => {
   const slackService = new SlackService();
   const webhookService = new WebhookService();
   const networkService = new NetworkService(got);
-  const statusService = new StatusService();
+  const statusService = new StatusService(monitorRepository);
   const settingsService = new SettingsService();
   const stripeService = new StripeService();
   const billingService = new BillingService();
@@ -68,11 +70,10 @@ export const initServices = async () => {
     maintenanceService,
     statsAggregationService
   );
-  const jobQueue = await JobQueue.create(jobGenerator);
+  const jobQueue = await JobQueue.create(jobGenerator, monitorRepository);
   const entitlementsProvider = EntitlementsFactory.create();
   const authService = new AuthService(jobQueue, entitlementsProvider);
   const meService = new MeService(entitlementsProvider);
-  const monitorRepository = new MongoMonitorRepository();
   const monitorService = new MonitorService(jobQueue, monitorRepository);
   const queueService = new QueueService(jobQueue);
   const teamService = new TeamService(jobQueue);

server/src/repositories/monitors/IMonitorRepoistory.ts

@@ -4,6 +4,20 @@ import type {
   MonitorStatus,
 } from "@/types/domain/monitor.js";
 
+export type UpdateMonitorPatch = Partial<
+  Pick<
+    Monitor,
+    | "name"
+    | "secret"
+    | "interval"
+    | "rejectUnauthorized"
+    | "n"
+    | "thresholds"
+    | "notificationChannels"
+    | "status"
+  >
+>;
+
 export interface TeamQueryConfig {
   search: string;
   sortField: string;
@@ -15,7 +29,16 @@ export interface TeamQueryConfig {
 }
 
 export interface IMonitorRepository {
+  // Create
+  create(
+    orgId: string,
+    userId: string,
+    currentTeamId: string,
+    monitorData: Monitor
+  ): Promise<Monitor>;
+
   // Single fetch
+  findAll(): Promise<Monitor[]>;
   findById(monitorId: string, teamId: string): Promise<Monitor | null>;
 
   // Collection fetches
@@ -34,8 +57,23 @@ export interface IMonitorRepository {
   findByOrgId(orgId: string): Promise<Monitor[]>;
 
   // Deletions
-  deleteById(id: string): Promise<boolean>;
-  deleteByOrgId(orgId: string): Promise<number>;
-}
+  deleteById(monitorId: string, teamId: string): Promise<boolean>;
+  deleteByOrgId(orgId: string): Promise<boolean>;
+
+  // Updates
+  togglePauseById(
+    monitorId: string,
+    teamId: string,
+    userId: string
+  ): Promise<Monitor | null>;
+
+  updateById(
+    monitorId: string,
+    teamId: string,
+    userId: string,
+    patch: UpdateMonitorPatch
+  ): Promise<Monitor | null>;
 
-export default IMonitorRepository;
+  // Counts
+  countByOrgId(orgId: string): Promise<number>;
+}

server/src/repositories/monitors/MongoMonitorRepository.ts

@@ -30,6 +30,27 @@ class MongoMonitorRepository implements IMonitorRepository {
     };
   };
 
+  // Create
+  create = async (
+    orgId: string,
+    userId: string,
+    currentTeamId: string,
+    monitorData: MonitorEntity
+  ) => {
+    const monitorLiteral: Partial<IMonitor> = {
+      ...monitorData,
+      orgId: new mongoose.Types.ObjectId(orgId),
+      teamId: new mongoose.Types.ObjectId(currentTeamId),
+      createdBy: new mongoose.Types.ObjectId(userId),
+      updatedBy: new mongoose.Types.ObjectId(userId),
+      notificationChannels: monitorData.notificationChannels?.map(
+        (id) => new mongoose.Types.ObjectId(id)
+      ),
+    };
+    const monitor = await Monitor.create(monitorLiteral);
+    return this.toEntity(monitor);
+  };
+
   // Single fetch
   findById = async (monitorId: string, teamId: string) => {
     const monitor = await Monitor.findOne({ _id: monitorId, teamId });
@@ -38,6 +59,11 @@ class MongoMonitorRepository implements IMonitorRepository {
   };
 
   // Collection fetches
+  findAll = async () => {
+    const monitors = await Monitor.find();
+    return monitors.map(this.toEntity);
+  };
+
   findByTeamId = async (teamId: string) => {
     const monitors = await Monitor.find({ teamId });
     return monitors.map(this.toEntity);
@@ -81,8 +107,10 @@ class MongoMonitorRepository implements IMonitorRepository {
 
     return monitors.map(this.toEntity);
   };
+
   findByOrgId = async (orgId: string) => {
-    return [];
+    const monitors = await Monitor.find({ orgId });
+    return monitors.map(this.toEntity);
   };
 
   findMonitorCountsByTeamId = async (teamId: string) => {
@@ -115,12 +143,69 @@ class MongoMonitorRepository implements IMonitorRepository {
   };
 
   // Deletions
-  deleteById = async (id: string) => {
-    return false;
+  deleteById = async (monitorId: string, teamId: string) => {
+    await Monitor.deleteOne({ _id: monitorId, teamId });
+    return true;
   };
   deleteByOrgId = async (orgId: string) => {
-    return 0;
+    const deleted = await Monitor.deleteMany({ orgId });
+    return deleted.acknowledged;
   };
+
+  togglePauseById = async (
+    monitorId: string,
+    teamId: string,
+    userId: string
+  ) => {
+    const monitor = await Monitor.findOneAndUpdate(
+      { _id: monitorId, teamId },
+      [
+        {
+          $set: {
+            status: {
+              $cond: {
+                if: { $eq: ["$status", "paused"] },
+                then: "resuming",
+                else: "paused",
+              },
+            },
+            updatedBy: userId,
+            updatedAt: new Date(),
+          },
+        },
+      ],
+      { new: true }
+    );
+    if (!monitor) return null;
+    return this.toEntity(monitor);
+  };
+
+  // Update
+  updateById = async (
+    monitorId: string,
+    teamId: string,
+    userId: string,
+    patch: Partial<MonitorEntity>
+  ) => {
+    const updatedMonitor = await Monitor.findOneAndUpdate(
+      { _id: monitorId, teamId },
+      {
+        $set: {
+          ...patch,
+          updatedAt: new Date(),
+          updatedBy: userId,
+        },
+      },
+      { new: true, runValidators: true }
+    );
+    if (!updatedMonitor) return null;
+    return this.toEntity(updatedMonitor);
+  };
+
+  // Counts
+  countByOrgId(orgId: string): Promise<number> {
+    return Monitor.countDocuments({ orgId });
+  }
 }
 
 export default MongoMonitorRepository;