- Add ability to bypass key() surrounding and make that default

1 files changed, 10 insertions, 2 deletions

diff --git a/ext/spl/examples/recursivetreeiterator.inc b/ext/spl/examples/recursivetreeiterator.inc
index 2d2210e704..42d217fa76 100755
--- a/ext/spl/examples/recursivetreeiterator.inc
+++ b/ext/spl/examples/recursivetreeiterator.inc
@@ -19,6 +19,7 @@
 class RecursiveTreeIterator extends RecursiveIteratorIterator
 {
 	const BYPASS_CURRENT = 0x00000004;
+	const BYPASS_KEY     = 0x00000008;
 
 	private $rit_flags;
 
@@ -28,7 +29,7 @@ class RecursiveTreeIterator extends RecursiveIteratorIterator
 	 * @param cit_flags  flags passed to RecursiveCachingIterator (for hasNext)
 	 * @param mode       mode  passed to RecursiveIteratoIterator (parent)
 	 */
-	function __construct(RecursiveIterator $it, $rit_flags = 0, $cit_flags = CachingIterator::CATCH_GET_CHILD, $mode = self::SELF_FIRST)
+	function __construct(RecursiveIterator $it, $rit_flags = self::BYPASS_KEY, $cit_flags = CachingIterator::CATCH_GET_CHILD, $mode = self::SELF_FIRST)
 	{
 		parent::__construct(new RecursiveCachingIterator($it, $cit_flags), $mode, $rit_flags);
 		$this->rit_flags = $rit_flags;
@@ -91,7 +92,14 @@ class RecursiveTreeIterator extends RecursiveIteratorIterator
 	 */
 	function key()
 	{
-		return $this->getPrefix() . parent::key() .  $this->getPostfix();
+		if ($this->rit_flags & self::BYPASS_KEY)
+		{
+			return parent::key();
+		}
+		else
+		{
+			return $this->getPrefix() . parent::key() .  $this->getPostfix();
+		}
 	}
 
 	/** Aggregates the inner iterator