Support Limit and ExclusiveStartKey

Author: kitar

src/Kitar/Dynamodb/Query/Builder.php

@@ -41,6 +41,12 @@ class Builder extends BaseBuilder
         'remove' => []
     ];
 
+    /**
+     * LastEvaluatedKey attribute.
+     * @var array|null
+     */
+    public $exclusive_start_key;
+
     /**
      * ConsistentRead option.
      * @var boolean|null
@@ -131,6 +137,20 @@ public function key(array $key)
         return $this;
     }
 
+    /**
+     * Set the ExclusiveStartKey option.
+     * Unlike other methods, this $key should be marshaled　beforehand.
+     *
+     * @param array $key
+     * @return $this
+     */
+    public function exclusiveStartKey($key)
+    {
+        $this->exclusive_start_key = $key;
+
+        return $this;
+    }
+
     /**
      * Set the ConsistentRead option.
      * @param bool $active
@@ -215,7 +235,6 @@ public function deleteItem($key)
     public function updateItem($item)
     {
         foreach ($item as $name => $value) {
-
             $name = $this->expression_attributes->addName($name);
 
             // If value is null, it will pass to REMOVE actions.
@@ -413,6 +432,8 @@ protected function process($query_method, $processor_method)
             $this->grammar->compileKey($this->key),
             $this->grammar->compileItem($this->item),
             $this->grammar->compileUpdates($this->updates),
+            $this->grammar->compileDynamodbLimit($this->limit),
+            $this->grammar->compileExclusiveStartKey($this->exclusive_start_key),
             $this->grammar->compileConsistentRead($this->consistent_read),
             $this->grammar->compileExpressionAttributes($this->expression_attributes),
         );

src/Kitar/Dynamodb/Query/Grammar.php

@@ -130,6 +130,34 @@ public function compileUpdates($updates)
         ];
     }
 
+    /**
+     * Compile the Limit attribute.
+     *
+     * @param int|null $limit
+     * @return array
+     */
+    public function compileDynamodbLimit($limit)
+    {
+        if ($limit === null) {
+            return [];
+        }
+
+        return [
+            'Limit' => $limit
+        ];
+    }
+
+    public function compileExclusiveStartKey($key)
+    {
+        if (empty($key)) {
+            return [];
+        }
+
+        return [
+            'ExclusiveStartKey' => $key
+        ];
+    }
+
     /**
      * Compile the ConsistentRead attribute.
      * @param bool $bool

tests/Query/BuilderTest.php

@@ -74,6 +74,43 @@ public function it_can_set_key()
         $this->assertEquals($params, $query['params']);
     }
 
+    /** @test */
+    public function it_can_set_limit()
+    {
+        $params = [
+            'TableName' => 'ProductCatalog',
+            'Limit' => 5
+        ];
+
+        $query = $this->newQuery('ProductCatalog')
+                      ->limit(5)
+                      ->scan();
+
+        $this->assertEquals($params, $query['params']);
+    }
+
+    /** @test */
+    public function it_can_set_exclusive_start_key()
+    {
+        $params = [
+            'TableName' => 'ProductCatalog',
+            'ExclusiveStartKey' => [
+                'Id' => [
+                    'N' => '101'
+                ]
+            ]
+        ];
+
+        $query = $this->newQuery('ProductCatalog')
+                      ->ExclusiveStartKey([
+                        'Id' => [
+                            'N' => '101'
+                        ]
+                      ])->scan();
+
+        $this->assertEquals($params, $query['params']);
+    }
+
     /** @test */
     public function it_can_set_consistent_read()
     {