blob: 518a363a884a4907d1a217be4e2c96ac205e5933 [file]
// 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 <fmt/format.h>
#include <fmt/ranges.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/metrics/doris_metrics.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 "storage/merger.h"
#include "storage/rowset/rowset_reader.h"
#include "storage/rowset/rowset_writer.h"
#include "storage/tablet/tablet_schema.h"
#include "util/debug_points.h"
#include "util/trace.h"
#include "util/uuid_generator.h"
namespace doris {
using namespace ErrorCode;
namespace cloud {
bool is_single_rowset_compaction_candidate(const RowsetSharedPtr& rowset) {
const auto& rowset_meta = rowset->rowset_meta();
const int64_t overlap_unit_count =
rowset_meta->segments_overlap() == NONOVERLAPPING_WITHIN_GROUP
? static_cast<int64_t>(rowset_meta->segment_group_sizes().size())
: rowset->num_segments();
return !rowset_meta->has_delete_predicate() && rowset_meta->is_segments_overlapping() &&
overlap_unit_count >= config::cloud_single_rowset_compaction_min_segments;
}
bool should_use_single_rowset_grouped_compaction(const std::vector<RowsetSharedPtr>& input_rowsets,
const TabletSchema& tablet_schema,
std::string_view compaction_policy) {
return compaction_policy == CUMULATIVE_SIZE_BASED_POLICY &&
tablet_schema.num_key_columns() > 0 && tablet_schema.cluster_key_uids().empty() &&
config::enable_cloud_single_rowset_compaction && input_rowsets.size() == 1 &&
is_single_rowset_compaction_candidate(input_rowsets.front());
}
std::vector<SegmentGroupMergeRange> build_segment_group_merge_ranges(const RowsetMeta& rowset_meta,
int64_t segment_group_size) {
DORIS_CHECK_GT(segment_group_size, 1);
DORIS_CHECK_GT(rowset_meta.num_segments(), 0);
std::vector<SegmentGroupMergeRange> ranges;
if (rowset_meta.segments_overlap() == NONOVERLAPPING_WITHIN_GROUP) {
const auto& input_segment_group_sizes = rowset_meta.segment_group_sizes();
const int64_t input_group_count = cast_set<int64_t>(input_segment_group_sizes.size());
DORIS_CHECK_GT(input_group_count, 0);
ranges.reserve(cast_set<size_t>((input_group_count + segment_group_size - 1) /
segment_group_size));
int64_t segment_end = 0;
for (int64_t group_start = 0; group_start < input_group_count;
group_start += segment_group_size) {
const int64_t group_end = std::min(group_start + segment_group_size, input_group_count);
const int64_t segment_start = segment_end;
for (int64_t group_index = group_start; group_index < group_end; ++group_index) {
const int32_t input_group_size =
input_segment_group_sizes.Get(cast_set<int>(group_index));
DORIS_CHECK_GT(input_group_size, 0);
segment_end += input_group_size;
}
ranges.push_back({.segment_start = segment_start,
.segment_end = segment_end,
.merge_way_num = group_end - group_start});
}
DORIS_CHECK_EQ(segment_end, rowset_meta.num_segments());
} else {
ranges.reserve(cast_set<size_t>((rowset_meta.num_segments() + segment_group_size - 1) /
segment_group_size));
for (int64_t segment_start = 0; segment_start < rowset_meta.num_segments();
segment_start += segment_group_size) {
const int64_t segment_end =
std::min(segment_start + segment_group_size, rowset_meta.num_segments());
ranges.push_back({.segment_start = segment_start,
.segment_end = segment_end,
.merge_way_num = segment_end - segment_start});
}
}
return ranges;
}
} // namespace cloud
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;
}
bool CloudCumulativeCompaction::should_calculate_new_cumulative_point(
int64_t input_cumulative_point) const {
if (!_single_rowset_compaction_segment_group_size.has_value()) {
return true;
}
DORIS_CHECK_EQ(_input_rowsets.size(), 1);
DORIS_CHECK(_output_rowset != nullptr);
return _input_rowsets.front()->start_version() == input_cumulative_point &&
_output_rowset->rowset_meta()->segments_overlap() == NONOVERLAPPING;
}
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 (should_calculate_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();
_single_rowset_compaction_segment_group_size.reset();
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);
const int64_t segment_group_size =
config::cloud_single_rowset_compaction_segment_group_size;
if (config::enable_cloud_single_rowset_compaction && segment_group_size > 1) {
for (const auto& rowset : _input_rowsets) {
if (cloud::should_use_single_rowset_grouped_compaction(
{rowset}, *cloud_tablet()->tablet_schema(), compaction_policy)) {
auto grouped_input_rowset = rowset;
_input_rowsets = {std::move(grouped_input_rowset)};
_single_rowset_compaction_segment_group_size = segment_group_size;
return Status::OK();
}
}
}
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();
}
Status CloudCumulativeCompaction::prepare_merge_input_rowsets(MergeInputRowsetsResult* result) {
if (!_single_rowset_compaction_segment_group_size.has_value()) {
return Status::OK();
}
const int64_t segment_group_size = *_single_rowset_compaction_segment_group_size;
DORIS_CHECK_GT(segment_group_size, 1);
result->is_segment_grouped = true;
result->segment_group_size = segment_group_size;
return Status::OK();
}
Status CloudCumulativeCompaction::do_merge_input_rowsets(
const std::vector<RowsetReaderSharedPtr>& input_rs_readers,
MergeInputRowsetsResult* result) {
if (!result->is_segment_grouped) {
return Compaction::do_merge_input_rowsets(input_rs_readers, result);
}
const int64_t segment_group_size = result->segment_group_size;
const auto& input_rowset = _input_rowsets.front();
const auto segment_ranges = cloud::build_segment_group_merge_ranges(
*input_rowset->rowset_meta(), segment_group_size);
for (size_t range_index = 0; range_index < segment_ranges.size(); ++range_index) {
const auto& range = segment_ranges[range_index];
const int32_t output_segment_start = _output_rs_writer->get_allocated_segment_id();
RowsetReaderSharedPtr rs_reader;
RETURN_IF_ERROR(input_rowset->create_reader(&rs_reader));
std::vector<RowsetReaderSharedPtr> group_readers;
group_readers.push_back(std::move(rs_reader));
Merger::Statistics group_stats;
group_stats.rowid_conversion = _stats.rowid_conversion;
RETURN_IF_ERROR(execute_merge(group_readers, range.merge_way_num, &group_stats,
std::make_pair(range.segment_start, range.segment_end),
{.total_ranges = cast_set<int64_t>(segment_ranges.size()),
.range_index = cast_set<int64_t>(range_index)}));
_stats.output_rows += group_stats.output_rows;
_stats.merged_rows += group_stats.merged_rows;
_stats.filtered_rows += group_stats.filtered_rows;
_stats.bytes_read_from_local += group_stats.bytes_read_from_local;
_stats.bytes_read_from_remote += group_stats.bytes_read_from_remote;
_stats.cached_bytes_total += group_stats.cached_bytes_total;
_stats.cloud_local_read_time += group_stats.cloud_local_read_time;
_stats.cloud_remote_read_time += group_stats.cloud_remote_read_time;
const int32_t output_segment_end = _output_rs_writer->get_allocated_segment_id();
const int32_t output_group_size = output_segment_end - output_segment_start;
if (output_group_size > 0) {
result->output_segment_group_sizes.push_back(output_group_size);
}
}
return Status::OK();
}
void CloudCumulativeCompaction::update_output_rowset_after_build(
const MergeInputRowsetsResult& result) {
if (!result.is_segment_grouped) {
return;
}
if (result.output_segment_group_sizes.size() > 1) {
_output_rowset->rowset_meta()->set_segments_overlap(NONOVERLAPPING_WITHIN_GROUP);
_output_rowset->rowset_meta()->set_segment_group_sizes(result.output_segment_group_sizes);
}
const auto& input_rowset = _input_rowsets.front();
LOG_INFO("finish single rowset grouped compaction, tablet_id={}, version=[{}-{}]",
_tablet->tablet_id(), input_rowset->start_version(), input_rowset->end_version())
.tag("job_id", _uuid)
.tag("input_segments", input_rowset->num_segments())
.tag("segment_group_size", result.segment_group_size)
.tag("output_segments", _output_rowset->num_segments())
.tag("output_groups", result.output_segment_group_sizes.size())
.tag("output_segment_group_sizes",
fmt::format("[{}]", fmt::join(result.output_segment_group_sizes, ", ")));
}
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