Cache

Author: phongnguyend

src/EntityFrameworkCore.SqlServer.SimpleBulks/Extensions/DbContextExtensions.cs

@@ -13,20 +13,24 @@ namespace EntityFrameworkCore.SqlServer.SimpleBulks.Extensions;
 
 public static class DbContextExtensions
 {
-    private static readonly ConcurrentDictionary<Type, IReadOnlyList<ColumnInfor>> _propertiesCache = [];
-    private static readonly ConcurrentDictionary<Type, TableInfor> _tableInfoCache = [];
-    private static readonly ConcurrentDictionary<Type, IReadOnlyDictionary<string, string>> _columnNamesCache = [];
-    private static readonly ConcurrentDictionary<Type, IReadOnlyDictionary<string, string>> _columnTypesCache = [];
-    private static readonly ConcurrentDictionary<Type, IReadOnlyList<string>> _primaryKeysCache = [];
-    private static readonly ConcurrentDictionary<Type, ColumnInfor> _outputIdCache = [];
-    private static readonly ConcurrentDictionary<Type, IReadOnlyList<string>> _insertablePropertyNamesCache = [];
-    private static readonly ConcurrentDictionary<Type, IReadOnlyList<string>> _allPropertyNamesCache = [];
+    private readonly record struct CacheKey(Type DbContextType, Type EntityType);
+
+    private static readonly ConcurrentDictionary<CacheKey, IReadOnlyList<ColumnInfor>> _propertiesCache = [];
+    private static readonly ConcurrentDictionary<CacheKey, TableInfor> _tableInfoCache = [];
+    private static readonly ConcurrentDictionary<CacheKey, IReadOnlyDictionary<string, string>> _columnNamesCache = [];
+    private static readonly ConcurrentDictionary<CacheKey, IReadOnlyDictionary<string, string>> _columnTypesCache = [];
+    private static readonly ConcurrentDictionary<CacheKey, IReadOnlyList<string>> _primaryKeysCache = [];
+    private static readonly ConcurrentDictionary<CacheKey, ColumnInfor> _outputIdCache = [];
+    private static readonly ConcurrentDictionary<CacheKey, IReadOnlyList<string>> _insertablePropertyNamesCache = [];
+    private static readonly ConcurrentDictionary<CacheKey, IReadOnlyList<string>> _allPropertyNamesCache = [];
+    private static readonly ConcurrentDictionary<CacheKey, IReadOnlyList<string>> _allPropertyNamesWithoutRowVersionsCache = [];
 
     public static TableInfor GetTableInfor(this DbContext dbContext, Type type)
     {
-        return _tableInfoCache.GetOrAdd(type, (type) =>
+        var cacheKey = new CacheKey(dbContext.GetType(), type);
+        return _tableInfoCache.GetOrAdd(cacheKey, (key) =>
         {
-            var entityType = dbContext.Model.FindEntityType(type);
+            var entityType = dbContext.Model.FindEntityType(key.EntityType);
 
             var schema = entityType.GetSchema();
             var tableName = entityType.GetTableName();
@@ -54,10 +58,11 @@ public static SqlTransaction GetCurrentSqlTransaction(this DbContext dbContext)
 
     public static IReadOnlyList<ColumnInfor> GetProperties(this DbContext dbContext, Type type)
     {
-        return _propertiesCache.GetOrAdd(type, (type) =>
+        var cacheKey = new CacheKey(dbContext.GetType(), type);
+        return _propertiesCache.GetOrAdd(cacheKey, (key) =>
         {
-            var typeProperties = type.GetProperties().Select(x => new { x.Name, x.PropertyType });
-            var entityProperties = dbContext.Model.FindEntityType(type)
+            var typeProperties = key.EntityType.GetProperties().Select(x => new { x.Name, x.PropertyType });
+            var entityProperties = dbContext.Model.FindEntityType(key.EntityType)
                            .GetProperties();
 
             var data = typeProperties.Join(entityProperties,
@@ -80,58 +85,74 @@ public static IReadOnlyList<ColumnInfor> GetProperties(this DbContext dbContext,
 
     public static IReadOnlyDictionary<string, string> GetColumnNames(this DbContext dbContext, Type type)
     {
-        return _columnNamesCache.GetOrAdd(type, (type) =>
+        var cacheKey = new CacheKey(dbContext.GetType(), type);
+        return _columnNamesCache.GetOrAdd(cacheKey, (key) =>
         {
-            var properties = dbContext.GetProperties(type);
+            var properties = dbContext.GetProperties(key.EntityType);
             return properties.ToDictionary(x => x.PropertyName, x => x.ColumnName);
         });
     }
 
     public static IReadOnlyDictionary<string, string> GetColumnTypes(this DbContext dbContext, Type type)
     {
-        return _columnTypesCache.GetOrAdd(type, (type) =>
+        var cacheKey = new CacheKey(dbContext.GetType(), type);
+        return _columnTypesCache.GetOrAdd(cacheKey, (key) =>
         {
-            var properties = dbContext.GetProperties(type);
+            var properties = dbContext.GetProperties(key.EntityType);
             return properties.ToDictionary(x => x.PropertyName, x => x.ColumnType);
         });
     }
 
     public static IReadOnlyList<string> GetPrimaryKeys(this DbContext dbContext, Type type)
     {
-        return _primaryKeysCache.GetOrAdd(type, (type) =>
+        var cacheKey = new CacheKey(dbContext.GetType(), type);
+        return _primaryKeysCache.GetOrAdd(cacheKey, (key) =>
         {
-            var properties = dbContext.GetProperties(type);
+            var properties = dbContext.GetProperties(key.EntityType);
             return properties.Where(x => x.IsPrimaryKey).Select(x => x.PropertyName).ToArray();
         });
     }
 
     public static ColumnInfor GetOutputId(this DbContext dbContext, Type type)
     {
-        return _outputIdCache.GetOrAdd(type, (type) =>
+        var cacheKey = new CacheKey(dbContext.GetType(), type);
+        return _outputIdCache.GetOrAdd(cacheKey, (key) =>
         {
-            var properties = dbContext.GetProperties(type);
+            var properties = dbContext.GetProperties(key.EntityType);
             return properties.Where(x => x.IsPrimaryKey && x.ValueGenerated == ValueGenerated.OnAdd).FirstOrDefault();
         });
     }
 
     public static IReadOnlyList<string> GetInsertablePropertyNames(this DbContext dbContext, Type type)
     {
-        return _insertablePropertyNamesCache.GetOrAdd(type, (type) =>
+        var cacheKey = new CacheKey(dbContext.GetType(), type);
+        return _insertablePropertyNamesCache.GetOrAdd(cacheKey, (key) =>
         {
-            var properties = dbContext.GetProperties(type);
+            var properties = dbContext.GetProperties(key.EntityType);
             return properties.Where(x => x.ValueGenerated == ValueGenerated.Never).Select(x => x.PropertyName).ToArray();
         });
     }
 
     public static IReadOnlyList<string> GetAllPropertyNames(this DbContext dbContext, Type type)
     {
-        return _allPropertyNamesCache.GetOrAdd(type, (type) =>
+        var cacheKey = new CacheKey(dbContext.GetType(), type);
+        return _allPropertyNamesCache.GetOrAdd(cacheKey, (key) =>
         {
-            var properties = dbContext.GetProperties(type);
+            var properties = dbContext.GetProperties(key.EntityType);
             return properties.Select(x => x.PropertyName).ToArray();
         });
     }
 
+    public static IReadOnlyList<string> GetAllPropertyNamesWithoutRowVersions(this DbContext dbContext, Type type)
+    {
+        var cacheKey = new CacheKey(dbContext.GetType(), type);
+        return _allPropertyNamesWithoutRowVersionsCache.GetOrAdd(cacheKey, (key) =>
+        {
+            var properties = dbContext.GetProperties(key.EntityType);
+            return properties.Where(x => !x.IsRowVersion).Select(x => x.PropertyName).ToArray();
+        });
+    }
+
     public static IDbCommand CreateTextCommand(this DbContext dbContext, string commandText, BulkOptions options = null)
     {
         return dbContext.GetSqlConnection().CreateTextCommand(dbContext.GetCurrentSqlTransaction(), commandText, options);