fix: correct return type of ArrowFileSystemFileIO functions( MakeMockFileIO, MakeLocalFileIO ) (#161)

change the return type of MakeMockFileIO, MakeLocalFileIO to FileIO

---------

Co-authored-by: nullccxsy <[email protected]>

Author: nullccxsy

src/iceberg/arrow/arrow_fs_file_io.cc

@@ -69,13 +69,15 @@ Status ArrowFileSystemFileIO::DeleteFile(const std::string& file_location) {
   return {};
 }
 
-std::unique_ptr<::arrow::fs::FileSystem> ArrowFileSystemFileIO::MakeMockFileIO() {
-  return std::make_unique<::arrow::fs::internal::MockFileSystem>(
-      std::chrono::system_clock::now());
+std::unique_ptr<FileIO> ArrowFileSystemFileIO::MakeMockFileIO() {
+  return std::make_unique<ArrowFileSystemFileIO>(
+      std::make_shared<::arrow::fs::internal::MockFileSystem>(
+          std::chrono::system_clock::now()));
 }
 
-std::unique_ptr<::arrow::fs::FileSystem> ArrowFileSystemFileIO::MakeLocalFileIO() {
-  return std::make_unique<::arrow::fs::LocalFileSystem>();
+std::unique_ptr<FileIO> ArrowFileSystemFileIO::MakeLocalFileIO() {
+  return std::make_unique<ArrowFileSystemFileIO>(
+      std::make_shared<::arrow::fs::LocalFileSystem>());
 }
 
 }  // namespace iceberg::arrow

src/iceberg/arrow/arrow_fs_file_io_internal.h

@@ -35,10 +35,10 @@ class ICEBERG_BUNDLE_EXPORT ArrowFileSystemFileIO : public FileIO {
       : arrow_fs_(std::move(arrow_fs)) {}
 
   /// \brief Make an in-memory FileIO backed by arrow::fs::internal::MockFileSystem.
-  static std::unique_ptr<::arrow::fs::FileSystem> MakeMockFileIO();
+  static std::unique_ptr<FileIO> MakeMockFileIO();
 
   /// \brief Make a local FileIO backed by arrow::fs::LocalFileSystem.
-  static std::unique_ptr<::arrow::fs::FileSystem> MakeLocalFileIO();
+  static std::unique_ptr<FileIO> MakeLocalFileIO();
 
   ~ArrowFileSystemFileIO() override = default;