缓存搭建

Yi.Framework.Net6/Yi.Framework.Core/ConsulExtend/WeightDispatcher.cs

@@ -0,0 +1,67 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using Consul;
+using Microsoft.Extensions.Options;
+using Yi.Framework.Common.IOCOptions;
+
+namespace Yi.Framework.Core.ConsulExtend
+{
+    /// <summary>
+    /// 权重
+    /// </summary>
+    public class WeightDispatcher : AbstractConsulDispatcher
+    {
+        #region Identity
+        private static int _iTotalCount = 0;
+        private static int iTotalCount
+        {
+            get
+            {
+                return _iTotalCount;
+            }
+            set
+            {
+                _iTotalCount = value >= Int32.MaxValue ? 0 : value;
+            }
+        }
+        public WeightDispatcher(IOptionsMonitor<ConsulClientOption> consulClientOption) : base(consulClientOption)
+        {
+
+        }
+        #endregion
+
+        protected override string ChooseAddress(string serviceName)
+        {
+            ConsulClient client = new ConsulClient(c =>
+            {
+                c.Address = new Uri($"http://{base._ConsulClientOption.IP}:{base._ConsulClientOption.Port}/");
+                c.Datacenter = base._ConsulClientOption.Datacenter;
+            });
+            AgentService agentService = null;
+            var response = client.Agent.Services().Result.Response;
+
+            this._CurrentAgentServiceDictionary = response.Where(s => s.Value.Service.Equals(serviceName, StringComparison.OrdinalIgnoreCase)).ToArray();
+
+
+            var serviceDictionaryNew = new List<AgentService>();
+            foreach (var service in base._CurrentAgentServiceDictionary)
+            {
+                serviceDictionaryNew.AddRange(Enumerable.Repeat(service.Value, int.TryParse(service.Value.Tags?[0], out int iWeight) ? 1 : iWeight));
+            }
+            int index = new Random(DateTime.Now.Millisecond).Next(0, int.MaxValue) % serviceDictionaryNew.Count;
+            agentService = serviceDictionaryNew[index];
+
+            return $"{agentService.Address}:{agentService.Port}";
+        }
+        /// <summary>
+        /// 不需要了
+        /// </summary>
+        /// <returns></returns>
+        protected override int GetIndex()
+        {
+            throw new NotImplementedException();
+        }
+    }
+}