LINQ表达式返回属性值?

23

我正在尝试创建一个通用函数,帮助我使用LINQ to SQL从本地列表中选择数千条记录。SQL Server(至少是2005版)限制查询到2100个参数,而我希望选择更多的记录。

以下是一个很好的使用示例:

var some_product_numbers = new int[] { 1,2,3 ... 9999 };

Products.SelectByParameterList(some_product_numbers, p => p.ProductNumber);

以下是我的(不可用的)实现:

public static IEnumerable<T> SelectByParameterList<T, PropertyType>(Table<T> items, 

IEnumerable<PropertyType> parameterList, Expression<Func<T, PropertyType>> property) where T : class
{
    var groups = parameterList
        .Select((Parameter, index) =>
            new
            {
                GroupID = index / 2000, //2000 parameters per request
                Parameter
            }
        )
        .GroupBy(x => x.GroupID)
        .AsEnumerable();

    var results = groups
    .Select(g => new { Group = g, Parameters = g.Select(x => x.Parameter) } )
    .SelectMany(g => 
        /* THIS PART FAILS MISERABLY */
        items.Where(item => g.Parameters.Contains(property.Compile()(item)))
    );

    return results;
}

我看到了很多使用表达式构建谓词的例子。在这种情况下,我只想执行委托以返回当前ProductNumber的值。或者说,我想将其转换为SQL查询(非泛型形式可以正常工作)。
我知道编译Expression只会让我回到原点(将委托作为Func传递),但我不确定如何向“未编译”的表达式传递参数。
感谢您的帮助!
****编辑:** 让我进一步澄清:
以下是我想要概括的工作示例:
var local_refill_ids = Refills.Select(r => r.Id).Take(20).ToArray();

var groups = local_refill_ids
    .Select((Parameter, index) =>
        new
        {
            GroupID = index / 5, //5 parameters per request
            Parameter
        }
    )
    .GroupBy(x => x.GroupID)
    .AsEnumerable();

var results = groups
.Select(g => new { Group = g, Parameters = g.Select(x => x.Parameter) } )
.SelectMany(g => 
    Refills.Where(r => g.Parameters.Contains(r.Id))
)
.ToArray()
;

这段SQL代码的结果:

SELECT [t0].[Id], ... [t0].[Version]
FROM [Refill] AS [t0]
WHERE [t0].[Id] IN (@p0, @p1, @p2, @p3, @p4)

... That query 4 more times (20 / 5 = 4)
5个回答

46

我想到了一种将查询分成多个部分的方法——即给它4000个值,因此它可能会执行4个请求,每个请求1000个值;同时提供了完整的Northwind示例。请注意,这种方法可能无法在Entity Framework上工作,因为涉及到Expression.Invoke,但在LINQ to SQL上是可行的。

using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Reflection;

namespace ConsoleApplication5 {
    /// SAMPLE USAGE
    class Program {
        static void Main(string[] args) {
            // get some ids to play with...
            string[] ids;
            using(var ctx = new DataClasses1DataContext()) {
                ids = ctx.Customers.Select(x => x.CustomerID)
                    .Take(100).ToArray();
            }

            // now do our fun select - using a deliberately small
            // batch size to prove it...
            using (var ctx = new DataClasses1DataContext()) {
                ctx.Log = Console.Out;
                foreach(var cust in ctx.Customers
                        .InRange(x => x.CustomerID, 5, ids)) {
                    Console.WriteLine(cust.CompanyName);
                }
            }
        }
    }

    /// THIS IS THE INTERESTING BIT
    public static class QueryableChunked {
        public static IEnumerable<T> InRange<T, TValue>(
                this IQueryable<T> source,
                Expression<Func<T, TValue>> selector,
                int blockSize,
                IEnumerable<TValue> values) {
            MethodInfo method = null;
            foreach(MethodInfo tmp in typeof(Enumerable).GetMethods(
                    BindingFlags.Public | BindingFlags.Static)) {
                if(tmp.Name == "Contains" && tmp.IsGenericMethodDefinition
                        && tmp.GetParameters().Length == 2) {
                    method = tmp.MakeGenericMethod(typeof (TValue));
                    break;
                }
            }
            if(method==null) throw new InvalidOperationException(
                "Unable to locate Contains");
            foreach(TValue[] block in values.GetBlocks(blockSize)) {
                var row = Expression.Parameter(typeof (T), "row");
                var member = Expression.Invoke(selector, row);
                var keys = Expression.Constant(block, typeof (TValue[]));
                var predicate = Expression.Call(method, keys, member);
                var lambda = Expression.Lambda<Func<T,bool>>(
                      predicate, row);
                foreach(T record in source.Where(lambda)) {
                    yield return record;
                }
            }
        }
        public static IEnumerable<T[]> GetBlocks<T>(
                this IEnumerable<T> source, int blockSize) {
            List<T> list = new List<T>(blockSize);
            foreach(T item in source) {
                list.Add(item);
                if(list.Count == blockSize) {
                    yield return list.ToArray();
                    list.Clear();
                }
            }
            if(list.Count > 0) {
                yield return list.ToArray();
            }
        }
    }
}

2
这段代码处理了 queryable.Where(o => values.Contains(o.propertyToTest)) 这种情况,通过将其替换为 queryable.InRange(o => o.propertyToTest, blockSize, values)(如果我理解正确的话)。但我在关注类似于2100个参数限制的溢出问题,例如 queryable.Where(o => !values.Contains(o.propertyToTest))。我正在尝试修改InRange()函数以获得NotInRange()等效函数,并且我不确定如何进行布尔否定。我的想法是在 foreach (T record in source.Where(lambda)) 的一行里进行? - Matt Sach
2
实际上,在经过多次搜索后,我认为我已经找到了所需的内容,而且是从您一个月前回答此问题的答案中得到的:https://dev59.com/7XRB5IYBdhLWcg3w9Lvn,参考“这也适用于否定单个操作:” - Matt Sach
1
@Marc - 在VB中如何处理这个问题?显然,我们没有yield return - Code Maverick
批量大小可配置的原因是什么?我不是总希望它像2090这样吗?这样我就可以在2100限制以下留有一些余地了。 - Dilbert789
不必使用Invoke。您可以重复使用选择器lambda和参数,尽管我承认这有点不干净。避免调用还将使其可在其他ORM上使用。 (我使用以下内容使其在LLBLGen Pro上工作:https://gist.github.com/FransBouma/5e7031fe557df4b5b688) - Frans Bouma

9
最简单的方法是:使用LINQKit(免费,非限制性许可证)。
代码的工作版本:
public static IEnumerable<T> SelectByParameterList<T, PropertyType>(this Table<T> items, IEnumerable<PropertyType> parameterList, Expression<Func<T, PropertyType>> propertySelector, int blockSize) where T : class
{
    var groups = parameterList
        .Select((Parameter, index) =>
            new
            {
                GroupID = index / blockSize, //# of parameters per request
                Parameter
            }
        )
        .GroupBy(x => x.GroupID)
        .AsEnumerable();

    var selector = LinqKit.Linq.Expr(propertySelector);

    var results = groups
    .Select(g => new { Group = g, Parameters = g.Select(x => x.Parameter) } )
    .SelectMany(g => 
        /* AsExpandable() extension method requires LinqKit DLL */
        items.AsExpandable().Where(item => g.Parameters.Contains(selector.Invoke(item)))
    );

    return results;
}

使用示例:

    Guid[] local_refill_ids = Refills.Select(r => r.Id).Take(20).ToArray();

    IEnumerable<Refill> results = Refills.SelectByParameterList(local_refill_ids, r => r.Id, 10); //runs 2 SQL queries with 10 parameters each

再次感谢您的帮助!


1
我对那个TSQL的操作很感兴趣,想看看它与我的InRange回复有何不同... - Marc Gravell
SELECT [t0].[Id], ... [t0].[Version] FROM [Refill] AS [t0] WHERE [t0].[Id] IN (@p0, @p1, @p2, @p3, @p4, @p5, @p6, @p7, @p8, @p9)... 该查询被执行了两次(20/10=2)。 - kwcto
您对优化使用LinqToSql的查询而言,关于blockSize有何建议?换句话说,是减少查询数量并增加块大小更好,还是增加查询数量并减少块大小更好? - ni5ni6
2
通常情况下,您希望尽可能少地使用块。主要问题是SQL Server每个查询/命令只允许2100个参数。我通常会使用2000的块大小。 - kwcto

3
LINQ-to-SQL仍然通过标准SQL参数工作,因此编写花哨的表达式并不能帮助解决问题。这里有三个常见的选择:
  • 将id打包成(例如)csv/tsv;将其作为varchar(max)传递并使用UDF(在服务器上)将其拆分成表变量;连接到表变量
  • 在SQL Server 2008中使用表值参数
  • 在服务器上有一个表可以将id推入其中(可能通过SqlBulkCopy)(可能带有类似的“会话guid”);连接到此表
第一种方法最简单;获取“split csv udf”很容易(只需搜索)。将udf拖放到数据上下文并从那里使用。

2
这不是必要的。请看下面我的回答。 - kwcto

0

你可以创建自己的查询提供程序

public class QueryProvider : IQueryProvider
{
    // Translates LINQ query to SQL.
    private readonly Func<IQueryable, DbCommand> _translator;

    // Executes the translated SQL and retrieves results.
    private readonly Func<Type, string, object[], IEnumerable> _executor;

    public QueryProvider(
        Func<IQueryable, DbCommand> translator,
        Func<Type, string, object[], IEnumerable> executor)
    {

        this._translator = translator;
        this._executor = executor;
    }

    #region IQueryProvider Members

    public IQueryable<TElement> CreateQuery<TElement>(Expression expression)
    {
        return new Queryable<TElement>(this, expression);
    }

    public IQueryable CreateQuery(Expression expression)
    {
        throw new NotImplementedException();
    }

    public TResult Execute<TResult>(Expression expression)
    {
        bool isCollection = typeof(TResult).IsGenericType &&
            typeof(TResult).GetGenericTypeDefinition() == typeof(IEnumerable<>);
        var itemType = isCollection
            // TResult is an IEnumerable`1 collection.
            ? typeof(TResult).GetGenericArguments().Single()
            // TResult is not an IEnumerable`1 collection, but a single item.
            : typeof(TResult);
        var queryable = Activator.CreateInstance(
            typeof(Queryable<>).MakeGenericType(itemType), this, expression) as IQueryable;

        IEnumerable queryResult;

        // Translates LINQ query to SQL.
        using (var command = this._translator(queryable))
        {
            var parameters = command.Parameters.OfType<DbParameter>()
                .Select(parameter => parameter)
                .ToList();

            var query = command.CommandText;
            var newParameters = GetNewParameterList(ref query, parameters);

            queryResult = _executor(itemType,query,newParameters);
        }

        return isCollection
            ? (TResult)queryResult // Returns an IEnumerable`1 collection.
            : queryResult.OfType<TResult>()
                         .SingleOrDefault(); // Returns a single item.
    }       

    public object Execute(Expression expression)
    {
        throw new NotImplementedException();
    }

    #endregion

     private static object[] GetNewParameterList(ref string query, List<DbParameter> parameters)
    {
        var newParameters = new List<DbParameter>(parameters);

        foreach (var dbParameter in parameters.Where(p => p.DbType == System.Data.DbType.Int32))
        {
            var name = dbParameter.ParameterName;
            var value = dbParameter.Value != null ? dbParameter.Value.ToString() : "NULL";
            var pattern = String.Format("{0}[^0-9]", dbParameter.ParameterName);
            query = Regex.Replace(query, pattern, match => value + match.Value.Replace(name, ""));
            newParameters.Remove(dbParameter);
        }

        for (var i = 0; i < newParameters.Count; i++)
        {
            var parameter = newParameters[i];
            var oldName = parameter.ParameterName;
            var pattern = String.Format("{0}[^0-9]", oldName);
            var newName = "@p" + i;
            query = Regex.Replace(query, pattern, match => newName + match.Value.Replace(oldName, ""));
        }      

        return newParameters.Select(x => x.Value).ToArray();
    }
}


    static void Main(string[] args)
    {
        using (var dc=new DataContext())
        {
            var provider = new QueryProvider(dc.GetCommand, dc.ExecuteQuery);

            var serviceIds = Enumerable.Range(1, 2200).ToArray();

            var tasks = new Queryable<Task>(provider, dc.Tasks).Where(x => serviceIds.Contains(x.ServiceId) && x.CreatorId==37 && x.Creator.Name=="12312").ToArray();

        }

    }

0
请将 IQuerable 传递给 Contains 函数,而不是列表或数组。请参考以下示例:
var df_handsets = db.DataFeed_Handsets.Where(m => m.LaunchDate != null).
                  Select(m => m.Name);
var Make = (from m in db.MobilePhones
    where (m.IsDeleted != true || m.IsDeleted == null)
        && df_handsets.Contains(m.Name)
    orderby m.Make
    select new { Value = m.Make, Text = m.Make }).Distinct();

当您传递列表或数组时,它以参数的形式传递,并且当列表项计数大于2100时,其超出计数。

2
你假设要匹配的集合是直接来自数据库的。但这并不总是正确的。 - kwcto

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接