扫描所有类和方法以获取自定义属性的最佳实践

Question

我第一次真正需要自己手动进行装配扫描。我遇到了 C# - 如何使用自定义类属性枚举所有类？ 为我设置了

var typesWithMyAttribute =
(from assembly in AppDomain.CurrentDomain.GetAssemblies()
    from type in assembly.GetTypes()
    let attributes = type.GetCustomAttributes(typeof(SomeAttribute), true)
    where attributes != null && attributes.Length > 0
    select new { Type = type, Attributes = attributes.Cast<SomeAttribute>() })
    .ToList();

Which is simple enough to spread out to the method level

var methodsWithAttributes =
    (from assembly in AppDomain.CurrentDomain.GetAssemblies()
    from type in assembly.GetTypes()
    from method in type.GetMethods()
    let attributes = method.GetCustomAttributes(typeof(SomeAttribute), true)
    where attributes != null && attributes.Length > 0
    select new { Type = type, Method = method, 
            Attributes = attributes.Cast<SomeAttribute>() })
    .ToList();

我应该尝试将这两个结合起来在一次扫描中完成此操作，还是只是陷入早期优化？ （扫描仅在应用程序启动时执行）

是否有一些不同的方法可以更优化地扫描方法，因为程序集中的方法比类型多得多？

Answer 1

反思非常慢...

我认为你已经掌握了基础知识。我建议您稍微更改代码以避免发生额外的完整扫描。

如果您必须多次执行此操作，我还建议您考虑在适当的时间段内缓存结果。

像这样的伪代码：

... (optional caches) ...
IDictionary<Type, IEnumerable<Attributes>> typeAttributeCache = new ...
IDictionary<MethodInfo, IEnumerable<Attributes>> methodAttributeCache = new ...

... (in another method or class) ...
foreach assembly in GetAssemblies()
  foreach type in assembly.GetTypes()        
    typeAttributes = typeAttributeCache.TryGet(...) // you know the correct syntax, trying to be brief

    if (typeAttributes is null)
      typeAttributes = type.GetCustomAttributes().OfType<TypeImLookingFor>();
      typeAttributeCache[type] = typeAttributes;

    foreach methodInfo in type.GetMethods()        
      methodAttributes = methodAttributeCache.TryGet(...) // same as above

      if (methodAttributes is null)
        methodAttributes = methodInfo.GetCustomAttributes().OfType<TypeImLookingFor>();
        methodAttributeCache[type] = methodAttributes;

    // do what you need to do

Answer 2

我认为您可以对此进行优化，但这取决于属性如何放置在方法和类型上。如果您知道所有具有特殊属性的类型和/或方法都是在特定程序集中定义的，则可以仅扫描这些程序集。

您还可以定义一些方法，例如：

 - IEnumerable<Type> GetAllTypesFromAssemblyByAttribute<TAttribute>(Assembly assembly) where TAttribute : Attribute
 - IEnumerable<MethodInfo> GetAllMethodsFromTypeByAttribute<TAttribute>(Type type) where TAttribute : Attribute

并在主扫描方法中使用这些方法。

所以你的结果扫描方法可能如下所示：

private void ScanAndDoSmth<TAttribute>(IEnumerable<Assembly> assemblies)
where TAttribute : Attribute
{
    var result =
        from assembly in assemblies
        from type in GetAllTypesFromAssemblyByAttribute<TAttribute>(assembly)
        let attributes = type.GetCustomAttributes(typeof(TAttribute), true)
        where attributes != null && attributes.Length > 0
        select new { Type = type, Attributes = attributes.Cast<TAttribute>();
}