Add ability to set the source for GeneratingIterator

1 files changed, 13 insertions, 0 deletions

diff --git a/src/main/java/bjc/data/GeneratingIterator.java b/src/main/java/bjc/data/GeneratingIterator.java
index ffa92cf..f926833 100644
--- a/src/main/java/bjc/data/GeneratingIterator.java
+++ b/src/main/java/bjc/data/GeneratingIterator.java
@@ -58,4 +58,17 @@ public class GeneratingIterator<E> implements Iterator<E> {
 
 		return state;
 	}
+	
+	/**
+	 * Sets the state of this iterator.
+	 * 
+	 * @param newState The new state value.
+	 * 
+	 * @return The old state value.
+	 */
+	public E setState(E newState) {
+		E oldState = this.state;
+		this.state = newState;
+		return oldState;
+	}
 }