1 /*
2 * Licensed to the Apache Software Foundation (ASF) under one or more
3 * contributor license agreements. See the NOTICE file distributed with
4 * this work for additional information regarding copyright ownership.
5 * The ASF licenses this file to You under the Apache License, Version 2.0
6 * (the "License"); you may not use this file except in compliance with
7 * the License. You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 */
17 package org.apache.commons.collections4.iterators;
18
19 import java.util.Iterator;
20
21 import org.apache.commons.collections4.ResettableIterator;
22
23 /**
24 * Adapter to make an {@link Iterator Iterator} instance appear to be an
25 * {@link Iterable Iterable} instance. The iterable can be constructed in one
26 * of two variants: single use, multiple use.
27 *
28 * <p>
29 * In the single use iterable case, the iterable is only usable for one
30 * iterative operation over the source iterator. Subsequent iterative
31 * operations use the same, exhausted source iterator. To create a single use
32 * iterable, construct a new {@link IteratorIterable} using a {@link Iterator}
33 * that is NOT a {@link ResettableIterator} iterator:
34 * </p>
35 *
36 * <pre>
37 * Iterator<Integer> iterator = // some non-resettable iterator
38 * Iterable<Integer> iterable = new IteratorIterable<Integer>(iterator);
39 * </pre>
40 *
41 * <p>
42 * In the multiple use iterable case, the iterable is usable for any number of
43 * iterative operations over the source iterator. Of special note, even though
44 * the iterable supports multiple iterations, it does not support concurrent
45 * iterations. To implicitly create a multiple use iterable, construct a new
46 * {@link IteratorIterable} using a {@link ResettableIterator} iterator:
47 * </p>
48 *
49 * <pre>
50 * Integer[] array = {Integer.valueOf(1),Integer.valueOf(2),Integer.valueOf(3)};
51 * Iterator<Integer> iterator = IteratorUtils.arrayIterator(array); // a resettable iterator
52 * Iterable<Integer> iterable = new IteratorIterable<Integer>(iterator);
53 * </pre>
54 *
55 * <p>
56 * A multiple use iterable can also be explicitly constructed using any
57 * {@link Iterator} and specifying {@code true} for the
58 * {@code multipleUse} flag:
59 * </p>
60 *
61 * <pre>
62 * Iterator<Integer> iterator = // some non-resettable iterator
63 * Iterable<Integer> iterable = new IteratorIterable<Integer>(iterator, true);
64 * </pre>
65 *
66 * @param <E> the type of elements returned by this iterator.
67 * @since 4.0
68 */
69 public class IteratorIterable<E> implements Iterable<E> {
70
71 /**
72 * Factory method to create an {@link Iterator Iterator} from another
73 * iterator over objects of a different subtype.
74 */
75 private static <E> Iterator<E> createTypesafeIterator(final Iterator<? extends E> iterator) {
76 return new Iterator<E>() {
77 @Override
78 public boolean hasNext() {
79 return iterator.hasNext();
80 }
81
82 @Override
83 public E next() {
84 return iterator.next();
85 }
86
87 @Override
88 public void remove() {
89 iterator.remove();
90 }
91 };
92 }
93
94 /** The iterator being adapted into an iterable. */
95 private final Iterator<? extends E> iterator;
96
97 /** The iterator parameterized as the {@link #iterator()} return type. */
98 private final Iterator<E> typeSafeIterator;
99
100 /**
101 * Constructs a new {@code IteratorIterable} that will use the given
102 * iterator.
103 *
104 * @param iterator the iterator to use.
105 */
106 public IteratorIterable(final Iterator<? extends E> iterator) {
107 this(iterator, false);
108 }
109
110 /**
111 * Constructs a new {@code IteratorIterable} that will use the given
112 * iterator.
113 *
114 * @param iterator the iterator to use.
115 * @param multipleUse {@code true} if the new iterable can be used in multiple iterations
116 */
117 public IteratorIterable(final Iterator<? extends E> iterator, final boolean multipleUse) {
118 if (multipleUse && !(iterator instanceof ResettableIterator)) {
119 this.iterator = new ListIteratorWrapper<>(iterator);
120 } else {
121 this.iterator = iterator;
122 }
123 this.typeSafeIterator = createTypesafeIterator(this.iterator);
124 }
125
126 /**
127 * Gets the iterator wrapped by this iterable.
128 *
129 * @return the iterator
130 */
131 @Override
132 public Iterator<E> iterator() {
133 if (iterator instanceof ResettableIterator) {
134 ((ResettableIterator<? extends E>) iterator).reset();
135 }
136 return typeSafeIterator;
137 }
138 }