QueryBuilder::update() implemented custom pipelines.

TCB13 · Jul 17, 2019 · 5f82dda · 5f82dda
1 parent 86ec665
commit 5f82dda
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/QueryBuilder.php b/src/QueryBuilder.php
@@ -509,7 +509,12 @@ public function update($update): UpdateResult
             throw new Exception("You must set a filter (where query) to update records.");
         }
 
-        $pipeline = [];
+        // Build the pipeline
+        if (is_object($this->customPipeline) && !empty($this->customPipeline->pipeline) && $this->customPipeline->beforeQb) {
+            $pipeline = $this->customPipeline->pipeline;
+        } else {
+            $pipeline = [];
+        }
 
         if (is_array($update)) {
             $inc    = [];
@@ -562,6 +567,10 @@ public function update($update): UpdateResult
             }
         }
 
+        if (is_object($this->customPipeline) && !empty($this->customPipeline->pipeline) && !$this->customPipeline->beforeQb) {
+            $pipeline = array_merge($pipeline, $this->customPipeline->pipeline);
+        }
+
         // Run the update query
         return $this->collection->updateMany($this->getNormalizedFilters(), $pipeline);
     }