@oneuptime/common 7.0.3831 → 7.0.3864

package/Server/Services/UserSmsService.ts

@@ -13,12 +13,14 @@ import ObjectID from "../../Types/ObjectID";
 import Text from "../../Types/Text";
 import Project from "Common/Models/DatabaseModels/Project";
 import Model from "Common/Models/DatabaseModels/UserSMS";
+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>> {
@@ -55,6 +57,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onBeforeCreate(
     createBy: CreateBy<Model>,
   ): Promise<OnCreate<Model>> {
@@ -97,6 +100,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/UserTwoFactorAuthService.ts

@@ -8,12 +8,14 @@ import BadDataException from "../../Types/Exception/BadDataException";
 import User from "Common/Models/DatabaseModels/User";
 import DeleteBy from "../Types/Database/DeleteBy";
 import LIMIT_MAX from "../../Types/Database/LimitMax";
+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>> {
@@ -54,6 +56,7 @@ export class Service extends DatabaseService<Model> {
     };
   }
 
+  @CaptureSpan()
   protected override async onBeforeDelete(
     deleteBy: DeleteBy<Model>,
   ): Promise<OnDelete<Model>> {

package/Server/Services/WorkflowService.ts

@@ -8,6 +8,7 @@ import Route from "../../Types/API/Route";
 import URL from "../../Types/API/URL";
 import { JSONObject } from "../../Types/JSON";
 import ObjectID from "../../Types/ObjectID";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 import {
   ComponentType,
   NodeDataProp,
@@ -22,6 +23,7 @@ export class Service extends DatabaseService<Model> {
     super(Model);
   }
 
+  @CaptureSpan()
   protected override async onUpdateSuccess(
     onUpdate: OnUpdate<Model>,
     _updatedItemIds: ObjectID[],

package/Server/Services/WorkspaceNotificationRuleService.ts

@@ -14,8 +14,6 @@ import Alert from "../../Models/DatabaseModels/Alert";
 import AlertService from "./AlertService";
 import ScheduledMaintenanceService from "./ScheduledMaintenanceService";
 import ScheduledMaintenance from "../../Models/DatabaseModels/ScheduledMaintenance";
-import MonitorStatusTimeline from "Common/Models/DatabaseModels/MonitorStatusTimeline";
-import MonitorStatusTimelineService from "./MonitorStatusTimelineService";
 import { WorkspaceNotificationRuleUtil } from "../../Types/Workspace/NotificationRules/NotificationRuleUtil";
 import TeamMemberService from "./TeamMemberService";
 import User from "../../Models/DatabaseModels/User";
@@ -25,7 +23,7 @@ import { WorkspaceChannel } from "../Utils/Workspace/WorkspaceBase";
 import WorkspaceUtil from "../Utils/Workspace/Workspace";
 import WorkspaceUserAuthToken from "../../Models/DatabaseModels/WorkspaceUserAuthToken";
 import WorkspaceUserAuthTokenService from "./WorkspaceUserAuthTokenService";
-import {
+import WorkspaceMessagePayload, {
   WorkspaceMessageBlock,
   WorkspacePayloadMarkdown,
 } from "../../Types/Workspace/WorkspaceMessagePayload";
@@ -38,6 +36,8 @@ import logger from "../Utils/Logger";
 import NotificationRuleWorkspaceChannel from "../../Types/Workspace/NotificationRules/NotificationRuleWorkspaceChannel";
 import WorkspaceNotificationRule from "Common/Models/DatabaseModels/WorkspaceNotificationRule";
 import UserService from "./UserService";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import Monitor from "../../Models/DatabaseModels/Monitor";
 
 export interface MessageBlocksByWorkspaceType {
   workspaceType: WorkspaceType;
@@ -48,7 +48,7 @@ export interface NotificationFor {
   incidentId?: ObjectID | undefined;
   alertId?: ObjectID | undefined;
   scheduledMaintenanceId?: ObjectID | undefined;
-  monitorStatusTimelineId?: ObjectID | undefined;
+  monitorId?: ObjectID | undefined;
 }
 
 export class Service extends DatabaseService<WorkspaceNotificationRule> {
@@ -56,6 +56,114 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     super(WorkspaceNotificationRule);
   }
 
+  @CaptureSpan()
+  public async sendWorkspaceMarkdownNotification(data: {
+    projectId: ObjectID;
+    notificationFor: NotificationFor;
+    feedInfoInMarkdown: string;
+    workspaceNotification: {
+      notifyUserId?: ObjectID | undefined; // this is oneuptime user id.
+      sendWorkspaceNotification: boolean;
+      appendMessageBlocks?: Array<MessageBlocksByWorkspaceType> | undefined;
+    };
+  }): Promise<void> {
+    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 this.getExistingChannelNamesBasedOnEventType({
+          projectId: data.projectId,
+          notificationRuleEventType: NotificationRuleEventType.Monitor,
+          workspaceType: messageBlocksByWorkspaceType.workspaceType,
+        });
+
+      let monitorChannels: Array<WorkspaceChannel> = [];
+
+      if (data.notificationFor.monitorId) {
+        monitorChannels = await MonitorService.getWorkspaceChannelForMonitor({
+          monitorId: data.notificationFor.monitorId,
+          workspaceType: messageBlocksByWorkspaceType.workspaceType,
+        });
+      }
+
+      // incidents
+      if (data.notificationFor.incidentId) {
+        monitorChannels = await IncidentService.getWorkspaceChannelForIncident({
+          incidentId: data.notificationFor.incidentId,
+          workspaceType: messageBlocksByWorkspaceType.workspaceType,
+        });
+      }
+
+      // alerts
+      if (data.notificationFor.alertId) {
+        monitorChannels = await AlertService.getWorkspaceChannelForAlert({
+          alertId: data.notificationFor.alertId,
+          workspaceType: messageBlocksByWorkspaceType.workspaceType,
+        });
+      }
+
+      // scheduled maintenance
+
+      if (data.notificationFor.scheduledMaintenanceId) {
+        monitorChannels =
+          await ScheduledMaintenanceService.getWorkspaceChannelForScheduledMaintenance(
+            {
+              scheduledMaintenanceId:
+                data.notificationFor.scheduledMaintenanceId,
+              workspaceType: messageBlocksByWorkspaceType.workspaceType,
+            },
+          );
+      }
+
+      const workspaceMessagePayload: WorkspaceMessagePayload = {
+        _type: "WorkspaceMessagePayload",
+        workspaceType: messageBlocksByWorkspaceType.workspaceType,
+        messageBlocks: messageBlocksByWorkspaceType.messageBlocks,
+        channelNames: existingChannels,
+        channelIds:
+          monitorChannels.map((channel: WorkspaceChannel) => {
+            return channel.id;
+          }) || [],
+      };
+
+      workspaceNotificationPaylaods.push(workspaceMessagePayload);
+    }
+
+    await WorkspaceUtil.postMessageToAllWorkspaceChannelsAsBot({
+      projectId: data.projectId,
+      messagePayloadsByWorkspace: workspaceNotificationPaylaods,
+    });
+  }
+
+  @CaptureSpan()
   public static getAllWorkspaceTypes(): Array<WorkspaceType> {
     return [WorkspaceType.Slack, WorkspaceType.MicrosoftTeams];
   }
@@ -85,6 +193,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     throw new BadDataException("Workspace type not supported");
   }
 
+  @CaptureSpan()
   public async getExistingChannelNamesBasedOnEventType(data: {
     projectId: ObjectID;
     workspaceType: WorkspaceType;
@@ -118,6 +227,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     return existingChannelNames;
   }
 
+  @CaptureSpan()
   public async createChannelsAndInviteUsersToChannelsBasedOnRules(data: {
     projectId: ObjectID;
     notificationRuleEventType: NotificationRuleEventType;
@@ -232,6 +342,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     };
   }
 
+  @CaptureSpan()
   public async getNotificationRulesWhereOnCallIsTrue(data: {
     projectId: ObjectID;
     notificationFor: NotificationFor;
@@ -263,6 +374,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     return result;
   }
 
+  @CaptureSpan()
   public async getNotificationRulesWhereInviteOwnersIsTrue(data: {
     projectId: ObjectID;
     notificationFor: NotificationFor;
@@ -294,6 +406,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     return result;
   }
 
+  @CaptureSpan()
   public async inviteUsersAndTeamsToChannelsBasedOnRules(data: {
     projectId: ObjectID;
     projectAuth: WorkspaceProjectAuthToken;
@@ -402,6 +515,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     logger.debug("Users invited to channels successfully");
   }
 
+  @CaptureSpan()
   public async inviteUsersBasedOnRulesAndWorkspaceChannels(data: {
     workspaceChannels: Array<NotificationRuleWorkspaceChannel>;
     projectId: ObjectID;
@@ -555,6 +669,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     }
   }
 
+  @CaptureSpan()
   public async inviteTeamsBasedOnRulesAndWorkspaceChannels(data: {
     workspaceChannels: Array<NotificationRuleWorkspaceChannel>;
     projectId: ObjectID;
@@ -584,6 +699,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     });
   }
 
+  @CaptureSpan()
   public async getWorkspaceUserIdFromOneUptimeUserId(data: {
     projectId: ObjectID;
     workspaceType: WorkspaceType;
@@ -618,6 +734,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     return userAuth.workspaceUserId?.toString() || null;
   }
 
+  @CaptureSpan()
   public async createChannelsBasedOnRules(data: {
     projectOrUserAuthTokenForWorkspace: string;
     workspaceType: WorkspaceType;
@@ -690,6 +807,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     return createdWorkspaceChannels;
   }
 
+  @CaptureSpan()
   public async getUsersIdsToInviteToChannel(data: {
     notificationRules: Array<WorkspaceNotificationRule>;
   }): Promise<
@@ -1159,46 +1277,41 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
       };
     }
 
-    if (data.notificationFor.monitorStatusTimelineId) {
+    if (data.notificationFor.monitorId) {
       logger.debug("Fetching monitor status timeline details for ID:");
-      logger.debug(data.notificationFor.monitorStatusTimelineId);
+      logger.debug(data.notificationFor.monitorId);
 
-      const monitorStatusTimeline: MonitorStatusTimeline | null =
-        await MonitorStatusTimelineService.findOneById({
-          id: data.notificationFor.monitorStatusTimelineId,
-          select: {
-            monitor: {
-              name: true,
-              labels: true,
-              monitorType: true,
-            },
-            monitorStatus: true,
-          },
-          props: {
-            isRoot: true,
-          },
-        });
+      const monitor: Monitor | null = await MonitorService.findOneById({
+        id: data.notificationFor.monitorId,
+        select: {
+          name: true,
+          labels: true,
+          monitorType: true,
+          currentMonitorStatusId: true,
+        },
+        props: {
+          isRoot: true,
+        },
+      });
 
-      if (!monitorStatusTimeline) {
-        logger.debug("Monitor status timeline not found for ID:");
-        logger.debug(data.notificationFor.monitorStatusTimelineId);
-        throw new BadDataException("Monitor Status Timeline ID not found");
+      if (!monitor) {
+        logger.debug("Monitor not found for ID:");
+        logger.debug(data.notificationFor.monitorId);
+        throw new BadDataException("Monitor ID not found");
       }
 
-      const monitorLabels: Array<Label> =
-        monitorStatusTimeline.monitor?.labels || [];
+      const monitorLabels: Array<Label> = monitor?.labels || [];
 
       return {
-        [NotificationRuleConditionCheckOn.MonitorName]:
-          monitorStatusTimeline.monitor?.name || "",
+        [NotificationRuleConditionCheckOn.MonitorName]: monitor?.name || "",
         [NotificationRuleConditionCheckOn.IncidentTitle]: undefined,
         [NotificationRuleConditionCheckOn.IncidentDescription]: undefined,
         [NotificationRuleConditionCheckOn.IncidentSeverity]: undefined,
         [NotificationRuleConditionCheckOn.IncidentState]: undefined,
         [NotificationRuleConditionCheckOn.MonitorType]:
-          monitorStatusTimeline.monitor?.monitorType || undefined,
+          monitor?.monitorType || undefined,
         [NotificationRuleConditionCheckOn.MonitorStatus]:
-          monitorStatusTimeline.monitorStatus?._id?.toString() || "",
+          monitor?.currentMonitorStatusId?.toString() || "",
         [NotificationRuleConditionCheckOn.AlertTitle]: undefined,
         [NotificationRuleConditionCheckOn.AlertDescription]: undefined,
         [NotificationRuleConditionCheckOn.AlertSeverity]: undefined,
@@ -1216,7 +1329,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
         [NotificationRuleConditionCheckOn.ScheduledMaintenanceLabels]:
           undefined,
         [NotificationRuleConditionCheckOn.Monitors]: [
-          monitorStatusTimeline.monitor?._id?.toString() || "",
+          monitor?._id?.toString() || "",
         ],
       };
     }
@@ -1224,6 +1337,7 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     throw new BadDataException("NotificationFor is not supported");
   }
 
+  @CaptureSpan()
   public async getMatchingNotificationRules(data: {
     projectId: ObjectID;
     workspaceType: WorkspaceType;

package/Server/Services/WorkspaceProjectAuthTokenService.ts

@@ -5,12 +5,14 @@ import Model, {
   SlackMiscData,
 } from "Common/Models/DatabaseModels/WorkspaceProjectAuthToken";
 import { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   public async getProjectAuth(data: {
     projectId: ObjectID;
     workspaceType: WorkspaceType;
@@ -32,6 +34,7 @@ export class Service extends DatabaseService<Model> {
     });
   }
 
+  @CaptureSpan()
   public async getProjectAuths(data: {
     projectId: ObjectID;
   }): Promise<Array<Model>> {
@@ -53,6 +56,7 @@ export class Service extends DatabaseService<Model> {
     });
   }
 
+  @CaptureSpan()
   public async doesExist(data: {
     projectId: ObjectID;
     workspaceType: WorkspaceType;
@@ -60,6 +64,7 @@ export class Service extends DatabaseService<Model> {
     return Boolean(await this.getProjectAuth(data));
   }
 
+  @CaptureSpan()
   public async refreshAuthToken(data: {
     projectId: ObjectID;
     workspaceType: WorkspaceType;

package/Server/Services/WorkspaceSettingService.ts

@@ -4,12 +4,14 @@ import DatabaseService from "./DatabaseService";
 import Model, {
   SlackSettings,
 } from "Common/Models/DatabaseModels/WorkspaceSetting";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   public async doesExist(data: {
     projectId: ObjectID;
     workspaceType: WorkspaceType;
@@ -31,6 +33,7 @@ export class Service extends DatabaseService<Model> {
     );
   }
 
+  @CaptureSpan()
   public async refreshSetting(data: {
     projectId: ObjectID;
     workspaceType: WorkspaceType;

package/Server/Services/WorkspaceUserAuthTokenService.ts

@@ -4,12 +4,14 @@ import DatabaseService from "./DatabaseService";
 import Model, {
   SlackMiscData,
 } from "Common/Models/DatabaseModels/WorkspaceUserAuthToken";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  @CaptureSpan()
   public async getUserAuth(data: {
     projectId: ObjectID;
     userId: ObjectID;
@@ -33,6 +35,7 @@ export class Service extends DatabaseService<Model> {
     });
   }
 
+  @CaptureSpan()
   public async doesExist(data: {
     projectId: ObjectID;
     userId: ObjectID;
@@ -56,6 +59,7 @@ export class Service extends DatabaseService<Model> {
     );
   }
 
+  @CaptureSpan()
   public async refreshAuthToken(data: {
     projectId: ObjectID;
     userId: ObjectID;

package/Server/Types/AnalyticsDatabase/AggregateBy.ts

@@ -3,6 +3,7 @@ import CommonAggregateBy from "Common/Types/BaseDatabase/AggregateBy";
 import AnalyticsBaseModel from "Common/Models/AnalyticsModels/AnalyticsBaseModel/AnalyticsBaseModel";
 import DatabaseCommonInteractionProps from "Common/Types/BaseDatabase/DatabaseCommonInteractionProps";
 import OneUptimeDate from "../../../Types/Date";
+import CaptureSpan from "../../Utils/Telemetry/CaptureSpan";
 
 export default interface AggregateBy<TBaseModel extends AnalyticsBaseModel>
   extends CommonAggregateBy<TBaseModel> {
@@ -10,6 +11,7 @@ export default interface AggregateBy<TBaseModel extends AnalyticsBaseModel>
 }
 
 export class AggregateUtil {
+  @CaptureSpan()
   public static getAggregationInterval(data: {
     startDate: Date;
     endDate: Date;

package/Server/Types/AnalyticsDatabase/ModelPermission.ts

@@ -26,6 +26,7 @@ import Permission, {
   UserPermission,
 } from "Common/Types/Permission";
 import UserType from "Common/Types/UserType";
+import CaptureSpan from "../../Utils/Telemetry/CaptureSpan";
 
 export interface CheckReadPermissionType<TBaseModel extends BaseModel> {
   query: Query<TBaseModel>;
@@ -33,6 +34,7 @@ export interface CheckReadPermissionType<TBaseModel extends BaseModel> {
 }
 
 export default class ModelPermission {
+  @CaptureSpan()
   public static async checkDeletePermission<TBaseModel extends BaseModel>(
     modelType: { new (): TBaseModel },
     query: Query<TBaseModel>,
@@ -54,6 +56,7 @@ export default class ModelPermission {
     return query;
   }
 
+  @CaptureSpan()
   public static async checkUpdatePermissions<TBaseModel extends BaseModel>(
     modelType: { new (): TBaseModel },
     query: Query<TBaseModel>,
@@ -85,6 +88,7 @@ export default class ModelPermission {
     return query;
   }
 
+  @CaptureSpan()
   public static checkCreatePermissions<TBaseModel extends BaseModel>(
     modelType: { new (): TBaseModel },
     data: TBaseModel,
@@ -220,6 +224,7 @@ export default class ModelPermission {
     }
   }
 
+  @CaptureSpan()
   public static async checkReadPermission<TBaseModel extends BaseModel>(
     modelType: { new (): TBaseModel },
     query: Query<TBaseModel>,
@@ -581,6 +586,7 @@ export default class ModelPermission {
     return isPublicAllowed;
   }
 
+  @CaptureSpan()
   public static checkIfUserIsLoggedIn(
     modelType: AnalyticsBaseModelType,
     props: DatabaseCommonInteractionProps,

package/Server/Types/AnalyticsDatabase/QueryHelper.ts

@@ -4,20 +4,25 @@ import Includes from "Common/Types/BaseDatabase/Includes";
 import LessThan from "Common/Types/BaseDatabase/LessThan";
 import ObjectID from "Common/Types/ObjectID";
 import { CompareType } from "../../../Types/Database/CompareBase";
+import CaptureSpan from "../../Utils/Telemetry/CaptureSpan";
 
 export default class QueryHelper {
+  @CaptureSpan()
   public static lessThan<T extends CompareType>(value: T): LessThan<T> {
     return new LessThan(value);
   }
 
+  @CaptureSpan()
   public static any(values: Array<string> | Array<ObjectID>): Includes {
     return new Includes(values);
   }
 
+  @CaptureSpan()
   public static greaterThan<T extends CompareType>(value: T): GreaterThan<T> {
     return new GreaterThan<T>(value);
   }
 
+  @CaptureSpan()
   public static inBetween<T extends CompareType>(
     startValue: T,
     endValue: T,

package/Server/Types/Billing/MeteredPlan/ActiveMonitoringMeteredPlan.ts

@@ -8,12 +8,15 @@ import MonitorType from "Common/Types/Monitor/MonitorType";
 import ObjectID from "Common/Types/ObjectID";
 import PositiveNumber from "Common/Types/PositiveNumber";
 import Project from "Common/Models/DatabaseModels/Project";
+import CaptureSpan from "../../../Utils/Telemetry/CaptureSpan";
 
 export default class ActiveMonitoringMeteredPlan extends ServerMeteredPlan {
+  @CaptureSpan()
   public override getProductType(): ProductType {
     return ProductType.ActiveMonitoring;
   }
 
+  @CaptureSpan()
   public override async reportQuantityToBillingProvider(
     projectId: ObjectID,
     options?: {

package/Server/Types/Billing/MeteredPlan/AllMeteredPlans.ts

@@ -1,3 +1,4 @@
+import CaptureSpan from "../../../Utils/Telemetry/CaptureSpan";
 import ActiveMonitoringMeteredPlanType from "./ActiveMonitoringMeteredPlan";
 import ServerMeteredPlan from "./ServerMeteredPlan";
 import TelemetryMeteredPlanType from "./TelemetryMeteredPlan";
@@ -33,6 +34,7 @@ const AllMeteredPlans: Array<ServerMeteredPlan> = [
 ];
 
 export class MeteredPlanUtil {
+  @CaptureSpan()
   public static getMeteredPlanByProductType(
     productType: ProductType,
   ): ServerMeteredPlan {

package/Server/Types/Billing/MeteredPlan/ServerMeteredPlan.ts

@@ -3,6 +3,7 @@ import MeteredPlan from "Common/Types/Billing/MeteredPlan";
 import NotImplementedException from "Common/Types/Exception/NotImplementedException";
 import ProductType from "Common/Types/MeteredPlan/ProductType";
 import ObjectID from "Common/Types/ObjectID";
+import CaptureSpan from "../../../Utils/Telemetry/CaptureSpan";
 
 export default class ServerMeteredPlan {
   public getProductType(): ProductType {
@@ -16,6 +17,7 @@ export default class ServerMeteredPlan {
     return meteredPlan.getPricePerUnit() * quantity;
   }
 
+  @CaptureSpan()
   public async reportQuantityToBillingProvider(
     _projectId: ObjectID,
     _options: {

package/Server/Types/Billing/MeteredPlan/TelemetryMeteredPlan.ts

@@ -7,6 +7,7 @@ import ProductType from "Common/Types/MeteredPlan/ProductType";
 import ObjectID from "Common/Types/ObjectID";
 import Project from "Common/Models/DatabaseModels/Project";
 import TelemetryUsageBilling from "Common/Models/DatabaseModels/TelemetryUsageBilling";
+import CaptureSpan from "../../../Utils/Telemetry/CaptureSpan";
 
 export default class TelemetryMeteredPlan extends ServerMeteredPlan {
   private _productType!: ProductType;
@@ -41,10 +42,12 @@ export default class TelemetryMeteredPlan extends ServerMeteredPlan {
     return data.dataIngestedInGB * data.retentionInDays * this.unitCostInUSD;
   }
 
+  @CaptureSpan()
   public override getProductType(): ProductType {
     return this.productType;
   }
 
+  @CaptureSpan()
   public override async reportQuantityToBillingProvider(
     projectId: ObjectID,
     options?: {

package/Server/Types/Database/Permissions/AccessControlPermission.ts

@@ -19,8 +19,10 @@ import Permission, {
   PermissionHelper,
   UserPermission,
 } from "Common/Types/Permission";
+import CaptureSpan from "../../../Utils/Telemetry/CaptureSpan";
 
 export default class AccessControlPermission {
+  @CaptureSpan()
   public static async checkAccessControlBlockPermissionByModel<
     TBaseModel extends BaseModel,
   >(data: {
@@ -123,6 +125,7 @@ export default class AccessControlPermission {
     }
   }
 
+  @CaptureSpan()
   public static async checkAccessControlPermissionByModel<
     TBaseModel extends BaseModel,
   >(data: {
@@ -213,6 +216,7 @@ export default class AccessControlPermission {
     }
   }
 
+  @CaptureSpan()
   public static async addAccessControlIdsToQuery<TBaseModel extends BaseModel>(
     modelType: { new (): TBaseModel },
     query: Query<TBaseModel>,
@@ -236,6 +240,7 @@ export default class AccessControlPermission {
     return query;
   }
 
+  @CaptureSpan()
   public static getAccessControlIdsForModel(
     modelType: DatabaseBaseModelType,
     props: DatabaseCommonInteractionProps,
@@ -260,6 +265,7 @@ export default class AccessControlPermission {
     return distinctLabelIds;
   }
 
+  @CaptureSpan()
   public static getAccessControlIdsForQuery<TBaseModel extends BaseModel>(
     modelType: DatabaseBaseModelType,
     query: Query<TBaseModel>,