HostApp配置，新增配置雪花漂移算法

zhontai · Jul 18, 2022 · ef0bab3 · ef0bab3
1 parent 358e989
commit ef0bab3
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/src/platform/ZhonTai.Admin/Core/HostApp.cs b/src/platform/ZhonTai.Admin/Core/HostApp.cs
@@ -154,7 +154,9 @@ private void ConfigureServices(IServiceCollection services, IWebHostEnvironment
             _hostAppOptions?.ConfigurePreServices?.Invoke(hostAppContext);
 
             //雪花漂移算法
-            YitIdHelper.SetIdGenerator(new IdGeneratorOptions(1) { WorkerIdBitLength = 6 });
+            var idGeneratorOptions = new IdGeneratorOptions(1) { WorkerIdBitLength = 6 };
+            _hostAppOptions?.ConfigureIdGenerator?.Invoke(idGeneratorOptions);
+            YitIdHelper.SetIdGenerator(idGeneratorOptions);
 
             //权限处理
             services.AddScoped<IPermissionHandler, PermissionHandler>();

diff --git a/src/platform/ZhonTai.Admin/Core/Startup/HostAppOptions.cs b/src/platform/ZhonTai.Admin/Core/Startup/HostAppOptions.cs
@@ -1,5 +1,6 @@
 using FreeSql;
 using System;
+using Yitter.IdGenerator;
 using ZhonTai.DynamicApi;
 
 namespace ZhonTai.Admin.Core.Startup
@@ -53,5 +54,11 @@ public class HostAppOptions
         /// 配置动态Api
         /// </summary>
         public Action<DynamicApiOptions> ConfigureDynamicApi { get; set; }
+
+        /// <summary>
+        /// 配置雪花漂移算法
+        /// </summary>
+        public Action<IdGeneratorOptions> ConfigureIdGenerator { get; set; }
+
     }
 }
diff --git a/src/platform/ZhonTai.Admin/ZhonTai.Admin.xml b/src/platform/ZhonTai.Admin/ZhonTai.Admin.xml