如何通过LINQ查询EF DBSet仅返回携带指定自定义特性的属性

你可以通过反射搭配缓存的方式实现该扩展方法，同时支持内存集合（IEnumerable）和EF查询（IQueryable，可直接在数据库层面过滤字段，避免全表字段加载），完整实现如下：

1. 先定义自定义特性 #

// 导出标记特性
[AttributeUsage(AttributeTargets.Property)]
public class ExportAttribute : Attribute
{
}

// 禁止导出标记特性
[AttributeUsage(AttributeTargets.Property)]
public class NeverExportAttribute : Attribute
{
}

2. 实现IEnumerable扩展方法（适配内存集合） #

using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Dynamic;
using System.Reflection;
using System.Linq;

public static class AttributeSelectionExtensions
{
    // 缓存类型与对应特性的属性映射，避免重复反射
    private static readonly ConcurrentDictionary<(Type EntityType, Type AttributeType), PropertyInfo[]> _propertyCache = new();

    public static IEnumerable<object> GetByAttribute<T>(this IEnumerable<T> list, Type attributeType)
    {
        if (list == null) throw new ArgumentNullException(nameof(list));
        if (attributeType == null) throw new ArgumentNullException(nameof(attributeType));
        if (!typeof(Attribute).IsAssignableFrom(attributeType)) 
            throw new ArgumentException("传入的类型必须是特性类型", nameof(attributeType));

        // 从缓存读取目标属性，无缓存则反射生成
        var targetProperties = _propertyCache.GetOrAdd((typeof(T), attributeType), key =>
        {
            return key.EntityType.GetProperties(BindingFlags.Public | BindingFlags.Instance)
                .Where(prop => prop.IsDefined(key.AttributeType, inherit: true))
                .ToArray();
        });

        foreach (var item in list)
        {
            if (item == null)
            {
                yield return null;
                continue;
            }
            // 构建带指定属性的动态对象
            dynamic exportObj = new ExpandoObject();
            var propertyDict = (IDictionary<string, object>)exportObj;
            foreach (var prop in targetProperties)
            {
                propertyDict[prop.Name] = prop.GetValue(item);
            }
            yield return exportObj;
        }
    }
}

3. 调用示例 #

var people = PersonRepository.GetAll().ToList(); // 数据已加载到内存
// 筛选所有带Export特性的属性
var exportResults = people.GetByAttribute(typeof(ExportAttribute));

4. EF场景优化（可选，数据库层面过滤字段） #

如果你是直接对EF的DBSet（IQueryable类型）调用该方法，建议额外实现IQueryable版本的扩展，可以让EF生成的SQL只查询需要的字段，避免全表字段加载，大幅提升大数据量导出的性能：

public static IQueryable<Dictionary<string, object>> GetByAttributeQueryable<T>(this IQueryable<T> query, Type attributeType)
{
    if (query == null) throw new ArgumentNullException(nameof(query));
    if (attributeType == null) throw new ArgumentNullException(nameof(attributeType));
    if (!typeof(Attribute).IsAssignableFrom(attributeType))
        throw new ArgumentException("传入的类型必须是特性类型", nameof(attributeType));

    var targetProperties = _propertyCache.GetOrAdd((typeof(T), attributeType), key =>
    {
        return key.EntityType.GetProperties(BindingFlags.Public | BindingFlags.Instance)
            .Where(prop => prop.IsDefined(key.AttributeType, inherit: true))
            .ToArray();
    });

    // 动态构建Select表达式，返回字典类型
    var parameter = Expression.Parameter(typeof(T), "e");
    var addMethod = typeof(Dictionary<string, object>).GetMethod("Add", new[] { typeof(string), typeof(object) });
    var elementInitExpressions = targetProperties.Select(prop =>
        Expression.ElementInit(addMethod,
            Expression.Constant(prop.Name),
            Expression.Convert(Expression.Property(parameter, prop), typeof(object))
        ));
    var newDictExpr = Expression.New(typeof(Dictionary<string, object>));
    var memberInitExpr = Expression.ListInit(newDictExpr, elementInitExpressions);
    var selector = Expression.Lambda<Func<T, Dictionary<string, object>>>(memberInitExpr, parameter);

    return query.Select(selector);
}

调用方式和IEnumerable版本一致，只需要直接传入DBSet即可，不需要先加载所有数据到内存：

var exportResults = dbContext.Persons.GetByAttributeQueryable(typeof(ExportAttribute)).ToList();

注意事项 #

• 返回的动态对象/字典可以直接适配绝大多数Excel、CSV导出库，不需要额外转换
• 内置的缓存机制已经把反射的性能损耗降到最低，适合高频导出场景使用
• 不要对IQueryable直接调用IEnumerable版本的扩展，否则会触发全表数据加载，数据量大时会出现性能问题

内容的提问来源于stack exchange，提问作者Red_Phoenix