@oneuptime/common 7.0.3831 → 7.0.3864

package/Server/Services/ProjectService.ts

@@ -70,6 +70,7 @@ import AlertStateService from "./AlertStateService";
 import SlackUtil from "../Utils/Workspace/Slack/Slack";
 import URL from "../../Types/API/URL";
 import Exception from "../../Types/Exception/Exception";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export interface CurrentPlan {
   plan: PlanType | null;
@@ -89,6 +90,7 @@ export class ProjectService extends DatabaseService<Model> {
     return SubscriptionPlan.getPlanType(planId);
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     data: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -266,6 +268,7 @@ export class ProjectService extends DatabaseService<Model> {
     return Promise.resolve({ createBy: data, carryForward: null });
   }
 
+  @CaptureSpan()
   protected override async onBeforeUpdate(
     updateBy: UpdateBy<Model>,
   ): Promise<OnUpdate<Model>> {
@@ -552,6 +555,7 @@ export class ProjectService extends DatabaseService<Model> {
     return createdItem;
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     _onCreate: OnCreate<Model>,
     createdItem: Model,
@@ -732,6 +736,7 @@ export class ProjectService extends DatabaseService<Model> {
     return createdItem;
   }
 
+  @CaptureSpan()
   public async addDefaultAlertState(createdItem: Model): Promise<Model> {
     let createdAlertState: AlertState = new AlertState();
     createdAlertState.name = "Identified";
@@ -784,6 +789,7 @@ export class ProjectService extends DatabaseService<Model> {
     return createdItem;
   }
 
+  @CaptureSpan()
   public async addDefaultAlertSeverity(createdItem: Model): Promise<Model> {
     let highSeverity: AlertSeverity = new AlertSeverity();
     highSeverity.name = "High";
@@ -1057,6 +1063,7 @@ export class ProjectService extends DatabaseService<Model> {
     return createdItem;
   }
 
+  @CaptureSpan()
   public async updateLastActive(projectId: ObjectID): Promise<void> {
     await this.updateOneById({
       id: projectId,
@@ -1069,6 +1076,7 @@ export class ProjectService extends DatabaseService<Model> {
     });
   }
 
+  @CaptureSpan()
   public async getOwners(projectId: ObjectID): Promise<Array<User>> {
     if (!projectId) {
       throw new BadDataException("Project ID is required");
@@ -1104,6 +1112,7 @@ export class ProjectService extends DatabaseService<Model> {
     return TeamMemberService.getUsersInTeams(teamIds);
   }
 
+  @CaptureSpan()
   protected override async onBeforeFind(
     findBy: FindBy<Model>,
   ): Promise<OnFind<Model>> {
@@ -1120,6 +1129,7 @@ export class ProjectService extends DatabaseService<Model> {
     return { findBy, carryForward: null };
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<Model>,
   ): Promise<OnDelete<Model>> {
@@ -1145,6 +1155,7 @@ export class ProjectService extends DatabaseService<Model> {
     return { deleteBy, carryForward: projects };
   }
 
+  @CaptureSpan()
   protected override async onDeleteSuccess(
     onDelete: OnDelete<Model>,
     _itemIdsBeforeDelete: ObjectID[],
@@ -1208,6 +1219,7 @@ export class ProjectService extends DatabaseService<Model> {
     return onDelete;
   }
 
+  @CaptureSpan()
   public async getCurrentPlan(projectId: ObjectID): Promise<CurrentPlan> {
     if (!IsBillingEnabled) {
       return { plan: null, isSubscriptionUnpaid: false };
@@ -1251,6 +1263,7 @@ export class ProjectService extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   public async sendEmailToProjectOwners(
     projectId: ObjectID,
     subject: string,
@@ -1286,6 +1299,7 @@ export class ProjectService extends DatabaseService<Model> {
     }
   }
 
+  @CaptureSpan()
   public async reactiveSubscription(projectId: ObjectID): Promise<void> {
     logger.debug("Reactivating subscription for project " + projectId);
 
@@ -1401,6 +1415,7 @@ export class ProjectService extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   public async isSMSNotificationsEnabled(
     projectId: ObjectID,
   ): Promise<boolean> {

package/Server/Services/ProjectUserService.ts

@@ -4,12 +4,14 @@ import ObjectID from "../../Types/ObjectID";
 import DatabaseService from "./DatabaseService";
 import Model from "Common/Models/DatabaseModels/ProjectUser";
 import TeamMemberService from "./TeamMemberService";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   public async refreshProjectUsersByProject(data: {
     projectId: ObjectID;
   }): Promise<void> {

package/Server/Services/ScheduledMaintenanceFeedService.ts

@@ -9,15 +9,10 @@ import DatabaseService from "./DatabaseService";
 import Model, {
   ScheduledMaintenanceFeedEventType,
 } from "Common/Models/DatabaseModels/ScheduledMaintenanceFeed";
-import NotificationRuleEventType from "../../Types/Workspace/NotificationRules/EventType";
-import ScheduledMaintenanceService from "./ScheduledMaintenanceService";
-import { WorkspaceChannel } from "../Utils/Workspace/WorkspaceBase";
-import WorkspaceUtil from "../Utils/Workspace/Workspace";
-import WorkspaceType from "../../Types/Workspace/WorkspaceType";
 import WorkspaceNotificationRuleService, {
   MessageBlocksByWorkspaceType,
 } from "./WorkspaceNotificationRuleService";
-import WorkspaceMessagePayload from "../../Types/Workspace/WorkspaceMessagePayload";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
@@ -28,6 +23,7 @@ export class Service extends DatabaseService<Model> {
     }
   }
 
+  @CaptureSpan()
   public async createScheduledMaintenanceFeedItem(data: {
     scheduledMaintenanceId: ObjectID;
     feedInfoInMarkdown: string;
@@ -101,81 +97,11 @@ export class Service extends DatabaseService<Model> {
       try {
         // send notification to slack and teams
         if (data.workspaceNotification?.sendWorkspaceNotification) {
-          let messageBlocksByWorkspaceTypes: Array<MessageBlocksByWorkspaceType> =
-            [];
-
-          // use markdown to create blocks
-          messageBlocksByWorkspaceTypes =
-            await WorkspaceUtil.getMessageBlocksByMarkdown({
-              userId: data.workspaceNotification.notifyUserId,
-              markdown: data.feedInfoInMarkdown,
-              projectId: data.projectId,
-            });
-
-          if (data.workspaceNotification.appendMessageBlocks) {
-            for (const messageBlocksByWorkspaceType of data
-              .workspaceNotification.appendMessageBlocks) {
-              const workspaceType: WorkspaceType =
-                messageBlocksByWorkspaceType.workspaceType;
-
-              messageBlocksByWorkspaceTypes
-                .find(
-                  (
-                    messageBlocksByWorkspaceType: MessageBlocksByWorkspaceType,
-                  ) => {
-                    return (
-                      messageBlocksByWorkspaceType.workspaceType ===
-                      workspaceType
-                    );
-                  },
-                )
-                ?.messageBlocks.push(
-                  ...messageBlocksByWorkspaceType.messageBlocks,
-                );
-            }
-          }
-
-          const workspaceNotificationPaylaods: Array<WorkspaceMessagePayload> =
-            [];
-
-          for (const messageBlocksByWorkspaceType of messageBlocksByWorkspaceTypes) {
-            const existingChannels: Array<string> =
-              await WorkspaceNotificationRuleService.getExistingChannelNamesBasedOnEventType(
-                {
-                  projectId: data.projectId,
-                  notificationRuleEventType:
-                    NotificationRuleEventType.ScheduledMaintenance,
-                  workspaceType: messageBlocksByWorkspaceType.workspaceType,
-                },
-              );
-
-            const scheduledMaintenanceChannels: Array<WorkspaceChannel> =
-              await ScheduledMaintenanceService.getWorkspaceChannelForScheduledMaintenance(
-                {
-                  scheduledMaintenanceId: data.scheduledMaintenanceId,
-                  workspaceType: messageBlocksByWorkspaceType.workspaceType,
-                },
-              );
-
-            const workspaceMessagePayload: WorkspaceMessagePayload = {
-              _type: "WorkspaceMessagePayload",
-              workspaceType: messageBlocksByWorkspaceType.workspaceType,
-              messageBlocks: messageBlocksByWorkspaceType.messageBlocks,
-              channelNames: existingChannels,
-              channelIds:
-                scheduledMaintenanceChannels.map(
-                  (channel: WorkspaceChannel) => {
-                    return channel.id;
-                  },
-                ) || [],
-            };
-
-            workspaceNotificationPaylaods.push(workspaceMessagePayload);
-          }
-
-          await WorkspaceUtil.postMessageToAllWorkspaceChannelsAsBot({
+          await this.sendWorkspaceNotification({
             projectId: data.projectId,
-            messagePayloadsByWorkspace: workspaceNotificationPaylaods,
+            scheduledMaintenanceId: data.scheduledMaintenanceId,
+            feedInfoInMarkdown: data.feedInfoInMarkdown,
+            workspaceNotification: data.workspaceNotification,
           });
         }
       } catch (e) {
@@ -192,6 +118,29 @@ export class Service extends DatabaseService<Model> {
       // we dont want to throw the error here, as this is not critical but we still log it.
     }
   }
+
+  @CaptureSpan()
+  public async sendWorkspaceNotification(data: {
+    projectId: ObjectID;
+    scheduledMaintenanceId: ObjectID;
+    feedInfoInMarkdown: string;
+    workspaceNotification: {
+      notifyUserId?: ObjectID | undefined; // this is oneuptime user id.
+      sendWorkspaceNotification: boolean;
+      appendMessageBlocks?: Array<MessageBlocksByWorkspaceType> | undefined;
+    };
+  }): Promise<void> {
+    return await WorkspaceNotificationRuleService.sendWorkspaceMarkdownNotification(
+      {
+        projectId: data.projectId,
+        notificationFor: {
+          scheduledMaintenanceId: data.scheduledMaintenanceId,
+        },
+        feedInfoInMarkdown: data.feedInfoInMarkdown,
+        workspaceNotification: data.workspaceNotification,
+      },
+    );
+  }
 }
 
 export default new Service();

package/Server/Services/ScheduledMaintenanceInternalNoteService.ts

@@ -8,12 +8,14 @@ import { Blue500 } from "../../Types/BrandColors";
 import { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
 import ScheduledMaintenance from "../../Models/DatabaseModels/ScheduledMaintenance";
 import ScheduledMaintenanceService from "./ScheduledMaintenanceService";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   public async addNote(data: {
     userId: ObjectID;
     scheduledMaintenanceId: ObjectID;
@@ -34,6 +36,7 @@ export class Service extends DatabaseService<Model> {
     });
   }
 
+  @CaptureSpan()
   public override async onCreateSuccess(
     _onCreate: OnCreate<Model>,
     createdItem: Model,
@@ -70,6 +73,7 @@ ${createdItem.note}
     return createdItem;
   }
 
+  @CaptureSpan()
   public override async onUpdateSuccess(
     onUpdate: OnUpdate<Model>,
     _updatedItemIds: Array<ObjectID>,

package/Server/Services/ScheduledMaintenanceOwnerTeamService.ts

@@ -13,12 +13,14 @@ import WorkspaceNotificationRuleService from "./WorkspaceNotificationRuleService
 import WorkspaceNotificationRule from "../../Models/DatabaseModels/WorkspaceNotificationRule";
 import logger from "../Utils/Logger";
 import ScheduledMaintenanceService from "./ScheduledMaintenanceService";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<Model>,
   ): Promise<OnDelete<Model>> {
@@ -44,6 +46,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onDeleteSuccess(
     onDelete: OnDelete<Model>,
     _itemIdsBeforeDelete: Array<ObjectID>,
@@ -89,6 +92,7 @@ export class Service extends DatabaseService<Model> {
     return onDelete;
   }
 
+  @CaptureSpan()
   public override async onCreateSuccess(
     onCreate: OnCreate<Model>,
     createdItem: Model,

package/Server/Services/ScheduledMaintenanceOwnerUserService.ts

@@ -13,12 +13,14 @@ import WorkspaceNotificationRuleService from "./WorkspaceNotificationRuleService
 import NotificationRuleEventType from "../../Types/Workspace/NotificationRules/EventType";
 import ScheduledMaintenanceService from "./ScheduledMaintenanceService";
 import logger from "../Utils/Logger";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<Model>,
   ): Promise<OnDelete<Model>> {
@@ -44,6 +46,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onDeleteSuccess(
     onDelete: OnDelete<Model>,
     _itemIdsBeforeDelete: Array<ObjectID>,
@@ -90,6 +93,7 @@ export class Service extends DatabaseService<Model> {
     return onDelete;
   }
 
+  @CaptureSpan()
   public override async onCreateSuccess(
     onCreate: OnCreate<Model>,
     createdItem: Model,

package/Server/Services/ScheduledMaintenancePublicNoteService.ts

@@ -10,12 +10,14 @@ import { Blue500, Indigo500 } from "../../Types/BrandColors";
 import { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
 import ScheduledMaintenanceService from "./ScheduledMaintenanceService";
 import ScheduledMaintenance from "../../Models/DatabaseModels/ScheduledMaintenance";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -29,6 +31,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   public override async onCreateSuccess(
     _onCreate: OnCreate<Model>,
     createdItem: Model,
@@ -64,6 +67,7 @@ ${createdItem.note}
     return createdItem;
   }
 
+  @CaptureSpan()
   public override async onUpdateSuccess(
     onUpdate: OnUpdate<Model>,
     _updatedItemIds: Array<ObjectID>,
@@ -118,6 +122,7 @@ ${updatedItem.note}
     return onUpdate;
   }
 
+  @CaptureSpan()
   public async addNote(data: {
     userId: ObjectID;
     scheduledMaintenanceId: ObjectID;

package/Server/Services/ScheduledMaintenanceService.ts

@@ -56,6 +56,7 @@ import WorkspaceType from "../../Types/Workspace/WorkspaceType";
 import NotificationRuleWorkspaceChannel from "../../Types/Workspace/NotificationRules/NotificationRuleWorkspaceChannel";
 import { MessageBlocksByWorkspaceType } from "./WorkspaceNotificationRuleService";
 import ScheduledMaintenanceWorkspaceMessages from "../Utils/Workspace/WorkspaceMessages/ScheduledMaintenance";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
@@ -65,6 +66,7 @@ export class Service extends DatabaseService<Model> {
     }
   }
 
+  @CaptureSpan()
   public async getExistingScheduledMaintenanceNumberForProject(data: {
     projectId: ObjectID;
   }): Promise<number> {
@@ -88,9 +90,12 @@ export class Service extends DatabaseService<Model> {
       return 0;
     }
 
-    return lastScheduledMaintenance.scheduledMaintenanceNumber || 0;
+    return lastScheduledMaintenance.scheduledMaintenanceNumber
+      ? Number(lastScheduledMaintenance.scheduledMaintenanceNumber)
+      : 0;
   }
 
+  @CaptureSpan()
   public async notififySubscribersOnEventScheduled(
     scheduledEvents: Array<Model>,
   ): Promise<void> {
@@ -306,6 +311,7 @@ export class Service extends DatabaseService<Model> {
     );
   }
 
+  @CaptureSpan()
   protected override async onBeforeUpdate(
     updateBy: UpdateBy<Model>,
   ): Promise<OnUpdate<Model>> {
@@ -348,6 +354,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<Model>,
   ): Promise<OnDelete<Model>> {
@@ -375,6 +382,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onDeleteSuccess(
     onDelete: OnDelete<Model>,
     _deletedItemIds: ObjectID[],
@@ -421,6 +429,7 @@ export class Service extends DatabaseService<Model> {
     return recurringDate;
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -486,6 +495,7 @@ export class Service extends DatabaseService<Model> {
     return { createBy, carryForward: null };
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     onCreate: OnCreate<Model>,
     createdItem: Model,
@@ -536,6 +546,9 @@ export class Service extends DatabaseService<Model> {
           name: true,
           _id: true,
         },
+        labels: {
+          name: true,
+        },
       },
       props: {
         isRoot: true,
@@ -645,6 +658,7 @@ ${createdItem.description || "No description provided."}
     return createdItem;
   }
 
+  @CaptureSpan()
   public async addOwners(
     projectId: ObjectID,
     scheduledMaintenanceId: ObjectID,
@@ -688,6 +702,7 @@ ${createdItem.description || "No description provided."}
     }
   }
 
+  @CaptureSpan()
   public async getScheduledMaintenanceLinkInDashboard(
     projectId: ObjectID,
     scheduledMaintenanceId: ObjectID,
@@ -699,6 +714,7 @@ ${createdItem.description || "No description provided."}
     );
   }
 
+  @CaptureSpan()
   public async findOwners(
     scheduledMaintenanceId: ObjectID,
   ): Promise<Array<User>> {
@@ -775,6 +791,7 @@ ${createdItem.description || "No description provided."}
     return users;
   }
 
+  @CaptureSpan()
   public async changeAttachedMonitorStates(
     item: Model,
     props: DatabaseCommonInteractionProps,
@@ -803,6 +820,7 @@ ${createdItem.description || "No description provided."}
     }
   }
 
+  @CaptureSpan()
   protected override async onUpdateSuccess(
     onUpdate: OnUpdate<Model>,
     updatedItemIds: ObjectID[],
@@ -1053,6 +1071,7 @@ ${labels
     return onUpdate;
   }
 
+  @CaptureSpan()
   public async changeScheduledMaintenanceState(data: {
     projectId: ObjectID;
     scheduledMaintenanceId: ObjectID;
@@ -1143,6 +1162,7 @@ ${labels
     });
   }
 
+  @CaptureSpan()
   public async isScheduledMaintenanceCompleted(data: {
     scheduledMaintenanceId: ObjectID;
   }): Promise<boolean> {
@@ -1194,6 +1214,7 @@ ${labels
     return false;
   }
 
+  @CaptureSpan()
   public async getScheduledMaintenanceNumber(data: {
     scheduledMaintenanceId: ObjectID;
   }): Promise<number | null> {
@@ -1211,9 +1232,12 @@ ${labels
       throw new BadDataException("ScheduledMaintenance not found.");
     }
 
-    return scheduledMaintenance.scheduledMaintenanceNumber || null;
+    return scheduledMaintenance.scheduledMaintenanceNumber
+      ? Number(scheduledMaintenance.scheduledMaintenanceNumber)
+      : null;
   }
 
+  @CaptureSpan()
   public async isScheduledMaintenanceOngoing(data: {
     scheduledMaintenanceId: ObjectID;
   }): Promise<boolean> {
@@ -1264,6 +1288,7 @@ ${labels
     return false;
   }
 
+  @CaptureSpan()
   public async markScheduledMaintenanceAsComplete(
     scheduledMaintenanceId: ObjectID,
     resolvedByUserId: ObjectID,
@@ -1325,6 +1350,7 @@ ${labels
     return scheduledMaintenance;
   }
 
+  @CaptureSpan()
   public async markScheduledMaintenanceAsOngoing(
     scheduledMaintenanceId: ObjectID,
     markedByUserId: ObjectID,
@@ -1386,6 +1412,7 @@ ${labels
     return scheduledMaintenance;
   }
 
+  @CaptureSpan()
   public async getWorkspaceChannelForScheduledMaintenance(data: {
     scheduledMaintenanceId: ObjectID;
     workspaceType?: WorkspaceType | null;

package/Server/Services/ScheduledMaintenanceStateService.ts

@@ -10,12 +10,14 @@ import BadDataException from "../../Types/Exception/BadDataException";
 import ObjectID from "../../Types/ObjectID";
 import DatabaseCommonInteractionProps from "../../Types/BaseDatabase/DatabaseCommonInteractionProps";
 import ScheduledMaintenanceState from "Common/Models/DatabaseModels/ScheduledMaintenanceState";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<ScheduledMaintenanceState> {
   public constructor() {
     super(ScheduledMaintenanceState);
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<ScheduledMaintenanceState>,
   ): Promise<OnCreate<ScheduledMaintenanceState>> {
@@ -43,6 +45,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceState> {
     };
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<ScheduledMaintenanceState>,
   ): Promise<OnDelete<ScheduledMaintenanceState>> {
@@ -73,6 +76,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceState> {
     };
   }
 
+  @CaptureSpan()
   protected override async onDeleteSuccess(
     onDelete: OnDelete<ScheduledMaintenanceState>,
     _itemIdsBeforeDelete: ObjectID[],
@@ -101,6 +105,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceState> {
     };
   }
 
+  @CaptureSpan()
   protected override async onBeforeUpdate(
     updateBy: UpdateBy<ScheduledMaintenanceState>,
   ): Promise<OnUpdate<ScheduledMaintenanceState>> {
@@ -162,6 +167,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceState> {
     }
   }
 
+  @CaptureSpan()
   public async getCompletedScheduledMaintenanceState(data: {
     projectId: ObjectID;
     props: DatabaseCommonInteractionProps;
@@ -189,6 +195,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceState> {
     return resolvedScheduledMaintenanceState;
   }
 
+  @CaptureSpan()
   public async getAllScheduledMaintenanceStates(data: {
     projectId: ObjectID;
     props: DatabaseCommonInteractionProps;
@@ -217,6 +224,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceState> {
     return scheduledMaintenanceStates;
   }
 
+  @CaptureSpan()
   public async getOngoingScheduledMaintenanceState(data: {
     projectId: ObjectID;
     props: DatabaseCommonInteractionProps;

package/Server/Services/ScheduledMaintenanceStateTimelineService.ts

@@ -26,6 +26,7 @@ import { IsBillingEnabled } from "../EnvironmentConfig";
 import ScheduledMaintenanceFeedService from "./ScheduledMaintenanceFeedService";
 import { ScheduledMaintenanceFeedEventType } from "../../Models/DatabaseModels/ScheduledMaintenanceFeed";
 import logger from "../Utils/Logger";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<ScheduledMaintenanceStateTimeline> {
   public constructor() {
@@ -35,6 +36,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceStateTimeline>
     }
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<ScheduledMaintenanceStateTimeline>,
   ): Promise<OnCreate<ScheduledMaintenanceStateTimeline>> {
@@ -134,6 +136,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceStateTimeline>
     };
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     onCreate: OnCreate<ScheduledMaintenanceStateTimeline>,
     createdItem: ScheduledMaintenanceStateTimeline,
@@ -361,6 +364,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceStateTimeline>
     return createdItem;
   }
 
+  @CaptureSpan()
   public async enableActiveMonitoringForMonitors(
     scheduledMaintenanceEvent: ScheduledMaintenance,
   ): Promise<void> {
@@ -446,6 +450,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceStateTimeline>
     }
   }
 
+  @CaptureSpan()
   public async hasThisMonitorMoreOngoingScheduledMaintenanceEvents(
     id: ObjectID,
   ): Promise<boolean> {
@@ -468,6 +473,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceStateTimeline>
     return false;
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<ScheduledMaintenanceStateTimeline>,
   ): Promise<OnDelete<ScheduledMaintenanceStateTimeline>> {
@@ -608,6 +614,7 @@ export class Service extends DatabaseService<ScheduledMaintenanceStateTimeline>
     return { deleteBy, carryForward: null };
   }
 
+  @CaptureSpan()
   protected override async onDeleteSuccess(
     onDelete: OnDelete<ScheduledMaintenanceStateTimeline>,
     _itemIdsBeforeDelete: ObjectID[],