Added Closure interface to BranchingLoader

flow-php · Jan 28, 2024 · e97aeec · e97aeec
1 parent 6790bc0
commit e97aeec
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/core/etl/src/Flow/ETL/Loader/BranchingLoader.php b/src/core/etl/src/Flow/ETL/Loader/BranchingLoader.php
@@ -10,14 +10,21 @@
 use Flow\ETL\Rows;
 use Flow\ETL\Transformer\ScalarFunctionFilterTransformer;
 
-final class BranchingLoader implements Loader, OverridingLoader
+final class BranchingLoader implements Closure, Loader, OverridingLoader
 {
     public function __construct(
         private readonly ScalarFunction $condition,
         private readonly Loader $loader
     ) {
     }
 
+    public function closure(FlowContext $context) : void
+    {
+        if ($this->loader instanceof Closure) {
+            $this->loader->closure($context);
+        }
+    }
+
     public function load(Rows $rows, FlowContext $context) : void
     {
         $this->loader->load(