|
26 | 26 | #include <vector> |
27 | 27 |
|
28 | 28 | #include "arrow/api.h" |
| 29 | +#include "arrow/c/bridge.h" |
29 | 30 | #include "arrow/ipc/json_simple.h" |
30 | 31 | #include "gtest/gtest.h" |
31 | 32 | #include "paimon/common/data/binary_row.h" |
|
34 | 35 | #include "paimon/common/reader/concat_batch_reader.h" |
35 | 36 | #include "paimon/common/types/data_field.h" |
36 | 37 | #include "paimon/common/utils/arrow/mem_utils.h" |
| 38 | +#include "paimon/common/utils/arrow/status_utils.h" |
37 | 39 | #include "paimon/core/core_options.h" |
38 | 40 | #include "paimon/core/deletionvectors/bitmap_deletion_vector.h" |
39 | 41 | #include "paimon/core/global_index/indexed_split_impl.h" |
@@ -95,8 +97,10 @@ class RawVectorSearchTestReader final : public FileIndexReader { |
95 | 97 | class RawVectorSearchTestIndexer final : public FileIndexer { |
96 | 98 | public: |
97 | 99 | Result<std::shared_ptr<FileIndexReader>> CreateReader( |
98 | | - ::ArrowSchema*, int32_t, int32_t, const std::shared_ptr<InputStream>&, |
| 100 | + ::ArrowSchema* arrow_schema, int32_t, int32_t, const std::shared_ptr<InputStream>&, |
99 | 101 | const std::shared_ptr<MemoryPool>&) const override { |
| 102 | + PAIMON_ASSIGN_OR_RAISE_FROM_ARROW([[maybe_unused]] std::shared_ptr<arrow::Schema> schema, |
| 103 | + arrow::ImportSchema(arrow_schema)); |
100 | 104 | return std::make_shared<RawVectorSearchTestReader>(); |
101 | 105 | } |
102 | 106 |
|
@@ -141,8 +145,10 @@ class RawFullTextSearchTestReader final : public FileIndexReader { |
141 | 145 | class RawFullTextSearchTestIndexer final : public FileIndexer { |
142 | 146 | public: |
143 | 147 | Result<std::shared_ptr<FileIndexReader>> CreateReader( |
144 | | - ::ArrowSchema*, int32_t, int32_t, const std::shared_ptr<InputStream>&, |
| 148 | + ::ArrowSchema* arrow_schema, int32_t, int32_t, const std::shared_ptr<InputStream>&, |
145 | 149 | const std::shared_ptr<MemoryPool>&) const override { |
| 150 | + PAIMON_ASSIGN_OR_RAISE_FROM_ARROW([[maybe_unused]] std::shared_ptr<arrow::Schema> schema, |
| 151 | + arrow::ImportSchema(arrow_schema)); |
146 | 152 | return std::make_shared<RawFullTextSearchTestReader>(); |
147 | 153 | } |
148 | 154 |
|
|
0 commit comments