From 0c65565b1c0e78ba29369b6f919a77d630e01220 Mon Sep 17 00:00:00 2001
From: Bingyi Sun <sunbingyi1992@gmail.com>
Date: Wed, 18 Oct 2023 18:41:59 +0800
Subject: [PATCH] [Cpp] Add Schema function (#79)

Signed-off-by: sunby <sunbingyi1992@gmail.com>
---
 .gitignore                                 | 1 +
 cpp/include/milvus-storage/storage/space.h | 4 ++++
 cpp/src/storage/space.cpp                  | 2 ++
 3 files changed, 7 insertions(+)

diff --git a/.gitignore b/.gitignore
index fd2db39..c540786 100644
--- 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
index c3d6ff3..61edc59 100644
--- 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
index 2872973..d705f98 100644
--- 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