// // QueryOrderByNode.cs // // Author: // Jérémie "Garuma" Laval // // Copyright (c) 2010 Jérémie "Garuma" Laval // // Permission is hereby granted, free of charge, to any person obtaining a copy // of this software and associated documentation files (the "Software"), to deal // in the Software without restriction, including without limitation the rights // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell // copies of the Software, and to permit persons to whom the Software is // furnished to do so, subject to the following conditions: // // The above copyright notice and this permission notice shall be included in // all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN // THE SOFTWARE. #if NET_4_0 using System; using System.Threading; using System.Collections; using System.Collections.Generic; using System.Collections.Concurrent; namespace System.Linq.Parallel.QueryNodes { internal class QueryGroupByNode : QueryStreamNode, TSource> { Func keySelector; Func elementSelector; IEqualityComparer comparer; public QueryGroupByNode (QueryBaseNode parent, Func keySelector, Func elementSelector, IEqualityComparer comparer) : base (parent, false) { this.keySelector = keySelector; this.elementSelector = elementSelector; this.comparer = comparer; } internal override IEnumerable> GetSequential () { IEnumerable src = Parent.GetSequential (); return src.GroupBy (keySelector, elementSelector, comparer); } internal override IList>> GetEnumerables (QueryOptions options) { return ParallelPartitioner.CreateForChunks (GetGroupedElements ()).GetPartitions (options.PartitionCount).Wrap (); } internal override IList>>> GetOrderedEnumerables (QueryOptions options) { return ParallelPartitioner.CreateForChunks (GetGroupedElements ()).GetOrderablePartitions (options.PartitionCount).Wrap (); } internal IEnumerable> GetGroupedElements () { return GetStore ().Select (e => (IGrouping)new ConcurrentGrouping (e.Key, e.Value)); } internal ConcurrentDictionary> GetStore () { var store = new ConcurrentDictionary> (); Func> queueFactory = (_) => new ConcurrentQueue (); ParallelExecuter.ProcessAndBlock (Parent, (e, c) => { ConcurrentQueue queue = store.GetOrAdd (keySelector (e), queueFactory); queue.Enqueue (elementSelector (e)); }); return store; } } } #endif