fix columnar wscg on empty recordbatch (#36)

Signed-off-by: Yuan Zhou <yuan.zhou@intel.com>
oap-project · Jan 14, 2021 · 8f01e2e · 8f01e2e
1 parent 4ee0d1b
commit 8f01e2e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/core/src/main/scala/com/intel/oap/execution/ColumnarWholeStageCodegenExec.scala b/core/src/main/scala/com/intel/oap/execution/ColumnarWholeStageCodegenExec.scala
@@ -344,6 +344,7 @@ case class ColumnarWholeStageCodegenExec(child: SparkPlan)(val codegenStageId: I
             var build_elapse_internal: Long = 0
             while (depIter.hasNext) {
               val dep_cb = depIter.next()
+              if (dep_cb.numRows > 0) {
               (0 until dep_cb.numCols).toList.foreach(i =>
                 dep_cb.column(i).asInstanceOf[ArrowWritableColumnVector].retain())
               buildRelationBatchHolder += dep_cb
@@ -353,6 +354,7 @@ case class ColumnarWholeStageCodegenExec(child: SparkPlan)(val codegenStageId: I
               ConverterUtils.releaseArrowRecordBatch(dep_rb)
               build_elapse += System.nanoTime() - beforeEval
               build_elapse_internal += System.nanoTime() - beforeEval
+              }
             }
             buildTime += (build_elapse_internal / 1000000)
             dependentKernels += hashRelationKernel