.NET Core统一设置路由前缀

拼搏现实的明天。 2022-11-02 11:58 259阅读 0赞

NetCore WebAPI 如何设置一个统一的路由前缀

创建一个类，遍历所有的controller,添加一个路由前缀

public class RouteConvention : IApplicationModelConvention
        {
            private readonly AttributeRouteModel _routePrefix;
            /// <summary>
            /// 构造方法
            /// </summary>
            /// <param name="routeTemplateProvider"></param>
            public RouteConvention(IRouteTemplateProvider routeTemplateProvider)
            {
                _routePrefix = new AttributeRouteModel(routeTemplateProvider);
            }
    
            /// <summary>
            /// 实现接口的Apply方法
            /// </summary>
            /// <param name="application"></param>
            public void Apply(ApplicationModel application)
            {
                //遍历所有的 Controller
                foreach (var controller in application.Controllers)
                {
                    // 已经标记了 RouteAttribute 的 Controller
                    var matchedSelectors = controller.Selectors.Where(x => x.AttributeRouteModel != null).ToList();
                    if (matchedSelectors.Any())
                    {
                        foreach (var selectorModel in matchedSelectors)
                        {
                            // 在当前路由上 再 添加一个路由前缀
                            selectorModel.AttributeRouteModel = AttributeRouteModel.CombineAttributeRouteModel(_routePrefix, selectorModel.AttributeRouteModel);
                        }
                    }
    
                    // 没有标记 RouteAttribute 的 Controller
                    var unmatchedSelectors = controller.Selectors.Where(x => x.AttributeRouteModel == null).ToList();
                    if (unmatchedSelectors.Any())
                    {
                        foreach (var selectorModel in unmatchedSelectors)
                        {
                            // 添加一个路由前缀
                            selectorModel.AttributeRouteModel = _routePrefix;                       
                        }
                    }
                }
            }
        }

使用自定义路由扩展方法

/// <summary>
        /// 自定义路由扩展类
        /// </summary>
        public static class MvcOptionsExtensions
        {
            /// <summary>
            /// 使用自定义路由扩展方法
            /// </summary>
            /// <param name="opts"></param>
            /// <param name="routeAttribute"></param>
            public static void UseCentralRoutePrefix(this MvcOptions opts, IRouteTemplateProvider routeAttribute)
            {
                // 添加我们自定义 实现IApplicationModelConvention的RouteConvention
                opts.Conventions.Insert(0, new RouteConvention(routeAttribute));
            }
        }

编辑startup.cs

public void ConfigureServices(IServiceCollection services)
    {
         services.AddControllers(opt =>
         {
              // 统一设置路由前缀
              opt.UseCentralRoutePrefix(new RouteAttribute("testapi"));
         });
    }