@utaba/deep-memory-storage-sqlserver 0.2.0 → 0.3.0

package/dist/index.d.cts

@@ -1,6 +1,7 @@
+import * as _utaba_deep_memory_types from '@utaba/deep-memory/types';
+import { StorageRepositoryConfig, StoredRepository, RepositoryFilter, PaginatedResult, StoredRepositorySummary, RepositoryUpdate, RepositoryStats, MemoryVocabulary, PaginationOptions, VocabularyChangeRecord, StoredEntity, StoredEntityUpdate, StorageFindQuery, StoredRelationship, RelationshipQueryOptions, StorageExploreOptions, StorageNeighbourhood, StoragePathOptions, StoragePathResult, StorageTimelineOptions, StorageTimelineResult, ExportChunk, ImportChunk, BulkImportResult, SearchOptions, SearchHit } from '@utaba/deep-memory/types';
 import sql from 'mssql';
 import { StorageProvider, EnsureSchemaResult, SearchProvider, SearchableEntity } from '@utaba/deep-memory/providers';
-import { StorageRepositoryConfig, StoredRepository, RepositoryFilter, PaginatedResult, StoredRepositorySummary, RepositoryUpdate, RepositoryStats, MemoryVocabulary, PaginationOptions, VocabularyChangeRecord, StoredEntity, StoredEntityUpdate, StorageFindQuery, StoredRelationship, RelationshipQueryOptions, StorageExploreOptions, StorageNeighbourhood, StoragePathOptions, StoragePathResult, StorageTimelineOptions, StorageTimelineResult, ExportChunk, ImportChunk, BulkImportResult, SearchOptions, SearchHit } from '@utaba/deep-memory/types';
 
 /** Configuration for SqlServerStorageProvider */
 interface SqlServerStorageProviderConfig {
@@ -32,7 +33,11 @@ declare class SqlServerStorageProvider implements StorageProvider {
     getRepository(repositoryId: string): Promise<StoredRepository | null>;
     listRepositories(filter?: RepositoryFilter): Promise<PaginatedResult<StoredRepositorySummary>>;
     updateRepository(repositoryId: string, updates: RepositoryUpdate): Promise<StoredRepository>;
-    deleteRepository(repositoryId: string): Promise<void>;
+    deleteRepository(repositoryId: string, _onProgress?: _utaba_deep_memory_types.DeleteProgressCallback): Promise<void>;
+    deleteAllContents(repositoryId: string, _onProgress?: _utaba_deep_memory_types.DeleteProgressCallback): Promise<{
+        deletedEntities: number;
+        deletedRelationships: number;
+    }>;
     getRepositoryStats(repositoryId: string): Promise<RepositoryStats>;
     getVocabulary(repositoryId: string): Promise<MemoryVocabulary>;
     saveVocabulary(repositoryId: string, vocabulary: MemoryVocabulary): Promise<void>;
@@ -64,7 +69,7 @@ declare class SqlServerStorageProvider implements StorageProvider {
     findPaths(repositoryId: string, sourceId: string, targetId: string, options: StoragePathOptions): Promise<StoragePathResult>;
     getTimeline(repositoryId: string, entityId: string, options: StorageTimelineOptions): Promise<StorageTimelineResult>;
     exportAll(repositoryId: string): AsyncIterable<ExportChunk>;
-    importBulk(repositoryId: string, data: ImportChunk[]): Promise<BulkImportResult>;
+    importBulk(repositoryId: string, data: ImportChunk[], _options?: _utaba_deep_memory_types.BulkImportOptions): Promise<BulkImportResult>;
     /** Creates a TVP table from an array of string IDs (eliminates plan cache bloat from IN clauses). */
     private createIdListTvp;
     /** Returns a StoredEntity without the embedding column (lighter I/O). */

package/dist/index.d.ts

@@ -1,6 +1,7 @@
+import * as _utaba_deep_memory_types from '@utaba/deep-memory/types';
+import { StorageRepositoryConfig, StoredRepository, RepositoryFilter, PaginatedResult, StoredRepositorySummary, RepositoryUpdate, RepositoryStats, MemoryVocabulary, PaginationOptions, VocabularyChangeRecord, StoredEntity, StoredEntityUpdate, StorageFindQuery, StoredRelationship, RelationshipQueryOptions, StorageExploreOptions, StorageNeighbourhood, StoragePathOptions, StoragePathResult, StorageTimelineOptions, StorageTimelineResult, ExportChunk, ImportChunk, BulkImportResult, SearchOptions, SearchHit } from '@utaba/deep-memory/types';
 import sql from 'mssql';
 import { StorageProvider, EnsureSchemaResult, SearchProvider, SearchableEntity } from '@utaba/deep-memory/providers';
-import { StorageRepositoryConfig, StoredRepository, RepositoryFilter, PaginatedResult, StoredRepositorySummary, RepositoryUpdate, RepositoryStats, MemoryVocabulary, PaginationOptions, VocabularyChangeRecord, StoredEntity, StoredEntityUpdate, StorageFindQuery, StoredRelationship, RelationshipQueryOptions, StorageExploreOptions, StorageNeighbourhood, StoragePathOptions, StoragePathResult, StorageTimelineOptions, StorageTimelineResult, ExportChunk, ImportChunk, BulkImportResult, SearchOptions, SearchHit } from '@utaba/deep-memory/types';
 
 /** Configuration for SqlServerStorageProvider */
 interface SqlServerStorageProviderConfig {
@@ -32,7 +33,11 @@ declare class SqlServerStorageProvider implements StorageProvider {
     getRepository(repositoryId: string): Promise<StoredRepository | null>;
     listRepositories(filter?: RepositoryFilter): Promise<PaginatedResult<StoredRepositorySummary>>;
     updateRepository(repositoryId: string, updates: RepositoryUpdate): Promise<StoredRepository>;
-    deleteRepository(repositoryId: string): Promise<void>;
+    deleteRepository(repositoryId: string, _onProgress?: _utaba_deep_memory_types.DeleteProgressCallback): Promise<void>;
+    deleteAllContents(repositoryId: string, _onProgress?: _utaba_deep_memory_types.DeleteProgressCallback): Promise<{
+        deletedEntities: number;
+        deletedRelationships: number;
+    }>;
     getRepositoryStats(repositoryId: string): Promise<RepositoryStats>;
     getVocabulary(repositoryId: string): Promise<MemoryVocabulary>;
     saveVocabulary(repositoryId: string, vocabulary: MemoryVocabulary): Promise<void>;
@@ -64,7 +69,7 @@ declare class SqlServerStorageProvider implements StorageProvider {
     findPaths(repositoryId: string, sourceId: string, targetId: string, options: StoragePathOptions): Promise<StoragePathResult>;
     getTimeline(repositoryId: string, entityId: string, options: StorageTimelineOptions): Promise<StorageTimelineResult>;
     exportAll(repositoryId: string): AsyncIterable<ExportChunk>;
-    importBulk(repositoryId: string, data: ImportChunk[]): Promise<BulkImportResult>;
+    importBulk(repositoryId: string, data: ImportChunk[], _options?: _utaba_deep_memory_types.BulkImportOptions): Promise<BulkImportResult>;
     /** Creates a TVP table from an array of string IDs (eliminates plan cache bloat from IN clauses). */
     private createIdListTvp;
     /** Returns a StoredEntity without the embedding column (lighter I/O). */

package/dist/index.js

@@ -667,13 +667,23 @@ var SqlServerStorageProvider = class {
     if (!updated) throw new RepositoryNotFoundError(repositoryId);
     return updated;
   }
-  async deleteRepository(repositoryId) {
+  async deleteRepository(repositoryId, _onProgress) {
     const pool = this.getPool();
     const result = await pool.request().input("id", sql.UniqueIdentifier, repositoryId).query(`DELETE FROM ${this.t("dm_repositories")} WHERE [repository_id] = @id`);
     if (result.rowsAffected[0] === 0) {
       throw new RepositoryNotFoundError(repositoryId);
     }
   }
+  async deleteAllContents(repositoryId, _onProgress) {
+    await this.assertRepository(repositoryId);
+    const pool = this.getPool();
+    const relResult = await pool.request().input("id", sql.UniqueIdentifier, repositoryId).query(`DELETE FROM ${this.t("dm_relationships")} WHERE [repository_id] = @id`);
+    const entityResult = await pool.request().input("id", sql.UniqueIdentifier, repositoryId).query(`DELETE FROM ${this.t("dm_entities")} WHERE [repository_id] = @id`);
+    return {
+      deletedEntities: entityResult.rowsAffected[0] ?? 0,
+      deletedRelationships: relResult.rowsAffected[0] ?? 0
+    };
+  }
   async getRepositoryStats(repositoryId) {
     await this.assertRepository(repositoryId);
     const pool = this.getPool();
@@ -1363,7 +1373,7 @@ var SqlServerStorageProvider = class {
       }
     }
   }
-  async importBulk(repositoryId, data) {
+  async importBulk(repositoryId, data, _options) {
     await this.assertRepository(repositoryId);
     const pool = this.getPool();
     let entitiesImported = 0;