From 65109fca1b4889fa538c85fbe0df28cca7805ae4 Mon Sep 17 00:00:00 2001
From: Dotan Simha <dotansimha@gmail.com>
Date: Tue, 20 Aug 2024 18:32:50 +0300
Subject: [PATCH] added more traces

---
 .../api/src/modules/schema/providers/registry-checks.ts        | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/packages/services/api/src/modules/schema/providers/registry-checks.ts b/packages/services/api/src/modules/schema/providers/registry-checks.ts
index 122ed6f8341..1ff495f1008 100644
--- a/packages/services/api/src/modules/schema/providers/registry-checks.ts
+++ b/packages/services/api/src/modules/schema/providers/registry-checks.ts
@@ -5,6 +5,7 @@ import hashObject from 'object-hash';
 import { CriticalityLevel } from '@graphql-inspector/core';
 import type { CheckPolicyResponse } from '@hive/policy';
 import type { CompositionFailureError, ContractsInputType } from '@hive/schema';
+import { traceFn } from '@hive/service-common';
 import {
   HiveSchemaChangeModel,
   type RegistryServiceUrlChangeSerializableChange,
@@ -346,6 +347,7 @@ export class RegistryChecks {
     return existingSchemaResult.sdl ?? null;
   }
 
+  @traceFn('RegistryChecks.policyCheck')
   async policyCheck({
     selector,
     modifiedSdl,
@@ -397,6 +399,7 @@ export class RegistryChecks {
    * Diff incoming and existing SDL and generate a list of changes.
    * Uses usage stats to determine whether a change is safe or not (if available).
    */
+  @traceFn('RegistryChecks.diff')
   async diff(args: {
     /** The existing SDL */
     existingSdl: string | null;