Support setting a field multiple times within an update statement (#7)

packaged · May 23, 2019 · 672f991 · 672f991
1 parent bf9abcb
commit 672f991
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 1 deletion.
diff --git a/src/Builder/Traits/SetTrait.php b/src/Builder/Traits/SetTrait.php
@@ -4,6 +4,7 @@
 use Packaged\QueryBuilder\Clause\SetClause;
 use Packaged\QueryBuilder\Predicate\EqualPredicate;
 use Packaged\QueryBuilder\Statement\IStatement;
+use function is_array;
 
 trait SetTrait
 {
@@ -19,7 +20,17 @@ public function set($field, $value)
       $this->addClause($set);
     }
 
-    $set->addPredicate(EqualPredicate::create($field, $value)->forceOperator());
+    if(is_array($value))
+    {
+      foreach($value as $exp)
+      {
+        $set->addPredicate(EqualPredicate::create($field, $exp)->forceOperator());
+      }
+    }
+    else
+    {
+      $set->addPredicate(EqualPredicate::create($field, $value)->forceOperator());
+    }
 
     return $this;
   }

diff --git a/tests/Statement/CQL/CqlUpdateStatementTest.php b/tests/Statement/CQL/CqlUpdateStatementTest.php
@@ -3,6 +3,8 @@
 
 use Packaged\QueryBuilder\Assembler\CQL\CqlAssembler;
 use Packaged\QueryBuilder\Builder\CQL\CqlQueryBuilder;
+use Packaged\QueryBuilder\Expression\DecrementExpression;
+use Packaged\QueryBuilder\Expression\IncrementExpression;
 
 class CqlUpdateStatementTest extends \PHPUnit_Framework_TestCase
 {
@@ -22,6 +24,30 @@ public function testUpdate()
     $this->assertEquals(['value1'], $assembler->getParameters());
   }
 
+  public function testUpdateInitCounter()
+  {
+    $stmt = CqlQueryBuilder::update(
+      'tbl',
+      [
+        'counter' => [
+          IncrementExpression::create('counter', 1),
+          DecrementExpression::create('counter', 2),
+        ],
+      ]
+    );
+    $this->assertEquals(
+      'UPDATE "tbl" SET "counter" = "counter" + 1, "counter" = "counter" - 2',
+      CqlAssembler::stringify($stmt)
+    );
+
+    $assembler = new CqlAssembler($stmt);
+    $this->assertEquals(
+      'UPDATE "tbl" SET "counter" = "counter" + ?, "counter" = "counter" - ?',
+      $assembler->getQuery()
+    );
+    $this->assertEquals([1, 2], $assembler->getParameters());
+  }
+
   public function testUpdateTtl()
   {
     $stmt = CqlQueryBuilder::update('tbl', ['field1' => 'value1'])