ds8-solution-pro/ds-wms-service/DS.WMS.Gateway/JwtSafeMiddleware.cs

using DS.Module.Core;

namespace DS.WMS.Gateway;

public class JwtSafeMiddleware
{
    private readonly RequestDelegate _next;
    public IConfiguration _configuration;

    public JwtSafeMiddleware(RequestDelegate next, IConfiguration configuration)
    {
        _next = next;
        _configuration = configuration;
    }

    public async Task Invoke(HttpContext context)
    {
        //表示如果RequestTokenServer1配置在网关下，则访问它获取token的请求不走jwt校验哦
        //if(!context.Request.Path.Value.StartsWith("/auth"))

        if (context.Request.Method == "GET" || context.Request.Method == "POST")
        {
            // Console.WriteLine(context.Request.Path);
            var path = context.Request.Path.ToString();
            if (path.Contains("swagger") || path.Contains("login"))
            {
                //跳过swagger及login
            }
            else
            {
                string jwtStr = context.Request.Headers["Authorization"].FirstOrDefault();
                // Console.WriteLine(jwtStr);
                if (string.IsNullOrEmpty(jwtStr))
                {
                    context.Response.StatusCode = 401; //401未授权
                    await context.Response.WriteAsync("token为空");
                    return;
                }
                //校验auth的正确性
                var result = JwtHelper.SerializeJwt(jwtStr);
                if (result == "expired")
                {
                    context.Response.StatusCode = 401; //401未授权
                    await context.Response.WriteAsync("非法请求，参数已经过期");
                    return;
                }
                else if (result == "invalid")
                {
                    context.Response.StatusCode = 401; //401未授权
                    await context.Response.WriteAsync("非法请求，未通过校验");
                    return;
                }
                else if (result == "error")
                {
                    context.Response.StatusCode = 401; //401未授权
                    await context.Response.WriteAsync("非法请求，未通过校验");
                    return;
                }
                else
                {
                    //表示校验通过
                }
            }
        }
        await _next.Invoke(context);
    }
}