[VL] Fix failed to get iterator exception from file format writer (#3641

)
apache · Nov 8, 2023 · b60fe75 · b60fe75
1 parent e3eff1d
commit b60fe75
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/IteratorApiImpl.scala b/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/IteratorApiImpl.scala
@@ -150,6 +150,7 @@ class IteratorApiImpl extends IteratorApi with Logging {
 
     Iterators
       .wrap(resIter.asScala)
+      .protectInvocationFlow()
       .recycleIterator {
         updateNativeMetrics(resIter.getMetrics)
         updateInputMetrics(TaskContext.get().taskMetrics().inputMetrics)