Add new iterator to handle differences

Author: Jamie Hannaford

lib/OpenCloud/LoadBalancer/Collection/LoadBalancerIterator.php

@@ -0,0 +1,43 @@
+<?php
+
+namespace OpenCloud\LoadBalancer\Collection;
+
+use OpenCloud\Common\Collection\PaginatedIterator;
+
+class LoadBalancerIterator extends PaginatedIterator
+{
+    private $nextElement;
+
+    public function constructNextUrl()
+    {
+        $url = parent::constructNextUrl();
+
+        // We need to return n+1 items in order to grab the relevant marker value
+        $query = $url->getQuery();
+        $query['limit'] = $query['limit'] + 1;
+        $url->setQuery($query);
+
+        return $url;
+    }
+
+    public function updateMarkerToCurrent()
+    {
+        $this->setMarkerFromElement($this->nextElement);
+    }
+
+    public function parseResponseBody($body)
+    {
+        $response = parent::parseResponseBody($body);
+
+        if (count($response) >= $this->getOption('limit.page')) {
+            // Save last element (we will need it for the next marker)
+            $this->nextElement = end($response);
+
+            // Since we previously asked for n+1 elements, pop the unwanted element
+            array_pop($response);
+            reset($response);
+        }
+
+        return $response;
+    }
+}