[VL] Refactor defaultLeafPool usage

cpp/velox/benchmarks/PlanValidatorUtil.cc

@@ -44,8 +44,8 @@ int main(int argc, char** argv) {
   std::unordered_map<std::string, std::string> conf;
   conf.insert({kDebugModeEnabled, "true"});
   initVeloxBackend(conf);
-  auto pool = defaultLeafVeloxMemoryPool().get();
-  SubstraitToVeloxPlanValidator planValidator(pool);
+  auto pool = defaultLeafVeloxMemoryPool();
+  SubstraitToVeloxPlanValidator planValidator(pool.get());
 
   ::substrait::Plan subPlan;
   parseProtobuf(reinterpret_cast<uint8_t*>(plan.data()), plan.size(), &subPlan);

cpp/velox/jni/VeloxJniWrapper.cc

@@ -194,8 +194,8 @@ Java_org_apache_gluten_vectorized_PlanEvaluatorJniWrapper_nativeValidateWithFail
     }
   }
 
-  const auto pool = defaultLeafVeloxMemoryPool().get();
-  SubstraitToVeloxPlanValidator planValidator(pool);
+  auto pool = defaultLeafVeloxMemoryPool();
+  SubstraitToVeloxPlanValidator planValidator(pool.get());
   ::substrait::Plan subPlan;
   parseProtobuf(planData, planSize, &subPlan);
 
@@ -252,8 +252,8 @@ JNIEXPORT jboolean JNICALL Java_org_apache_gluten_vectorized_PlanEvaluatorJniWra
     env->DeleteLocalRef(mapping);
   }
 
-  auto pool = defaultLeafVeloxMemoryPool().get();
-  SubstraitToVeloxPlanValidator planValidator(pool);
+  auto pool = defaultLeafVeloxMemoryPool();
+  SubstraitToVeloxPlanValidator planValidator(pool.get());
   auto inputType = SubstraitParser::parseType(inputSubstraitType);
   if (inputType->kind() != TypeKind::ROW) {
     throw GlutenException("Input type is not a RowType.");

cpp/velox/substrait/SubstraitToVeloxPlan.cc

@@ -1473,10 +1473,11 @@ core::PlanNodePtr SubstraitToVeloxPlanConverter::constructValuesNode(
   // It loads all data from the iterator at plan construction time
   VELOX_CHECK_LT(streamIdx, inputIters_.size(), "Could not find stream index {} in input iterator list.", streamIdx);
   const auto iter = std::move(inputIters_[streamIdx]);
+  auto pool = defaultLeafVeloxMemoryPool();
   std::vector<RowVectorPtr> rowVectors;
   while (iter->hasNext()) {
     auto batch = iter->next();
-    auto veloxBatch = VeloxColumnarBatch::from(defaultLeafVeloxMemoryPool().get(), batch);
+    auto veloxBatch = VeloxColumnarBatch::from(pool.get(), batch);
     rowVectors.emplace_back(veloxBatch->getRowVector());
   }
   auto node = std::make_shared<facebook::velox::core::ValuesNode>(nextPlanNodeId(), std::move(rowVectors));

cpp/velox/utils/VeloxArrowUtils.cc

@@ -53,7 +53,8 @@ arrow::Result<std::shared_ptr<ColumnarBatch>> recordBatch2VeloxColumnarBatch(con
   ArrowArray arrowArray;
   ArrowSchema arrowSchema;
   RETURN_NOT_OK(arrow::ExportRecordBatch(rb, &arrowArray, &arrowSchema));
-  auto vp = velox::importFromArrowAsOwner(arrowSchema, arrowArray, gluten::defaultLeafVeloxMemoryPool().get());
+  auto pool = gluten::defaultLeafVeloxMemoryPool();
+  auto vp = velox::importFromArrowAsOwner(arrowSchema, arrowArray, pool.get());
   return std::make_shared<VeloxColumnarBatch>(std::dynamic_pointer_cast<velox::RowVector>(vp));
 }