[Cpp] Add Schema function (#79)

Signed-off-by: sunby <[email protected]>
milvus-io · Oct 18, 2023 · 0c65565 · 0c65565
1 parent bef9bb3
commit 0c65565
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -6,3 +6,4 @@ go/.vscode/*
 go/.idea/*
 compile_commands.json
 CMakeUserPresets.json
+.vscode/*
diff --git a/cpp/include/milvus-storage/storage/space.h b/cpp/include/milvus-storage/storage/space.h
@@ -1,9 +1,11 @@
 #pragma once
 #include <arrow/record_batch.h>
+#include <arrow/type.h>
 #include <atomic>
 #include <mutex>
 
 #include "storage/manifest.h"
+#include "storage/schema.h"
 #include "storage/space.h"
 #include "file/delete_fragment.h"
 namespace milvus_storage {
@@ -42,6 +44,8 @@ class Space {
 
   std::vector<Blob> StatisticsBlobs();
 
+  arrow::Schema Schema();
+
   private:
   Status Init();
 

diff --git a/cpp/src/storage/space.cpp b/cpp/src/storage/space.cpp
@@ -307,4 +307,6 @@ Result<std::shared_ptr<arrow::RecordBatchReader>> Space::ScanData() {
   return RecordReader::MakeScanDataReader(manifest_, fs_);
 }
 
+arrow::Schema Space::Schema() { return *manifest_->schema()->schema(); }
+
 }  // namespace milvus_storage