如何检查方法是否具有属性

Question

我有一个示例类

public class MyClass{

    ActionResult Method1(){
        ....
    } 

    [Authorize]
    ActionResult Method2(){
       ....
    }

    [Authorize]    
    ActionResult Method3(int value){
       ....
    }

}

现在我想要的是编写一个返回 true/false 的函数，可以像这样执行

var controller = new MyClass();

Assert.IsFalse(MethodHasAuthorizeAttribute(controller.Method1));
Assert.IsTrue(MethodHasAuthorizeAttribute(controller.Method2));
Assert.IsTrue(MethodHasAuthorizeAttribute(controller.Method3));

地步

public bool MethodHasAuthorizeAttribute(Func<int, ActionResult> function)
{
    return function.Method.GetCustomAttributes(typeof(AuthorizeAttribute), false).Length > 0;
}

我已经到了适用于 Method3 的 。现在我怎样才能以一种将字符串和类作为参数的方式来实现这个通用呢？

Answer 1

您的代码的问题是 public bool MethodHasAuthorizeAttribute(Funcfunction) 的签名。 MethodHasAuthorizeAttribute 只能与与您指定的委托签名匹配的参数一起使用。在本例中，方法返回带有 int 类型参数的 ActionResult。

当您调用此方法（如 MethodHasAuthorizeAttribute(controller.Method3)）时，编译器将执行方法组转换。这可能并不总是理想的，并且可能会产生意想不到的结果（方法组转换并不总是直接的）。如果您尝试调用 MethodHasAuthorizeAttribute(controller.Method1)，您将收到编译器错误，因为没有转换。

可以使用表达式树和著名的“MethodOf”技巧构建更通用的解决方案。它使用编译器生成的表达式树来查找调用目标：

public static MethodInfo MethodOf( Expression<System.Action> expression )
{
    MethodCallExpression body = (MethodCallExpression)expression.Body;
    return body.Method;
}

您可以像这样使用它，但它也可以与任何方法一起使用：

MethodInfo method = MethodOf( () => controller.Method3( default( int ) ) );

有了它，我们就可以构建一个通用实现：

public static bool MethodHasAuthorizeAttribute( Expression<System.Action> expression )
{
    var method = MethodOf( expression );

    const bool includeInherited = false;
    return method.GetCustomAttributes( typeof( AuthorizeAttribute ), includeInherited ).Any();
}

好的，这就是方法。现在，如果您想对类或字段应用属性检查（我将保留属性，因为它们实际上是方法），我们需要对 MemberInfo 执行检查，它是 的继承根类型、FieldInfo 和MethodInfo。这就像将属性搜索提取到单独的方法中并提供具有好名称的适当适配器方法一样简单：

public static bool MethodHasAuthorizeAttribute( Expression<System.Action> expression )
{
    MemberInfo member = MethodOf( expression );
    return MemberHasAuthorizeAttribute( member );
}

public static bool TypeHasAuthorizeAttribute( Type t)
{
    return MemberHasAuthorizeAttribute( t );
}

private static bool MemberHasAuthorizeAttribute( MemberInfo member )
{
    const bool includeInherited = false;
    return member.GetCustomAttributes( typeof( AuthorizeAttribute ), includeInherited ).Any();
}

我将把字段的实现作为练习，您可以采用与 MethodOf 相同的技巧。

Answer 2

与当前 .NET/C# 版本（4.6.1、C#6）的其他解决方案相比，有一种更简单的解决方案：

如果您只有一个具有该名称的方法：

var method = typeof(TestClass).GetMethods()
  .SingleOrDefault(x => x.Name == nameof(TestClass.TestMethod));

var attribute = method?.GetCustomAttributes(typeof(MethodAttribute), true)
  .Single() as MethodAttribute;

现在检查您是否已设置该属性方法：

bool isDefined = attribute != null;

如果你想访问属性的属性，你可以简单地做到这一点：

var someInfo = attribute.SomeMethodInfo

如果有多个同名的方法，你可以继续使用 method.GetParameters() 并检查参数，而不是.GetMethods().Single...

如果您知道您的方法没有参数，则此检查很容易：

var method = typeof(TestClass).GetMethods()
    .SingleOrDefault(
      x => x.Name == nameof(TestClass.TestMethod) 
      && x.GetParameters().Length == 0
);

如果不知道，这将更加复杂（检查参数等）并且其他解决方案使用起来更简单、更可靠。

因此：如果您没有方法重载，或者只想从具有指定数量参数的方法中读取属性，请使用此方法。
否则，请使用此处其他答案提供的 MethodOf 。

Answer 3

我做了类似的事情：

public static bool MethodHasAuthorizeAttribute(this Delegate pMethod, string pRoleAccess)
{
    var mi = pMethod.GetMethodInfo();
    const bool includeInherited = false;
    var atr = mi.GetCustomAttributes(typeof(AuthorizeAttribute), includeInherited)
                .Select(t => (AuthorizeAttribute)t)
                .Where(t => pRoleAccess.Length>0?t.Roles == pRoleAccess:true);
    if (pRoleAccess == String.Empty)
    {
        return !atr.Any();
    }
    else
    {
        return atr.Any();
    }
}

public static bool MethodHasAllowAnonymousAttribute(this Delegate pMethod)
{
    var mi = pMethod.GetMethodInfo();
    const bool includeInherited = false;
    var atr = mi.GetCustomAttributes(typeof(AllowAnonymousAttribute), includeInherited);
    return atr.Any();
}

调用它如下

Func<string,System.Web.Mvc.ActionResult> func = controller.Login;
bool atrAuthorize = func.MethodHasAuthorizeAttribute(String.Empty);

Answer 4

如果您使用 FluentAssertions 您可以执行以下操作：

var classInstance = new MyClass();
Func<ActionResult> method1 = classInstance.Method1;
method1.GetMethodInfo().Should().BeDecoratedWith<AuthorizeAttribute>();

Func<ActionResult> method2 = classInstance.Method2;
method2.GetMethodInfo().Should().BeDecoratedWith<AuthorizeAttribute>();

Func<int, ActionResult> method3 = classInstance.Method3;
method3.GetMethodInfo().Should().BeDecoratedWith<AuthorizeAttribute>();

Answer 5

查找一些示例，在其中我找到应用了指定属性的类中的方法。

private static void GetMethodInfo(object className)
        {
            var methods = className.GetType().GetMethods(BindingFlags.Instance | BindingFlags.Public);

            foreach(var m in methods)
            {
                var parameters = m.GetParameters();
                var att = m.GetCustomAttributes(typeof (CustomAttribute), true);
            }
        }

传递的参数是类的实例。您可以修改代码以满足您的要求，这应该很容易。