Merge pull request #56 from lifekent/feature/offset

Offset clause for the \InfluxDB\Query\Builder class

README.md

@@ -67,6 +67,7 @@ $result = $database->getQueryBuilder()
 	->select('cpucount')
 	->from('test_metric')
 	->limit(2)
+	->offset(2)
 	->getResultSet()
 	->getPoints();
 

src/InfluxDB/Query/Builder.php

@@ -65,6 +65,11 @@ class Builder
      */
     protected $limitClause = '';
 
+    /**
+     * @var string
+     */
+    protected $offsetClause = '';
+
     /**
      * @var array
      */
@@ -248,6 +253,20 @@ public function limit($count)
         return $this;
     }
 
+    /**
+     * Offset the ResultSet to n records
+     *
+     * @param int $count
+     *
+     * @return $this
+     */
+    public function offset($count)
+    {
+        $this->offsetClause = sprintf(' OFFSET %s', (int) $count);
+
+        return $this;
+    }
+
     /**
      * Add retention policy to query
      *
@@ -322,6 +341,10 @@ protected function parseQuery()
             $query .= $this->limitClause;
         }
 
+        if ($this->offsetClause) {
+            $query .= $this->offsetClause;
+        }
+
         return $query;
     }
 }

tests/unit/Query/BuilderTest.php

@@ -0,0 +1,73 @@
+<?php
+
+    namespace InfluxDB\Test\unit\Query;
+
+    use InfluxDB\Query\Builder;
+    use InfluxDB\Test\unit\AbstractTest;
+
+    class BuilderTest extends AbstractTest
+    {
+        /**
+         * @var Builder
+         */
+        protected $queryBuilder;
+
+        public function setUp()
+        {
+            parent::setUp();
+
+            $this->queryBuilder = $this->database->getQueryBuilder();
+        }
+
+        /**
+         * @return array
+         */
+        public function provideLimit()
+        {
+            return [
+                ['test_metric', 2],
+            ];
+        }
+
+        /**
+         * @dataProvider provideLimit
+         * @param $from
+         * @param $limit
+         */
+        public function testLimit($from, $limit)
+        {
+            $this->assertEquals(
+                sprintf('SELECT * FROM "%s" LIMIT %s', $from, $limit),
+                $this->database->getQueryBuilder()
+                    ->from($from)
+                    ->limit($limit)
+                    ->getQuery()
+            );
+        }
+
+        /**
+         * @return array
+         */
+        public function provideOffset()
+        {
+            return [
+                ['test_metric', 2],
+            ];
+        }
+
+        /**
+         * @dataProvider provideOffset
+         * @param $offset
+         * @param $from
+         */
+        public function testOffset($from, $offset)
+        {
+            $this->assertEquals(
+                sprintf('SELECT * FROM "%s" OFFSET %s', $from, $offset),
+                $this->database->getQueryBuilder()
+                    ->from($from)
+                    ->offset($offset)
+                    ->getQuery()
+            );
+        }
+    }