♻ refactor(IDistributedCache): Optimize connections under Isolation and Attempt to reconnect

src/BuildingBlocks/Caching/Masa.BuildingBlocks.Caching/Internal/DistributedCacheClientCache.cs

@@ -0,0 +1,21 @@
+// Copyright (c) MASA Stack All rights reserved.
+// Licensed under the MIT License. See LICENSE.txt in the project root for license information.
+
+namespace Masa.BuildingBlocks.Caching;
+
+internal class DistributedCacheClientCache
+{
+    public static ConcurrentDictionary<string, IManualDistributedCacheClient> CacheClients { get; set; } = new();
+
+    public IManualDistributedCacheClient GetCacheClient(IServiceProvider serviceProvider)
+    {
+        var multiEnvironmentContext = serviceProvider.GetRequiredService<IMultiEnvironmentContext>();
+        var environment = multiEnvironmentContext.CurrentEnvironment;
+
+        return CacheClients.GetOrAdd(environment, env =>
+        {
+            var cacheClient = serviceProvider.GetRequiredService<ScopedService<IManualDistributedCacheClient>>().Service;
+            return cacheClient;
+        });
+    }
+}

src/BuildingBlocks/Caching/Masa.BuildingBlocks.Caching/Internal/ServiceCollectionExtensions.Core.cs

@@ -23,8 +23,9 @@ public static void TryAddDistributedCache(
         services.TryAddTransient<IManualDistributedCacheClient>(serviceProvider =>
         {
             var cacheClient = serviceProvider.EnableIsolation() ?
-                serviceProvider.GetRequiredService<ScopedService<IManualDistributedCacheClient>>().Service :
+                serviceProvider.GetRequiredService<DistributedCacheClientCache>().GetCacheClient(serviceProvider) :
                 serviceProvider.GetRequiredService<SingletonService<IManualDistributedCacheClient>>().Service;
+
             return new DefaultDistributedCacheClient(cacheClient);
         });
         services.TryAddTransient<IDistributedCacheClient>(serviceProvider
@@ -65,6 +66,8 @@ private static void AddTypeAlias(
 
     private static void AddCaching(this IServiceCollection services)
     {
+        services.TryAddSingleton<DistributedCacheClientCache>();
+
         services.TryAddSingleton<SingletonService<IManualDistributedCacheClient>>(serviceProvider =>
             new SingletonService<IManualDistributedCacheClient>(serviceProvider.GetRequiredService<IDistributedCacheClientFactory>()
                 .Create()));

src/Contrib/Caching/Distributed/Masa.Contrib.Caching.Distributed.StackExchangeRedis/RedisCacheClientBase.cs

@@ -7,29 +7,29 @@ public abstract class RedisCacheClientBase : DistributedCacheClientBase
 {
     protected readonly string? InstanceId;
     protected static readonly Guid UniquelyIdentifies = Guid.NewGuid();
-    protected readonly ISubscriber Subscriber;
+    protected ISubscriber Subscriber;
 
     protected IDatabase Db
     {
         get
         {
-            if (_connection.IsConnected || _connection.IsConnecting)
-                return _connection.GetDatabase();
-
-            throw new NotSupportedException("Redis service has been disconnected, please wait for reconnection and try again");
+            return EnsureDbConnection();
         }
     }
 
-    private readonly IConnectionMultiplexer _connection;
+    private IConnectionMultiplexer _connection;
     protected readonly JsonSerializerOptions GlobalJsonSerializerOptions;
     private readonly CacheEntryOptions _globalCacheEntryOptions;
     private readonly CacheOptions _globalCacheOptions;
 
+    private readonly RedisConfigurationOptions _redisConfigurationOptions;
+
     protected RedisCacheClientBase(
         RedisConfigurationOptions redisConfigurationOptions,
         JsonSerializerOptions? jsonSerializerOptions)
         : this(redisConfigurationOptions.GlobalCacheOptions, redisConfigurationOptions, jsonSerializerOptions)
     {
+        _redisConfigurationOptions = redisConfigurationOptions;
         var redisConfiguration = redisConfigurationOptions.GetAvailableRedisOptions();
         _connection = ConnectionMultiplexer.Connect(redisConfiguration);
         Subscriber = _connection.GetSubscriber();
@@ -51,6 +51,28 @@ private RedisCacheClientBase(
         GlobalJsonSerializerOptions = jsonSerializerOptions ?? new JsonSerializerOptions().EnableDynamicTypes();
     }
 
+    protected IDatabase EnsureDbConnection()
+    {
+        if (_connection.IsConnected || _connection.IsConnecting)
+        {
+            return _connection.GetDatabase();
+        }
+
+        // Attempt to reconnect
+        var redisConfiguration = _redisConfigurationOptions.GetAvailableRedisOptions();
+        _connection = ConnectionMultiplexer.Connect(redisConfiguration);
+        Subscriber = _connection.GetSubscriber();
+
+        if (_connection.IsConnected || _connection.IsConnecting)
+        {
+            return _connection.GetDatabase();
+        }
+        else
+        {
+            throw new NotSupportedException("Unable to reconnect to Redis, please check the connection settings and try again.");
+        }
+    }
+
     protected T? ConvertToValue<T>(RedisValue value, out bool isExist)
     {
         if (value is { HasValue: true, IsNullOrEmpty: false })