database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity
Files
ebbe6f650ca572b9dfcce60951b7dfc8d134b4a6
doris/be/test/olap/cumulative_compaction_policy_test.cpp
Xinyi Zou e17aef9467 [refactor] refactor the implement of MemTracker, and related usage (#8322) 
Modify the implementation of MemTracker:
1. Simplify a lot of useless logic;
2. Added MemTrackerTaskPool, as the ancestor of all query and import trackers, This is used to track the local memory usage of all tasks executing;
3. Add cosume/release cache, trigger a cosume/release when the memory accumulation exceeds the parameter mem_tracker_consume_min_size_bytes;
4. Add a new memory leak detection mode (Experimental feature), throw an exception when the remaining statistical value is greater than the specified range when the MemTracker is destructed, and print the accurate statistical value in HTTP, the parameter memory_leak_detection
5. Added Virtual MemTracker, cosume/release will not sync to parent. It will be used when introducing TCMalloc Hook to record memory later, to record the specified memory independently;
6. Modify the GC logic, register the buffer cached in DiskIoMgr as a GC function, and add other GC functions later;
7. Change the global root node from Root MemTracker to Process MemTracker, and remove Process MemTracker in exec_env;
8. Modify the macro that detects whether the memory has reached the upper limit, modify the parameters and default behavior of creating MemTracker, modify the error message format in mem_limit_exceeded, extend and apply transfer_to, remove Metric in MemTracker, etc.;

Modify where MemTracker is used:
1. MemPool adds a constructor to create a temporary tracker to avoid a lot of redundant code;
2. Added trackers for global objects such as ChunkAllocator and StorageEngine;
3. Added more fine-grained trackers such as ExprContext;
4. RuntimeState removes FragmentMemTracker, that is, PlanFragmentExecutor mem_tracker, which was previously used for independent statistical scan process memory, and replaces it with _scanner_mem_tracker in OlapScanNode;
5. MemTracker is no longer recorded in ReservationTracker, and ReservationTracker will be removed later;
2022-03-11 22:04:23 +08:00

1070 lines
37 KiB
C++
Raw Blame History

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
#include "olap/cumulative_compaction_policy.h"
#include <gtest/gtest.h>
#include <sstream>
#include "olap/cumulative_compaction.h"
#include "olap/rowset/rowset_meta.h"
#include "olap/tablet_meta.h"
namespace doris {
class TestNumBasedCumulativeCompactionPolicy : public testing::Test {
public:
TestNumBasedCumulativeCompactionPolicy() {}
void SetUp() {
_tablet_meta = static_cast<TabletMetaSharedPtr>(
new TabletMeta(1, 2, 15673, 4, 5, TTabletSchema(), 6, {{7, 8}}, UniqueId(9, 10),
TTabletType::TABLET_TYPE_DISK, TStorageMedium::HDD));
_json_rowset_meta = R"({
"rowset_id": 540081,
"tablet_id": 15673,
"txn_id": 4042,
"tablet_schema_hash": 567997577,
"rowset_type": "BETA_ROWSET",
"rowset_state": "VISIBLE",
"start_version": 2,
"end_version": 2,
"num_rows": 3929,
"total_disk_size": 84699,
"data_disk_size": 84464,
"index_disk_size": 235,
"empty": false,
"load_id": {
"hi": -5350970832824939812,
"lo": -6717994719194512122
},
"creation_time": 1553765670,
"alpha_rowset_extra_meta_pb": {
"segment_groups": [
{
"segment_group_id": 0,
"num_segments": 2,
"index_size": 132,
"data_size": 576,
"num_rows": 5,
"zone_maps": [
{
"min": "MQ==",
"max": "NQ==",
"null_flag": false
},
{
"min": "MQ==",
"max": "Mw==",
"null_flag": false
},
{
"min": "J2J1c2gn",
"max": "J3RvbSc=",
"null_flag": false
}
],
"empty": false
},
{
"segment_group_id": 1,
"num_segments": 1,
"index_size": 132,
"data_size": 576,
"num_rows": 5,
"zone_maps": [
{
"min": "MQ==",
"max": "NQ==",
"null_flag": false
},
{
"min": "MQ==",
"max": "Mw==",
"null_flag": false
},
{
"min": "J2J1c2gn",
"max": "J3RvbSc=",
"null_flag": false
}
],
"empty": false
}
]
}
})";
}
void TearDown() {}
void init_rs_meta(RowsetMetaSharedPtr& pb1, int64_t start, int64_t end) {
pb1->init_from_json(_json_rowset_meta);
pb1->set_start_version(start);
pb1->set_end_version(end);
pb1->set_creation_time(10000);
}
void init_all_rs_meta(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 0);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 1, 1);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 2, 2);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 3, 3);
rs_metas->push_back(ptr4);
RowsetMetaSharedPtr ptr5(new RowsetMeta());
init_rs_meta(ptr5, 4, 4);
rs_metas->push_back(ptr5);
}
void init_all_rs_meta_cal_point(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 1);
ptr1->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 2, 3);
ptr2->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 4, 4);
ptr3->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 5, 5);
ptr4->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr4);
}
void init_all_rs_meta_delete(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 1);
ptr1->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 2, 3);
ptr2->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 4, 4);
ptr3->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 5, 5);
DeletePredicatePB del;
del.add_sub_predicates("a = 1");
del.set_version(5);
ptr4->set_delete_predicate(del);
ptr4->set_segments_overlap(OVERLAP_UNKNOWN);
rs_metas->push_back(ptr4);
RowsetMetaSharedPtr ptr5(new RowsetMeta());
init_rs_meta(ptr5, 6, 6);
ptr5->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr5);
}
protected:
std::string _json_rowset_meta;
TabletMetaSharedPtr _tablet_meta;
};
TEST_F(TestNumBasedCumulativeCompactionPolicy, calc_cumulative_compaction_score) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_all_rs_meta(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_NUM_BASED_POLICY));
_tablet->init();
std::shared_ptr<CumulativeCompactionPolicy> cumulative_compaction_policy =
CumulativeCompactionPolicyFactory::create_cumulative_compaction_policy(
CUMULATIVE_NUM_BASED_POLICY);
const uint32_t score = _tablet->calc_compaction_score(CompactionType::CUMULATIVE_COMPACTION,
cumulative_compaction_policy);
ASSERT_EQ(15, score);
}
TEST_F(TestNumBasedCumulativeCompactionPolicy, calculate_cumulative_point) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_all_rs_meta_cal_point(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_NUM_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
ASSERT_EQ(4, _tablet->cumulative_layer_point());
}
TEST_F(TestNumBasedCumulativeCompactionPolicy, pick_candidate_rowsets) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_all_rs_meta_cal_point(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_NUM_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
ASSERT_EQ(2, candidate_rowsets.size());
}
TEST_F(TestNumBasedCumulativeCompactionPolicy, pick_input_rowsets_normal) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_all_rs_meta_cal_point(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_NUM_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
NumBasedCumulativeCompactionPolicy policy;
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
std::vector<RowsetSharedPtr> input_rowsets;
Version last_delete_version{-1, -1};
size_t compaction_score = 0;
policy.pick_input_rowsets(_tablet.get(), candidate_rowsets, 10, 5, &input_rowsets,
&last_delete_version, &compaction_score);
ASSERT_EQ(2, input_rowsets.size());
ASSERT_EQ(6, compaction_score);
ASSERT_EQ(-1, last_delete_version.first);
ASSERT_EQ(-1, last_delete_version.second);
}
TEST_F(TestNumBasedCumulativeCompactionPolicy, pick_input_rowsets_delete) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_all_rs_meta_delete(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_NUM_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
NumBasedCumulativeCompactionPolicy policy;
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
std::vector<RowsetSharedPtr> input_rowsets;
Version last_delete_version{-1, -1};
size_t compaction_score = 0;
policy.pick_input_rowsets(_tablet.get(), candidate_rowsets, 10, 5, &input_rowsets,
&last_delete_version, &compaction_score);
ASSERT_EQ(1, input_rowsets.size());
ASSERT_EQ(3, compaction_score);
ASSERT_EQ(5, last_delete_version.first);
ASSERT_EQ(5, last_delete_version.second);
}
class TestSizeBasedCumulativeCompactionPolicy : public testing::Test {
public:
TestSizeBasedCumulativeCompactionPolicy() {}
void SetUp() {
config::cumulative_size_based_promotion_size_mbytes = 1024;
config::cumulative_size_based_promotion_ratio = 0.05;
config::cumulative_size_based_promotion_min_size_mbytes = 64;
config::cumulative_size_based_compaction_lower_size_mbytes = 64;
_tablet_meta = static_cast<TabletMetaSharedPtr>(
new TabletMeta(1, 2, 15673, 4, 5, TTabletSchema(), 6, {{7, 8}}, UniqueId(9, 10),
TTabletType::TABLET_TYPE_DISK, TStorageMedium::HDD));
_json_rowset_meta = R"({
"rowset_id": 540081,
"tablet_id": 15673,
"txn_id": 4042,
"tablet_schema_hash": 567997577,
"rowset_type": "BETA_ROWSET",
"rowset_state": "VISIBLE",
"start_version": 2,
"end_version": 2,
"num_rows": 3929,
"total_disk_size": 41,
"data_disk_size": 41,
"index_disk_size": 235,
"empty": false,
"load_id": {
"hi": -5350970832824939812,
"lo": -6717994719194512122
},
"creation_time": 1553765670,
"alpha_rowset_extra_meta_pb": {
"segment_groups": [
{
"segment_group_id": 0,
"num_segments": 2,
"index_size": 132,
"data_size": 576,
"num_rows": 5,
"zone_maps": [
{
"min": "MQ==",
"max": "NQ==",
"null_flag": false
},
{
"min": "MQ==",
"max": "Mw==",
"null_flag": false
},
{
"min": "J2J1c2gn",
"max": "J3RvbSc=",
"null_flag": false
}
],
"empty": false
},
{
"segment_group_id": 1,
"num_segments": 1,
"index_size": 132,
"data_size": 576,
"num_rows": 5,
"zone_maps": [
{
"min": "MQ==",
"max": "NQ==",
"null_flag": false
},
{
"min": "MQ==",
"max": "Mw==",
"null_flag": false
},
{
"min": "J2J1c2gn",
"max": "J3RvbSc=",
"null_flag": false
}
],
"empty": false
}
]
}
})";
}
void TearDown() {}
void init_rs_meta(RowsetMetaSharedPtr& pb1, int64_t start, int64_t end) {
pb1->init_from_json(_json_rowset_meta);
pb1->set_start_version(start);
pb1->set_end_version(end);
pb1->set_total_disk_size(41);
pb1->set_creation_time(10000);
}
void init_rs_meta_small_base(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 0);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 1, 1);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 2, 2);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 3, 3);
rs_metas->push_back(ptr4);
RowsetMetaSharedPtr ptr5(new RowsetMeta());
init_rs_meta(ptr5, 4, 4);
rs_metas->push_back(ptr5);
}
void init_rs_meta_big_base(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 1);
ptr1->set_total_disk_size(1024 * 1024 * 1024);
ptr1->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 2, 3);
ptr2->set_total_disk_size(65 * 1024 * 1024);
ptr2->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 4, 5);
ptr3->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 6, 6);
ptr4->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr4);
RowsetMetaSharedPtr ptr5(new RowsetMeta());
init_rs_meta(ptr5, 7, 7);
rs_metas->push_back(ptr5);
}
void init_rs_meta_pick_promotion(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 1);
ptr1->set_total_disk_size(1024 * 1024 * 1024);
ptr1->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 2, 3);
ptr2->set_total_disk_size(65 * 1024 * 1024);
ptr2->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 4, 5);
ptr3->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 6, 6);
ptr4->set_total_disk_size(65 * 1024 * 1024);
ptr4->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr4);
}
void init_rs_meta_pick_not_same_level(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 1);
ptr1->set_total_disk_size(21474836480L); // 20G
ptr1->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 2, 3);
ptr2->set_total_disk_size(129 * 1024 * 1024);
ptr2->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 4, 5);
ptr3->set_total_disk_size(12 * 1024 * 1024);
ptr3->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 6, 6);
ptr4->set_segments_overlap(OVERLAPPING);
ptr4->set_total_disk_size(12 * 1024 * 1024);
rs_metas->push_back(ptr4);
RowsetMetaSharedPtr ptr5(new RowsetMeta());
init_rs_meta(ptr5, 7, 7);
rs_metas->push_back(ptr5);
RowsetMetaSharedPtr ptr6(new RowsetMeta());
init_rs_meta(ptr6, 8, 8);
rs_metas->push_back(ptr6);
}
void init_rs_meta_pick_empty(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 1);
ptr1->set_total_disk_size(21474836480L); // 20G
ptr1->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 2, 3);
ptr2->set_total_disk_size(257 * 1024 * 1024);
ptr2->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 4, 5);
ptr3->set_total_disk_size(129 * 1024 * 1024);
ptr3->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
ptr4->set_total_disk_size(65 * 1024 * 1024);
init_rs_meta(ptr4, 6, 6);
ptr4->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr4);
}
void init_rs_meta_pick_empty_not_reach_min_limit(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 1);
ptr1->set_total_disk_size(21474836480L); // 20G
ptr1->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 2, 3);
ptr2->set_total_disk_size(257 * 1024 * 1024);
ptr2->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 4, 5);
ptr3->set_total_disk_size(1 * 1024 * 1024);
ptr3->set_num_segments(1);
ptr3->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 6, 6);
ptr4->set_total_disk_size(1 * 1024 * 1024);
ptr4->set_num_segments(1);
ptr4->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr4);
RowsetMetaSharedPtr ptr5(new RowsetMeta());
init_rs_meta(ptr5, 7, 7);
ptr5->set_total_disk_size(1 * 1024 * 1024);
ptr5->set_num_segments(1);
ptr5->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr5);
}
void init_all_rs_meta_cal_point(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 1);
ptr1->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 2, 3);
ptr2->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 4, 4);
ptr3->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 5, 5);
ptr4->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr4);
}
void init_all_rs_meta_delete(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 1);
ptr1->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 2, 3);
ptr2->set_segments_overlap(NONOVERLAPPING);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 4, 4);
ptr3->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr4(new RowsetMeta());
init_rs_meta(ptr4, 5, 5);
DeletePredicatePB del;
del.add_sub_predicates("a = 1");
del.set_version(5);
ptr4->set_delete_predicate(del);
ptr4->set_segments_overlap(OVERLAP_UNKNOWN);
rs_metas->push_back(ptr4);
RowsetMetaSharedPtr ptr5(new RowsetMeta());
init_rs_meta(ptr5, 6, 6);
ptr5->set_segments_overlap(OVERLAPPING);
rs_metas->push_back(ptr5);
}
void init_rs_meta_missing_version(std::vector<RowsetMetaSharedPtr>* rs_metas) {
RowsetMetaSharedPtr ptr1(new RowsetMeta());
init_rs_meta(ptr1, 0, 0);
rs_metas->push_back(ptr1);
RowsetMetaSharedPtr ptr2(new RowsetMeta());
init_rs_meta(ptr2, 1, 1);
rs_metas->push_back(ptr2);
RowsetMetaSharedPtr ptr3(new RowsetMeta());
init_rs_meta(ptr3, 2, 2);
rs_metas->push_back(ptr3);
RowsetMetaSharedPtr ptr5(new RowsetMeta());
init_rs_meta(ptr5, 4, 4);
rs_metas->push_back(ptr5);
}
protected:
std::string _json_rowset_meta;
TabletMetaSharedPtr _tablet_meta;
};
TEST_F(TestSizeBasedCumulativeCompactionPolicy, calc_cumulative_compaction_score) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_small_base(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::shared_ptr<CumulativeCompactionPolicy> cumulative_compaction_policy =
CumulativeCompactionPolicyFactory::create_cumulative_compaction_policy(
CUMULATIVE_SIZE_BASED_POLICY);
const uint32_t score = _tablet->calc_compaction_score(CompactionType::CUMULATIVE_COMPACTION,
cumulative_compaction_policy);
ASSERT_EQ(15, score);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, calc_cumulative_compaction_score_big_base) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_big_base(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::shared_ptr<CumulativeCompactionPolicy> cumulative_compaction_policy =
CumulativeCompactionPolicyFactory::create_cumulative_compaction_policy(
CUMULATIVE_SIZE_BASED_POLICY);
const uint32_t score = _tablet->calc_compaction_score(CompactionType::CUMULATIVE_COMPACTION,
cumulative_compaction_policy);
ASSERT_EQ(7, score);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, calculate_cumulative_point_big_base) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_big_base(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
ASSERT_EQ(4, _tablet->cumulative_layer_point());
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, calculate_cumulative_point_overlap) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_all_rs_meta_cal_point(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
ASSERT_EQ(2, _tablet->cumulative_layer_point());
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, pick_candidate_rowsets) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_all_rs_meta_cal_point(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
ASSERT_EQ(3, candidate_rowsets.size());
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, pick_candidate_rowsets_big_base) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_big_base(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
ASSERT_EQ(3, candidate_rowsets.size());
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, pick_input_rowsets_normal) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_small_base(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
std::vector<RowsetSharedPtr> input_rowsets;
Version last_delete_version{-1, -1};
size_t compaction_score = 0;
_tablet->_cumulative_compaction_policy->pick_input_rowsets(
_tablet.get(), candidate_rowsets, 10, 5, &input_rowsets, &last_delete_version,
&compaction_score);
ASSERT_EQ(4, input_rowsets.size());
ASSERT_EQ(12, compaction_score);
ASSERT_EQ(-1, last_delete_version.first);
ASSERT_EQ(-1, last_delete_version.second);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, pick_input_rowsets_big_base) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_big_base(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
std::vector<RowsetSharedPtr> input_rowsets;
Version last_delete_version{-1, -1};
size_t compaction_score = 0;
_tablet->_cumulative_compaction_policy->pick_input_rowsets(
_tablet.get(), candidate_rowsets, 10, 5, &input_rowsets, &last_delete_version,
&compaction_score);
ASSERT_EQ(3, input_rowsets.size());
ASSERT_EQ(7, compaction_score);
ASSERT_EQ(-1, last_delete_version.first);
ASSERT_EQ(-1, last_delete_version.second);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, pick_input_rowsets_promotion) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_pick_promotion(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
std::vector<RowsetSharedPtr> input_rowsets;
Version last_delete_version{-1, -1};
size_t compaction_score = 0;
_tablet->_cumulative_compaction_policy->pick_input_rowsets(
_tablet.get(), candidate_rowsets, 10, 5, &input_rowsets, &last_delete_version,
&compaction_score);
ASSERT_EQ(2, input_rowsets.size());
ASSERT_EQ(4, compaction_score);
ASSERT_EQ(-1, last_delete_version.first);
ASSERT_EQ(-1, last_delete_version.second);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, pick_input_rowsets_not_same_level) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_pick_not_same_level(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
std::vector<RowsetSharedPtr> input_rowsets;
Version last_delete_version{-1, -1};
size_t compaction_score = 0;
_tablet->_cumulative_compaction_policy->pick_input_rowsets(
_tablet.get(), candidate_rowsets, 10, 5, &input_rowsets, &last_delete_version,
&compaction_score);
ASSERT_EQ(4, input_rowsets.size());
ASSERT_EQ(10, compaction_score);
ASSERT_EQ(-1, last_delete_version.first);
ASSERT_EQ(-1, last_delete_version.second);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, pick_input_rowsets_empty) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_pick_empty(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
std::vector<RowsetSharedPtr> input_rowsets;
Version last_delete_version{-1, -1};
size_t compaction_score = 0;
_tablet->_cumulative_compaction_policy->pick_input_rowsets(
_tablet.get(), candidate_rowsets, 10, 5, &input_rowsets, &last_delete_version,
&compaction_score);
ASSERT_EQ(0, input_rowsets.size());
ASSERT_EQ(0, compaction_score);
ASSERT_EQ(-1, last_delete_version.first);
ASSERT_EQ(-1, last_delete_version.second);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, pick_input_rowsets_not_reach_min_limit) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_pick_empty_not_reach_min_limit(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
std::vector<RowsetSharedPtr> input_rowsets;
Version last_delete_version{-1, -1};
size_t compaction_score = 0;
_tablet->_cumulative_compaction_policy->pick_input_rowsets(
_tablet.get(), candidate_rowsets, 10, 5, &input_rowsets, &last_delete_version,
&compaction_score);
ASSERT_EQ(0, input_rowsets.size());
ASSERT_EQ(0, compaction_score);
ASSERT_EQ(-1, last_delete_version.first);
ASSERT_EQ(-1, last_delete_version.second);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, pick_input_rowsets_delete) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_all_rs_meta_delete(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
std::vector<RowsetSharedPtr> candidate_rowsets;
_tablet->pick_candidate_rowsets_to_cumulative_compaction(1000, &candidate_rowsets);
std::vector<RowsetSharedPtr> input_rowsets;
Version last_delete_version{-1, -1};
size_t compaction_score = 0;
_tablet->_cumulative_compaction_policy->pick_input_rowsets(
_tablet.get(), candidate_rowsets, 10, 5, &input_rowsets, &last_delete_version,
&compaction_score);
ASSERT_EQ(2, input_rowsets.size());
ASSERT_EQ(4, compaction_score);
ASSERT_EQ(5, last_delete_version.first);
ASSERT_EQ(5, last_delete_version.second);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, _calc_promotion_size_big) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_pick_not_same_level(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
SizeBasedCumulativeCompactionPolicy* policy =
dynamic_cast<SizeBasedCumulativeCompactionPolicy*>(
_tablet->_cumulative_compaction_policy.get());
ASSERT_EQ(1073741824, policy->_tablet_size_based_promotion_size);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, _calc_promotion_size_small) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_small_base(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
_tablet->calculate_cumulative_point();
SizeBasedCumulativeCompactionPolicy* policy =
dynamic_cast<SizeBasedCumulativeCompactionPolicy*>(
_tablet->_cumulative_compaction_policy.get());
ASSERT_EQ(67108864, policy->_tablet_size_based_promotion_size);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, _level_size) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_small_base(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
SizeBasedCumulativeCompactionPolicy* policy =
dynamic_cast<SizeBasedCumulativeCompactionPolicy*>(
_tablet->_cumulative_compaction_policy.get());
ASSERT_EQ(4, policy->_levels.size());
ASSERT_EQ(536870912, policy->_levels[0]);
ASSERT_EQ(268435456, policy->_levels[1]);
ASSERT_EQ(134217728, policy->_levels[2]);
ASSERT_EQ(67108864, policy->_levels[3]);
}
TEST_F(TestSizeBasedCumulativeCompactionPolicy, _pick_missing_version_cumulative_compaction) {
std::vector<RowsetMetaSharedPtr> rs_metas;
init_rs_meta_missing_version(&rs_metas);
for (auto& rowset : rs_metas) {
_tablet_meta->add_rs_meta(rowset);
}
TabletSharedPtr _tablet(new Tablet(_tablet_meta, nullptr, CUMULATIVE_SIZE_BASED_POLICY));
_tablet->init();
// has miss version
std::vector<RowsetSharedPtr> rowsets;
rowsets.push_back(_tablet->get_rowset_by_version({0, 0}));
rowsets.push_back(_tablet->get_rowset_by_version({1, 1}));
rowsets.push_back(_tablet->get_rowset_by_version({2, 2}));
rowsets.push_back(_tablet->get_rowset_by_version({4, 4}));
std::shared_ptr<MemTracker> mem_tracker(new MemTracker());
CumulativeCompaction compaction(_tablet, mem_tracker);
compaction.find_longest_consecutive_version(&rowsets, nullptr);
ASSERT_EQ(3, rowsets.size());
ASSERT_EQ(2, rowsets[2]->end_version());
// no miss version
std::vector<RowsetSharedPtr> rowsets2;
rowsets2.push_back(_tablet->get_rowset_by_version({0, 0}));
compaction.find_longest_consecutive_version(&rowsets2, nullptr);
ASSERT_EQ(1, rowsets2.size());
ASSERT_EQ(0, rowsets[0]->end_version());
// no version
std::vector<RowsetSharedPtr> rowsets3;
compaction.find_longest_consecutive_version(&rowsets3, nullptr);
ASSERT_EQ(0, rowsets3.size());
}
} // namespace doris
// @brief Test Stub
int main(int argc, char** argv) {
testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}
Reference in New Issue View Git Blame Copy Permalink