ABP基于Attribute构建动态Api

背景

加的群里有人（这里暂且成为A君）有这样的需求：针对PC端和App端有不同的api需求，例如App端只可访问GetXXX，而PC端可访问GetXXX，UpdateXXX，CreateXXX等。

而我们也知道，Abp中是提供了这样的方法的，但写起来比较麻烦，写法如下：

DynamicApiControllerBuilder
    .For<ITaskAppService>("tasksystem/task")
    .ForMethod("CreateTask").DontCreateAction()
    .ForMethod("UpdateTask").DontCreateAction()
    .ForMethod("DeleteTask").DontCreateAction()
    .Build();

于是A君通过反射和Attribute相结合自己实现了一套，但遇到了些问题，于是在讨论的过程中我也试着在本机测试了下，最终测试通过，下面给出解决方案。

解决方案

1.定义一个Attribute，这个Attribute里边没什么代码，只是标识作用。

public class AppUsedAttribute : Attribute
{
}

2.实现类

public class BuildApiHelper
    {
        public static void BuildApi(Type moduleType, string servicePrefix, Type attrType)
        {
            var _assembly = Assembly.GetAssembly(moduleType);

            var types = from type in _assembly.GetTypes()
                        where type.IsPublic &&
                              type.IsInterface && typeof(IApplicationService).IsAssignableFrom(type) &&
                              IocManager.Instance.IsRegistered(type)
                        select type;

            foreach (var type in types)
            {
                var serviceName = servicePrefix + "/" + GetConventionalServiceName(type);

                var obj = typeof(DynamicApiControllerBuilder)
                    .GetMethod("For", BindingFlags.Public | BindingFlags.Static)
                    .MakeGenericMethod(type)
                    .Invoke(null, new object[] { serviceName });

                var methods = type.GetMethods().Where(n => n.DeclaringType == type).ToArray();

                foreach (MethodInfo item in methods)
                {
                    var attr = item.GetCustomAttribute(attrType);
                    if (attr == null)
                    {
                        var ationobj = obj.GetType().GetMethod("ForMethod").Invoke(obj, new object[] { item.Name });
                        obj = ationobj.GetType().GetMethod("DontCreateAction").Invoke(ationobj, null);
                    }
                }
                obj.GetType().GetMethod("Build").Invoke(obj, null);
            }
        }

        private static string GetConventionalServiceName(Type type)
        {
            var typeName = type.Name;

            if (typeName.EndsWith("ApplicationService"))
            {
                typeName = typeName.Substring(0, typeName.Length - "ApplicationService".Length);
            }
            else if (typeName.EndsWith("AppService"))
            {
                typeName = typeName.Substring(0, typeName.Length - "AppService".Length);
            }
            else if (typeName.EndsWith("Service"))
            {
                typeName = typeName.Substring(0, typeName.Length - "Service".Length);
            }

            if (typeName.Length > 1 && typeName.StartsWith("I") && char.IsUpper(typeName, 1))
            {
                typeName = typeName.Substring(1);
            }

            return typeName.ToCamelCase();
        }
    }

3.调用方式

public override void Initialize()
{ 
     IocManager.RegisterAssemblyByConvention(Assembly.GetExecutingAssembly());

     BuildApiHelper.BuildApi(typeof(MonkeyApplicationModule), "app", typeof(AppUsedAttribute));

}

 写在最后

这只是实现方式中的一种解决方案，如何你有更好的解决方案欢迎在评论中指出，欢迎大家一起讨论，共同进步