Implemented get parameters method for update queries

HarmonyIO · Apr 27, 2019 · d3287c6 · d3287c6
1 parent cd8907d
commit d3287c6
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/src/QueryBuilder/Statement/Set.php b/src/QueryBuilder/Statement/Set.php
@@ -19,4 +19,12 @@ public function toSql(): string
     {
         return $this->condition->toSql();
     }
+
+    /**
+     * @return mixed[]
+     */
+    public function getParameters(): array
+    {
+        return $this->condition->getParameters();
+    }
 }
diff --git a/src/QueryBuilder/Statement/Update.php b/src/QueryBuilder/Statement/Update.php
@@ -230,4 +230,30 @@ public function getQuery(): string
 
         return 'UPDATE ' . $this->table->toSql() . $set . $joins . $condition . $orders . $limit . $offset;
     }
+
+    /**
+     * @return mixed[]
+     */
+    public function getParameters(): array
+    {
+        $parameters = [];
+
+        if ($this->set) {
+            foreach ($this->set as $set) {
+                $parameters = array_merge($parameters, $set->getParameters());
+            }
+        }
+
+        if ($this->joins) {
+            foreach ($this->joins as $join) {
+                $parameters = array_merge($parameters, $join->getParameters());
+            }
+        }
+
+        if ($this->condition !== null) {
+            $parameters = array_merge($parameters, $this->condition->getParameters());
+        }
+
+        return $parameters;
+    }
 }