author | psandoz |
Thu, 21 Dec 2017 13:52:20 -0800 | |
changeset 48593 | fca88bbbafb9 |
parent 47216 | 71c04702a3d5 |
permissions | -rw-r--r-- |
17182 | 1 |
/* |
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
2 |
* Copyright (c) 2012, 2017, Oracle and/or its affiliates. All rights reserved. |
17182 | 3 |
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
4 |
* |
|
5 |
* This code is free software; you can redistribute it and/or modify it |
|
6 |
* under the terms of the GNU General Public License version 2 only, as |
|
7 |
* published by the Free Software Foundation. Oracle designates this |
|
8 |
* particular file as subject to the "Classpath" exception as provided |
|
9 |
* by Oracle in the LICENSE file that accompanied this code. |
|
10 |
* |
|
11 |
* This code is distributed in the hope that it will be useful, but WITHOUT |
|
12 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
13 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
14 |
* version 2 for more details (a copy is included in the LICENSE file that |
|
15 |
* accompanied this code). |
|
16 |
* |
|
17 |
* You should have received a copy of the GNU General Public License version |
|
18 |
* 2 along with this work; if not, write to the Free Software Foundation, |
|
19 |
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
|
20 |
* |
|
21 |
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
|
22 |
* or visit www.oracle.com if you need additional information or have any |
|
23 |
* questions. |
|
24 |
*/ |
|
25 |
package java.util.stream; |
|
26 |
||
27 |
import java.util.ArrayList; |
|
28 |
import java.util.Arrays; |
|
29 |
import java.util.Comparator; |
|
30 |
import java.util.Objects; |
|
31 |
import java.util.Spliterator; |
|
32 |
import java.util.function.IntFunction; |
|
33 |
||
34 |
||
35 |
/** |
|
36 |
* Factory methods for transforming streams into sorted streams. |
|
37 |
* |
|
38 |
* @since 1.8 |
|
39 |
*/ |
|
40 |
final class SortedOps { |
|
41 |
||
42 |
private SortedOps() { } |
|
43 |
||
44 |
/** |
|
45 |
* Appends a "sorted" operation to the provided stream. |
|
46 |
* |
|
47 |
* @param <T> the type of both input and output elements |
|
48 |
* @param upstream a reference stream with element type T |
|
49 |
*/ |
|
50 |
static <T> Stream<T> makeRef(AbstractPipeline<?, T, ?> upstream) { |
|
51 |
return new OfRef<>(upstream); |
|
52 |
} |
|
53 |
||
54 |
/** |
|
55 |
* Appends a "sorted" operation to the provided stream. |
|
56 |
* |
|
57 |
* @param <T> the type of both input and output elements |
|
58 |
* @param upstream a reference stream with element type T |
|
59 |
* @param comparator the comparator to order elements by |
|
60 |
*/ |
|
61 |
static <T> Stream<T> makeRef(AbstractPipeline<?, T, ?> upstream, |
|
62 |
Comparator<? super T> comparator) { |
|
63 |
return new OfRef<>(upstream, comparator); |
|
64 |
} |
|
65 |
||
66 |
/** |
|
67 |
* Appends a "sorted" operation to the provided stream. |
|
68 |
* |
|
69 |
* @param <T> the type of both input and output elements |
|
70 |
* @param upstream a reference stream with element type T |
|
71 |
*/ |
|
72 |
static <T> IntStream makeInt(AbstractPipeline<?, Integer, ?> upstream) { |
|
73 |
return new OfInt(upstream); |
|
74 |
} |
|
75 |
||
76 |
/** |
|
77 |
* Appends a "sorted" operation to the provided stream. |
|
78 |
* |
|
79 |
* @param <T> the type of both input and output elements |
|
80 |
* @param upstream a reference stream with element type T |
|
81 |
*/ |
|
82 |
static <T> LongStream makeLong(AbstractPipeline<?, Long, ?> upstream) { |
|
83 |
return new OfLong(upstream); |
|
84 |
} |
|
85 |
||
86 |
/** |
|
87 |
* Appends a "sorted" operation to the provided stream. |
|
88 |
* |
|
89 |
* @param <T> the type of both input and output elements |
|
90 |
* @param upstream a reference stream with element type T |
|
91 |
*/ |
|
92 |
static <T> DoubleStream makeDouble(AbstractPipeline<?, Double, ?> upstream) { |
|
93 |
return new OfDouble(upstream); |
|
94 |
} |
|
95 |
||
96 |
/** |
|
97 |
* Specialized subtype for sorting reference streams |
|
98 |
*/ |
|
99 |
private static final class OfRef<T> extends ReferencePipeline.StatefulOp<T, T> { |
|
100 |
/** |
|
101 |
* Comparator used for sorting |
|
102 |
*/ |
|
103 |
private final boolean isNaturalSort; |
|
104 |
private final Comparator<? super T> comparator; |
|
105 |
||
106 |
/** |
|
107 |
* Sort using natural order of {@literal <T>} which must be |
|
108 |
* {@code Comparable}. |
|
109 |
*/ |
|
110 |
OfRef(AbstractPipeline<?, T, ?> upstream) { |
|
111 |
super(upstream, StreamShape.REFERENCE, |
|
112 |
StreamOpFlag.IS_ORDERED | StreamOpFlag.IS_SORTED); |
|
113 |
this.isNaturalSort = true; |
|
114 |
// Will throw CCE when we try to sort if T is not Comparable |
|
22297 | 115 |
@SuppressWarnings("unchecked") |
116 |
Comparator<? super T> comp = (Comparator<? super T>) Comparator.naturalOrder(); |
|
117 |
this.comparator = comp; |
|
17182 | 118 |
} |
119 |
||
120 |
/** |
|
121 |
* Sort using the provided comparator. |
|
122 |
* |
|
123 |
* @param comparator The comparator to be used to evaluate ordering. |
|
124 |
*/ |
|
125 |
OfRef(AbstractPipeline<?, T, ?> upstream, Comparator<? super T> comparator) { |
|
126 |
super(upstream, StreamShape.REFERENCE, |
|
127 |
StreamOpFlag.IS_ORDERED | StreamOpFlag.NOT_SORTED); |
|
128 |
this.isNaturalSort = false; |
|
129 |
this.comparator = Objects.requireNonNull(comparator); |
|
130 |
} |
|
131 |
||
132 |
@Override |
|
19593 | 133 |
public Sink<T> opWrapSink(int flags, Sink<T> sink) { |
17182 | 134 |
Objects.requireNonNull(sink); |
135 |
||
136 |
// If the input is already naturally sorted and this operation |
|
137 |
// also naturally sorted then this is a no-op |
|
138 |
if (StreamOpFlag.SORTED.isKnown(flags) && isNaturalSort) |
|
139 |
return sink; |
|
140 |
else if (StreamOpFlag.SIZED.isKnown(flags)) |
|
141 |
return new SizedRefSortingSink<>(sink, comparator); |
|
142 |
else |
|
143 |
return new RefSortingSink<>(sink, comparator); |
|
144 |
} |
|
145 |
||
146 |
@Override |
|
147 |
public <P_IN> Node<T> opEvaluateParallel(PipelineHelper<T> helper, |
|
148 |
Spliterator<P_IN> spliterator, |
|
149 |
IntFunction<T[]> generator) { |
|
150 |
// If the input is already naturally sorted and this operation |
|
151 |
// naturally sorts then collect the output |
|
152 |
if (StreamOpFlag.SORTED.isKnown(helper.getStreamAndOpFlags()) && isNaturalSort) { |
|
153 |
return helper.evaluate(spliterator, false, generator); |
|
154 |
} |
|
155 |
else { |
|
156 |
// @@@ Weak two-pass parallel implementation; parallel collect, parallel sort |
|
157 |
T[] flattenedData = helper.evaluate(spliterator, true, generator).asArray(generator); |
|
158 |
Arrays.parallelSort(flattenedData, comparator); |
|
159 |
return Nodes.node(flattenedData); |
|
160 |
} |
|
161 |
} |
|
162 |
} |
|
163 |
||
164 |
/** |
|
165 |
* Specialized subtype for sorting int streams. |
|
166 |
*/ |
|
167 |
private static final class OfInt extends IntPipeline.StatefulOp<Integer> { |
|
168 |
OfInt(AbstractPipeline<?, Integer, ?> upstream) { |
|
169 |
super(upstream, StreamShape.INT_VALUE, |
|
170 |
StreamOpFlag.IS_ORDERED | StreamOpFlag.IS_SORTED); |
|
171 |
} |
|
172 |
||
173 |
@Override |
|
22297 | 174 |
public Sink<Integer> opWrapSink(int flags, Sink<Integer> sink) { |
17182 | 175 |
Objects.requireNonNull(sink); |
176 |
||
177 |
if (StreamOpFlag.SORTED.isKnown(flags)) |
|
178 |
return sink; |
|
179 |
else if (StreamOpFlag.SIZED.isKnown(flags)) |
|
180 |
return new SizedIntSortingSink(sink); |
|
181 |
else |
|
182 |
return new IntSortingSink(sink); |
|
183 |
} |
|
184 |
||
185 |
@Override |
|
186 |
public <P_IN> Node<Integer> opEvaluateParallel(PipelineHelper<Integer> helper, |
|
187 |
Spliterator<P_IN> spliterator, |
|
188 |
IntFunction<Integer[]> generator) { |
|
189 |
if (StreamOpFlag.SORTED.isKnown(helper.getStreamAndOpFlags())) { |
|
190 |
return helper.evaluate(spliterator, false, generator); |
|
191 |
} |
|
192 |
else { |
|
193 |
Node.OfInt n = (Node.OfInt) helper.evaluate(spliterator, true, generator); |
|
194 |
||
18171 | 195 |
int[] content = n.asPrimitiveArray(); |
17182 | 196 |
Arrays.parallelSort(content); |
197 |
||
198 |
return Nodes.node(content); |
|
199 |
} |
|
200 |
} |
|
201 |
} |
|
202 |
||
203 |
/** |
|
204 |
* Specialized subtype for sorting long streams. |
|
205 |
*/ |
|
206 |
private static final class OfLong extends LongPipeline.StatefulOp<Long> { |
|
207 |
OfLong(AbstractPipeline<?, Long, ?> upstream) { |
|
208 |
super(upstream, StreamShape.LONG_VALUE, |
|
209 |
StreamOpFlag.IS_ORDERED | StreamOpFlag.IS_SORTED); |
|
210 |
} |
|
211 |
||
212 |
@Override |
|
19220
d3d40ccb544e
8022476: cleanup some raw types and unchecked warnings in java.util.stream
mduigou
parents:
18571
diff
changeset
|
213 |
public Sink<Long> opWrapSink(int flags, Sink<Long> sink) { |
17182 | 214 |
Objects.requireNonNull(sink); |
215 |
||
216 |
if (StreamOpFlag.SORTED.isKnown(flags)) |
|
217 |
return sink; |
|
218 |
else if (StreamOpFlag.SIZED.isKnown(flags)) |
|
219 |
return new SizedLongSortingSink(sink); |
|
220 |
else |
|
221 |
return new LongSortingSink(sink); |
|
222 |
} |
|
223 |
||
224 |
@Override |
|
225 |
public <P_IN> Node<Long> opEvaluateParallel(PipelineHelper<Long> helper, |
|
226 |
Spliterator<P_IN> spliterator, |
|
227 |
IntFunction<Long[]> generator) { |
|
228 |
if (StreamOpFlag.SORTED.isKnown(helper.getStreamAndOpFlags())) { |
|
229 |
return helper.evaluate(spliterator, false, generator); |
|
230 |
} |
|
231 |
else { |
|
232 |
Node.OfLong n = (Node.OfLong) helper.evaluate(spliterator, true, generator); |
|
233 |
||
18171 | 234 |
long[] content = n.asPrimitiveArray(); |
17182 | 235 |
Arrays.parallelSort(content); |
236 |
||
237 |
return Nodes.node(content); |
|
238 |
} |
|
239 |
} |
|
240 |
} |
|
241 |
||
242 |
/** |
|
243 |
* Specialized subtype for sorting double streams. |
|
244 |
*/ |
|
245 |
private static final class OfDouble extends DoublePipeline.StatefulOp<Double> { |
|
246 |
OfDouble(AbstractPipeline<?, Double, ?> upstream) { |
|
247 |
super(upstream, StreamShape.DOUBLE_VALUE, |
|
248 |
StreamOpFlag.IS_ORDERED | StreamOpFlag.IS_SORTED); |
|
249 |
} |
|
250 |
||
251 |
@Override |
|
19220
d3d40ccb544e
8022476: cleanup some raw types and unchecked warnings in java.util.stream
mduigou
parents:
18571
diff
changeset
|
252 |
public Sink<Double> opWrapSink(int flags, Sink<Double> sink) { |
17182 | 253 |
Objects.requireNonNull(sink); |
254 |
||
255 |
if (StreamOpFlag.SORTED.isKnown(flags)) |
|
256 |
return sink; |
|
257 |
else if (StreamOpFlag.SIZED.isKnown(flags)) |
|
258 |
return new SizedDoubleSortingSink(sink); |
|
259 |
else |
|
260 |
return new DoubleSortingSink(sink); |
|
261 |
} |
|
262 |
||
263 |
@Override |
|
264 |
public <P_IN> Node<Double> opEvaluateParallel(PipelineHelper<Double> helper, |
|
265 |
Spliterator<P_IN> spliterator, |
|
266 |
IntFunction<Double[]> generator) { |
|
267 |
if (StreamOpFlag.SORTED.isKnown(helper.getStreamAndOpFlags())) { |
|
268 |
return helper.evaluate(spliterator, false, generator); |
|
269 |
} |
|
270 |
else { |
|
271 |
Node.OfDouble n = (Node.OfDouble) helper.evaluate(spliterator, true, generator); |
|
272 |
||
18171 | 273 |
double[] content = n.asPrimitiveArray(); |
17182 | 274 |
Arrays.parallelSort(content); |
275 |
||
276 |
return Nodes.node(content); |
|
277 |
} |
|
278 |
} |
|
279 |
} |
|
280 |
||
281 |
/** |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
282 |
* Abstract {@link Sink} for implementing sort on reference streams. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
283 |
* |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
284 |
* <p> |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
285 |
* Note: documentation below applies to reference and all primitive sinks. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
286 |
* <p> |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
287 |
* Sorting sinks first accept all elements, buffering then into an array |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
288 |
* or a re-sizable data structure, if the size of the pipeline is known or |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
289 |
* unknown respectively. At the end of the sink protocol those elements are |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
290 |
* sorted and then pushed downstream. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
291 |
* This class records if {@link #cancellationRequested} is called. If so it |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
292 |
* can be inferred that the source pushing source elements into the pipeline |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
293 |
* knows that the pipeline is short-circuiting. In such cases sub-classes |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
294 |
* pushing elements downstream will preserve the short-circuiting protocol |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
295 |
* by calling {@code downstream.cancellationRequested()} and checking the |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
296 |
* result is {@code false} before an element is pushed. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
297 |
* <p> |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
298 |
* Note that the above behaviour is an optimization for sorting with |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
299 |
* sequential streams. It is not an error that more elements, than strictly |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
300 |
* required to produce a result, may flow through the pipeline. This can |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
301 |
* occur, in general (not restricted to just sorting), for short-circuiting |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
302 |
* parallel pipelines. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
303 |
*/ |
32649
2ee9017c7597
8136583: Core libraries should use blessed modifier order
martin
parents:
25859
diff
changeset
|
304 |
private abstract static class AbstractRefSortingSink<T> extends Sink.ChainedReference<T, T> { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
305 |
protected final Comparator<? super T> comparator; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
306 |
// @@@ could be a lazy final value, if/when support is added |
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
307 |
// true if cancellationRequested() has been called |
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
308 |
protected boolean cancellationRequestedCalled; |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
309 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
310 |
AbstractRefSortingSink(Sink<? super T> downstream, Comparator<? super T> comparator) { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
311 |
super(downstream); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
312 |
this.comparator = comparator; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
313 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
314 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
315 |
/** |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
316 |
* Records is cancellation is requested so short-circuiting behaviour |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
317 |
* can be preserved when the sorted elements are pushed downstream. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
318 |
* |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
319 |
* @return false, as this sink never short-circuits. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
320 |
*/ |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
321 |
@Override |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
322 |
public final boolean cancellationRequested() { |
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
323 |
// If this method is called then an operation within the stream |
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
324 |
// pipeline is short-circuiting (see AbstractPipeline.copyInto). |
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
325 |
// Note that we cannot differentiate between an upstream or |
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
326 |
// downstream operation |
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
327 |
cancellationRequestedCalled = true; |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
328 |
return false; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
329 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
330 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
331 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
332 |
/** |
20503
074dd13d9cdf
8025535: Unsafe typecast in java.util.stream.SortedOps
psandoz
parents:
19593
diff
changeset
|
333 |
* {@link Sink} for implementing sort on SIZED reference streams. |
17182 | 334 |
*/ |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
335 |
private static final class SizedRefSortingSink<T> extends AbstractRefSortingSink<T> { |
17182 | 336 |
private T[] array; |
337 |
private int offset; |
|
338 |
||
19593 | 339 |
SizedRefSortingSink(Sink<? super T> sink, Comparator<? super T> comparator) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
340 |
super(sink, comparator); |
17182 | 341 |
} |
342 |
||
343 |
@Override |
|
22297 | 344 |
@SuppressWarnings("unchecked") |
17182 | 345 |
public void begin(long size) { |
346 |
if (size >= Nodes.MAX_ARRAY_SIZE) |
|
20507
8498104f92c3
8025534: Unsafe typecast in java.util.stream.Streams.Nodes
psandoz
parents:
20503
diff
changeset
|
347 |
throw new IllegalArgumentException(Nodes.BAD_SIZE); |
17182 | 348 |
array = (T[]) new Object[(int) size]; |
349 |
} |
|
350 |
||
351 |
@Override |
|
352 |
public void end() { |
|
353 |
Arrays.sort(array, 0, offset, comparator); |
|
354 |
downstream.begin(offset); |
|
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
355 |
if (!cancellationRequestedCalled) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
356 |
for (int i = 0; i < offset; i++) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
357 |
downstream.accept(array[i]); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
358 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
359 |
else { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
360 |
for (int i = 0; i < offset && !downstream.cancellationRequested(); i++) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
361 |
downstream.accept(array[i]); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
362 |
} |
17182 | 363 |
downstream.end(); |
364 |
array = null; |
|
365 |
} |
|
366 |
||
367 |
@Override |
|
368 |
public void accept(T t) { |
|
369 |
array[offset++] = t; |
|
370 |
} |
|
371 |
} |
|
372 |
||
373 |
/** |
|
374 |
* {@link Sink} for implementing sort on reference streams. |
|
375 |
*/ |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
376 |
private static final class RefSortingSink<T> extends AbstractRefSortingSink<T> { |
17182 | 377 |
private ArrayList<T> list; |
378 |
||
19593 | 379 |
RefSortingSink(Sink<? super T> sink, Comparator<? super T> comparator) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
380 |
super(sink, comparator); |
17182 | 381 |
} |
382 |
||
383 |
@Override |
|
384 |
public void begin(long size) { |
|
20503
074dd13d9cdf
8025535: Unsafe typecast in java.util.stream.SortedOps
psandoz
parents:
19593
diff
changeset
|
385 |
if (size >= Nodes.MAX_ARRAY_SIZE) |
20507
8498104f92c3
8025534: Unsafe typecast in java.util.stream.Streams.Nodes
psandoz
parents:
20503
diff
changeset
|
386 |
throw new IllegalArgumentException(Nodes.BAD_SIZE); |
22078
bdec5d53e98c
8030851: Update code in java.util to use newer language features
psandoz
parents:
20507
diff
changeset
|
387 |
list = (size >= 0) ? new ArrayList<>((int) size) : new ArrayList<>(); |
17182 | 388 |
} |
389 |
||
390 |
@Override |
|
391 |
public void end() { |
|
392 |
list.sort(comparator); |
|
393 |
downstream.begin(list.size()); |
|
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
394 |
if (!cancellationRequestedCalled) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
395 |
list.forEach(downstream::accept); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
396 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
397 |
else { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
398 |
for (T t : list) { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
399 |
if (downstream.cancellationRequested()) break; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
400 |
downstream.accept(t); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
401 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
402 |
} |
17182 | 403 |
downstream.end(); |
404 |
list = null; |
|
405 |
} |
|
406 |
||
407 |
@Override |
|
408 |
public void accept(T t) { |
|
409 |
list.add(t); |
|
410 |
} |
|
411 |
} |
|
412 |
||
413 |
/** |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
414 |
* Abstract {@link Sink} for implementing sort on int streams. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
415 |
*/ |
32649
2ee9017c7597
8136583: Core libraries should use blessed modifier order
martin
parents:
25859
diff
changeset
|
416 |
private abstract static class AbstractIntSortingSink extends Sink.ChainedInt<Integer> { |
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
417 |
// true if cancellationRequested() has been called |
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
418 |
protected boolean cancellationRequestedCalled; |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
419 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
420 |
AbstractIntSortingSink(Sink<? super Integer> downstream) { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
421 |
super(downstream); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
422 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
423 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
424 |
@Override |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
425 |
public final boolean cancellationRequested() { |
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
426 |
cancellationRequestedCalled = true; |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
427 |
return false; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
428 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
429 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
430 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
431 |
/** |
17182 | 432 |
* {@link Sink} for implementing sort on SIZED int streams. |
433 |
*/ |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
434 |
private static final class SizedIntSortingSink extends AbstractIntSortingSink { |
17182 | 435 |
private int[] array; |
436 |
private int offset; |
|
437 |
||
19593 | 438 |
SizedIntSortingSink(Sink<? super Integer> downstream) { |
17182 | 439 |
super(downstream); |
440 |
} |
|
441 |
||
442 |
@Override |
|
443 |
public void begin(long size) { |
|
444 |
if (size >= Nodes.MAX_ARRAY_SIZE) |
|
20507
8498104f92c3
8025534: Unsafe typecast in java.util.stream.Streams.Nodes
psandoz
parents:
20503
diff
changeset
|
445 |
throw new IllegalArgumentException(Nodes.BAD_SIZE); |
17182 | 446 |
array = new int[(int) size]; |
447 |
} |
|
448 |
||
449 |
@Override |
|
450 |
public void end() { |
|
451 |
Arrays.sort(array, 0, offset); |
|
452 |
downstream.begin(offset); |
|
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
453 |
if (!cancellationRequestedCalled) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
454 |
for (int i = 0; i < offset; i++) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
455 |
downstream.accept(array[i]); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
456 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
457 |
else { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
458 |
for (int i = 0; i < offset && !downstream.cancellationRequested(); i++) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
459 |
downstream.accept(array[i]); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
460 |
} |
17182 | 461 |
downstream.end(); |
462 |
array = null; |
|
463 |
} |
|
464 |
||
465 |
@Override |
|
466 |
public void accept(int t) { |
|
467 |
array[offset++] = t; |
|
468 |
} |
|
469 |
} |
|
470 |
||
471 |
/** |
|
472 |
* {@link Sink} for implementing sort on int streams. |
|
473 |
*/ |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
474 |
private static final class IntSortingSink extends AbstractIntSortingSink { |
17182 | 475 |
private SpinedBuffer.OfInt b; |
476 |
||
19593 | 477 |
IntSortingSink(Sink<? super Integer> sink) { |
17182 | 478 |
super(sink); |
479 |
} |
|
480 |
||
481 |
@Override |
|
482 |
public void begin(long size) { |
|
20503
074dd13d9cdf
8025535: Unsafe typecast in java.util.stream.SortedOps
psandoz
parents:
19593
diff
changeset
|
483 |
if (size >= Nodes.MAX_ARRAY_SIZE) |
20507
8498104f92c3
8025534: Unsafe typecast in java.util.stream.Streams.Nodes
psandoz
parents:
20503
diff
changeset
|
484 |
throw new IllegalArgumentException(Nodes.BAD_SIZE); |
17182 | 485 |
b = (size > 0) ? new SpinedBuffer.OfInt((int) size) : new SpinedBuffer.OfInt(); |
486 |
} |
|
487 |
||
488 |
@Override |
|
489 |
public void end() { |
|
18171 | 490 |
int[] ints = b.asPrimitiveArray(); |
17182 | 491 |
Arrays.sort(ints); |
492 |
downstream.begin(ints.length); |
|
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
493 |
if (!cancellationRequestedCalled) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
494 |
for (int anInt : ints) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
495 |
downstream.accept(anInt); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
496 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
497 |
else { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
498 |
for (int anInt : ints) { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
499 |
if (downstream.cancellationRequested()) break; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
500 |
downstream.accept(anInt); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
501 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
502 |
} |
17182 | 503 |
downstream.end(); |
504 |
} |
|
505 |
||
506 |
@Override |
|
507 |
public void accept(int t) { |
|
508 |
b.accept(t); |
|
509 |
} |
|
510 |
} |
|
511 |
||
512 |
/** |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
513 |
* Abstract {@link Sink} for implementing sort on long streams. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
514 |
*/ |
32649
2ee9017c7597
8136583: Core libraries should use blessed modifier order
martin
parents:
25859
diff
changeset
|
515 |
private abstract static class AbstractLongSortingSink extends Sink.ChainedLong<Long> { |
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
516 |
// true if cancellationRequested() has been called |
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
517 |
protected boolean cancellationRequestedCalled; |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
518 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
519 |
AbstractLongSortingSink(Sink<? super Long> downstream) { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
520 |
super(downstream); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
521 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
522 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
523 |
@Override |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
524 |
public final boolean cancellationRequested() { |
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
525 |
cancellationRequestedCalled = true; |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
526 |
return false; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
527 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
528 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
529 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
530 |
/** |
17182 | 531 |
* {@link Sink} for implementing sort on SIZED long streams. |
532 |
*/ |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
533 |
private static final class SizedLongSortingSink extends AbstractLongSortingSink { |
17182 | 534 |
private long[] array; |
535 |
private int offset; |
|
536 |
||
19593 | 537 |
SizedLongSortingSink(Sink<? super Long> downstream) { |
17182 | 538 |
super(downstream); |
539 |
} |
|
540 |
||
541 |
@Override |
|
542 |
public void begin(long size) { |
|
543 |
if (size >= Nodes.MAX_ARRAY_SIZE) |
|
20507
8498104f92c3
8025534: Unsafe typecast in java.util.stream.Streams.Nodes
psandoz
parents:
20503
diff
changeset
|
544 |
throw new IllegalArgumentException(Nodes.BAD_SIZE); |
17182 | 545 |
array = new long[(int) size]; |
546 |
} |
|
547 |
||
548 |
@Override |
|
549 |
public void end() { |
|
550 |
Arrays.sort(array, 0, offset); |
|
551 |
downstream.begin(offset); |
|
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
552 |
if (!cancellationRequestedCalled) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
553 |
for (int i = 0; i < offset; i++) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
554 |
downstream.accept(array[i]); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
555 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
556 |
else { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
557 |
for (int i = 0; i < offset && !downstream.cancellationRequested(); i++) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
558 |
downstream.accept(array[i]); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
559 |
} |
17182 | 560 |
downstream.end(); |
561 |
array = null; |
|
562 |
} |
|
563 |
||
564 |
@Override |
|
565 |
public void accept(long t) { |
|
566 |
array[offset++] = t; |
|
567 |
} |
|
568 |
} |
|
569 |
||
570 |
/** |
|
571 |
* {@link Sink} for implementing sort on long streams. |
|
572 |
*/ |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
573 |
private static final class LongSortingSink extends AbstractLongSortingSink { |
17182 | 574 |
private SpinedBuffer.OfLong b; |
575 |
||
19593 | 576 |
LongSortingSink(Sink<? super Long> sink) { |
17182 | 577 |
super(sink); |
578 |
} |
|
579 |
||
580 |
@Override |
|
581 |
public void begin(long size) { |
|
20503
074dd13d9cdf
8025535: Unsafe typecast in java.util.stream.SortedOps
psandoz
parents:
19593
diff
changeset
|
582 |
if (size >= Nodes.MAX_ARRAY_SIZE) |
20507
8498104f92c3
8025534: Unsafe typecast in java.util.stream.Streams.Nodes
psandoz
parents:
20503
diff
changeset
|
583 |
throw new IllegalArgumentException(Nodes.BAD_SIZE); |
17182 | 584 |
b = (size > 0) ? new SpinedBuffer.OfLong((int) size) : new SpinedBuffer.OfLong(); |
585 |
} |
|
586 |
||
587 |
@Override |
|
588 |
public void end() { |
|
18171 | 589 |
long[] longs = b.asPrimitiveArray(); |
17182 | 590 |
Arrays.sort(longs); |
591 |
downstream.begin(longs.length); |
|
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
592 |
if (!cancellationRequestedCalled) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
593 |
for (long aLong : longs) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
594 |
downstream.accept(aLong); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
595 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
596 |
else { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
597 |
for (long aLong : longs) { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
598 |
if (downstream.cancellationRequested()) break; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
599 |
downstream.accept(aLong); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
600 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
601 |
} |
17182 | 602 |
downstream.end(); |
603 |
} |
|
604 |
||
605 |
@Override |
|
606 |
public void accept(long t) { |
|
607 |
b.accept(t); |
|
608 |
} |
|
609 |
} |
|
610 |
||
611 |
/** |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
612 |
* Abstract {@link Sink} for implementing sort on long streams. |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
613 |
*/ |
32649
2ee9017c7597
8136583: Core libraries should use blessed modifier order
martin
parents:
25859
diff
changeset
|
614 |
private abstract static class AbstractDoubleSortingSink extends Sink.ChainedDouble<Double> { |
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
615 |
// true if cancellationRequested() has been called |
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
616 |
protected boolean cancellationRequestedCalled; |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
617 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
618 |
AbstractDoubleSortingSink(Sink<? super Double> downstream) { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
619 |
super(downstream); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
620 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
621 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
622 |
@Override |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
623 |
public final boolean cancellationRequested() { |
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
624 |
cancellationRequestedCalled = true; |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
625 |
return false; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
626 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
627 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
628 |
|
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
629 |
/** |
17182 | 630 |
* {@link Sink} for implementing sort on SIZED double streams. |
631 |
*/ |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
632 |
private static final class SizedDoubleSortingSink extends AbstractDoubleSortingSink { |
17182 | 633 |
private double[] array; |
634 |
private int offset; |
|
635 |
||
19593 | 636 |
SizedDoubleSortingSink(Sink<? super Double> downstream) { |
17182 | 637 |
super(downstream); |
638 |
} |
|
639 |
||
640 |
@Override |
|
641 |
public void begin(long size) { |
|
642 |
if (size >= Nodes.MAX_ARRAY_SIZE) |
|
20507
8498104f92c3
8025534: Unsafe typecast in java.util.stream.Streams.Nodes
psandoz
parents:
20503
diff
changeset
|
643 |
throw new IllegalArgumentException(Nodes.BAD_SIZE); |
17182 | 644 |
array = new double[(int) size]; |
645 |
} |
|
646 |
||
647 |
@Override |
|
648 |
public void end() { |
|
649 |
Arrays.sort(array, 0, offset); |
|
650 |
downstream.begin(offset); |
|
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
651 |
if (!cancellationRequestedCalled) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
652 |
for (int i = 0; i < offset; i++) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
653 |
downstream.accept(array[i]); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
654 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
655 |
else { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
656 |
for (int i = 0; i < offset && !downstream.cancellationRequested(); i++) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
657 |
downstream.accept(array[i]); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
658 |
} |
17182 | 659 |
downstream.end(); |
660 |
array = null; |
|
661 |
} |
|
662 |
||
663 |
@Override |
|
664 |
public void accept(double t) { |
|
665 |
array[offset++] = t; |
|
666 |
} |
|
667 |
} |
|
668 |
||
669 |
/** |
|
670 |
* {@link Sink} for implementing sort on double streams. |
|
671 |
*/ |
|
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
672 |
private static final class DoubleSortingSink extends AbstractDoubleSortingSink { |
17182 | 673 |
private SpinedBuffer.OfDouble b; |
674 |
||
19593 | 675 |
DoubleSortingSink(Sink<? super Double> sink) { |
17182 | 676 |
super(sink); |
677 |
} |
|
678 |
||
679 |
@Override |
|
680 |
public void begin(long size) { |
|
20503
074dd13d9cdf
8025535: Unsafe typecast in java.util.stream.SortedOps
psandoz
parents:
19593
diff
changeset
|
681 |
if (size >= Nodes.MAX_ARRAY_SIZE) |
20507
8498104f92c3
8025534: Unsafe typecast in java.util.stream.Streams.Nodes
psandoz
parents:
20503
diff
changeset
|
682 |
throw new IllegalArgumentException(Nodes.BAD_SIZE); |
17182 | 683 |
b = (size > 0) ? new SpinedBuffer.OfDouble((int) size) : new SpinedBuffer.OfDouble(); |
684 |
} |
|
685 |
||
686 |
@Override |
|
687 |
public void end() { |
|
18171 | 688 |
double[] doubles = b.asPrimitiveArray(); |
17182 | 689 |
Arrays.sort(doubles); |
690 |
downstream.begin(doubles.length); |
|
48593
fca88bbbafb9
8075939: Stream.flatMap() causes breaking of short-circuiting of terminal operations
psandoz
parents:
47216
diff
changeset
|
691 |
if (!cancellationRequestedCalled) { |
24258
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
692 |
for (double aDouble : doubles) |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
693 |
downstream.accept(aDouble); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
694 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
695 |
else { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
696 |
for (double aDouble : doubles) { |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
697 |
if (downstream.cancellationRequested()) break; |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
698 |
downstream.accept(aDouble); |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
699 |
} |
0e9ab834f44a
8042355: stream with sorted() causes downstream ops not to be lazy
psandoz
parents:
22297
diff
changeset
|
700 |
} |
17182 | 701 |
downstream.end(); |
702 |
} |
|
703 |
||
704 |
@Override |
|
705 |
public void accept(double t) { |
|
706 |
b.accept(t); |
|
707 |
} |
|
708 |
} |
|
709 |
} |