@oneuptime/common 7.0.3831 → 7.0.3864

package/Server/Services/StatusPageSubscriberService.ts

@@ -6,6 +6,7 @@ import {
 } from "../EnvironmentConfig";
 import ProjectSMTPConfigService from "../Services/ProjectSmtpConfigService";
 import CreateBy from "../Types/Database/CreateBy";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 import { OnCreate } from "../Types/Database/Hooks";
 import QueryHelper from "../Types/Database/QueryHelper";
 import logger from "../Utils/Logger";
@@ -36,6 +37,7 @@ export class Service extends DatabaseService<Model> {
     super(Model);
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     data: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -215,6 +217,7 @@ export class Service extends DatabaseService<Model> {
     return { createBy: data, carryForward: statuspage };
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     onCreate: OnCreate<Model>,
     createdItem: Model,
@@ -262,7 +265,8 @@ export class Service extends DatabaseService<Model> {
             _id: true,
             twilioAccountSID: true,
             twilioAuthToken: true,
-            twilioPhoneNumber: true,
+            twilioPrimaryPhoneNumber: true,
+            twilioSecondaryPhoneNumbers: true,
           },
         },
         props: {
@@ -331,6 +335,7 @@ export class Service extends DatabaseService<Model> {
     return createdItem;
   }
 
+  @CaptureSpan()
   public async sendConfirmSubscriptionEmail(data: {
     subscriberId: ObjectID;
   }): Promise<void> {
@@ -464,6 +469,7 @@ export class Service extends DatabaseService<Model> {
     }
   }
 
+  @CaptureSpan()
   public async sendYouHaveSubscribedEmail(data: {
     subscriberId: ObjectID;
   }): Promise<void> {
@@ -610,6 +616,7 @@ export class Service extends DatabaseService<Model> {
     return confirmSubscriptionLink;
   }
 
+  @CaptureSpan()
   public async getSubscribersByStatusPage(
     statusPageId: ObjectID,
     props: DatabaseCommonInteractionProps,
@@ -755,6 +762,7 @@ export class Service extends DatabaseService<Model> {
     return shouldSendNotification;
   }
 
+  @CaptureSpan()
   public async getStatusPagesToSendNotification(
     statusPageIds: Array<ObjectID>,
   ): Promise<Array<StatusPage>> {
@@ -795,7 +803,8 @@ export class Service extends DatabaseService<Model> {
           _id: true,
           twilioAccountSID: true,
           twilioAuthToken: true,
-          twilioPhoneNumber: true,
+          twilioPrimaryPhoneNumber: true,
+          twilioSecondaryPhoneNumbers: true,
         },
         subscriberTimezones: true,
         reportDataInDays: true,

package/Server/Services/TeamMemberService.ts

@@ -4,6 +4,7 @@ import CreateBy from "../Types/Database/CreateBy";
 import DeleteBy from "../Types/Database/DeleteBy";
 import { OnCreate, OnDelete, OnUpdate } from "../Types/Database/Hooks";
 import QueryHelper from "../Types/Database/QueryHelper";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 import Select from "../Types/Database/Select";
 import UpdateBy from "../Types/Database/UpdateBy";
 import Errors from "../Utils/Errors";
@@ -39,6 +40,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     super(TeamMember);
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<TeamMember>,
   ): Promise<OnCreate<TeamMember>> {
@@ -165,6 +167,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     return { createBy, carryForward: null };
   }
 
+  @CaptureSpan()
   public async refreshTokens(
     userId: ObjectID,
     projectId: ObjectID,
@@ -178,6 +181,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     );
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     onCreate: OnCreate<TeamMember>,
     createdItem: TeamMember,
@@ -200,6 +204,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     return createdItem;
   }
 
+  @CaptureSpan()
   protected override async onUpdateSuccess(
     onUpdate: OnUpdate<TeamMember>,
     updatedItemIds: Array<ObjectID>,
@@ -250,6 +255,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     return { updateBy, carryForward: onUpdate.carryForward };
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<TeamMember>,
   ): Promise<OnDelete<TeamMember>> {
@@ -305,6 +311,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     };
   }
 
+  @CaptureSpan()
   protected override async onDeleteSuccess(
     onDelete: OnDelete<TeamMember>,
   ): Promise<OnDelete<TeamMember>> {
@@ -329,6 +336,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     return onDelete;
   }
 
+  @CaptureSpan()
   public async getUniqueTeamMemberCountInProject(
     projectId: ObjectID,
   ): Promise<number> {
@@ -357,6 +365,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     return [...new Set(memberIds)].length; //get unique member ids.
   }
 
+  @CaptureSpan()
   public async getUsersInTeams(teamIds: Array<ObjectID>): Promise<Array<User>> {
     const members: Array<TeamMember> = await this.findBy({
       query: {
@@ -394,6 +403,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     });
   }
 
+  @CaptureSpan()
   public async getUsersInTeam(teamId: ObjectID): Promise<Array<User>> {
     const members: Array<TeamMember> = await this.findBy({
       query: {
@@ -420,6 +430,7 @@ export class TeamMemberService extends DatabaseService<TeamMember> {
     });
   }
 
+  @CaptureSpan()
   public async updateSubscriptionSeatsByUniqueTeamMembersInProject(
     projectId: ObjectID,
   ): Promise<void> {

package/Server/Services/TeamPermissionService.ts

@@ -6,6 +6,7 @@ import AccessTokenService from "./AccessTokenService";
 import DatabaseService from "./DatabaseService";
 import TeamMemberService from "./TeamMemberService";
 import TeamService from "./TeamService";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 import LIMIT_MAX, { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
 import BadDataException from "../../Types/Exception/BadDataException";
 import ObjectID from "../../Types/ObjectID";
@@ -18,6 +19,7 @@ export class Service extends DatabaseService<Model> {
     super(Model);
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -116,6 +118,7 @@ export class Service extends DatabaseService<Model> {
     return { createBy, carryForward: null };
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     onCreate: OnCreate<Model>,
     createdItem: Model,
@@ -150,6 +153,7 @@ export class Service extends DatabaseService<Model> {
     return createdItem;
   }
 
+  @CaptureSpan()
   protected override async onBeforeUpdate(
     updateBy: UpdateBy<Model>,
   ): Promise<OnUpdate<Model>> {
@@ -240,6 +244,7 @@ export class Service extends DatabaseService<Model> {
     return { updateBy, carryForward: teamPermissions };
   }
 
+  @CaptureSpan()
   protected override async onUpdateSuccess(
     onUpdate: OnUpdate<Model>,
     _updatedItemIds: ObjectID[],
@@ -283,6 +288,7 @@ export class Service extends DatabaseService<Model> {
     return onUpdate;
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<Model>,
   ): Promise<OnDelete<Model>> {
@@ -335,6 +341,7 @@ export class Service extends DatabaseService<Model> {
     return { deleteBy, carryForward: teamMembers };
   }
 
+  @CaptureSpan()
   protected override async onDeleteSuccess(
     onDelete: OnDelete<Model>,
     _itemIdsBeforeDelete: ObjectID[],

package/Server/Services/TeamService.ts

@@ -5,12 +5,14 @@ import DatabaseService from "./DatabaseService";
 import LIMIT_MAX from "../../Types/Database/LimitMax";
 import BadDataException from "../../Types/Exception/BadDataException";
 import Model from "Common/Models/DatabaseModels/Team";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   protected override async onBeforeUpdate(
     updateBy: UpdateBy<Model>,
   ): Promise<OnUpdate<Model>> {
@@ -41,6 +43,7 @@ export class Service extends DatabaseService<Model> {
     return { updateBy, carryForward: null };
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<Model>,
   ): Promise<OnDelete<Model>> {

package/Server/Services/TelemetryAttributeService.ts

@@ -4,12 +4,14 @@ import AnalyticsDatabaseService from "./AnalyticsDatabaseService";
 import TelemetryAttribute from "Common/Models/AnalyticsModels/TelemetryAttribute";
 import { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
 import ObjectID from "../../Types/ObjectID";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class TelemetryAttributeService extends AnalyticsDatabaseService<TelemetryAttribute> {
   public constructor(clickhouseDatabase?: ClickhouseDatabase | undefined) {
     super({ modelType: TelemetryAttribute, database: clickhouseDatabase });
   }
 
+  @CaptureSpan()
   public async fetchAttributes(data: {
     projectId: ObjectID;
     telemetryType: TelemetryType;
@@ -40,6 +42,7 @@ export class TelemetryAttributeService extends AnalyticsDatabaseService<Telemetr
     return dbAttributes.sort();
   }
 
+  @CaptureSpan()
   public async refreshAttributes(data: {
     projectId: ObjectID;
     telemetryType: TelemetryType;

package/Server/Services/TelemetryIngestionKeyService.ts

@@ -3,12 +3,14 @@ import { OnCreate } from "../Types/Database/Hooks";
 import DatabaseService from "./DatabaseService";
 import ObjectID from "../../Types/ObjectID";
 import Model from "Common/Models/DatabaseModels/TelemetryIngestionKey";
+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/TelemetryServiceService.ts

@@ -6,12 +6,14 @@ import { BrightColors } from "../../Types/BrandColors";
 import BadDataException from "../../Types/Exception/BadDataException";
 import ObjectID from "../../Types/ObjectID";
 import Model from "Common/Models/DatabaseModels/TelemetryService";
+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 +28,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   public async getTelemetryDataRetentionInDays(
     telemetryServiceId: ObjectID,
   ): Promise<number> {

package/Server/Services/TelemetryUsageBillingService.ts

@@ -11,6 +11,7 @@ import ProductType from "../../Types/MeteredPlan/ProductType";
 import ObjectID from "../../Types/ObjectID";
 import Model from "Common/Models/DatabaseModels/TelemetryUsageBilling";
 import { IsBillingEnabled } from "../EnvironmentConfig";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
@@ -20,6 +21,7 @@ export class Service extends DatabaseService<Model> {
     }
   }
 
+  @CaptureSpan()
   public async getUnreportedUsageBilling(data: {
     projectId: ObjectID;
     productType: ProductType;
@@ -45,6 +47,7 @@ export class Service extends DatabaseService<Model> {
     });
   }
 
+  @CaptureSpan()
   public async updateUsageBilling(data: {
     projectId: ObjectID;
     productType: ProductType;

package/Server/Services/UserCallService.ts

@@ -14,12 +14,14 @@ import ObjectID from "../../Types/ObjectID";
 import Text from "../../Types/Text";
 import Project from "Common/Models/DatabaseModels/Project";
 import Model from "Common/Models/DatabaseModels/UserCall";
+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>> {
@@ -56,6 +58,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -98,6 +101,7 @@ export class Service extends DatabaseService<Model> {
     return { carryForward: null, createBy };
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     _onCreate: OnCreate<Model>,
     createdItem: Model,
@@ -108,6 +112,7 @@ export class Service extends DatabaseService<Model> {
     return createdItem;
   }
 
+  @CaptureSpan()
   public async resendVerificationCode(itemId: ObjectID): Promise<void> {
     const item: Model | null = await this.findOneById({
       id: itemId,

package/Server/Services/UserEmailService.ts

@@ -11,12 +11,14 @@ import BadDataException from "../../Types/Exception/BadDataException";
 import ObjectID from "../../Types/ObjectID";
 import Text from "../../Types/Text";
 import Model from "Common/Models/DatabaseModels/UserEmail";
+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>> {
@@ -53,6 +55,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -66,6 +69,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     _onCreate: OnCreate<Model>,
     createdItem: Model,
@@ -78,6 +82,7 @@ export class Service extends DatabaseService<Model> {
     return createdItem;
   }
 
+  @CaptureSpan()
   public async resendVerificationCode(itemId: ObjectID): Promise<void> {
     const item: Model | null = await this.findOneById({
       id: itemId,

package/Server/Services/UserNotificationRuleService.ts

@@ -48,12 +48,14 @@ import WorkspaceNotificationRuleService from "./WorkspaceNotificationRuleService
 import NotificationRuleEventType from "../../Types/Workspace/NotificationRules/EventType";
 import NotificationRuleWorkspaceChannel from "../../Types/Workspace/NotificationRules/NotificationRuleWorkspaceChannel";
 import logger from "../Utils/Logger";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   public async executeNotificationRuleItem(
     userNotificationRuleId: ObjectID,
     options: {
@@ -585,6 +587,7 @@ export class Service extends DatabaseService<Model> {
     }
   }
 
+  @CaptureSpan()
   public async generateCallTemplateForAlertCreated(
     to: Phone,
     alert: Alert,
@@ -635,6 +638,7 @@ export class Service extends DatabaseService<Model> {
     return callRequest;
   }
 
+  @CaptureSpan()
   public async generateCallTemplateForIncidentCreated(
     to: Phone,
     incident: Incident,
@@ -685,6 +689,7 @@ export class Service extends DatabaseService<Model> {
     return callRequest;
   }
 
+  @CaptureSpan()
   public async generateSmsTemplateForAlertCreated(
     to: Phone,
     alert: Alert,
@@ -714,6 +719,7 @@ export class Service extends DatabaseService<Model> {
     return sms;
   }
 
+  @CaptureSpan()
   public async generateSmsTemplateForIncidentCreated(
     to: Phone,
     incident: Incident,
@@ -743,6 +749,7 @@ export class Service extends DatabaseService<Model> {
     return sms;
   }
 
+  @CaptureSpan()
   public async generateEmailTemplateForAlertCreated(
     to: Email,
     alert: Alert,
@@ -782,6 +789,7 @@ export class Service extends DatabaseService<Model> {
     return emailMessage;
   }
 
+  @CaptureSpan()
   public async generateEmailTemplateForIncidentCreated(
     to: Email,
     incident: Incident,
@@ -826,6 +834,7 @@ export class Service extends DatabaseService<Model> {
     return emailMessage;
   }
 
+  @CaptureSpan()
   public async startUserNotificationRulesExecution(
     userId: ObjectID,
     options: {
@@ -839,6 +848,7 @@ export class Service extends DatabaseService<Model> {
       userBelongsToTeamId?: ObjectID | undefined;
       onCallDutyPolicyExecutionLogTimelineId?: ObjectID | undefined;
       onCallScheduleId?: ObjectID | undefined;
+      overridedByUserId?: ObjectID | undefined;
     },
   ): Promise<void> {
     // add user notification log.
@@ -887,6 +897,10 @@ export class Service extends DatabaseService<Model> {
     userOnCallLog.status = UserNotificationExecutionStatus.Scheduled;
     userOnCallLog.statusMessage = "Scheduled";
 
+    if (options.overridedByUserId) {
+      userOnCallLog.overridedByUserId = options.overridedByUserId;
+    }
+
     await UserOnCallLogService.create({
       data: userOnCallLog,
       props: {
@@ -906,6 +920,7 @@ export class Service extends DatabaseService<Model> {
     });
   }
 
+  @CaptureSpan()
   public async runWorkspaceRulesForOnCallNotification(data: {
     projectId: ObjectID;
     incidentId?: ObjectID | undefined;
@@ -964,6 +979,7 @@ export class Service extends DatabaseService<Model> {
     });
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -984,6 +1000,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   public async addDefaultIncidentNotificationRuleForUser(data: {
     projectId: ObjectID;
     userId: ObjectID;
@@ -1044,6 +1061,7 @@ export class Service extends DatabaseService<Model> {
     }
   }
 
+  @CaptureSpan()
   public async addDefaultAlertNotificationRulesForUser(data: {
     projectId: ObjectID;
     userId: ObjectID;
@@ -1104,6 +1122,7 @@ export class Service extends DatabaseService<Model> {
     }
   }
 
+  @CaptureSpan()
   public async addDefaultNotificationRuleForUser(
     projectId: ObjectID,
     userId: ObjectID,

package/Server/Services/UserNotificationSettingService.ts

@@ -21,12 +21,14 @@ import UserCall from "Common/Models/DatabaseModels/UserCall";
 import UserEmail from "Common/Models/DatabaseModels/UserEmail";
 import UserNotificationSetting from "Common/Models/DatabaseModels/UserNotificationSetting";
 import UserSMS from "Common/Models/DatabaseModels/UserSMS";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<UserNotificationSetting> {
   public constructor() {
     super(UserNotificationSetting);
   }
 
+  @CaptureSpan()
   public async sendUserNotification(data: {
     userId: ObjectID;
     projectId: ObjectID;
@@ -158,6 +160,7 @@ export class Service extends DatabaseService<UserNotificationSetting> {
     }
   }
 
+  @CaptureSpan()
   public async removeDefaultNotificationSettingsForUser(
     userId: ObjectID,
     projectId: ObjectID,
@@ -189,6 +192,7 @@ export class Service extends DatabaseService<UserNotificationSetting> {
     }
   }
 
+  @CaptureSpan()
   public async addDefaultNotificationSettingsForUser(
     userId: ObjectID,
     projectId: ObjectID,
@@ -464,6 +468,7 @@ export class Service extends DatabaseService<UserNotificationSetting> {
     }
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<UserNotificationSetting>,
   ): Promise<OnCreate<UserNotificationSetting>> {

package/Server/Services/UserOnCallLogService.ts

@@ -8,6 +8,7 @@ import { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
 import BadDataException from "../../Types/Exception/BadDataException";
 import NotificationRuleType from "../../Types/NotificationRule/NotificationRuleType";
 import ObjectID from "../../Types/ObjectID";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 import OnCallDutyExecutionLogTimelineStatus from "../../Types/OnCallDutyPolicy/OnCalDutyExecutionLogTimelineStatus";
 import PositiveNumber from "../../Types/PositiveNumber";
 import UserNotificationEventType from "../../Types/UserNotification/UserNotificationEventType";
@@ -27,6 +28,7 @@ export class Service extends DatabaseService<Model> {
     }
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -38,6 +40,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onUpdateSuccess(
     onUpdate: OnUpdate<Model>,
     _updatedItemIds: ObjectID[],
@@ -95,6 +98,7 @@ export class Service extends DatabaseService<Model> {
     return onUpdate;
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     _onCreate: OnCreate<Model>,
     createdItem: Model,

package/Server/Services/UserOnCallLogTimelineService.ts

@@ -8,6 +8,7 @@ import UserService from "./UserService";
 import { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
 import BadDataException from "../../Types/Exception/BadDataException";
 import ObjectID from "../../Types/ObjectID";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 import OnCallDutyExecutionLogTimelineStatus from "../../Types/OnCallDutyPolicy/OnCalDutyExecutionLogTimelineStatus";
 import OnCallDutyPolicyStatus from "../../Types/OnCallDutyPolicy/OnCallDutyPolicyStatus";
 import UserNotificationExecutionStatus from "../../Types/UserNotification/UserNotificationExecutionStatus";
@@ -20,6 +21,7 @@ export class Service extends DatabaseService<Model> {
     super(Model);
   }
 
+  @CaptureSpan()
   protected override async onUpdateSuccess(
     onUpdate: OnUpdate<Model>,
     _updatedItemIds: ObjectID[],

package/Server/Services/UserService.ts

@@ -33,12 +33,14 @@ import UserTwoFactorAuth from "Common/Models/DatabaseModels/UserTwoFactorAuth";
 import UserTwoFactorAuthService from "./UserTwoFactorAuthService";
 import BadDataException from "../../Types/Exception/BadDataException";
 import Name from "../../Types/Name";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   public async getUserMarkdownString(data: {
     userId: ObjectID;
     projectId: ObjectID;
@@ -63,6 +65,7 @@ export class Service extends DatabaseService<Model> {
     return `[${user.name?.toString() || user.email?.toString() || "User"}](${(await this.getUserLinkInDashboard(data.projectId, data.userId)).toString()})`;
   }
 
+  @CaptureSpan()
   public async getUserLinkInDashboard(
     projectId: ObjectID,
     userId: ObjectID,
@@ -74,6 +77,7 @@ export class Service extends DatabaseService<Model> {
     );
   }
 
+  @CaptureSpan()
   protected override async onCreateSuccess(
     _onCreate: OnCreate<Model>,
     createdItem: Model,
@@ -96,6 +100,7 @@ export class Service extends DatabaseService<Model> {
     return Promise.resolve(createdItem);
   }
 
+  @CaptureSpan()
   public async findByEmail(
     email: Email,
     props: DatabaseCommonInteractionProps,
@@ -111,6 +116,7 @@ export class Service extends DatabaseService<Model> {
     });
   }
 
+  @CaptureSpan()
   protected override async onBeforeUpdate(
     updateBy: UpdateBy<Model>,
   ): Promise<OnUpdate<Model>> {
@@ -175,6 +181,7 @@ export class Service extends DatabaseService<Model> {
     return { updateBy, carryForward: carryForward };
   }
 
+  @CaptureSpan()
   protected override async onUpdateSuccess(
     onUpdate: OnUpdate<Model>,
     _updatedItemIds: ObjectID[],
@@ -326,6 +333,7 @@ export class Service extends DatabaseService<Model> {
     return onUpdate;
   }
 
+  @CaptureSpan()
   public async createByEmail(data: {
     email: Email;
     name: Name | undefined;