init

DouyinApi.Extensions/ServiceExtensions/CacheSetup.cs

@@ -0,0 +1,58 @@
+using System.Threading.Tasks;
+using DouyinApi.Common;
+using DouyinApi.Common.Caches;
+using DouyinApi.Common.Caches.Distributed;
+using DouyinApi.Common.Caches.Interface;
+using DouyinApi.Common.Option;
+using Microsoft.Extensions.Caching.Distributed;
+using Microsoft.Extensions.Caching.Memory;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.DependencyInjection.Extensions;
+using Serilog;
+using StackExchange.Redis;
+
+namespace DouyinApi.Extensions.ServiceExtensions;
+
+public static class CacheSetup
+{
+    /// <summary>
+    /// 统一注册缓存
+    /// </summary>
+    /// <param name="services"></param>
+    public static void AddCacheSetup(this IServiceCollection services)
+    {
+        var cacheOptions = App.GetOptions<RedisOptions>();
+        if (cacheOptions.Enable)
+        {
+            // 配置启动Redis服务，虽然可能影响项目启动速度，但是不能在运行的时候报错，所以是合理的
+            services.AddSingleton<IConnectionMultiplexer>(sp =>
+            {
+                //获取连接字符串
+                var configuration = ConfigurationOptions.Parse(cacheOptions.ConnectionString, true);
+                configuration.ResolveDns = true;
+                return ConnectionMultiplexer.Connect(configuration);
+            });
+            services.AddSingleton(p => p.GetService<IConnectionMultiplexer>() as ConnectionMultiplexer);
+            //使用Redis
+            services.AddStackExchangeRedisCache(options =>
+            {
+                options.ConnectionMultiplexerFactory =
+                    () => Task.FromResult(App.GetService<IConnectionMultiplexer>(false));
+                if (!cacheOptions.InstanceName.IsNullOrEmpty()) options.InstanceName = cacheOptions.InstanceName;
+            });
+
+            services.AddTransient<IRedisBasketRepository, RedisBasketRepository>();
+        }
+        else
+        {
+            //使用内存
+            services.Remove(services.FirstOrDefault(x => x.ServiceType == typeof(IMemoryCache)));
+            services.AddSingleton<MemoryCacheManager>();
+            services.AddSingleton<IMemoryCache>(provider => provider.GetService<MemoryCacheManager>());
+            services.AddOptions();
+            services.AddSingleton<IDistributedCache, CommonMemoryDistributedCache>();
+        }
+
+        services.AddSingleton<ICaching, Caching>();
+    }
+}