|
33 | 33 | #include "io/fs/local_file_system.h"
|
34 | 34 | #include "olap/rowset/segment_v2/inverted_index_compound_reader.h"
|
35 | 35 | #include "olap/rowset/segment_v2/inverted_index_desc.h"
|
36 |
| -#include "olap/rowset/segment_v2/inverted_index_file_writer.h" |
37 | 36 | #include "olap/rowset/segment_v2/inverted_index_fs_directory.h"
|
38 | 37 | #include "olap/rowset/segment_v2/inverted_index_writer.h"
|
39 | 38 | #include "olap/rowset/segment_v2/x_index_file_reader.h"
|
| 39 | +#include "olap/rowset/segment_v2/x_index_file_writer.h" |
40 | 40 | #include "olap/rowset/segment_v2/zone_map_index.h"
|
41 | 41 | #include "olap/tablet_schema.h"
|
42 | 42 | #include "olap/tablet_schema_helper.h"
|
|
54 | 54 | #include "vec/olap/olap_data_convertor.h"
|
55 | 55 |
|
56 | 56 | using namespace lucene::index;
|
57 |
| -using doris::segment_v2::InvertedIndexFileWriter; |
| 57 | +using doris::segment_v2::XIndexFileWriter; |
58 | 58 |
|
59 | 59 | namespace doris {
|
60 | 60 | namespace segment_v2 {
|
@@ -119,7 +119,7 @@ class InvertedIndexArrayTest : public testing::Test {
|
119 | 119 | std::cout << "==================================" << std::endl;
|
120 | 120 | lucene::store::Directory* dir = compound_reader.get();
|
121 | 121 |
|
122 |
| - IndexReader* r = IndexReader::open(dir); |
| 122 | + lucene::index::IndexReader* r = lucene::index::IndexReader::open(dir); |
123 | 123 |
|
124 | 124 | printf("Max Docs: %d\n", r->maxDoc());
|
125 | 125 | printf("Num Docs: %d\n", r->numDocs());
|
@@ -220,12 +220,12 @@ class InvertedIndexArrayTest : public testing::Test {
|
220 | 220 | TabletIndex idx_meta;
|
221 | 221 | idx_meta.index_type();
|
222 | 222 | idx_meta.init_from_pb(*index_meta_pb.get());
|
223 |
| - auto index_file_writer = std::make_unique<InvertedIndexFileWriter>( |
224 |
| - fs, index_path_prefix, std::string {rowset_id}, seg_id, |
225 |
| - InvertedIndexStorageFormatPB::V1); |
226 |
| - std::unique_ptr<segment_v2::InvertedIndexColumnWriter> _inverted_index_builder = nullptr; |
227 |
| - EXPECT_EQ(InvertedIndexColumnWriter::create(field, &_inverted_index_builder, |
228 |
| - index_file_writer.get(), &idx_meta), |
| 223 | + auto index_file_writer = |
| 224 | + std::make_unique<XIndexFileWriter>(fs, index_path_prefix, std::string {rowset_id}, |
| 225 | + seg_id, InvertedIndexStorageFormatPB::V1); |
| 226 | + std::unique_ptr<segment_v2::IndexColumnWriter> _inverted_index_builder = nullptr; |
| 227 | + EXPECT_EQ(IndexColumnWriter::create(field, &_inverted_index_builder, index_file_writer.get(), |
| 228 | + &idx_meta), |
229 | 229 | Status::OK());
|
230 | 230 |
|
231 | 231 | // Construct two arrays: The first row is ["amory","doris"], and the second row is ["amory", "commiter"]
|
@@ -305,12 +305,12 @@ class InvertedIndexArrayTest : public testing::Test {
|
305 | 305 | TabletIndex idx_meta;
|
306 | 306 | idx_meta.index_type();
|
307 | 307 | idx_meta.init_from_pb(*index_meta_pb.get());
|
308 |
| - auto index_file_writer = std::make_unique<InvertedIndexFileWriter>( |
| 308 | + auto index_file_writer = std::make_unique<XIndexFileWriter>( |
309 | 309 | fs, index_path_prefix, std::string {rowset_id}, seg_id,
|
310 | 310 | InvertedIndexStorageFormatPB::V1);
|
311 |
| - std::unique_ptr<segment_v2::InvertedIndexColumnWriter> _inverted_index_builder = nullptr; |
312 |
| - EXPECT_EQ(InvertedIndexColumnWriter::create(field, &_inverted_index_builder, |
313 |
| - index_file_writer.get(), &idx_meta), |
| 311 | + std::unique_ptr<segment_v2::IndexColumnWriter> _inverted_index_builder = nullptr; |
| 312 | + EXPECT_EQ(IndexColumnWriter::create(field, &_inverted_index_builder, index_file_writer.get(), |
| 313 | + &idx_meta), |
314 | 314 | Status::OK());
|
315 | 315 |
|
316 | 316 | // Construct two arrays: The first row is ["amory","doris"], and the second row is [NULL, "amory", "commiter"]
|
@@ -394,12 +394,12 @@ class InvertedIndexArrayTest : public testing::Test {
|
394 | 394 | io::FileWriterOptions opts;
|
395 | 395 | Status sts = fs->create_file(index_path, &file_writer, &opts);
|
396 | 396 | ASSERT_TRUE(sts.ok());
|
397 |
| - auto index_file_writer = std::make_unique<InvertedIndexFileWriter>( |
| 397 | + auto index_file_writer = std::make_unique<XIndexFileWriter>( |
398 | 398 | fs, index_path_prefix, std::string {rowset_id}, seg_id,
|
399 | 399 | InvertedIndexStorageFormatPB::V2, std::move(file_writer));
|
400 |
| - std::unique_ptr<segment_v2::InvertedIndexColumnWriter> _inverted_index_builder = nullptr; |
401 |
| - EXPECT_EQ(InvertedIndexColumnWriter::create(field, &_inverted_index_builder, |
402 |
| - index_file_writer.get(), &idx_meta), |
| 400 | + std::unique_ptr<segment_v2::IndexColumnWriter> _inverted_index_builder = nullptr; |
| 401 | + EXPECT_EQ(IndexColumnWriter::create(field, &_inverted_index_builder, index_file_writer.get(), |
| 402 | + &idx_meta), |
403 | 403 | Status::OK());
|
404 | 404 |
|
405 | 405 | // Simulate outer null cases: 5 rows, outer null map = {1, 0, 0, 1, 0}, i.e., rows 0 and 3 are null
|
@@ -508,12 +508,12 @@ class InvertedIndexArrayTest : public testing::Test {
|
508 | 508 | TabletIndex idx_meta;
|
509 | 509 | idx_meta.index_type();
|
510 | 510 | idx_meta.init_from_pb(*index_meta_pb.get());
|
511 |
| - auto index_file_writer = std::make_unique<InvertedIndexFileWriter>( |
| 511 | + auto index_file_writer = std::make_unique<XIndexFileWriter>( |
512 | 512 | fs, index_path_prefix, std::string {rowset_id}, seg_id,
|
513 | 513 | InvertedIndexStorageFormatPB::V1);
|
514 |
| - std::unique_ptr<segment_v2::InvertedIndexColumnWriter> _inverted_index_builder = nullptr; |
515 |
| - EXPECT_EQ(InvertedIndexColumnWriter::create(field, &_inverted_index_builder, |
516 |
| - index_file_writer.get(), &idx_meta), |
| 514 | + std::unique_ptr<segment_v2::IndexColumnWriter> _inverted_index_builder = nullptr; |
| 515 | + EXPECT_EQ(IndexColumnWriter::create(field, &_inverted_index_builder, index_file_writer.get(), |
| 516 | + &idx_meta), |
517 | 517 | Status::OK());
|
518 | 518 |
|
519 | 519 | // Simulate outer null cases: 5 rows, outer null map = {1, 0, 0, 1, 0}, i.e., rows 0 and 3 are null
|
@@ -621,11 +621,11 @@ class InvertedIndexArrayTest : public testing::Test {
|
621 | 621 |
|
622 | 622 | TabletIndex idx_meta;
|
623 | 623 | idx_meta.init_from_pb(*index_meta_pb.get());
|
624 |
| - auto index_file_writer = std::make_unique<InvertedIndexFileWriter>( |
| 624 | + auto index_file_writer = std::make_unique<XIndexFileWriter>( |
625 | 625 | fs, index_path_prefix, "multi_block", 0, InvertedIndexStorageFormatPB::V1);
|
626 |
| - std::unique_ptr<segment_v2::InvertedIndexColumnWriter> _inverted_index_builder = nullptr; |
627 |
| - EXPECT_EQ(InvertedIndexColumnWriter::create(field, &_inverted_index_builder, |
628 |
| - index_file_writer.get(), &idx_meta), |
| 626 | + std::unique_ptr<segment_v2::IndexColumnWriter> _inverted_index_builder = nullptr; |
| 627 | + EXPECT_EQ(IndexColumnWriter::create(field, &_inverted_index_builder, index_file_writer.get(), |
| 628 | + &idx_meta), |
629 | 629 | Status::OK());
|
630 | 630 |
|
631 | 631 | ExpectedDocMap merged_expected;
|
@@ -818,12 +818,12 @@ class InvertedIndexArrayTest : public testing::Test {
|
818 | 818 |
|
819 | 819 | TabletIndex idx_meta;
|
820 | 820 | idx_meta.init_from_pb(*index_meta_pb.get());
|
821 |
| - auto index_file_writer = std::make_unique<InvertedIndexFileWriter>( |
| 821 | + auto index_file_writer = std::make_unique<XIndexFileWriter>( |
822 | 822 | fs, index_path_prefix, std::string {rowset_id}, seg_id,
|
823 | 823 | InvertedIndexStorageFormatPB::V1);
|
824 |
| - std::unique_ptr<segment_v2::InvertedIndexColumnWriter> _inverted_index_builder = nullptr; |
825 |
| - EXPECT_EQ(InvertedIndexColumnWriter::create(field, &_inverted_index_builder, |
826 |
| - index_file_writer.get(), &idx_meta), |
| 824 | + std::unique_ptr<segment_v2::IndexColumnWriter> _inverted_index_builder = nullptr; |
| 825 | + EXPECT_EQ(IndexColumnWriter::create(field, &_inverted_index_builder, index_file_writer.get(), |
| 826 | + &idx_meta), |
827 | 827 | Status::OK());
|
828 | 828 |
|
829 | 829 | vectorized::DataTypePtr inner_int = std::make_shared<vectorized::DataTypeInt32>();
|
@@ -960,12 +960,12 @@ class InvertedIndexArrayTest : public testing::Test {
|
960 | 960 |
|
961 | 961 | TabletIndex idx_meta;
|
962 | 962 | idx_meta.init_from_pb(*index_meta_pb.get());
|
963 |
| - auto index_file_writer = std::make_unique<InvertedIndexFileWriter>( |
| 963 | + auto index_file_writer = std::make_unique<XIndexFileWriter>( |
964 | 964 | fs, index_path_prefix, std::string {rowset_id}, seg_id,
|
965 | 965 | InvertedIndexStorageFormatPB::V1);
|
966 |
| - std::unique_ptr<segment_v2::InvertedIndexColumnWriter> _inverted_index_builder = nullptr; |
967 |
| - EXPECT_EQ(InvertedIndexColumnWriter::create(field, &_inverted_index_builder, |
968 |
| - index_file_writer.get(), &idx_meta), |
| 966 | + std::unique_ptr<segment_v2::IndexColumnWriter> _inverted_index_builder = nullptr; |
| 967 | + EXPECT_EQ(IndexColumnWriter::create(field, &_inverted_index_builder, index_file_writer.get(), |
| 968 | + &idx_meta), |
969 | 969 | Status::OK());
|
970 | 970 |
|
971 | 971 | // Construct inner array type: DataTypeArray(DataTypeNullable(DataTypeString))
|
|
0 commit comments