| // 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 "cloud/cloud_cumulative_compaction.h" |
| |
| #include <gen_cpp/cloud.pb.h> |
| |
| #include <random> |
| |
| #include "cloud/cloud_meta_mgr.h" |
| #include "cloud/cloud_tablet_mgr.h" |
| #include "cloud/config.h" |
| #include "common/config.h" |
| #include "common/logging.h" |
| #include "common/status.h" |
| #include "cpp/sync_point.h" |
| #include "service/backend_options.h" |
| #include "storage/compaction/compaction.h" |
| #include "storage/compaction/cumulative_compaction_policy.h" |
| #include "storage/compaction/cumulative_compaction_time_series_policy.h" |
| #include "util/debug_points.h" |
| #include "util/trace.h" |
| #include "util/uuid_generator.h" |
| |
| namespace doris { |
| using namespace ErrorCode; |
| |
| bvar::Adder<uint64_t> cumu_output_size("cumu_compaction", "output_size"); |
| bvar::LatencyRecorder g_cu_compaction_hold_delete_bitmap_lock_time_ms( |
| "cu_compaction_hold_delete_bitmap_lock_time_ms"); |
| |
| CloudCumulativeCompaction::CloudCumulativeCompaction(CloudStorageEngine& engine, |
| CloudTabletSPtr tablet) |
| : CloudCompactionMixin(engine, tablet, |
| "BaseCompaction:" + std::to_string(tablet->tablet_id())), |
| _enable_parallel_cumu_compaction(config::enable_parallel_cumu_compaction) {} |
| |
| CloudCumulativeCompaction::~CloudCumulativeCompaction() = default; |
| |
| Status CloudCumulativeCompaction::prepare_compact() { |
| DBUG_EXECUTE_IF("CloudCumulativeCompaction.prepare_compact.sleep", { sleep(5); }) |
| Status st; |
| Defer defer_set_st([&] { |
| if (!st.ok()) { |
| cloud_tablet()->set_last_cumu_compaction_status(st.to_string()); |
| } |
| }); |
| if (_tablet->tablet_state() != TABLET_RUNNING && |
| (!config::enable_new_tablet_do_compaction || |
| static_cast<CloudTablet*>(_tablet.get())->alter_version() == -1)) { |
| st = Status::InternalError("invalid tablet state. tablet_id={}", _tablet->tablet_id()); |
| return st; |
| } |
| |
| std::vector<std::shared_ptr<CloudCumulativeCompaction>> cumu_compactions; |
| _engine.get_cumu_compaction(_tablet->tablet_id(), cumu_compactions); |
| for (const auto& cumu : cumu_compactions) { |
| _min_conflict_version = |
| std::min(_min_conflict_version, cumu->_input_rowsets.front()->start_version()); |
| _max_conflict_version = |
| std::max(_max_conflict_version, cumu->_input_rowsets.back()->end_version()); |
| } |
| |
| bool need_sync_tablet = true; |
| { |
| std::shared_lock rlock(_tablet->get_header_lock()); |
| // If number of rowsets is equal to approximate_num_rowsets, it is very likely that this tablet has been |
| // synchronized with meta-service. |
| if (_tablet->tablet_meta()->all_rs_metas().size() >= |
| cloud_tablet()->fetch_add_approximate_num_rowsets(0) && |
| cloud_tablet()->last_sync_time_s > 0) { |
| need_sync_tablet = false; |
| } |
| } |
| if (need_sync_tablet) { |
| st = cloud_tablet()->sync_rowsets(); |
| RETURN_IF_ERROR(st); |
| } |
| |
| // pick rowsets to compact |
| st = pick_rowsets_to_compact(); |
| if (!st.ok()) { |
| if (_last_delete_version.first != -1) { |
| // we meet a delete version, should increase the cumulative point to let base compaction handle the delete version. |
| // plus 1 to skip the delete version. |
| // NOTICE: after that, the cumulative point may be larger than max version of this tablet, but it doesn't matter. |
| // The picker only preserves a delete version reached continuously from the |
| // cumulative point. |
| DORIS_CHECK_LE(_picked_cumulative_point, _last_delete_version.first); |
| update_cumulative_point(_picked_cumulative_point, _last_delete_version.first + 1); |
| if (!config::enable_sleep_between_delete_cumu_compaction) { |
| st = Status::Error<CUMULATIVE_MEET_DELETE_VERSION>( |
| "cumulative compaction meet delete version"); |
| } |
| } |
| return st; |
| } |
| |
| for (auto& rs : _input_rowsets) { |
| _input_row_num += rs->num_rows(); |
| _input_segments += rs->num_segments(); |
| _input_rowsets_data_size += rs->data_disk_size(); |
| _input_rowsets_index_size += rs->index_disk_size(); |
| _input_rowsets_total_size += rs->total_disk_size(); |
| } |
| LOG_INFO("start CloudCumulativeCompaction, tablet_id={}, range=[{}-{}]", _tablet->tablet_id(), |
| _input_rowsets.front()->start_version(), _input_rowsets.back()->end_version()) |
| .tag("job_id", _uuid) |
| .tag("input_rowsets", _input_rowsets.size()) |
| .tag("input_rows", _input_row_num) |
| .tag("input_segments", _input_segments) |
| .tag("input_rowsets_data_size", _input_rowsets_data_size) |
| .tag("input_rowsets_index_size", _input_rowsets_index_size) |
| .tag("input_rowsets_total_size", _input_rowsets_total_size) |
| .tag("tablet_max_version", cloud_tablet()->max_version_unlocked()) |
| .tag("cumulative_point", cloud_tablet()->cumulative_layer_point()) |
| .tag("num_rowsets", cloud_tablet()->fetch_add_approximate_num_rowsets(0)) |
| .tag("cumu_num_rowsets", cloud_tablet()->fetch_add_approximate_cumu_num_rowsets(0)); |
| return st; |
| } |
| |
| Status CloudCumulativeCompaction::request_global_lock() { |
| // prepare compaction job |
| cloud::TabletJobInfoPB job; |
| auto idx = job.mutable_idx(); |
| idx->set_tablet_id(_tablet->tablet_id()); |
| idx->set_table_id(_tablet->table_id()); |
| idx->set_index_id(_tablet->index_id()); |
| idx->set_partition_id(_tablet->partition_id()); |
| auto compaction_job = job.add_compaction(); |
| compaction_job->set_id(_uuid); |
| compaction_job->set_initiator(BackendOptions::get_localhost() + ':' + |
| std::to_string(config::heartbeat_service_port)); |
| compaction_job->set_type(cloud::TabletCompactionJobPB::CUMULATIVE); |
| compaction_job->set_base_compaction_cnt(_base_compaction_cnt); |
| compaction_job->set_cumulative_compaction_cnt(_cumulative_compaction_cnt); |
| using namespace std::chrono; |
| int64_t now = duration_cast<seconds>(system_clock::now().time_since_epoch()).count(); |
| _expiration = now + config::compaction_timeout_seconds; |
| compaction_job->set_expiration(_expiration); |
| compaction_job->set_lease(now + config::lease_compaction_interval_seconds * 4); |
| |
| compaction_job->add_input_versions(_input_rowsets.front()->start_version()); |
| compaction_job->add_input_versions(_input_rowsets.back()->end_version()); |
| // Set input version range to let meta-service check version range conflict |
| compaction_job->set_check_input_versions_range(_enable_parallel_cumu_compaction); |
| cloud::StartTabletJobResponse resp; |
| Status st = _engine.meta_mgr().prepare_tablet_job(job, &resp); |
| if (!st.ok()) { |
| if (resp.status().code() == cloud::STALE_TABLET_CACHE) { |
| // set last_sync_time to 0 to force sync tablet next time |
| cloud_tablet()->last_sync_time_s = 0; |
| } else if (resp.status().code() == cloud::TABLET_NOT_FOUND) { |
| // tablet not found |
| cloud_tablet()->clear_cache(); |
| } else if (resp.status().code() == cloud::JOB_TABLET_BUSY) { |
| LOG_WARNING("failed to prepare cumu compaction") |
| .tag("job_id", _uuid) |
| .tag("msg", resp.status().msg()); |
| return Status::Error<CUMULATIVE_NO_SUITABLE_VERSION>( |
| "cumu no suitable versions: job tablet busy"); |
| } else if (resp.status().code() == cloud::JOB_CHECK_ALTER_VERSION) { |
| (static_cast<CloudTablet*>(_tablet.get()))->set_alter_version(resp.alter_version()); |
| std::stringstream ss; |
| ss << "failed to prepare cumu compaction. Check compaction input versions " |
| "failed in schema change. " |
| "input_version_start=" |
| << compaction_job->input_versions(0) |
| << " input_version_end=" << compaction_job->input_versions(1) |
| << " schema_change_alter_version=" << resp.alter_version(); |
| std::string msg = ss.str(); |
| LOG(WARNING) << msg; |
| return Status::InternalError(msg); |
| } |
| } |
| return st; |
| } |
| |
| Status CloudCumulativeCompaction::execute_compact() { |
| TEST_SYNC_POINT_RETURN_WITH_VALUE("CloudCumulativeCompaction::execute_compact_impl", |
| Status::OK(), this); |
| |
| SCOPED_ATTACH_TASK(_mem_tracker); |
| |
| using namespace std::chrono; |
| auto start = steady_clock::now(); |
| Status st; |
| Defer defer_set_st([&] { |
| cloud_tablet()->set_last_cumu_compaction_status(st.to_string()); |
| if (!st.ok()) { |
| cloud_tablet()->set_last_cumu_compaction_failure_time(UnixMillis()); |
| } else { |
| cloud_tablet()->set_last_cumu_compaction_success_time(UnixMillis()); |
| } |
| }); |
| st = CloudCompactionMixin::execute_compact(); |
| if (!st.ok()) { |
| LOG(WARNING) << "fail to do " << compaction_name() << ". res=" << st |
| << ", tablet=" << _tablet->tablet_id() |
| << ", output_version=" << _output_version; |
| return st; |
| } |
| LOG_INFO("finish CloudCumulativeCompaction, tablet_id={}, cost={}ms, range=[{}-{}]", |
| _tablet->tablet_id(), duration_cast<milliseconds>(steady_clock::now() - start).count(), |
| _input_rowsets.front()->start_version(), _input_rowsets.back()->end_version()) |
| .tag("job_id", _uuid) |
| .tag("input_rowsets", _input_rowsets.size()) |
| .tag("input_rows", _input_row_num) |
| .tag("input_segments", _input_segments) |
| .tag("input_rowsets_data_size", _input_rowsets_data_size) |
| .tag("input_rowsets_index_size", _input_rowsets_index_size) |
| .tag("input_rowsets_total_size", _input_rowsets_total_size) |
| .tag("output_rows", _output_rowset->num_rows()) |
| .tag("output_segments", _output_rowset->num_segments()) |
| .tag("output_rowset_data_size", _output_rowset->data_disk_size()) |
| .tag("output_rowset_index_size", _output_rowset->index_disk_size()) |
| .tag("output_rowset_total_size", _output_rowset->total_disk_size()) |
| .tag("tablet_max_version", _tablet->max_version_unlocked()) |
| .tag("cumulative_point", cloud_tablet()->cumulative_layer_point()) |
| .tag("num_rowsets", cloud_tablet()->fetch_add_approximate_num_rowsets(0)) |
| .tag("cumu_num_rowsets", cloud_tablet()->fetch_add_approximate_cumu_num_rowsets(0)) |
| .tag("local_read_time_us", _stats.cloud_local_read_time) |
| .tag("remote_read_time_us", _stats.cloud_remote_read_time) |
| .tag("local_read_bytes", _local_read_bytes_total) |
| .tag("remote_read_bytes", _remote_read_bytes_total); |
| |
| _state = CompactionState::SUCCESS; |
| |
| DorisMetrics::instance()->cumulative_compaction_deltas_total->increment(_input_rowsets.size()); |
| DorisMetrics::instance()->cumulative_compaction_bytes_total->increment( |
| _input_rowsets_total_size); |
| cumu_output_size << _output_rowset->total_disk_size(); |
| |
| st = Status::OK(); |
| return st; |
| } |
| |
| Status CloudCumulativeCompaction::modify_rowsets() { |
| // calculate new cumulative point |
| int64_t input_cumulative_point; |
| int64_t proposal_base_compaction_cnt; |
| int64_t proposal_cumulative_compaction_cnt; |
| TabletState input_tablet_state; |
| int64_t input_alter_version; |
| { |
| std::shared_lock rlock(_tablet->get_header_lock()); |
| input_cumulative_point = cloud_tablet()->cumulative_layer_point(); |
| proposal_base_compaction_cnt = cloud_tablet()->base_compaction_cnt(); |
| proposal_cumulative_compaction_cnt = cloud_tablet()->cumulative_compaction_cnt(); |
| input_tablet_state = _tablet->tablet_state(); |
| input_alter_version = cloud_tablet()->alter_version(); |
| } |
| auto compaction_policy = cloud_tablet()->tablet_meta()->compaction_policy(); |
| int64_t new_cumulative_point = input_cumulative_point; |
| if (!_enable_parallel_cumu_compaction && input_tablet_state == TABLET_NOTREADY && |
| _output_rowset->start_version() > input_cumulative_point) { |
| // Historical rowsets are absent from a schema-change target until conversion finishes. |
| DORIS_CHECK_LE(input_cumulative_point, input_alter_version); |
| DORIS_CHECK_GT(_output_rowset->start_version(), input_alter_version); |
| } else if (!_enable_parallel_cumu_compaction || |
| _output_rowset->start_version() == input_cumulative_point) { |
| new_cumulative_point = |
| _engine.cumu_compaction_policy(compaction_policy) |
| ->new_cumulative_point(cloud_tablet(), _output_rowset, _last_delete_version, |
| input_cumulative_point); |
| } |
| // commit compaction job |
| cloud::TabletJobInfoPB job; |
| auto idx = job.mutable_idx(); |
| idx->set_tablet_id(_tablet->tablet_id()); |
| idx->set_table_id(_tablet->table_id()); |
| idx->set_index_id(_tablet->index_id()); |
| idx->set_partition_id(_tablet->partition_id()); |
| auto compaction_job = job.add_compaction(); |
| compaction_job->set_id(_uuid); |
| compaction_job->set_initiator(BackendOptions::get_localhost() + ':' + |
| std::to_string(config::heartbeat_service_port)); |
| compaction_job->set_type(cloud::TabletCompactionJobPB::CUMULATIVE); |
| compaction_job->set_base_compaction_cnt(proposal_base_compaction_cnt); |
| compaction_job->set_cumulative_compaction_cnt(proposal_cumulative_compaction_cnt); |
| compaction_job->set_input_cumulative_point(input_cumulative_point); |
| compaction_job->set_output_cumulative_point(new_cumulative_point); |
| compaction_job->set_num_input_rows(_input_row_num); |
| compaction_job->set_num_output_rows(_output_rowset->num_rows()); |
| compaction_job->set_size_input_rowsets(_input_rowsets_total_size); |
| compaction_job->set_size_output_rowsets(_output_rowset->total_disk_size()); |
| compaction_job->set_num_input_segments(_input_segments); |
| compaction_job->set_num_output_segments(_output_rowset->num_segments()); |
| compaction_job->set_num_input_rowsets(num_input_rowsets()); |
| compaction_job->set_num_output_rowsets(1); |
| compaction_job->add_input_versions(_input_rowsets.front()->start_version()); |
| compaction_job->add_input_versions(_input_rowsets.back()->end_version()); |
| compaction_job->add_output_versions(_output_rowset->end_version()); |
| compaction_job->add_txn_id(_output_rowset->txn_id()); |
| compaction_job->add_output_rowset_ids(_output_rowset->rowset_id().to_string()); |
| compaction_job->set_index_size_input_rowsets(_input_rowsets_index_size); |
| compaction_job->set_segment_size_input_rowsets(_input_rowsets_data_size); |
| compaction_job->set_index_size_output_rowsets(_output_rowset->index_disk_size()); |
| compaction_job->set_segment_size_output_rowsets(_output_rowset->data_disk_size()); |
| |
| DBUG_EXECUTE_IF("CloudCumulativeCompaction::modify_rowsets.enable_spin_wait", { |
| LOG(INFO) << "CloudCumulativeCompaction::modify_rowsets.enable_spin_wait, start"; |
| while (DebugPoints::instance()->is_enable( |
| "CloudCumulativeCompaction::modify_rowsets.block")) { |
| std::this_thread::sleep_for(std::chrono::milliseconds(50)); |
| } |
| LOG(INFO) << "CloudCumulativeCompaction::modify_rowsets.enable_spin_wait, exit"; |
| }); |
| |
| // Block only NOTREADY tablets (SC new tablets) before compaction commit. |
| // RUNNING tablets (system tables, base tablets) are not affected. |
| DBUG_EXECUTE_IF("CloudCumulativeCompaction::modify_rowsets.block_notready", { |
| if (_tablet->tablet_state() == TABLET_NOTREADY) { |
| LOG(INFO) << "block NOTREADY tablet compaction before commit" |
| << ", tablet_id=" << _tablet->tablet_id() << ", output=[" |
| << _input_rowsets.front()->start_version() << "-" |
| << _input_rowsets.back()->end_version() << "]"; |
| while (DebugPoints::instance()->is_enable( |
| "CloudCumulativeCompaction::modify_rowsets.block_notready")) { |
| std::this_thread::sleep_for(std::chrono::milliseconds(50)); |
| } |
| LOG(INFO) << "release NOTREADY tablet compaction, tablet_id=" << _tablet->tablet_id(); |
| } |
| }); |
| |
| DBUG_EXECUTE_IF("CloudCumulativeCompaction::modify_rowsets.random_sleep", { |
| auto probability = dp->param("probability", dp->param("percent", 0.0)); |
| DORIS_CHECK(probability >= 0.0 && probability <= 1.0); |
| static thread_local std::mt19937 gen(std::random_device {}()); |
| std::bernoulli_distribution inject_sleep {probability}; |
| if (inject_sleep(gen)) { |
| auto max_sleep_ms = dp->param<int64_t>( |
| "max_sleep_ms", dp->param<int64_t>("max_sleep_time_ms", |
| dp->param<int64_t>("max_sleep_time", 0))); |
| DORIS_CHECK(max_sleep_ms >= 0); |
| std::uniform_int_distribution<int64_t> sleep_dist(0, max_sleep_ms); |
| auto sleep_ms = sleep_dist(gen); |
| LOG(INFO) << "CloudCumulativeCompaction::modify_rowsets.random_sleep" |
| << ", tablet_id=" << _tablet->tablet_id() << ", sleep_ms=" << sleep_ms |
| << ", probability=" << probability; |
| std::this_thread::sleep_for(std::chrono::milliseconds(sleep_ms)); |
| } |
| }); |
| |
| DBUG_EXECUTE_IF("CloudCumulativeCompaction::modify_rowsets.random_fail", { |
| auto probability = dp->param("probability", dp->param("percent", 0.0)); |
| DORIS_CHECK(probability >= 0.0 && probability <= 1.0); |
| static thread_local std::mt19937 gen(std::random_device {}()); |
| std::bernoulli_distribution inject_fail {probability}; |
| if (inject_fail(gen)) { |
| LOG(WARNING) << "CloudCumulativeCompaction::modify_rowsets.random_fail" |
| << ", tablet_id=" << _tablet->tablet_id() |
| << ", probability=" << probability; |
| return Status::InternalError( |
| "debug cloud cumulative compaction modify rowsets random failed"); |
| } |
| }); |
| |
| DeleteBitmapPtr output_rowset_delete_bitmap = nullptr; |
| int64_t initiator = this->initiator(); |
| int64_t get_delete_bitmap_lock_start_time = 0; |
| if (_tablet->keys_type() == KeysType::UNIQUE_KEYS && |
| _tablet->enable_unique_key_merge_on_write()) { |
| RETURN_IF_ERROR(cloud_tablet()->calc_delete_bitmap_for_compaction( |
| _input_rowsets, _output_rowset, *_rowid_conversion, compaction_type(), |
| _stats.merged_rows, _stats.filtered_rows, initiator, output_rowset_delete_bitmap, |
| _allow_delete_in_cumu_compaction, get_delete_bitmap_lock_start_time)); |
| LOG_INFO("update delete bitmap in CloudCumulativeCompaction, tablet_id={}, range=[{}-{}]", |
| _tablet->tablet_id(), _input_rowsets.front()->start_version(), |
| _input_rowsets.back()->end_version()) |
| .tag("job_id", _uuid) |
| .tag("initiator", initiator) |
| .tag("input_rowsets", _input_rowsets.size()) |
| .tag("input_rows", _input_row_num) |
| .tag("input_segments", _input_segments) |
| .tag("number_output_delete_bitmap", |
| output_rowset_delete_bitmap->delete_bitmap.size()); |
| compaction_job->set_delete_bitmap_lock_initiator(initiator); |
| } |
| |
| DBUG_EXECUTE_IF("CumulativeCompaction.modify_rowsets.trigger_abort_job_failed", { |
| LOG(INFO) << "CumulativeCompaction.modify_rowsets.trigger_abort_job_failed for tablet_id" |
| << cloud_tablet()->tablet_id(); |
| return Status::InternalError( |
| "CumulativeCompaction.modify_rowsets.trigger_abort_job_failed for tablet_id {}", |
| cloud_tablet()->tablet_id()); |
| }); |
| cloud::FinishTabletJobResponse resp; |
| auto st = _engine.meta_mgr().commit_tablet_job(job, &resp); |
| if (_tablet->keys_type() == KeysType::UNIQUE_KEYS && |
| _tablet->enable_unique_key_merge_on_write()) { |
| int64_t hold_delete_bitmap_lock_time_ms = |
| (MonotonicMicros() - get_delete_bitmap_lock_start_time) / 1000; |
| g_cu_compaction_hold_delete_bitmap_lock_time_ms << hold_delete_bitmap_lock_time_ms; |
| } |
| if (resp.has_alter_version()) { |
| (static_cast<CloudTablet*>(_tablet.get()))->set_alter_version(resp.alter_version()); |
| } |
| if (!st.ok()) { |
| if (resp.status().code() == cloud::TABLET_NOT_FOUND) { |
| cloud_tablet()->clear_cache(); |
| } else if (resp.status().code() == cloud::JOB_CHECK_ALTER_VERSION) { |
| std::stringstream ss; |
| ss << "failed to prepare cumu compaction. Check compaction input versions " |
| "failed in schema change. " |
| "input_version_start=" |
| << compaction_job->input_versions(0) |
| << " input_version_end=" << compaction_job->input_versions(1) |
| << " schema_change_alter_version=" << resp.alter_version(); |
| std::string msg = ss.str(); |
| LOG(WARNING) << msg; |
| return Status::InternalError(msg); |
| } |
| return st; |
| } |
| |
| auto& stats = resp.stats(); |
| LOG(INFO) << "tablet stats=" << stats.ShortDebugString(); |
| { |
| std::unique_lock wrlock(_tablet->get_header_lock()); |
| // clang-format off |
| cloud_tablet()->set_last_base_compaction_success_time(std::max(cloud_tablet()->last_base_compaction_success_time(), stats.last_base_compaction_time_ms())); |
| cloud_tablet()->set_last_cumu_compaction_success_time(std::max(cloud_tablet()->last_cumu_compaction_success_time(), stats.last_cumu_compaction_time_ms())); |
| cloud_tablet()->set_last_full_compaction_success_time(std::max(cloud_tablet()->last_full_compaction_success_time(), stats.last_full_compaction_time_ms())); |
| // clang-format on |
| if (!should_apply_cumulative_compaction_result(stats.cumulative_compaction_cnt())) { |
| return Status::OK(); |
| } |
| // Try to make output rowset visible immediately in tablet cache, instead of waiting for next synchronization from meta-service. |
| if (_input_rowsets.size() == 1) { |
| DCHECK_EQ(_output_rowset->version(), _input_rowsets[0]->version()); |
| // MUST NOT move input rowset to stale path |
| cloud_tablet()->add_rowsets({_output_rowset}, true, wrlock, true); |
| } else { |
| cloud_tablet()->delete_rowsets(_input_rowsets, wrlock); |
| cloud_tablet()->add_rowsets({_output_rowset}, false, wrlock); |
| } |
| // ATTN: MUST NOT update `base_compaction_cnt` which are used when sync rowsets, otherwise may cause |
| // the tablet to be unable to synchronize the rowset meta changes generated by base compaction. |
| cloud_tablet()->set_cumulative_compaction_cnt(stats.cumulative_compaction_cnt()); |
| cloud_tablet()->set_cumulative_layer_point(stats.cumulative_point()); |
| if (output_rowset_delete_bitmap) { |
| _tablet->tablet_meta()->delete_bitmap().merge(*output_rowset_delete_bitmap); |
| } |
| if (stats.base_compaction_cnt() >= cloud_tablet()->base_compaction_cnt()) { |
| cloud_tablet()->reset_approximate_stats(stats.num_rowsets(), stats.num_segments(), |
| stats.num_rows(), stats.data_size()); |
| } |
| } |
| // agg delete bitmap for pre rowsets |
| if (config::enable_agg_and_remove_pre_rowsets_delete_bitmap && |
| _tablet->keys_type() == KeysType::UNIQUE_KEYS && |
| _tablet->enable_unique_key_merge_on_write() && _input_rowsets.size() != 1) { |
| OlapStopWatch watch; |
| std::vector<RowsetSharedPtr> pre_rowsets {}; |
| { |
| std::shared_lock rlock(_tablet->get_header_lock()); |
| for (const auto& it2 : cloud_tablet()->rowset_map()) { |
| if (it2.first.second < _output_rowset->start_version()) { |
| pre_rowsets.emplace_back(it2.second); |
| } |
| } |
| } |
| std::sort(pre_rowsets.begin(), pre_rowsets.end(), Rowset::comparator); |
| auto pre_rowsets_delete_bitmap = std::make_shared<DeleteBitmap>(_tablet->tablet_id()); |
| std::map<std::string, int64_t> pre_rowset_to_versions; |
| cloud_tablet()->agg_delete_bitmap_for_compaction( |
| _output_rowset->start_version(), _output_rowset->end_version(), pre_rowsets, |
| pre_rowsets_delete_bitmap, pre_rowset_to_versions); |
| // update delete bitmap to ms |
| DBUG_EXECUTE_IF( |
| "CumulativeCompaction.modify_rowsets.cloud_update_delete_bitmap_without_lock.block", |
| DBUG_BLOCK); |
| auto status = _engine.meta_mgr().cloud_update_delete_bitmap_without_lock( |
| *cloud_tablet(), pre_rowsets_delete_bitmap.get(), pre_rowset_to_versions, |
| cloud_tablet()->table_id(), _output_rowset->start_version(), |
| _output_rowset->end_version()); |
| if (!status.ok()) { |
| LOG(WARNING) << "failed to agg pre rowsets delete bitmap to ms. tablet_id=" |
| << _tablet->tablet_id() << ", pre rowset num=" << pre_rowsets.size() |
| << ", output version=" << _output_rowset->version().to_string() |
| << ", status=" << status.to_string(); |
| } else { |
| LOG(INFO) << "agg pre rowsets delete bitmap to ms. tablet_id=" << _tablet->tablet_id() |
| << ", pre rowset num=" << pre_rowsets.size() |
| << ", output version=" << _output_rowset->version().to_string() |
| << ", cost(us)=" << watch.get_elapse_time_us(); |
| } |
| } |
| DBUG_EXECUTE_IF("CumulativeCompaction.modify_rowsets.delete_expired_stale_rowset", { |
| LOG(INFO) << "delete_expired_stale_rowsets for tablet=" << _tablet->tablet_id(); |
| _engine.tablet_mgr().vacuum_stale_rowsets(CountDownLatch(1)); |
| }); |
| |
| _tablet->prefill_dbm_agg_cache_after_compaction(_output_rowset); |
| return Status::OK(); |
| } |
| |
| Status CloudCumulativeCompaction::garbage_collection() { |
| RETURN_IF_ERROR(CloudCompactionMixin::garbage_collection()); |
| cloud::TabletJobInfoPB job; |
| auto idx = job.mutable_idx(); |
| idx->set_tablet_id(_tablet->tablet_id()); |
| idx->set_table_id(_tablet->table_id()); |
| idx->set_index_id(_tablet->index_id()); |
| idx->set_partition_id(_tablet->partition_id()); |
| auto compaction_job = job.add_compaction(); |
| compaction_job->set_id(_uuid); |
| compaction_job->set_initiator(BackendOptions::get_localhost() + ':' + |
| std::to_string(config::heartbeat_service_port)); |
| compaction_job->set_type(cloud::TabletCompactionJobPB::CUMULATIVE); |
| if (_tablet->keys_type() == KeysType::UNIQUE_KEYS && |
| _tablet->enable_unique_key_merge_on_write()) { |
| compaction_job->set_delete_bitmap_lock_initiator(this->initiator()); |
| } |
| DBUG_EXECUTE_IF("CumulativeCompaction.modify_rowsets.trigger_abort_job_failed", { |
| LOG(INFO) << "CumulativeCompaction.modify_rowsets.abort_job_failed for tablet_id" |
| << cloud_tablet()->tablet_id(); |
| return Status::InternalError( |
| "CumulativeCompaction.modify_rowsets.abort_job_failed for tablet_id {}", |
| cloud_tablet()->tablet_id()); |
| }); |
| auto st = _engine.meta_mgr().abort_tablet_job(job); |
| if (!st.ok()) { |
| LOG_WARNING("failed to abort compaction job") |
| .tag("job_id", _uuid) |
| .tag("tablet_id", _tablet->tablet_id()) |
| .error(st); |
| } |
| return st; |
| } |
| |
| Status CloudCumulativeCompaction::advance_cumulative_point_before_pick( |
| int64_t min_conflict_version) { |
| if (!_enable_parallel_cumu_compaction) { |
| return Status::OK(); |
| } |
| |
| auto compaction_policy = |
| _engine.cumu_compaction_policy(cloud_tablet()->tablet_meta()->compaction_policy()); |
| int64_t input_cumulative_point; |
| int64_t output_cumulative_point; |
| std::vector<RowsetSharedPtr> candidates; |
| { |
| std::shared_lock rlock(_tablet->get_header_lock()); |
| input_cumulative_point = cloud_tablet()->cumulative_layer_point(); |
| output_cumulative_point = input_cumulative_point; |
| _base_compaction_cnt = cloud_tablet()->base_compaction_cnt(); |
| _cumulative_compaction_cnt = cloud_tablet()->cumulative_compaction_cnt(); |
| cloud_tablet()->traverse_rowsets_unlocked( |
| [&candidates, input_cumulative_point, |
| min_conflict_version](const RowsetSharedPtr& rs) { |
| if (rs->start_version() >= input_cumulative_point && |
| rs->end_version() < min_conflict_version) { |
| candidates.push_back(rs); |
| } |
| }); |
| } |
| std::sort(candidates.begin(), candidates.end(), Rowset::comparator); |
| const bool is_time_series_policy = compaction_policy->name() == CUMULATIVE_TIME_SERIES_POLICY; |
| Version no_delete_version {-1, -1}; |
| for (const auto& rowset : candidates) { |
| if (rowset->start_version() != output_cumulative_point) { |
| break; |
| } |
| |
| auto rowset_meta = rowset->rowset_meta(); |
| if (rowset_meta->has_delete_predicate()) { |
| output_cumulative_point = rowset->end_version() + 1; |
| continue; |
| } |
| // A time-series singleton is a raw delta. Its post-compaction point rule is not safe here. |
| if (rowset_meta->is_segments_overlapping() || |
| (is_time_series_policy && rowset_meta->is_singleton_delta())) { |
| break; |
| } |
| |
| int64_t new_cumulative_point = compaction_policy->new_cumulative_point( |
| cloud_tablet(), rowset, no_delete_version, output_cumulative_point); |
| if (new_cumulative_point == output_cumulative_point) { |
| break; |
| } |
| DORIS_CHECK_EQ(new_cumulative_point, rowset->end_version() + 1); |
| output_cumulative_point = new_cumulative_point; |
| } |
| if (output_cumulative_point == input_cumulative_point) { |
| return Status::OK(); |
| } |
| update_cumulative_point(input_cumulative_point, output_cumulative_point); |
| return Status::Error<CUMULATIVE_NO_SUITABLE_VERSION>( |
| "cumulative point advanced before picking rowsets"); |
| } |
| |
| Status CloudCumulativeCompaction::pick_rowsets_to_compact() { |
| _input_rowsets.clear(); |
| |
| int64_t min_conflict_version = _min_conflict_version; |
| int64_t max_conflict_version = _max_conflict_version; |
| RETURN_IF_ERROR(advance_cumulative_point_before_pick(min_conflict_version)); |
| |
| int64_t max_score = config::cumulative_compaction_max_deltas; |
| double process_memory_usage = |
| cast_set<double>(doris::GlobalMemoryArbitrator::process_memory_usage()); |
| bool memory_usage_high = |
| process_memory_usage > cast_set<double>(MemInfo::soft_mem_limit()) * 0.8; |
| if (cloud_tablet()->last_compaction_status.is<ErrorCode::MEM_LIMIT_EXCEEDED>() || |
| memory_usage_high) { |
| max_score = std::max(config::cumulative_compaction_max_deltas / |
| config::cumulative_compaction_max_deltas_factor, |
| config::cumulative_compaction_min_deltas + 1); |
| } |
| |
| auto compaction_policy = cloud_tablet()->tablet_meta()->compaction_policy(); |
| auto pick_from_candidates = [&](std::vector<RowsetSharedPtr>& candidates) { |
| _input_rowsets.clear(); |
| _last_delete_version = Version {-1, -1}; |
| |
| if (candidates.empty()) { |
| return Status::Error<CUMULATIVE_NO_SUITABLE_VERSION>( |
| "no suitable versions: candidate rowsets empty"); |
| } |
| std::sort(candidates.begin(), candidates.end(), Rowset::comparator); |
| if (auto st = check_version_continuity(candidates); !st.ok()) { |
| DCHECK(false) << st; |
| return st; |
| } |
| |
| size_t compaction_score = 0; |
| _engine.cumu_compaction_policy(compaction_policy) |
| ->pick_input_rowsets(cloud_tablet(), candidates, max_score, |
| config::cumulative_compaction_min_deltas, &_input_rowsets, |
| &_last_delete_version, &compaction_score); |
| |
| if (_input_rowsets.empty()) { |
| return Status::Error<CUMULATIVE_NO_SUITABLE_VERSION>( |
| "no suitable versions: input rowsets empty"); |
| } |
| if (_input_rowsets.size() == 1 && |
| !_input_rowsets.front()->rowset_meta()->is_segments_overlapping()) { |
| VLOG_DEBUG << "there is only one rowset and not overlapping. tablet_id=" |
| << _tablet->tablet_id() << ", version=" << _input_rowsets.front()->version(); |
| _input_rowsets.clear(); |
| return Status::Error<CUMULATIVE_NO_SUITABLE_VERSION>( |
| "no suitable versions: only one rowset and not overlapping"); |
| } |
| return Status::OK(); |
| }; |
| |
| auto pick_from_version_range = [&](int64_t start_version, int64_t end_version, |
| bool preserve_delete_version) { |
| std::vector<RowsetSharedPtr> candidates; |
| { |
| std::shared_lock rlock(_tablet->get_header_lock()); |
| _base_compaction_cnt = cloud_tablet()->base_compaction_cnt(); |
| _cumulative_compaction_cnt = cloud_tablet()->cumulative_compaction_cnt(); |
| _picked_cumulative_point = cloud_tablet()->cumulative_layer_point(); |
| int64_t range_start_version = |
| std::max(std::max(_picked_cumulative_point, start_version), |
| cloud_tablet()->alter_version() + 1); |
| cloud_tablet()->traverse_rowsets_unlocked( |
| [&candidates, range_start_version, end_version](const RowsetSharedPtr& rs) { |
| if (rs->start_version() >= range_start_version && |
| rs->start_version() < end_version) { |
| candidates.push_back(rs); |
| } |
| }); |
| } |
| auto st = pick_from_candidates(candidates); |
| if (_last_delete_version.first != -1) { |
| DORIS_CHECK(!candidates.empty()); |
| if (!preserve_delete_version || |
| candidates.front()->start_version() != _picked_cumulative_point) { |
| _last_delete_version = Version {-1, -1}; |
| } |
| } |
| return st; |
| }; |
| |
| auto st = pick_from_version_range(0, min_conflict_version, true); |
| if (!st.ok()) { |
| if (_last_delete_version.first != -1) { |
| return st; |
| } |
| if (!st.is<CUMULATIVE_NO_SUITABLE_VERSION>() || |
| min_conflict_version == std::numeric_limits<int64_t>::max()) { |
| return st; |
| } |
| st = pick_from_version_range(max_conflict_version + 1, std::numeric_limits<int64_t>::max(), |
| false); |
| RETURN_IF_ERROR(st); |
| } |
| |
| apply_txn_size_truncation_and_log("CloudCumulativeCompaction"); |
| return Status::OK(); |
| } |
| |
| void CloudCumulativeCompaction::update_cumulative_point(int64_t input_cumulative_point, |
| int64_t output_cumulative_point) { |
| DORIS_CHECK_LT(input_cumulative_point, output_cumulative_point); |
| cloud::TabletJobInfoPB job; |
| auto idx = job.mutable_idx(); |
| idx->set_tablet_id(_tablet->tablet_id()); |
| idx->set_table_id(_tablet->table_id()); |
| idx->set_index_id(_tablet->index_id()); |
| idx->set_partition_id(_tablet->partition_id()); |
| auto compaction_job = job.add_compaction(); |
| compaction_job->set_id(_uuid); |
| compaction_job->set_initiator(BackendOptions::get_localhost() + ':' + |
| std::to_string(config::heartbeat_service_port)); |
| compaction_job->set_type(cloud::TabletCompactionJobPB::EMPTY_CUMULATIVE); |
| compaction_job->set_base_compaction_cnt(_base_compaction_cnt); |
| compaction_job->set_cumulative_compaction_cnt(_cumulative_compaction_cnt); |
| compaction_job->add_input_versions(input_cumulative_point); |
| compaction_job->add_input_versions(output_cumulative_point - 1); |
| compaction_job->set_check_input_versions_range(_enable_parallel_cumu_compaction); |
| int64_t now = time(nullptr); |
| compaction_job->set_lease(now + config::lease_compaction_interval_seconds); |
| // No need to set expiration time, since there is no output rowset |
| cloud::StartTabletJobResponse start_resp; |
| auto st = _engine.meta_mgr().prepare_tablet_job(job, &start_resp); |
| if (!st.ok()) { |
| if (start_resp.status().code() == cloud::STALE_TABLET_CACHE) { |
| // set last_sync_time to 0 to force sync tablet next time |
| cloud_tablet()->last_sync_time_s = 0; |
| } else if (start_resp.status().code() == cloud::TABLET_NOT_FOUND) { |
| // tablet not found |
| cloud_tablet()->clear_cache(); |
| } |
| LOG_WARNING("failed to update cumulative point to meta srv") |
| .tag("job_id", _uuid) |
| .tag("tablet_id", _tablet->tablet_id()) |
| .error(st); |
| return; |
| } |
| compaction_job->set_input_cumulative_point(input_cumulative_point); |
| compaction_job->set_output_cumulative_point(output_cumulative_point); |
| cloud::FinishTabletJobResponse finish_resp; |
| st = _engine.meta_mgr().commit_tablet_job(job, &finish_resp); |
| if (!st.ok()) { |
| if (finish_resp.status().code() == cloud::TABLET_NOT_FOUND) { |
| cloud_tablet()->clear_cache(); |
| } |
| LOG_WARNING("failed to update cumulative point to meta srv") |
| .tag("job_id", _uuid) |
| .tag("tablet_id", _tablet->tablet_id()) |
| .error(st); |
| return; |
| } |
| LOG_INFO("do empty cumulative compaction to update cumulative point") |
| .tag("job_id", _uuid) |
| .tag("tablet_id", _tablet->tablet_id()) |
| .tag("input_cumulative_point", input_cumulative_point) |
| .tag("output_cumulative_point", output_cumulative_point); |
| auto& stats = finish_resp.stats(); |
| LOG(INFO) << "tablet stats=" << stats.ShortDebugString(); |
| { |
| std::lock_guard wrlock(_tablet->get_header_lock()); |
| // clang-format off |
| cloud_tablet()->set_last_base_compaction_success_time(std::max(cloud_tablet()->last_base_compaction_success_time(), stats.last_base_compaction_time_ms())); |
| cloud_tablet()->set_last_cumu_compaction_success_time(std::max(cloud_tablet()->last_cumu_compaction_success_time(), stats.last_cumu_compaction_time_ms())); |
| // clang-format on |
| if (!should_apply_cumulative_compaction_result(stats.cumulative_compaction_cnt())) { |
| return; |
| } |
| // ATTN: MUST NOT update `base_compaction_cnt` which are used when sync rowsets, otherwise may cause |
| // the tablet to be unable to synchronize the rowset meta changes generated by base compaction. |
| cloud_tablet()->set_cumulative_compaction_cnt(stats.cumulative_compaction_cnt()); |
| cloud_tablet()->set_cumulative_layer_point(stats.cumulative_point()); |
| if (stats.base_compaction_cnt() >= cloud_tablet()->base_compaction_cnt()) { |
| cloud_tablet()->reset_approximate_stats(stats.num_rowsets(), stats.num_segments(), |
| stats.num_rows(), stats.data_size()); |
| } |
| } |
| } |
| |
| void CloudCumulativeCompaction::do_lease() { |
| TEST_INJECTION_POINT_RETURN_WITH_VOID("CloudCumulativeCompaction::do_lease"); |
| if (_state == CompactionState::SUCCESS) { |
| return; |
| } |
| cloud::TabletJobInfoPB job; |
| auto idx = job.mutable_idx(); |
| idx->set_tablet_id(_tablet->tablet_id()); |
| idx->set_table_id(_tablet->table_id()); |
| idx->set_index_id(_tablet->index_id()); |
| idx->set_partition_id(_tablet->partition_id()); |
| auto compaction_job = job.add_compaction(); |
| compaction_job->set_id(_uuid); |
| using namespace std::chrono; |
| int64_t lease_time = duration_cast<seconds>(system_clock::now().time_since_epoch()).count() + |
| config::lease_compaction_interval_seconds * 4; |
| compaction_job->set_lease(lease_time); |
| auto st = _engine.meta_mgr().lease_tablet_job(job); |
| if (!st.ok()) { |
| LOG_WARNING("failed to lease compaction job") |
| .tag("job_id", _uuid) |
| .tag("tablet_id", _tablet->tablet_id()) |
| .error(st); |
| } |
| } |
| |
| } // namespace doris |
