@oneuptime/common 7.0.3840 → 7.0.3864

package/Server/Services/BillingService.ts

@@ -15,6 +15,7 @@ import BadDataException from "../../Types/Exception/BadDataException";
 import ProductType from "../../Types/MeteredPlan/ProductType";
 import ObjectID from "../../Types/ObjectID";
 import Stripe from "stripe";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export type SubscriptionItem = Stripe.SubscriptionItem;
 
@@ -49,6 +50,7 @@ export class BillingService extends BaseService {
   });
 
   // returns billing id of the customer.
+  @CaptureSpan()
   public async createCustomer(data: {
     name: string;
     id: ObjectID;
@@ -68,6 +70,7 @@ export class BillingService extends BaseService {
     return customer.id;
   }
 
+  @CaptureSpan()
   public async updateCustomerName(id: string, newName: string): Promise<void> {
     if (!this.isBillingEnabled()) {
       throw new BadDataException(Errors.BillingService.BILLING_NOT_ENABLED);
@@ -76,6 +79,7 @@ export class BillingService extends BaseService {
     await this.stripe.customers.update(id, { name: newName });
   }
 
+  @CaptureSpan()
   public async deleteCustomer(id: string): Promise<void> {
     if (!this.isBillingEnabled()) {
       throw new BadDataException(Errors.BillingService.BILLING_NOT_ENABLED);
@@ -92,6 +96,7 @@ export class BillingService extends BaseService {
     return SubscriptionStatusUtil.isSubscriptionActive(status);
   }
 
+  @CaptureSpan()
   public async subscribeToMeteredPlan(data: {
     projectId: ObjectID;
     customerId: string;
@@ -149,6 +154,7 @@ export class BillingService extends BaseService {
     return BillingPrivateKey.startsWith("sk_test");
   }
 
+  @CaptureSpan()
   public async generateCouponCode(data: {
     name: string;
     metadata?: Dictionary<string> | undefined;
@@ -168,6 +174,7 @@ export class BillingService extends BaseService {
     return coupon.id;
   }
 
+  @CaptureSpan()
   public async subscribeToPlan(data: {
     projectId: ObjectID;
     customerId: string;
@@ -247,6 +254,7 @@ export class BillingService extends BaseService {
     };
   }
 
+  @CaptureSpan()
   public async changeQuantity(
     subscriptionId: string,
     quantity: number,
@@ -285,6 +293,7 @@ export class BillingService extends BaseService {
     });
   }
 
+  @CaptureSpan()
   public async addOrUpdateMeteredPricingOnSubscription(
     subscriptionId: string,
     serverMeteredPlan: ServerMeteredPlan,
@@ -351,6 +360,7 @@ export class BillingService extends BaseService {
     // complete.
   }
 
+  @CaptureSpan()
   public async isPromoCodeValid(promoCode: string): Promise<boolean> {
     if (!this.isBillingEnabled()) {
       throw new BadDataException(Errors.BillingService.BILLING_NOT_ENABLED);
@@ -371,6 +381,7 @@ export class BillingService extends BaseService {
     }
   }
 
+  @CaptureSpan()
   public async removeSubscriptionItem(
     subscriptionId: string,
     subscriptionItemId: string,
@@ -406,6 +417,7 @@ export class BillingService extends BaseService {
     );
   }
 
+  @CaptureSpan()
   public async getSubscriptionItems(
     subscriptionId: string,
   ): Promise<Array<SubscriptionItem>> {
@@ -423,6 +435,7 @@ export class BillingService extends BaseService {
     return subscription.items.data;
   }
 
+  @CaptureSpan()
   public async changePlan(data: {
     projectId: ObjectID;
     subscriptionId: string;
@@ -518,6 +531,7 @@ export class BillingService extends BaseService {
     return value;
   }
 
+  @CaptureSpan()
   public async deletePaymentMethod(
     customerId: string,
     paymentMethodId: string,
@@ -538,6 +552,7 @@ export class BillingService extends BaseService {
     await this.stripe.paymentMethods.detach(paymentMethodId);
   }
 
+  @CaptureSpan()
   public async hasPaymentMethods(customerId: string): Promise<boolean> {
     if ((await this.getPaymentMethods(customerId)).length > 0) {
       return true;
@@ -546,6 +561,7 @@ export class BillingService extends BaseService {
     return false;
   }
 
+  @CaptureSpan()
   public async setDefaultPaymentMethod(
     customerId: string,
     paymentMethodId: string,
@@ -557,6 +573,7 @@ export class BillingService extends BaseService {
     });
   }
 
+  @CaptureSpan()
   public async getPaymentMethods(
     customerId: string,
   ): Promise<Array<PaymentMethod>> {
@@ -643,6 +660,7 @@ export class BillingService extends BaseService {
     return paymentMethods;
   }
 
+  @CaptureSpan()
   public async getSetupIntentSecret(customerId: string): Promise<string> {
     const setupIntent: Stripe.Response<Stripe.SetupIntent> =
       await this.stripe.setupIntents.create({
@@ -656,6 +674,7 @@ export class BillingService extends BaseService {
     return setupIntent.client_secret;
   }
 
+  @CaptureSpan()
   public async cancelSubscription(subscriptionId: string): Promise<void> {
     if (!this.isBillingEnabled()) {
       throw new BadDataException(Errors.BillingService.BILLING_NOT_ENABLED);
@@ -667,6 +686,7 @@ export class BillingService extends BaseService {
     }
   }
 
+  @CaptureSpan()
   public async getSubscriptionStatus(
     subscriptionId: string,
   ): Promise<SubscriptionStatus> {
@@ -675,6 +695,7 @@ export class BillingService extends BaseService {
     return subscription.status as SubscriptionStatus;
   }
 
+  @CaptureSpan()
   public async getSubscription(
     subscriptionId: string,
   ): Promise<Stripe.Subscription> {
@@ -688,6 +709,7 @@ export class BillingService extends BaseService {
     return subscription;
   }
 
+  @CaptureSpan()
   public async getInvoices(customerId: string): Promise<Array<Invoice>> {
     const invoices: Stripe.ApiList<Stripe.Invoice> =
       await this.stripe.invoices.list({
@@ -721,6 +743,7 @@ export class BillingService extends BaseService {
     return billingInvoices;
   }
 
+  @CaptureSpan()
   public async generateInvoiceAndChargeCustomer(
     customerId: string,
     itemText: string,
@@ -754,6 +777,7 @@ export class BillingService extends BaseService {
     }
   }
 
+  @CaptureSpan()
   public async voidInvoice(invoiceId: string): Promise<Stripe.Invoice> {
     const invoice: Stripe.Invoice =
       await this.stripe.invoices.voidInvoice(invoiceId);
@@ -761,6 +785,7 @@ export class BillingService extends BaseService {
     return invoice;
   }
 
+  @CaptureSpan()
   public async payInvoice(
     customerId: string,
     invoiceId: string,

package/Server/Services/CallService.ts

@@ -11,12 +11,14 @@ import TwilioConfig from "../../Types/CallAndSMS/TwilioConfig";
 import { JSONObject } from "../../Types/JSON";
 import ObjectID from "../../Types/ObjectID";
 import API from "Common/Utils/API";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class CallService extends BaseService {
   public constructor() {
     super();
   }
 
+  @CaptureSpan()
   public async makeCall(
     callRequest: CallRequest,
     options: {

package/Server/Services/CopilotActionTypePriorityService.ts

@@ -5,12 +5,13 @@ import CreateBy from "../Types/Database/CreateBy";
 import { OnCreate } from "../Types/Database/Hooks";
 import DatabaseService from "./DatabaseService";
 import Model from "Common/Models/DatabaseModels/CopilotActionTypePriority";
-
+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>> {

package/Server/Services/CopilotCodeRepositoryService.ts

@@ -9,12 +9,13 @@ import {
 } from "../../Types/Copilot/CopilotActionType";
 import CopilotActionTypePriority from "../../Models/DatabaseModels/CopilotActionTypePriority";
 import CopilotActionTypePriorityService from "./CopilotActionTypePriorityService";
-
+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>> {
@@ -26,6 +27,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     _onCreate: OnCreate<Model>,
     createdItem: Model,

package/Server/Services/DashboardService.ts

@@ -6,12 +6,13 @@ import Model from "Common/Models/DatabaseModels/Dashboard";
 import { IsBillingEnabled } from "../EnvironmentConfig";
 import { PlanType } from "../../Types/Billing/SubscriptionPlan";
 import DashboardViewConfigUtil from "../../Utils/Dashboard/DashboardViewConfig";
-
+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>> {

package/Server/Services/DatabaseService.ts

@@ -62,6 +62,7 @@ import { DataSource, Repository, SelectQueryBuilder } from "typeorm";
 import { FindWhere } from "../../Types/BaseDatabase/Query";
 import Realtime from "../Utils/Realtime";
 import ModelEventType from "../../Types/Realtime/ModelEventType";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
   public modelType!: { new (): TBaseModel };
@@ -526,6 +527,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     return data;
   }
 
+  @CaptureSpan()
   public async onTriggerRealtime(
     modelId: ObjectID,
     projectId: ObjectID,
@@ -577,6 +579,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     }
   }
 
+  @CaptureSpan()
   public async onTriggerWorkflow(
     id: ObjectID,
     projectId: ObjectID,
@@ -609,6 +612,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     }
   }
 
+  @CaptureSpan()
   public async create(createBy: CreateBy<TBaseModel>): Promise<TBaseModel> {
     const onCreate: OnCreate<TBaseModel> = createBy.props.ignoreHooks
       ? { createBy, carryForward: [] }
@@ -829,6 +833,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     return Promise.resolve(createBy);
   }
 
+  @CaptureSpan()
   public async countBy({
     query,
     skip,
@@ -905,6 +910,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     }
   }
 
+  @CaptureSpan()
   public async deleteOneById(deleteById: DeleteById): Promise<number> {
     await ModelPermission.checkDeletePermissionByModel({
       modelType: this.modelType,
@@ -940,16 +946,19 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     });
   }
 
+  @CaptureSpan()
   public async deleteOneBy(
     deleteOneBy: DeleteOneBy<TBaseModel>,
   ): Promise<number> {
     return await this._deleteBy({ ...deleteOneBy, limit: 1, skip: 0 });
   }
 
+  @CaptureSpan()
   public async deleteBy(deleteBy: DeleteBy<TBaseModel>): Promise<number> {
     return await this._deleteBy(deleteBy);
   }
 
+  @CaptureSpan()
   public async hardDeleteBy(deleteBy: DeleteBy<TBaseModel>): Promise<number> {
     try {
       const onDelete: OnDelete<TBaseModel> = deleteBy.props.ignoreHooks
@@ -1125,6 +1134,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     }
   }
 
+  @CaptureSpan()
   public async findBy(findBy: FindBy<TBaseModel>): Promise<Array<TBaseModel>> {
     return await this._findBy(findBy);
   }
@@ -1293,6 +1303,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     return items;
   }
 
+  @CaptureSpan()
   public async findOneBy(
     findOneBy: FindOneBy<TBaseModel>,
   ): Promise<TBaseModel | null> {
@@ -1308,6 +1319,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     return null;
   }
 
+  @CaptureSpan()
   public async findOneById(
     findOneById: FindOneByID<TBaseModel>,
   ): Promise<TBaseModel | null> {
@@ -1468,16 +1480,19 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     return true;
   }
 
+  @CaptureSpan()
   public async updateOneBy(
     updateOneBy: UpdateOneBy<TBaseModel>,
   ): Promise<number> {
     return await this._updateBy({ ...updateOneBy, limit: 1, skip: 0 });
   }
 
+  @CaptureSpan()
   public async updateBy(updateBy: UpdateBy<TBaseModel>): Promise<number> {
     return await this._updateBy(updateBy);
   }
 
+  @CaptureSpan()
   public async updateOneById(
     updateById: UpdateByID<TBaseModel>,
   ): Promise<void> {
@@ -1519,6 +1534,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     });
   }
 
+  @CaptureSpan()
   public async updateOneByIdAndFetch(
     updateById: UpdateByIDAndFetch<TBaseModel>,
   ): Promise<TBaseModel | null> {
@@ -1530,6 +1546,7 @@ class DatabaseService<TBaseModel extends BaseModel> extends BaseService {
     });
   }
 
+  @CaptureSpan()
   public async searchBy({
     skip,
     limit,

package/Server/Services/DomainService.ts

@@ -6,12 +6,13 @@ import DatabaseService from "./DatabaseService";
 import BadDataException from "../../Types/Exception/BadDataException";
 import Text from "../../Types/Text";
 import Model from "Common/Models/DatabaseModels/Domain";
-
+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>> {
@@ -20,6 +21,7 @@ export class Service extends DatabaseService<Model> {
     return Promise.resolve({ createBy, carryForward: null });
   }
 
+  @CaptureSpan()
   protected override async onBeforeUpdate(
     updateBy: UpdateBy<Model>,
   ): Promise<OnUpdate<Model>> {

package/Server/Services/FileService.ts

@@ -5,12 +5,13 @@ import UpdateBy from "../Types/Database/UpdateBy";
 import DatabaseService from "./DatabaseService";
 import NotAuthorizedException from "../../Types/Exception/NotAuthorizedException";
 import File from "Common/Models/DatabaseModels/File";
-
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 export class Service extends DatabaseService<File> {
   public constructor() {
     super(File);
   }
 
+  @CaptureSpan()
   protected override async onBeforeUpdate(
     updateBy: UpdateBy<File>,
   ): Promise<OnUpdate<File>> {
@@ -21,6 +22,7 @@ export class Service extends DatabaseService<File> {
     return { updateBy, carryForward: null };
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<File>,
   ): Promise<OnDelete<File>> {
@@ -31,6 +33,7 @@ export class Service extends DatabaseService<File> {
     return { deleteBy, carryForward: null };
   }
 
+  @CaptureSpan()
   protected override async onBeforeFind(
     findBy: FindBy<File>,
   ): Promise<OnFind<File>> {

package/Server/Services/IncidentFeedService.ts

@@ -3,7 +3,6 @@ import Color from "../../Types/Color";
 import OneUptimeDate from "../../Types/Date";
 import BadDataException from "../../Types/Exception/BadDataException";
 import ObjectID from "../../Types/ObjectID";
-import WorkspaceMessagePayload from "../../Types/Workspace/WorkspaceMessagePayload";
 import { IsBillingEnabled } from "../EnvironmentConfig";
 import logger from "../Utils/Logger";
 import DatabaseService from "./DatabaseService";
@@ -13,11 +12,7 @@ import IncidentFeed, {
 import WorkspaceNotificationRuleService, {
   MessageBlocksByWorkspaceType,
 } from "./WorkspaceNotificationRuleService";
-import NotificationRuleEventType from "../../Types/Workspace/NotificationRules/EventType";
-import IncidentService from "./IncidentService";
-import { WorkspaceChannel } from "../Utils/Workspace/WorkspaceBase";
-import WorkspaceUtil from "../Utils/Workspace/Workspace";
-import WorkspaceType from "../../Types/Workspace/WorkspaceType";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<IncidentFeed> {
   public constructor() {
@@ -28,6 +23,7 @@ export class Service extends DatabaseService<IncidentFeed> {
     }
   }
 
+  @CaptureSpan()
   public async createIncidentFeedItem(data: {
     incidentId: ObjectID;
     feedInfoInMarkdown: string;
@@ -102,77 +98,15 @@ export class Service extends DatabaseService<IncidentFeed> {
 
       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.Incident,
-                  workspaceType: messageBlocksByWorkspaceType.workspaceType,
-                },
-              );
-
-            const incidentChannels: Array<WorkspaceChannel> =
-              await IncidentService.getWorkspaceChannelForIncident({
-                incidentId: data.incidentId,
-                workspaceType: messageBlocksByWorkspaceType.workspaceType,
-              });
-
-            const workspaceMessagePayload: WorkspaceMessagePayload = {
-              _type: "WorkspaceMessagePayload",
-              workspaceType: messageBlocksByWorkspaceType.workspaceType,
-              messageBlocks: messageBlocksByWorkspaceType.messageBlocks,
-              channelNames: existingChannels,
-              channelIds:
-                incidentChannels.map((channel: WorkspaceChannel) => {
-                  return channel.id;
-                }) || [],
-            };
-
-            workspaceNotificationPaylaods.push(workspaceMessagePayload);
-          }
-
-          await WorkspaceUtil.postMessageToAllWorkspaceChannelsAsBot({
+        if (
+          data.workspaceNotification &&
+          data.workspaceNotification?.sendWorkspaceNotification
+        ) {
+          await this.sendWorkspaceNotification({
             projectId: data.projectId,
-            messagePayloadsByWorkspace: workspaceNotificationPaylaods,
+            incidentId: data.incidentId,
+            feedInfoInMarkdown: data.feedInfoInMarkdown,
+            workspaceNotification: data.workspaceNotification,
           });
         }
       } catch (e) {
@@ -188,6 +122,29 @@ export class Service extends DatabaseService<IncidentFeed> {
       // we dont throw this error as it is not a critical error
     }
   }
+
+  @CaptureSpan()
+  public async sendWorkspaceNotification(data: {
+    projectId: ObjectID;
+    incidentId: 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: {
+          incidentId: data.incidentId,
+        },
+        feedInfoInMarkdown: data.feedInfoInMarkdown,
+        workspaceNotification: data.workspaceNotification,
+      },
+    );
+  }
 }
 
 export default new Service();

package/Server/Services/IncidentInternalNoteService.ts

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

package/Server/Services/IncidentOwnerTeamService.ts

@@ -13,12 +13,14 @@ import WorkspaceNotificationRuleService from "./WorkspaceNotificationRuleService
 import NotificationRuleEventType from "../../Types/Workspace/NotificationRules/EventType";
 import WorkspaceNotificationRule from "../../Models/DatabaseModels/WorkspaceNotificationRule";
 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>,
@@ -94,6 +97,7 @@ export class Service extends DatabaseService<Model> {
     return onDelete;
   }
 
+  @CaptureSpan()
   public override async onCreateSuccess(
     onCreate: OnCreate<Model>,
     createdItem: Model,

package/Server/Services/IncidentOwnerUserService.ts

@@ -13,12 +13,14 @@ import WorkspaceNotificationRuleService from "./WorkspaceNotificationRuleService
 import NotificationRuleEventType from "../../Types/Workspace/NotificationRules/EventType";
 import WorkspaceNotificationRule from "../../Models/DatabaseModels/WorkspaceNotificationRule";
 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>,
@@ -95,6 +98,7 @@ export class Service extends DatabaseService<Model> {
     return onDelete;
   }
 
+  @CaptureSpan()
   public override async onCreateSuccess(
     onCreate: OnCreate<Model>,
     createdItem: Model,