支持按策略对Actor进行统一配置

Author: u-less

src/Vertex.Runtime/Actor/Attributes/FollowPolicyAttribute.cs

@@ -0,0 +1,15 @@
+using System;
+
+namespace Vertex.Runtime.Actor.Attributes
+{
+    [AttributeUsage(AttributeTargets.Class, AllowMultiple = false)]
+    public class FollowPolicyAttribute : Attribute
+    {
+        public FollowPolicyAttribute(string optionPolicy)
+        {
+            this.OptionPolicy = optionPolicy;
+        }
+
+        public string OptionPolicy { get; set; }
+    }
+}

src/Vertex.Runtime/Actor/Attributes/VertexPolicyAttribute.cs

@@ -0,0 +1,18 @@
+using System;
+
+namespace Vertex.Runtime.Actor.Attributes
+{
+    [AttributeUsage(AttributeTargets.Class, AllowMultiple = false)]
+    public class VertexPolicyAttribute : Attribute
+    {
+        public VertexPolicyAttribute(string optionPolicy, string archiveOptionPolicy)
+        {
+            this.OptionPolicy = optionPolicy;
+            this.ArchiveOptionPolicy = archiveOptionPolicy;
+        }
+
+        public string OptionPolicy { get; set; }
+
+        public string ArchiveOptionPolicy { get; set; }
+    }
+}

src/Vertex.Runtime/Actor/FlowActor.cs

@@ -4,6 +4,7 @@
 using System.Data;
 using System.Linq;
 using System.Linq.Expressions;
+using System.Reflection;
 using System.Reflection.Emit;
 using System.Runtime.CompilerServices;
 using System.Threading.Tasks;
@@ -340,7 +341,16 @@ static void GetInheritor(SwitchMethodEmit from, List<SwitchMethodEmit> list, Lis
         /// <returns></returns>
         protected virtual async ValueTask DependencyInjection()
         {
-            this.FlowOptions = this.ServiceProvider.GetService<IOptionsMonitor<FlowActorOptions>>().Get(this.ActorType.FullName);
+            var optionPolicy = this.ActorType.GetCustomAttribute<FollowPolicyAttribute>(true);
+            if (optionPolicy != default)
+            {
+                this.FlowOptions = this.ServiceProvider.GetService<IOptionsMonitor<FlowActorOptions>>().Get(optionPolicy.OptionPolicy);
+            }
+            else
+            {
+                this.FlowOptions = this.ServiceProvider.GetService<IOptionsMonitor<FlowActorOptions>>().CurrentValue;
+            }
+
             this.Serializer = this.ServiceProvider.GetService<ISerializer>();
             this.EventTypeContainer = this.ServiceProvider.GetService<IEventTypeContainer>();
             this.Logger = (ILogger)this.ServiceProvider.GetService(typeof(ILogger<>).MakeGenericType(this.ActorType));

src/Vertex.Runtime/Actor/VertexActor.cs

@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.Data;
 using System.Linq;
+using System.Reflection;
 using System.Runtime.CompilerServices;
 using System.Text;
 using System.Threading.Tasks;
@@ -16,6 +17,7 @@
 using Vertex.Abstractions.Snapshot;
 using Vertex.Abstractions.Storage;
 using Vertex.Protocol;
+using Vertex.Runtime.Actor.Attributes;
 using Vertex.Runtime.Event;
 using Vertex.Runtime.Exceptions;
 using Vertex.Runtime.Options;
@@ -67,8 +69,18 @@ public class VertexActor<TPrimaryKey, T> : ActorBase<TPrimaryKey>, IVertexActor
         /// <returns></returns>
         protected virtual async ValueTask DependencyInjection()
         {
-            this.VertexOptions = this.ServiceProvider.GetService<IOptionsMonitor<ActorOptions>>().Get(this.ActorType.FullName);
-            this.ArchiveOptions = this.ServiceProvider.GetService<IOptionsMonitor<ArchiveOptions>>().Get(this.ActorType.FullName);
+            var optionPolicy = this.ActorType.GetCustomAttribute<VertexPolicyAttribute>(true);
+            if (optionPolicy != default)
+            {
+                this.VertexOptions = this.ServiceProvider.GetService<IOptionsMonitor<ActorOptions>>().Get(optionPolicy.OptionPolicy);
+                this.ArchiveOptions = this.ServiceProvider.GetService<IOptionsMonitor<ArchiveOptions>>().Get(optionPolicy.ArchiveOptionPolicy);
+            }
+            else
+            {
+                this.VertexOptions = this.ServiceProvider.GetService<IOptionsMonitor<ActorOptions>>().CurrentValue;
+                this.ArchiveOptions = this.ServiceProvider.GetService<IOptionsMonitor<ArchiveOptions>>().CurrentValue;
+            }
+
             this.Logger = (ILogger)this.ServiceProvider.GetService(typeof(ILogger<>).MakeGenericType(this.ActorType));
             this.Serializer = this.ServiceProvider.GetService<ISerializer>();
             this.EventTypeContainer = this.ServiceProvider.GetService<IEventTypeContainer>();