@utaba/deep-memory-storage-sqlserver 0.8.1 → 0.9.0

package/dist/index.cjs

@@ -319,6 +319,42 @@ END;
 }
 
 // src/SqlServerStorageProvider.ts
+var PROVIDER_NAME = "sqlserver";
+var TRACKED_METHODS = {
+  ensureSchema: () => void 0,
+  createRepository: (args) => {
+    const cfg = args[0];
+    return cfg?.repositoryId;
+  },
+  getRepository: (args) => args[0],
+  listRepositories: () => void 0,
+  updateRepository: (args) => args[0],
+  deleteRepository: (args) => args[0],
+  deleteAllContents: (args) => args[0],
+  getRepositoryStats: (args) => args[0],
+  getVocabulary: (args) => args[0],
+  saveVocabulary: (args) => args[0],
+  getVocabularyChangeLog: (args) => args[0],
+  createEntity: (args) => args[0],
+  getEntity: (args) => args[0],
+  getEntityBySlug: (args) => args[0],
+  getEntities: (args) => args[0],
+  updateEntity: (args) => args[0],
+  deleteEntity: (args) => args[0],
+  deleteEntities: (args) => args[0],
+  deleteEntitiesByType: (args) => args[0],
+  findEntities: (args) => args[0],
+  createRelationship: (args) => args[0],
+  getRelationship: (args) => args[0],
+  getEntityRelationships: (args) => args[0],
+  deleteRelationship: (args) => args[0],
+  deleteRelationships: (args) => args[0],
+  deleteRelationshipsByType: (args) => args[0],
+  exploreNeighborhood: (args) => args[0],
+  findPaths: (args) => args[0],
+  getTimeline: (args) => args[0],
+  importBulk: (args) => args[0]
+};
 var ENTITY_COLS_LIGHT = [
   "entity_id",
   "slug",
@@ -430,6 +466,53 @@ var SqlServerStorageProvider = class {
     this.config = config;
     this.schema = config.schema ?? "dbo";
     this.ownsPool = !(config.connection instanceof import_mssql.default.ConnectionPool);
+    const safeSink = (0, import_deep_memory.createSafeSink)(config.reportUsage);
+    if (safeSink) {
+      return new Proxy(this, {
+        get(target, prop, receiver) {
+          const value = Reflect.get(target, prop, receiver);
+          if (typeof prop !== "string" || typeof value !== "function") return value;
+          const extractRepoId = TRACKED_METHODS[prop];
+          if (!extractRepoId) return value;
+          const method = value;
+          return (...args) => {
+            const start = Date.now();
+            const repositoryId = extractRepoId(args);
+            const emit = () => {
+              safeSink({
+                provider: PROVIDER_NAME,
+                operation: prop,
+                unit: "ms",
+                value: Date.now() - start,
+                ...repositoryId ? { repositoryId } : {},
+                timestamp: /* @__PURE__ */ new Date()
+              });
+            };
+            let result;
+            try {
+              result = method.apply(target, args);
+            } catch (err) {
+              emit();
+              throw err;
+            }
+            if (result && typeof result.then === "function") {
+              return result.then(
+                (v) => {
+                  emit();
+                  return v;
+                },
+                (err) => {
+                  emit();
+                  throw err;
+                }
+              );
+            }
+            emit();
+            return result;
+          };
+        }
+      });
+    }
   }
   t(table) {
     return `[${this.schema}].[${table}]`;