Filtering Iterator

With Iterators,you can do more than just present each item of the collection. You can also select what items are presented. Let's modify the Library::getIterator() to allow two additional iterator types.

  class Library {
    // ...
    function getIterator($type=false) {
      switch (strtolower($type)) {
      case 'media':
        $iterator_class = 'LibraryIterator';
        break;
      case 'available':
        $iterator_class = 'LibraryAvailableIterator';
        break;
      case 'released':
        $iterator_class = 'LibraryReleasedIterator';
        break;
      default:
        $iterator_class = 'LibraryGofIterator';
      }
      return new $iterator_class($this->collection);
    }
  }

The class LibraryAvailableIterator should only iterate over items that have a status of "library" (recall that the checkOut() method changes the status to "borrowed").

  class IteratorTestCase extends UnitTestCase {
    // ...
    function TestAvailableIteratorUsage() {
      $this->lib->add($dvd = new Media('test', 1999));
      $this->lib->add(new Media('name4', 1999));
      $this->assertIsA(
        $it = $this->lib->getIterator('available')
        ,'LibraryAvailableIterator');
      $output = '';
      while ($item = $it->next()) {
        $output .= $item->name;
      }
      $this->assertEqual('name1name2name3testname4', $output);
      $dvd->checkOut(ÔJasonÕ);
      $it = $this->lib->getIterator('available');
      $output = '';
      while ($item = $it->next()) {
        $output .= $item->name;
      }
      $this->assertEqual('name1name2name3name4', $output);
    }
  }

This test creates a new Media instance and stores it in the variable $dvd. The first highlighted assertEqual()assertion verifies that the new item is present when iterating with LibraryAvailableIterator. Next, the test uses the checkOut() method and verifies that the new item is missing from the display.

The code to implement filtering is very similar to LibraryIterator::next(), except filtering is done prior to returning the item. If the current item does not match the filter criteria, the code returns $this->next() instead.

  class LibraryAvailableIterator {
    protected $collection = array();
    protected $first=true;
    function __construct($collection) {
      $this->collection = $collection;
    }
    function next() {
      if ($this->first) {
        $this->first = false;
        $ret = current($this->collection);
      } else {
        $ret = next($this->collection);
      }
      if ($ret && 'library' != $ret->status) {
        return $this->next();
      }
      return $ret;
    }
  }