Expose getting base directory

Author: yuandagits

presto-native-execution/presto_cpp/main/TaskManager.cpp

@@ -351,6 +351,11 @@ void TaskManager::setBaseSpillDirectory(const std::string& baseSpillDirectory) {
       [&](auto& baseSpillDir) { baseSpillDir = baseSpillDirectory; });
 }
 
+std::string TaskManager::getBaseSpillDirectory() const {
+  return baseSpillDir_.withRLock(
+      [](const auto& baseSpillDir) { return baseSpillDir; });
+}
+
 bool TaskManager::emptyBaseSpillDirectory() const {
   return baseSpillDir_.withRLock(
       [](const auto& baseSpillDir) { return baseSpillDir.empty(); });

presto-native-execution/presto_cpp/main/TaskManager.h

@@ -42,6 +42,8 @@ class TaskManager {
 
   bool emptyBaseSpillDirectory() const;
 
+  std::string getBaseSpillDirectory() const;
+
   /// Sets the time (ms) that a task is considered to be old for cleanup since
   /// its completion.
   void setOldTaskCleanUpMs(int32_t oldTaskCleanUpMs);

presto-native-execution/presto_cpp/main/tests/TaskManagerTest.cpp

@@ -1221,6 +1221,11 @@ TEST_P(TaskManagerTest, timeoutOutOfOrderRequests) {
           .getVia(eventBase));
 }
 
+TEST_P(TaskManagerTest, getBaseSpillDirectory) {
+  taskManager_->setBaseSpillDirectory("dummy");
+  EXPECT_EQ("dummy", taskManager_->getBaseSpillDirectory());
+}
+
 TEST_P(TaskManagerTest, aggregationSpill) {
   // NOTE: we need to write more than one batches to each file (source split) to
   // trigger spill.