我可以改进这些分页扩展方法吗?

时间:2010-10-27 05:33:17

标签: c# .net linq

我刚刚编写了一些分页扩展方法,我很想知道我是否有任何改进。

我对基本分页方法非常满意,您提供页面大小和页码(如下所示)

    public static IEnumerable<T> Paginate<T>(this IEnumerable<T> source, int pageSize, int pageNumber)
    {
        if (pageSize == 0) throw new ArgumentOutOfRangeException("pageSize");
        if (pageNumber == 0) throw new ArgumentOutOfRangeException("pageNumber");

        return source.Skip(pageSize * (pageNumber - 1)).Take(pageSize);
    }

但是我想知道是否有更好的方法来执行“自动”分页,它会返回IEnumerable<IEnumerable<T>>

    public static IEnumerable<IEnumerable<T>> Paginate<T>(this IEnumerable<T> source, int pageSize)
    {
        source.ThrowIfNull("source");
        if (pageSize == 0) throw new ArgumentOutOfRangeException("pageSize");

        var pageCount = (int)Math.Ceiling(source.Count() / (double)pageSize);

        if (pageSize == 1)
            pageCount = source.Count();

        for (int i = 1; i <= pageCount; i++)
        {
            yield return source.Paginate(pageSize, i);
        }
    }

似乎有点怀疑必须迭代两次(一次用于计数,一次用于收益率返回。

有没有明显的方法可以改进这些方法?

1 个答案:

答案 0 :(得分:3)

查看MoreLinq Batch: - http://code.google.com/p/morelinq/source/browse/trunk/MoreLinq/Batch.cs?r=84

实现方式如下:

public static IEnumerable<IEnumerable<TSource>> Batch<TSource>(
              this IEnumerable<TSource> source, int size)
{
  TSource[] bucket = null;
  var count = 0;

  foreach (var item in source)
  {
      if (bucket == null)
          bucket = new TSource[size];

      bucket[count++] = item;
      if (count != size)
          continue;

      yield return bucket;

      bucket = null;
      count = 0;
  }

  if (bucket != null && count > 0)
      yield return bucket.Take(count);
}