框架分层

Yi.Framework/Yi.Framework.Core/ConsulExtend/ServerExtend/ConsulRegisterOption.cs

@@ -0,0 +1,44 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace CC.ElectronicCommerce.Core.ConsulExtend
+{
+    public class ConsulRegisterOption
+    {
+        /// <summary>
+        /// 服务自身IP
+        /// </summary>
+        public string IP { get; set; }
+        /// <summary>
+        /// 服务自身Port
+        /// </summary>
+        public int Port { get; set; }
+        /// <summary>
+        /// 组名称
+        /// </summary>
+        public string GroupName { get; set; }
+        /// <summary>
+        /// 心跳检查地址
+        /// </summary>
+        public string HealthCheckUrl { get; set; }
+        /// <summary>
+        /// 心跳频率
+        /// </summary>
+        public int Interval { get; set; }
+        /// <summary>
+        /// 心跳超时
+        /// </summary>
+        public int Timeout { get; set; }
+        /// <summary>
+        /// 移除延迟时间
+        /// </summary>
+        public int DeregisterCriticalServiceAfter { get; set; }
+        /// <summary>
+        /// 标签，额外信息，用于权重
+        /// </summary>
+        public string Tag { get; set; }
+    }
+}

Yi.Framework/Yi.Framework.Core/ConsulExtend/ServerExtend/ConsulRegiterExtend.cs

@@ -0,0 +1,69 @@
+using Consul;
+using Microsoft.AspNetCore.Builder;
+using Microsoft.Extensions.Configuration;
+using Newtonsoft.Json;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace CC.ElectronicCommerce.Core.ConsulExtend
+{
+    /// <summary>
+    /// HTTP模式
+    /// </summary>
+    public static class ConsulRegiterExtend
+    {
+        /// <summary>
+        /// 自动读取配置文件完成注册
+        /// </summary>
+        /// <param name="app"></param>
+        /// <param name="configuration"></param>
+        /// <returns></returns>
+        public static async Task UseConsulConfiguration(this IApplicationBuilder app, IConfiguration configuration)
+        {
+            ConsulRegisterOption consulRegisterOption = new ConsulRegisterOption();
+            configuration.Bind("ConsulRegisterOption", consulRegisterOption);
+
+            ConsulClientOption consulClientOption = new ConsulClientOption();
+            configuration.Bind("ConsulClientOption", consulClientOption);
+
+            await UseConsul(app, consulClientOption, consulRegisterOption);
+        }
+        /// <summary>
+        /// 基于提供信息完成注册
+        /// </summary>
+        /// <param name="app"></param>
+        /// <param name="healthService"></param>
+        /// <returns></returns>
+        public static async Task UseConsul(this IApplicationBuilder app, ConsulClientOption consulClientOption, ConsulRegisterOption consulRegisterOption)
+        {
+            using (ConsulClient client = new ConsulClient(c =>
+             {
+                 c.Address = new Uri($"http://{consulClientOption.IP}:{consulClientOption.Port}/");
+                 c.Datacenter = consulClientOption.Datacenter;
+             }))
+            {
+                await client.Agent.ServiceRegister(new AgentServiceRegistration()
+                {
+                    ID = $"{consulRegisterOption.IP}-{consulRegisterOption.Port}-{Guid.NewGuid()}",//唯一Id
+                    Name = consulRegisterOption.GroupName,//组名称-Group
+                    Address = consulRegisterOption.IP,
+                    Port = consulRegisterOption.Port,
+                    Tags = new string[] { consulRegisterOption.Tag },
+                    Check = new AgentServiceCheck()
+                    {
+                        Interval = TimeSpan.FromSeconds(consulRegisterOption.Interval),
+                        HTTP = $"http://{consulRegisterOption.IP}:{consulRegisterOption.Port}{consulRegisterOption.HealthCheckUrl}",
+                        Timeout = TimeSpan.FromSeconds(consulRegisterOption.Timeout),
+                        DeregisterCriticalServiceAfter = TimeSpan.FromSeconds(consulRegisterOption.DeregisterCriticalServiceAfter)
+                    }
+                });
+                Console.WriteLine($"{JsonConvert.SerializeObject(consulRegisterOption)} 完成注册");
+            }
+        }
+
+
+    }
+}

Yi.Framework/Yi.Framework.Core/ConsulExtend/ServerExtend/HealthCheckMiddleware.cs

@@ -0,0 +1,31 @@
+using Microsoft.AspNetCore.Builder;
+using Microsoft.AspNetCore.Http;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Net;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace CC.ElectronicCommerce.Core.ConsulExtend
+{
+    public static class HealthCheckMiddleware
+    {
+        /// <summary>
+        /// 设置心跳响应
+        /// </summary>
+        /// <param name="app"></param>
+        /// <param name="checkPath">默认是/Health</param>
+        /// <returns></returns>
+        public static void UseHealthCheckMiddleware(this IApplicationBuilder app, string checkPath = "/Health")
+        {
+            app.Map(checkPath, applicationBuilder => applicationBuilder.Run(async context =>
+            {
+                Console.WriteLine($"This is Health Check");
+                context.Response.StatusCode = (int)HttpStatusCode.OK;
+                await context.Response.WriteAsync("OK");
+            }));
+        }
+
+    }
+}