- DibiResultIterator is back (from rev. 123)

dibi/dibi.php

@@ -66,6 +66,7 @@ require_once dirname(__FILE__) . '/libs/DibiObject.php';
 require_once dirname(__FILE__) . '/libs/DibiException.php';
 require_once dirname(__FILE__) . '/libs/DibiConnection.php';
 require_once dirname(__FILE__) . '/libs/DibiResult.php';
+require_once dirname(__FILE__) . '/libs/DibiResultIterator.php';
 require_once dirname(__FILE__) . '/libs/DibiTranslator.php';
 require_once dirname(__FILE__) . '/libs/DibiVariable.php';
 require_once dirname(__FILE__) . '/libs/DibiTableX.php';

dibi/libs/DibiResult.php

@@ -587,11 +587,11 @@ class DibiResult extends DibiObject implements IDataSource
 	 * Required by the IteratorAggregate interface.
 	 * @param  int  offset
 	 * @param  int  limit
-	 * @return ArrayIterator
+	 * @return DibiResultIterator
 	 */
 	final public function getIterator($offset = NULL, $limit = NULL)
 	{
-		return new ArrayIterator($this->fetchAll($offset, $limit));
+		return new DibiResultIterator($this, $offset, $limit);
 	}
 
 

dibi/libs/DibiResultIterator.php

@@ -0,0 +1,136 @@
+<?php
+
+/**
+ * dibi - tiny'n'smart database abstraction layer
+ * ----------------------------------------------
+ *
+ * Copyright (c) 2005, 2009 David Grudl (http://davidgrudl.com)
+ *
+ * This source file is subject to the "dibi license" that is bundled
+ * with this package in the file license.txt.
+ *
+ * For more information please see http://dibiphp.com
+ *
+ * @copyright  Copyright (c) 2005, 2009 David Grudl
+ * @license    http://dibiphp.com/license  dibi license
+ * @link       http://dibiphp.com
+ * @package    dibi
+ * @version    $Id: DibiResult.php 175 2009-01-05 00:03:27Z david@grudl.com $
+ */
+
+
+
+/**
+ * External result set iterator.
+ *
+ * This can be returned by DibiResult::getIterator() method or using foreach
+ * <code>
+ * $result = dibi::query('SELECT * FROM table');
+ * foreach ($result as $row) {
+ *    print_r($row);
+ * }
+ * unset($result);
+ * </code>
+ *
+ * Optionally you can specify offset and limit:
+ * <code>
+ * foreach ($result->getIterator(2, 3) as $row) {
+ *     print_r($row);
+ * }
+ * </code>
+ *
+ * @author     David Grudl
+ * @copyright  Copyright (c) 2005, 2009 David Grudl
+ * @package    dibi
+ */
+class DibiResultIterator implements Iterator
+{
+	/** @var DibiResult */
+	private $result;
+
+	/** @var int */
+	private $offset;
+
+	/** @var int */
+	private $limit;
+
+	/** @var int */
+	private $row;
+
+	/** @var int */
+	private $pointer;
+
+
+	/**
+	 * @param  DibiResult
+	 * @param  int  offset
+	 * @param  int  limit
+	 */
+	public function __construct(DibiResult $result, $offset = NULL, $limit = NULL)
+	{
+		$this->result = $result;
+		$this->offset = (int) $offset;
+		$this->limit = $limit === NULL ? -1 : (int) $limit;
+	}
+
+
+
+	/**
+	 * Rewinds the iterator to the first element.
+	 * @return void
+	 */
+	public function rewind()
+	{
+		$this->pointer = 0;
+		$this->result->seek($this->offset);
+		$this->row = $this->result->fetch();
+	}
+
+
+
+	/**
+	 * Returns the key of the current element.
+	 * @return mixed
+	 */
+	public function key()
+	{
+		return $this->pointer;
+	}
+
+
+
+	/**
+	 * Returns the current element.
+	 * @return mixed
+	 */
+	public function current()
+	{
+		return $this->row;
+	}
+
+
+
+	/**
+	 * Moves forward to next element.
+	 * @return void
+	 */
+	public function next()
+	{
+		//$this->result->seek($this->offset + $this->pointer + 1);
+		$this->row = $this->result->fetch();
+		$this->pointer++;
+	}
+
+
+
+	/**
+	 * Checks if there is a current element after calls to rewind() or next().
+	 * @return bool
+	 */
+	public function valid()
+	{
+		return !empty($this->row) && ($this->limit < 0 || $this->pointer < $this->limit);
+	}
+
+
+}