src/Framework/Utils/Utils.Mvc/Helpers/MvcHelper.cs

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

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

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;
using System.Reflection;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.ApplicationParts;
using Microsoft.AspNetCore.Mvc.Controllers;
using NetModular.Lib.Utils.Core.Attributes;

namespace NetModular.Lib.Utils.Mvc.Helpers
{
    /// <summary>
    /// MVC帮助类
    /// </summary>
    [Singleton]
    public class MvcHelper
    {
        private List<ControllerDescriptor> _controllerDescriptors;
        private List<ActionDescriptor> _actionDescriptors;
        private readonly ApplicationPartManager _partManager;
        private static readonly object LockObject = new object();

        public MvcHelper(ApplicationPartManager partManager)
        {
            _partManager = partManager;
        }

        /// <summary>
        /// 获取所有Controller，不包含抽象的类
        /// </summary>
        /// <returns></returns>
        public List<ControllerDescriptor> GetAllController()
        {
            lock (LockObject)
            {
                if (_controllerDescriptors != null && _controllerDescriptors.Any())
                    return _controllerDescriptors;

                _controllerDescriptors = new List<ControllerDescriptor>();

                var controllerFeature = new ControllerFeature();
                _partManager.PopulateFeature(controllerFeature);
                var controllerTypes = controllerFeature.Controllers.ToList();
                foreach (var typeInfo in controllerTypes)
                {
                    if (typeInfo.IsAbstract)
                        continue;

                    var controller = new ControllerDescriptor
                    {
                        Name = typeInfo.Name.Replace("Controller", ""),
                        TypeInfo = typeInfo
                    };

                    var areaAttr = (AreaAttribute)Attribute.GetCustomAttribute(typeInfo, typeof(AreaAttribute));
                    if (areaAttr != null)
                    {
                        controller.Area = areaAttr.RouteValue;
                    }