src/Library/NetPro.Checker/CheckMiddlewareHandler.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Diagnostics.HealthChecks;
using NetTools;
using Newtonsoft.Json;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Threading.Tasks;

namespace NetPro.Checker
{
    public static class CheckMiddlewareHandler
    {
        private static readonly string DEFAULT_CONTENT_TYPE = "application/json";

        /// <summary>
        /// inclued: EnvCheck ;InfoCheck
        /// </summary>
        /// <param name="app"></param>
        /// <param name="envPath"></param>
        /// <param name="infoPath"></param>
        /// <param name="health"></param>
        public static void UseCheck(this IApplicationBuilder app, string envPath = "/env", string infoPath = "/info", string health = "/check")
        {
            app.UseEnvCheck(envPath);
            app.UseInfoCheck(infoPath);
            app.UseHealthCheck(health);
        }

        /// <summary>
        /// 
        /// </summary>
        /// <param name="app"></param>
        /// <param name="path"></param>
        public static void UseEnvCheck(this IApplicationBuilder app, string path = "/env")
        {
            var config = app.ApplicationServices.GetService<IConfiguration>();
            app.Map(path, s =>
            {
                s.Run(async context =>
                {
                    //ip可篡改，只作为基础限制，实际建议nginx处配置路由映射的方式限制
                    var remoteIp = context.Connection.RemoteIpAddress;
                    var rangeA = IPAddressRange.Parse("192.168.0.0 - 192.168.255.255");
                    var rangeB = IPAddressRange.Parse("172.16.0.0 - 172.31.255.255");
                    var rangeC = IPAddressRange.Parse("10.0.0.0 - 10.255.255.255");
                    if (IPAddress.IsLoopback(remoteIp) || rangeA.Contains(remoteIp) || rangeB.Contains(remoteIp) || rangeC.Contains(remoteIp))
                    {
                        var env = AppEnvironment.GetAppEnvironment();
                        context.Response.ContentType = DEFAULT_CONTENT_TYPE;
                        await context.Response.WriteAsync(Serialize(env));
                    }
                    else
                    {
                        context.Response.StatusCode = 403;
                        context.Response.ContentType = "application/html";