feat: opened efcorestore to allow overriding methods (#577)

Closes #574
Finbuckle · Aug 18, 2022 · 7dac251 · 7dac251
1 parent df55b24
commit 7dac251
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/Finbuckle.MultiTenant.EntityFrameworkCore/Stores/EFCoreStore/EFCoreStore.cs b/src/Finbuckle.MultiTenant.EntityFrameworkCore/Stores/EFCoreStore/EFCoreStore.cs
@@ -20,33 +20,33 @@ public EFCoreStore(TEFCoreStoreDbContext dbContext)
             this.dbContext = dbContext ?? throw new ArgumentNullException(nameof(dbContext));
         }
 
-        public async Task<TTenantInfo?> TryGetAsync(string id)
+        public virtual async Task<TTenantInfo?> TryGetAsync(string id)
         {
             return await dbContext.TenantInfo
                             .Where(ti => ti.Id == id)
                             .SingleOrDefaultAsync();
         }
 
-        public async Task<IEnumerable<TTenantInfo>> GetAllAsync()
+        public virtual async Task<IEnumerable<TTenantInfo>> GetAllAsync()
         {
             return await dbContext.TenantInfo.ToListAsync();
         }
 
-        public async Task<TTenantInfo?> TryGetByIdentifierAsync(string identifier)
+        public virtual async Task<TTenantInfo?> TryGetByIdentifierAsync(string identifier)
         {
             return await dbContext.TenantInfo
                             .Where(ti => ti.Identifier == identifier)
                             .SingleOrDefaultAsync();
         }
 
-        public async Task<bool> TryAddAsync(TTenantInfo tenantInfo)
+        public virtual async Task<bool> TryAddAsync(TTenantInfo tenantInfo)
         {
             await dbContext.TenantInfo.AddAsync(tenantInfo);
 
             return await dbContext.SaveChangesAsync() > 0;
         }
 
-        public async Task<bool> TryRemoveAsync(string identifier)
+        public virtual async Task<bool> TryRemoveAsync(string identifier)
         {
             var existing = await dbContext.TenantInfo
                 .Where(ti => ti.Identifier == identifier)
@@ -61,7 +61,7 @@ public async Task<bool> TryRemoveAsync(string identifier)
             return await dbContext.SaveChangesAsync() > 0;
         }
 
-        public async Task<bool> TryUpdateAsync(TTenantInfo tenantInfo)
+        public virtual async Task<bool> TryUpdateAsync(TTenantInfo tenantInfo)
         {
             var existingLocal = dbContext.TenantInfo.Local.Where(ti => ti.Id == tenantInfo.Id).SingleOrDefault();
             if(existingLocal != null)