用于检查属性的 C# 扩展方法

Question

抱歉，如果这是一个愚蠢的菜鸟问题，请对我温柔一点，我正在努力学习......

我想测试模型和控制器等事物的属性方法。主要是为了确保它们具有正确的属性，即必需的。但我也将其用作扩展方法和 Lambda 的实验。

我想要的是一种方法，在实现时看起来有点像

Controller controller = new Controller();
controller.MethodName(params).HasAttribute<AttributeName>();

我使用过的扩展方法，但没有达到这种程度。我确信这应该足够简单，但似乎无法让我的泛型等正确。

Answer 1

也许您正在寻找这个：

Controller controller = new Controller();
bool ok = controller.GetMethod(c => c.MethodName(null, null))
    .HasAttribute<AttributeName>();

像这样编写它的好处是您拥有完全的编译时支持。到目前为止，所有其他解决方案都使用字符串文字来定义方法。

以下是 GetMethod 和 HasAttribute 扩展方法的实现：

public static MethodInfo GetMethod<T>(this T instance,
    Expression<Func<T, object>> methodSelector)
{
    // Note: this is a bit simplistic implementation. It will
    // not work for all expressions.
    return ((MethodCallExpression)methodSelector.Body).Method;
}

public static MethodInfo GetMethod<T>(this T instance,
    Expression<Action<T>> methodSelector)
{
    return ((MethodCallExpression)methodSelector.Body).Method;
}

public static bool HasAttribute<TAttribute>(
    this MemberInfo member) 
    where TAttribute : Attribute
{
    return GetAttributes<TAttribute>(member).Length > 0;
}

public static TAttribute[] GetAttributes<TAttribute>(
    this MemberInfo member) 
    where TAttribute : Attribute
{
    var attributes = 
        member.GetCustomAttributes(typeof(TAttribute), true);

    return (TAttribute[])attributes;
}

Answer 2

我认为您无法完全按照您所描述的那样实施。语句的 MethodName(params) 部分将实际执行该方法，返回该方法返回的任何内容，而不是有关该方法的信息。

您想要做的是使用反射将 MethodInfo 传递到您的扩展类中。因此，您可能会得到类似以下内容的结果，而不是您的示例：

 controller.GetType().GetMethod(methodName).HasAttribute<AttributeName>();

您可以通过封装 GetType().GetMethod() 将其简化为控制器类上的单个扩展方法，如下所示：

 controller.MethodHasAttribute(methodName, attributeName);

Answer 3

使用扩展方法无法完全做到这一点，但这已经很接近了：

public static class Program
{
  static void Main(string[] args)
  {
     Controller c1 = new Controller();
     Action a1 = c1.Method1;
     Console.WriteLine(a1.HasAttribute<Controller.TestAttribute>());
  }

  public static bool HasAttribute<T>(this Action method)
  {
     return method.Method.GetCustomAttributes(typeof(T), false).Any();
  }
}

class Controller
{
  [AttributeUsage(AttributeTargets.Method)]
  public class TestAttribute : System.Attribute
  {
  }

  [Test()]
  public void Method1()
  {
  }
}

Answer 4

用法：

bool hasAttribute = controller.HasMethodAttribute<TestAttribute>( "Test" )

扩展：

public static bool HasMethodAttribute<TAttribute>( this object obj, string methodName )
{
    Type type = obj.GetType();

    MethodInfo method = type.GetMethod( methodName );
    if( method == null )
    {
        throw new ArgumentException( string.Format( 
            "Method '{0}' not found on object '{1}'", methodName, type.Name ) );
    }

    return method.GetCustomAttributes( typeof( TAttribute ), true ).Length > 0 ;
} 

Answer 5

您可以通过执行以下操作来检查方法是否具有特定属性：

typeof(Controller).GetMethod("MethodName").GetAttributes<AttributeName>().Any();

就扩展方法本身而言，假设您正在寻找 Controller 类型上的扩展方法，那么像这样：

public static bool HasAttribute<A>(this Controller controller, string methodName)
{
   return controller.GetType().GetMethod(methodName).GetCustomAttributes(typeof(A), true).Any();
}

记住，扩展方法被调用于一个实例，因此要使用它，您仍然需要一个 Controller 实例：

var controller = new Controller();
Assert.IsTrue(controller.HasAttribute<AttributeName>("Method1"));

Answer 6

您正在寻找 Reflection 类 -它允许您检查传递的对象。

但坦率地说，除了学习练习之外，如果您希望模型中有一个名为“Required”的字段，或者在控制器中有一个名为“IsRequired”的方法，而不是在接口中实现并要求实现该接口，那么这就是接口存在的目的。

具体属性请参见此处：

Type t = something;
System.Console.WriteLine("Author information for {0}", t);
System.Attribute[] attrs = System.Attribute.GetCustomAttributes(t);  // reflection

foreach (System.Attribute attr in attrs)
{
    if (attr is Author)
    {
        Author a = (Author)attr;
            System.Console.WriteLine("   {0}, version {1:f}", a.GetName(), a.version);
    }
}