Coverage Report

Created: 2026-06-02 20:24

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
be/src/storage/tablet/tablet.cpp
Line
Count
Source
1
// Licensed to the Apache Software Foundation (ASF) under one
2
// or more contributor license agreements.  See the NOTICE file
3
// distributed with this work for additional information
4
// regarding copyright ownership.  The ASF licenses this file
5
// to you under the Apache License, Version 2.0 (the
6
// "License"); you may not use this file except in compliance
7
// with the License.  You may obtain a copy of the License at
8
//
9
//   http://www.apache.org/licenses/LICENSE-2.0
10
//
11
// Unless required by applicable law or agreed to in writing,
12
// software distributed under the License is distributed on an
13
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14
// KIND, either express or implied.  See the License for the
15
// specific language governing permissions and limitations
16
// under the License.
17
18
#include "storage/tablet/tablet.h"
19
20
#include <butil/logging.h>
21
#include <bvar/reducer.h>
22
#include <bvar/window.h>
23
#include <fmt/format.h>
24
#include <gen_cpp/FrontendService_types.h>
25
#include <gen_cpp/MasterService_types.h>
26
#include <gen_cpp/Metrics_types.h>
27
#include <gen_cpp/olap_file.pb.h>
28
#include <gen_cpp/types.pb.h>
29
#include <glog/logging.h>
30
#include <rapidjson/document.h>
31
#include <rapidjson/encodings.h>
32
#include <rapidjson/prettywriter.h>
33
#include <rapidjson/rapidjson.h>
34
#include <rapidjson/stringbuffer.h>
35
36
#include <algorithm>
37
#include <atomic>
38
#include <boost/container/detail/std_fwd.hpp>
39
#include <cstdint>
40
#include <roaring/roaring.hh>
41
42
#include "common/compiler_util.h" // IWYU pragma: keep
43
// IWYU pragma: no_include <bits/chrono.h>
44
#include <chrono> // IWYU pragma: keep
45
#include <filesystem>
46
#include <iterator>
47
#include <limits>
48
#include <map>
49
#include <memory>
50
#include <mutex>
51
#include <set>
52
#include <shared_mutex>
53
#include <string>
54
#include <tuple>
55
#include <type_traits>
56
#include <unordered_map>
57
#include <unordered_set>
58
59
#include "agent/utils.h"
60
#include "common/config.h"
61
#include "common/consts.h"
62
#include "common/logging.h"
63
#include "common/metrics/doris_metrics.h"
64
#include "common/signal_handler.h"
65
#include "common/status.h"
66
#include "core/column/column.h"
67
#include "core/column/column_string.h"
68
#include "core/custom_allocator.h"
69
#include "core/data_type/data_type.h"
70
#include "core/data_type/data_type_factory.hpp"
71
#include "core/data_type_serde/data_type_serde.h"
72
#include "core/string_ref.h"
73
#include "io/fs/file_reader.h"
74
#include "io/fs/file_reader_writer_fwd.h"
75
#include "io/fs/file_system.h"
76
#include "io/fs/file_writer.h"
77
#include "io/fs/path.h"
78
#include "io/fs/remote_file_system.h"
79
#include "io/io_common.h"
80
#include "load/memtable/memtable.h"
81
#include "service/point_query_executor.h"
82
#include "storage/binlog.h"
83
#include "storage/compaction/base_compaction.h"
84
#include "storage/compaction/binlog_compaction.h"
85
#include "storage/compaction/binlog_compaction_policy.h"
86
#include "storage/compaction/cumulative_compaction.h"
87
#include "storage/compaction/cumulative_compaction_policy.h"
88
#include "storage/compaction/cumulative_compaction_time_series_policy.h"
89
#include "storage/compaction/full_compaction.h"
90
#include "storage/compaction/single_replica_compaction.h"
91
#include "storage/delete/delete_bitmap_calculator.h"
92
#include "storage/index/indexed_column_reader.h"
93
#include "storage/index/primary_key_index.h"
94
#include "storage/olap_common.h"
95
#include "storage/olap_define.h"
96
#include "storage/olap_meta.h"
97
#include "storage/rowset/beta_rowset.h"
98
#include "storage/rowset/group_rowset_writer.h"
99
#include "storage/rowset/rowset.h"
100
#include "storage/rowset/rowset_factory.h"
101
#include "storage/rowset/rowset_fwd.h"
102
#include "storage/rowset/rowset_meta.h"
103
#include "storage/rowset/rowset_meta_manager.h"
104
#include "storage/rowset/rowset_writer.h"
105
#include "storage/rowset/rowset_writer_context.h"
106
#include "storage/rowset/vertical_beta_rowset_writer.h"
107
#include "storage/schema_change/schema_change.h"
108
#include "storage/segment/column_reader.h"
109
#include "storage/segment/common.h"
110
#include "storage/segment/segment_loader.h"
111
#include "storage/storage_engine.h"
112
#include "storage/storage_policy.h"
113
#include "storage/tablet/base_tablet.h"
114
#include "storage/tablet/tablet.h"
115
#include "storage/tablet/tablet_manager.h"
116
#include "storage/tablet/tablet_meta.h"
117
#include "storage/tablet/tablet_meta_manager.h"
118
#include "storage/tablet/tablet_schema.h"
119
#include "storage/txn/txn_manager.h"
120
#include "storage/types.h"
121
#include "storage/utils.h"
122
#include "util/bvar_helper.h"
123
#include "util/debug_points.h"
124
#include "util/defer_op.h"
125
#include "util/jsonb/serialize.h"
126
#include "util/pretty_printer.h"
127
#include "util/stopwatch.hpp"
128
#include "util/threadpool.h"
129
#include "util/time.h"
130
#include "util/trace.h"
131
#include "util/uid_util.h"
132
#include "util/work_thread_pool.hpp"
133
134
namespace doris {
135
class TupleDescriptor;
136
137
class Block;
138
139
using namespace ErrorCode;
140
using namespace std::chrono_literals;
141
142
using std::pair;
143
using std::string;
144
using std::vector;
145
using io::FileSystemSPtr;
146
147
namespace {
148
149
bvar::Adder<uint64_t> exceed_version_limit_counter;
150
bvar::Window<bvar::Adder<uint64_t>> exceed_version_limit_counter_minute(
151
        &exceed_version_limit_counter, 60);
152
bvar::Adder<uint64_t> cooldown_pending_task("cooldown_pending_task");
153
bvar::Adder<uint64_t> cooldown_processing_task("cooldown_processing_task");
154
155
3.48k
void set_last_failure_time(Tablet* tablet, const Compaction& compaction, int64_t ms) {
156
3.48k
    switch (compaction.compaction_type()) {
157
3.49k
    case ReaderType::READER_CUMULATIVE_COMPACTION:
158
3.49k
        tablet->set_last_cumu_compaction_failure_time(ms);
159
3.49k
        return;
160
2
    case ReaderType::READER_BASE_COMPACTION:
161
2
        tablet->set_last_base_compaction_failure_time(ms);
162
2
        return;
163
0
    case ReaderType::READER_FULL_COMPACTION:
164
0
        tablet->set_last_full_compaction_failure_time(ms);
165
0
        return;
166
0
    case ReaderType::READER_BINLOG_COMPACTION:
167
0
        tablet->set_last_binlog_compaction_failure_time(ms);
168
0
        return;
169
0
    default:
170
0
        LOG(FATAL) << "invalid compaction type " << compaction.compaction_name()
171
0
                   << " tablet_id: " << tablet->tablet_id();
172
3.48k
    }
173
3.48k
};
174
175
} // namespace
176
177
bvar::Adder<uint64_t> unused_remote_rowset_num("unused_remote_rowset_num");
178
179
WriteCooldownMetaExecutors::WriteCooldownMetaExecutors(size_t executor_nums)
180
8
        : _executor_nums(executor_nums) {
181
48
    for (size_t i = 0; i < _executor_nums; i++) {
182
40
        std::unique_ptr<PriorityThreadPool> pool;
183
40
        static_cast<void>(ThreadPoolBuilder("WriteCooldownMetaExecutor")
184
40
                                  .set_min_threads(1)
185
40
                                  .set_max_threads(1)
186
40
                                  .set_max_queue_size(std::numeric_limits<int>::max())
187
40
                                  .build(&pool));
188
40
        _executors.emplace_back(std::move(pool));
189
40
    }
190
8
}
191
192
3
void WriteCooldownMetaExecutors::stop() {
193
15
    for (auto& pool_ptr : _executors) {
194
15
        if (pool_ptr) {
195
15
            pool_ptr->shutdown();
196
15
        }
197
15
    }
198
3
}
199
200
124k
void WriteCooldownMetaExecutors::WriteCooldownMetaExecutors::submit(TabletSharedPtr tablet) {
201
124k
    auto tablet_id = tablet->tablet_id();
202
203
124k
    {
204
124k
        std::shared_lock rdlock(tablet->get_header_lock());
205
124k
        if (!tablet->tablet_meta()->cooldown_meta_id().initialized()) {
206
0
            VLOG_NOTICE << "tablet " << tablet_id << " is not cooldown replica";
207
0
            return;
208
0
        }
209
124k
        if (tablet->tablet_state() == TABLET_SHUTDOWN) [[unlikely]] {
210
0
            LOG_INFO("tablet {} has been dropped, don't do cooldown", tablet_id);
211
0
            return;
212
0
        }
213
124k
    }
214
124k
    {
215
        // one tablet could at most have one cooldown task to be done
216
124k
        std::unique_lock<std::mutex> lck {_latch};
217
124k
        if (_pending_tablets.count(tablet_id) > 0) {
218
0
            return;
219
0
        }
220
124k
        _pending_tablets.insert(tablet_id);
221
124k
    }
222
223
124k
    auto async_write_task = [this, t = std::move(tablet)]() {
224
124k
        {
225
124k
            std::unique_lock<std::mutex> lck {_latch};
226
124k
            _pending_tablets.erase(t->tablet_id());
227
124k
        }
228
124k
        SCOPED_ATTACH_TASK(ExecEnv::GetInstance()->orphan_mem_tracker());
229
124k
        auto s = t->write_cooldown_meta();
230
124k
        if (s.ok()) {
231
7
            return;
232
7
        }
233
124k
        if (!s.is<ABORTED>()) {
234
124k
            LOG_EVERY_SECOND(WARNING)
235
14
                    << "write tablet " << t->tablet_id() << " cooldown meta failed because: " << s;
236
124k
            submit(t);
237
124k
            return;
238
124k
        }
239
0
        VLOG_DEBUG << "tablet " << t->tablet_id() << " is not cooldown replica";
240
0
    };
241
242
124k
    cooldown_pending_task << 1;
243
124k
    _executors[_get_executor_pos(tablet_id)]->offer([task = std::move(async_write_task)]() {
244
124k
        cooldown_pending_task << -1;
245
124k
        cooldown_processing_task << 1;
246
124k
        task();
247
124k
        cooldown_processing_task << -1;
248
124k
    });
249
124k
}
250
251
Tablet::Tablet(StorageEngine& engine, TabletMetaSharedPtr tablet_meta, DataDir* data_dir,
252
               const std::string_view& cumulative_compaction_type)
253
290k
        : BaseTablet(std::move(tablet_meta)),
254
290k
          _engine(engine),
255
290k
          _data_dir(data_dir),
256
290k
          _is_bad(false),
257
290k
          _last_cumu_compaction_failure_millis(0),
258
290k
          _last_base_compaction_failure_millis(0),
259
290k
          _last_full_compaction_failure_millis(0),
260
290k
          _last_cumu_compaction_success_millis(0),
261
290k
          _last_base_compaction_success_millis(0),
262
290k
          _last_full_compaction_success_millis(0),
263
290k
          _last_binlog_compaction_failure_millis(0),
264
290k
          _cumulative_point(K_INVALID_CUMULATIVE_POINT),
265
290k
          _newly_created_rowset_num(0),
266
290k
          _last_checkpoint_time(0),
267
290k
          _cumulative_compaction_type(cumulative_compaction_type),
268
290k
          _is_tablet_path_exists(true),
269
290k
          _last_missed_version(-1),
270
290k
          _last_missed_time_s(0) {
271
290k
    if (_data_dir != nullptr) {
272
290k
        _tablet_path = fmt::format("{}/{}/{}/{}/{}", _data_dir->path(), DATA_PREFIX,
273
290k
                                   _tablet_meta->shard_id(), tablet_id(), schema_hash());
274
290k
    }
275
290k
    _binlog_compaction_policy = std::make_shared<BinlogCompactionPolicy>();
276
290k
    init_last_binlog_compaction_success_and_failure_time();
277
290k
}
278
279
290k
void Tablet::init_last_binlog_compaction_success_and_failure_time() {
280
290k
    int64_t now = UnixMillis();
281
1.16M
    for (int8_t level = 0; level < BinlogCompactionPolicy::kBinlogCompactionMaxLevel; ++level) {
282
871k
        _last_binlog_compaction_success_millis[level] = now;
283
871k
    }
284
290k
    _last_binlog_compaction_failure_millis = 0;
285
290k
}
286
287
282k
bool Tablet::set_tablet_schema_into_rowset_meta() {
288
282k
    bool flag = false;
289
473k
    for (auto& [_, rowset_meta] : _tablet_meta->all_mutable_rs_metas()) {
290
473k
        if (!rowset_meta->tablet_schema()) {
291
0
            rowset_meta->set_tablet_schema(_tablet_meta->tablet_schema());
292
0
            flag = true;
293
0
        }
294
473k
    }
295
282k
    return flag;
296
282k
}
297
298
290k
Status Tablet::_init_once_action() {
299
290k
    Status res = Status::OK();
300
290k
    VLOG_NOTICE << "begin to load tablet. tablet=" << tablet_id()
301
398
                << ", version_size=" << _tablet_meta->version_count();
302
303
#ifdef BE_TEST
304
    // init cumulative compaction policy by type
305
    _cumulative_compaction_policy =
306
            CumulativeCompactionPolicyFactory::create_cumulative_compaction_policy(
307
                    _tablet_meta->compaction_policy());
308
#endif
309
310
477k
    for (const auto& [_, rs_meta] : _tablet_meta->all_rs_metas()) {
311
477k
        Version version = rs_meta->version();
312
477k
        RowsetSharedPtr rowset;
313
477k
        res = create_rowset(rs_meta, &rowset);
314
477k
        if (!res.ok()) {
315
0
            LOG(WARNING) << "fail to init rowset. tablet_id=" << tablet_id()
316
0
                         << ", schema_hash=" << schema_hash() << ", version=" << version
317
0
                         << ", res=" << res;
318
0
            return res;
319
0
        }
320
477k
        _rs_version_map[version] = std::move(rowset);
321
477k
    }
322
323
    // init stale rowset
324
290k
    int64_t now = ::time(nullptr);
325
290k
    for (const auto& [_, stale_rs_meta] : _tablet_meta->all_stale_rs_metas()) {
326
4.72k
        Version version = stale_rs_meta->version();
327
328
4.72k
        if (!stale_rs_meta->has_stale_at()) {
329
4.72k
            stale_rs_meta->set_stale_at(now);
330
4.72k
        }
331
332
4.72k
        RowsetSharedPtr rowset;
333
4.72k
        res = create_rowset(stale_rs_meta, &rowset);
334
4.72k
        if (!res.ok()) {
335
0
            LOG(WARNING) << "fail to init stale rowset. tablet_id:" << tablet_id()
336
0
                         << ", schema_hash:" << schema_hash() << ", version=" << version
337
0
                         << ", res:" << res;
338
0
            return res;
339
0
        }
340
4.72k
        _stale_rs_version_map[version] = std::move(rowset);
341
4.72k
    }
342
343
    // init row_binlog rowset
344
290k
    for (const auto& [_, row_binlog_rs_meta] : _tablet_meta->all_row_binlog_rs_metas()) {
345
70
        Version version = row_binlog_rs_meta->version();
346
70
        RowsetSharedPtr rowset;
347
70
        res = create_rowset(row_binlog_rs_meta, &rowset);
348
70
        if (!res.ok()) {
349
0
            LOG(WARNING) << "fail to init binlog<row> rowset. tablet_id:" << tablet_id()
350
0
                         << ", schema_hash:" << schema_hash() << ", version=" << version
351
0
                         << ", res:" << res;
352
0
            return res;
353
0
        }
354
70
        _row_binlog_rs_version_map[version] = std::move(rowset);
355
70
    }
356
357
290k
    return res;
358
290k
}
359
360
290k
Status Tablet::init() {
361
290k
    return _init_once.call([this] { return _init_once_action(); });
362
290k
}
363
364
// should save tablet meta to remote meta store
365
// if it's a primary replica
366
25.3k
void Tablet::save_meta() {
367
25.3k
    check_table_size_correctness();
368
25.3k
    auto res = _tablet_meta->save_meta(_data_dir);
369
25.3k
    CHECK_EQ(res, Status::OK()) << "fail to save tablet_meta. res=" << res
370
0
                                << ", root=" << _data_dir->path();
371
25.3k
}
372
373
// Caller should hold _meta_lock.
374
Status Tablet::revise_tablet_meta(const std::vector<RowsetSharedPtr>& to_add,
375
                                  const std::vector<RowsetSharedPtr>& to_delete,
376
0
                                  bool is_incremental_clone, bool copy_row_binlog) {
377
0
    LOG(INFO) << "begin to revise tablet. tablet_id=" << tablet_id();
378
    // 1. for incremental clone, we have to add the rowsets first to make it easy to compute
379
    //    all the delete bitmaps, and it's easy to delete them if we end up with a failure
380
    // 2. for full clone, we can calculate delete bitmaps on the cloned rowsets directly.
381
0
    if (is_incremental_clone) {
382
0
        CHECK(to_delete.empty()); // don't need to delete rowsets
383
0
        add_rowsets(to_add, copy_row_binlog);
384
        // reconstruct from tablet meta
385
0
        _timestamped_version_tracker.construct_versioned_tracker(_tablet_meta->all_rs_metas());
386
0
        if (copy_row_binlog) {
387
0
            _row_binlog_version_tracker.construct_versioned_tracker(
388
0
                    _tablet_meta->all_row_binlog_rs_metas());
389
0
        }
390
0
    }
391
392
0
    Status calc_bm_status;
393
0
    std::vector<RowsetSharedPtr> normal_rowsets;
394
0
    std::vector<RowsetSharedPtr> base_rowsets_for_full_clone;
395
0
    normal_rowsets.reserve(to_add.size());
396
0
    base_rowsets_for_full_clone.reserve(to_add.size());
397
0
    for (const auto& rs : to_add) {
398
0
        if (copy_row_binlog && rs->rowset_meta()->is_row_binlog()) {
399
0
            continue;
400
0
        }
401
0
        normal_rowsets.push_back(rs);
402
0
        base_rowsets_for_full_clone.push_back(rs);
403
0
    }
404
0
    while (keys_type() == UNIQUE_KEYS && enable_unique_key_merge_on_write()) {
405
0
        std::vector<RowsetSharedPtr> calc_delete_bitmap_rowsets;
406
0
        int64_t to_add_min_version = INT64_MAX;
407
0
        int64_t to_add_max_version = INT64_MIN;
408
0
        for (auto& rs : normal_rowsets) {
409
0
            if (to_add_min_version > rs->start_version()) {
410
0
                to_add_min_version = rs->start_version();
411
0
            }
412
0
            if (to_add_max_version < rs->end_version()) {
413
0
                to_add_max_version = rs->end_version();
414
0
            }
415
0
        }
416
0
        Version calc_delete_bitmap_ver;
417
0
        if (is_incremental_clone) {
418
            // From the rowset of to_add with smallest version, all other rowsets
419
            // need to recalculate the delete bitmap
420
            // For example:
421
            // local tablet: [0-1] [2-5] [6-6] [9-10]
422
            // clone tablet: [7-7] [8-8]
423
            // new tablet:   [0-1] [2-5] [6-6] [7-7] [8-8] [9-10]
424
            // [7-7] [8-8] [9-10] need to recalculate delete bitmap
425
0
            calc_delete_bitmap_ver = Version(to_add_min_version, max_version_unlocked());
426
0
        } else {
427
            // the delete bitmap of to_add's rowsets has clone from remote when full clone.
428
            // only other rowsets in local need to recalculate the delete bitmap.
429
            // For example:
430
            // local tablet: [0-1]x [2-5]x [6-6]x [7-7]x [9-10]
431
            // clone tablet: [0-1]  [2-4]  [5-6]  [7-8]
432
            // new tablet:   [0-1]  [2-4]  [5-6]  [7-8] [9-10]
433
            // only [9-10] need to recalculate delete bitmap
434
0
            CHECK_EQ(to_add_min_version, 0) << "to_add_min_version is: " << to_add_min_version;
435
0
            calc_delete_bitmap_ver = Version(to_add_max_version + 1, max_version_unlocked());
436
0
        }
437
438
0
        if (calc_delete_bitmap_ver.first <= calc_delete_bitmap_ver.second) {
439
0
            auto ret = capture_consistent_rowsets_unlocked(calc_delete_bitmap_ver,
440
0
                                                           CaptureRowsetOps {});
441
0
            if (!ret) {
442
0
                LOG(WARNING) << "fail to capture_consistent_rowsets, res: " << ret.error();
443
0
                calc_bm_status = std::move(ret.error());
444
0
                break;
445
0
            }
446
0
            calc_delete_bitmap_rowsets = std::move(ret->rowsets);
447
            // FIXME(plat1ko): Use `const TabletSharedPtr&` as parameter
448
0
            auto self = _engine.tablet_manager()->get_tablet(tablet_id());
449
0
            CHECK(self);
450
0
            for (auto rs : calc_delete_bitmap_rowsets) {
451
0
                if (is_incremental_clone) {
452
0
                    calc_bm_status = update_delete_bitmap_without_lock(self, rs);
453
0
                } else {
454
0
                    calc_bm_status = update_delete_bitmap_without_lock(
455
0
                            self, rs, &base_rowsets_for_full_clone);
456
0
                    base_rowsets_for_full_clone.push_back(rs);
457
0
                }
458
0
                if (!calc_bm_status.ok()) {
459
0
                    LOG(WARNING) << "fail to update_delete_bitmap_without_lock, res: "
460
0
                                 << calc_bm_status;
461
0
                    break;
462
0
                }
463
0
            }
464
0
        }
465
0
        break; // while (keys_type() == UNIQUE_KEYS && enable_unique_key_merge_on_write())
466
0
    }
467
468
0
    DBUG_EXECUTE_IF("Tablet.revise_tablet_meta_fail", {
469
0
        auto ptablet_id = dp->param("tablet_id", 0);
470
0
        if (tablet_id() == ptablet_id) {
471
0
            LOG(INFO) << "injected revies_tablet_meta failure for tabelt: " << ptablet_id;
472
0
            calc_bm_status = Status::InternalError("fault injection error");
473
0
        }
474
0
    });
475
476
    // error handling
477
0
    if (!calc_bm_status.ok()) {
478
0
        if (is_incremental_clone) {
479
0
            RETURN_IF_ERROR(delete_rowsets(to_add, false, copy_row_binlog));
480
0
            LOG(WARNING) << "incremental clone on tablet: " << tablet_id() << " failed due to "
481
0
                         << calc_bm_status.msg() << ", revert " << to_add.size()
482
0
                         << " rowsets added before.";
483
0
        } else {
484
0
            LOG(WARNING) << "full clone on tablet: " << tablet_id() << " failed due to "
485
0
                         << calc_bm_status.msg() << ", will not update tablet meta.";
486
0
        }
487
0
        return calc_bm_status;
488
0
    }
489
490
    // full clone, calculate delete bitmap succeeded, update rowset
491
0
    if (!is_incremental_clone) {
492
0
        RETURN_IF_ERROR(delete_rowsets(to_delete, false, copy_row_binlog));
493
0
        add_rowsets(to_add, copy_row_binlog);
494
        // reconstruct from tablet meta
495
0
        _timestamped_version_tracker.construct_versioned_tracker(_tablet_meta->all_rs_metas());
496
0
        if (copy_row_binlog) {
497
0
            _row_binlog_version_tracker.construct_versioned_tracker(
498
0
                    _tablet_meta->all_row_binlog_rs_metas());
499
0
        }
500
501
        // check the rowsets used for delete bitmap calculation is equal to the rowsets
502
        // that we can capture by version
503
0
        if (keys_type() == UNIQUE_KEYS && enable_unique_key_merge_on_write()) {
504
0
            Version full_version = Version(0, max_version_unlocked());
505
0
            auto ret = capture_consistent_rowsets_unlocked(full_version, CaptureRowsetOps {});
506
0
            DCHECK(ret) << ret.error();
507
0
            DCHECK_EQ(base_rowsets_for_full_clone.size(), ret->rowsets.size());
508
509
0
            if (ret && base_rowsets_for_full_clone.size() != ret->rowsets.size()) [[unlikely]] {
510
0
                LOG(WARNING) << "full clone succeeded, but the count("
511
0
                             << base_rowsets_for_full_clone.size()
512
0
                             << ") of base rowsets used for delete bitmap calculation is not match "
513
0
                                "expect count("
514
0
                             << ret->rowsets.size() << ") we capture from tablet meta";
515
0
            }
516
0
        }
517
0
    }
518
519
    // clear stale rowset
520
0
    for (auto& [v, rs] : _stale_rs_version_map) {
521
0
        _engine.add_unused_rowset(rs);
522
0
    }
523
0
    _stale_rs_version_map.clear();
524
0
    _tablet_meta->clear_stale_rowset();
525
0
    save_meta();
526
527
0
    LOG(INFO) << "finish to revise tablet. tablet_id=" << tablet_id();
528
0
    return Status::OK();
529
0
}
530
531
10.9k
Status Tablet::add_rowset(RowsetSharedPtr rowset, RowsetSharedPtr row_binlog_rowset) {
532
10.9k
    DCHECK(rowset != nullptr);
533
10.9k
    std::lock_guard<std::shared_mutex> wrlock(_meta_lock);
534
10.9k
    SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
535
    // If the rowset already exist, just return directly.  The rowset_id is an unique-id,
536
    // we can use it to check this situation.
537
10.9k
    if (_contains_rowset(rowset->rowset_id())) {
538
        // Ensure binlog<row> is also added on retry.
539
665
        return _add_row_binlog_rowset_unlocked(rowset, row_binlog_rowset);
540
665
    }
541
    // Otherwise, the version should be not contained in any existing rowset.
542
10.2k
    RETURN_IF_ERROR(_contains_version(rowset->version()));
543
544
10.0k
    RETURN_IF_ERROR(_tablet_meta->add_rs_meta(rowset->rowset_meta()));
545
10.0k
    _rs_version_map[rowset->version()] = rowset;
546
10.0k
    _timestamped_version_tracker.add_version(rowset->version());
547
10.0k
    add_compaction_score(rowset->rowset_meta()->get_compaction_score());
548
549
10.0k
    RETURN_IF_ERROR(_add_row_binlog_rowset_unlocked(rowset, row_binlog_rowset));
550
551
10.0k
    std::vector<RowsetSharedPtr> rowsets_to_delete;
552
    // yiguolei: temp code, should remove the rowset contains by this rowset
553
    // but it should be removed in multi path version
554
13.5k
    for (auto& it : _rs_version_map) {
555
13.5k
        if (rowset->version().contains(it.first) && rowset->version() != it.first) {
556
0
            CHECK(it.second != nullptr)
557
0
                    << "there exist a version=" << it.first
558
0
                    << " contains the input rs with version=" << rowset->version()
559
0
                    << ", but the related rs is null";
560
0
            rowsets_to_delete.push_back(it.second);
561
0
        }
562
13.5k
    }
563
10.0k
    std::vector<RowsetSharedPtr> empty_vec;
564
10.0k
    RETURN_IF_ERROR(modify_rowsets(empty_vec, rowsets_to_delete));
565
10.0k
    ++_newly_created_rowset_num;
566
10.0k
    return Status::OK();
567
10.0k
}
568
569
Status Tablet::_add_row_binlog_rowset_unlocked(const RowsetSharedPtr& rowset,
570
57.8k
                                               const RowsetSharedPtr& row_binlog_rowset) {
571
57.8k
    if (row_binlog_rowset == nullptr) {
572
57.8k
        return Status::OK();
573
57.8k
    }
574
18.4E
    DCHECK(rowset != nullptr);
575
18.4E
    DCHECK_EQ(row_binlog_rowset->version(), rowset->version());
576
577
18.4E
    const auto& version = row_binlog_rowset->version();
578
18.4E
    if (auto it = _row_binlog_rs_version_map.find(version);
579
18.4E
        it != _row_binlog_rs_version_map.end()) {
580
0
        if (it->second != nullptr && it->second->rowset_id() == row_binlog_rowset->rowset_id()) {
581
0
            return Status::OK();
582
0
        }
583
0
        return Status::Error<PUSH_VERSION_ALREADY_EXIST>(
584
0
                "binlog<row> version already exists. existing rowset_id={}, version={}, tablet={}",
585
0
                it->second != nullptr ? it->second->rowset_id().to_string() : "0",
586
0
                version.to_string(), tablet_id());
587
0
    }
588
589
18.4E
    RETURN_IF_ERROR(_tablet_meta->add_row_binlog_rs_meta(row_binlog_rowset->rowset_meta()));
590
18.4E
    _row_binlog_rs_version_map[version] = row_binlog_rowset;
591
18.4E
    _row_binlog_version_tracker.add_version(version);
592
18.4E
    return Status::OK();
593
18.4E
}
594
595
22.2k
bool Tablet::rowset_exists_unlocked(const RowsetSharedPtr& rowset) {
596
22.2k
    if (auto it = _rs_version_map.find(rowset->version()); it == _rs_version_map.end()) {
597
0
        return false;
598
22.2k
    } else if (rowset->rowset_id() != it->second->rowset_id()) {
599
0
        return false;
600
0
    }
601
22.2k
    return true;
602
22.2k
}
603
604
Status Tablet::modify_rowsets(std::vector<RowsetSharedPtr>& to_add,
605
13.4k
                              std::vector<RowsetSharedPtr>& to_delete, bool check_delete) {
606
    // the compaction process allow to compact the single version, eg: version[4-4].
607
    // this kind of "single version compaction" has same "input version" and "output version".
608
    // which means "to_add->version()" equals to "to_delete->version()".
609
    // So we should delete the "to_delete" before adding the "to_add",
610
    // otherwise, the "to_add" will be deleted from _rs_version_map, eventually.
611
    //
612
    // And if the version of "to_add" and "to_delete" are exactly same. eg:
613
    // to_add:      [7-7]
614
    // to_delete:   [7-7]
615
    // In this case, we no longer need to add the rowset in "to_delete" to
616
    // _stale_rs_version_map, but can delete it directly.
617
618
13.4k
    if (to_add.empty() && to_delete.empty()) {
619
10.1k
        return Status::OK();
620
10.1k
    }
621
622
3.51k
    if (check_delete) {
623
31.3k
        for (auto&& rs : to_delete) {
624
31.3k
            if (auto it = _rs_version_map.find(rs->version()); it == _rs_version_map.end()) {
625
0
                return Status::Error<DELETE_VERSION_ERROR>(
626
0
                        "try to delete not exist version {} from {}", rs->version().to_string(),
627
0
                        tablet_id());
628
31.3k
            } else if (rs->rowset_id() != it->second->rowset_id()) {
629
0
                return Status::Error<DELETE_VERSION_ERROR>(
630
0
                        "try to delete version {} from {}, but rowset id changed, delete "
631
0
                        "rowset id "
632
0
                        "is {}, exists rowsetid is {}",
633
0
                        rs->version().to_string(), tablet_id(), rs->rowset_id().to_string(),
634
0
                        it->second->rowset_id().to_string());
635
0
            }
636
31.3k
        }
637
3.51k
    }
638
639
3.38k
    bool same_version = true;
640
3.38k
    std::sort(to_add.begin(), to_add.end(), Rowset::comparator);
641
3.38k
    std::sort(to_delete.begin(), to_delete.end(), Rowset::comparator);
642
3.38k
    if (to_add.size() == to_delete.size()) {
643
32
        for (int i = 0; i < to_add.size(); ++i) {
644
16
            if (to_add[i]->version() != to_delete[i]->version()) {
645
0
                same_version = false;
646
0
                break;
647
0
            }
648
16
        }
649
3.36k
    } else {
650
3.36k
        same_version = false;
651
3.36k
    }
652
653
3.38k
    std::vector<RowsetMetaSharedPtr> rs_metas_to_delete;
654
3.38k
    int64_t now = ::time(nullptr);
655
31.3k
    for (auto& rs : to_delete) {
656
31.3k
        rs_metas_to_delete.push_back(rs->rowset_meta());
657
31.3k
        _rs_version_map.erase(rs->version());
658
659
31.3k
        if (!same_version) {
660
31.3k
            rs->rowset_meta()->set_stale_at(now);
661
            // put compaction rowsets in _stale_rs_version_map.
662
31.3k
            _stale_rs_version_map[rs->version()] = rs;
663
31.3k
        }
664
31.3k
    }
665
666
3.38k
    std::vector<RowsetMetaSharedPtr> rs_metas_to_add;
667
3.51k
    for (auto& rs : to_add) {
668
3.51k
        rs_metas_to_add.push_back(rs->rowset_meta());
669
3.51k
        _rs_version_map[rs->version()] = rs;
670
671
3.51k
        if (!same_version) {
672
            // If version are same, then _timestamped_version_tracker
673
            // already has this version, no need to add again.
674
3.49k
            _timestamped_version_tracker.add_version(rs->version());
675
3.49k
        }
676
3.51k
        ++_newly_created_rowset_num;
677
3.51k
    }
678
679
3.38k
    _tablet_meta->modify_rs_metas(rs_metas_to_add, rs_metas_to_delete, same_version);
680
681
3.49k
    if (!same_version) {
682
        // add rs_metas_to_delete to tracker
683
3.49k
        _timestamped_version_tracker.add_stale_path_version(rs_metas_to_delete);
684
18.4E
    } else {
685
        // delete rowset in "to_delete" directly
686
18.4E
        for (auto& rs : to_delete) {
687
16
            LOG(INFO) << "add unused rowset " << rs->rowset_id() << " because of same version";
688
16
            if (rs->is_local()) {
689
16
                _engine.add_unused_rowset(rs);
690
16
            }
691
16
        }
692
18.4E
    }
693
694
3.38k
    int32_t add_score = 0;
695
3.51k
    for (auto rs : to_add) {
696
3.51k
        add_score += rs->rowset_meta()->get_compaction_score();
697
3.51k
    }
698
3.38k
    int32_t sub_score = 0;
699
31.3k
    for (auto rs : to_delete) {
700
31.3k
        sub_score += rs->rowset_meta()->get_compaction_score();
701
31.3k
    }
702
3.38k
    add_compaction_score(add_score - sub_score);
703
704
3.38k
    return Status::OK();
705
3.38k
}
706
707
Status Tablet::modify_row_binlog_rowsets(std::vector<RowsetSharedPtr>& to_add,
708
0
                                         std::vector<RowsetSharedPtr>& to_delete) {
709
0
    if (to_add.empty() && to_delete.empty()) {
710
0
        return Status::OK();
711
0
    }
712
713
0
    for (auto&& rs : to_delete) {
714
0
        if (auto it = _row_binlog_rs_version_map.find(rs->version());
715
0
            it == _row_binlog_rs_version_map.end()) {
716
0
            return Status::Error<DELETE_VERSION_ERROR>(
717
0
                    "try to delete not exist row-binlog {} from {}", rs->version().to_string(),
718
0
                    tablet_id());
719
0
        } else if (rs->rowset_id() != it->second->rowset_id()) {
720
0
            return Status::Error<DELETE_VERSION_ERROR>(
721
0
                    "try to delete row-binlog {} from {}, but rowset id changed, delete rowset id "
722
0
                    "is {}, exists rowsetid is {}",
723
0
                    rs->version().to_string(), tablet_id(), rs->rowset_id().to_string(),
724
0
                    it->second->rowset_id().to_string());
725
0
        }
726
0
    }
727
728
0
    std::vector<RowsetMetaSharedPtr> rs_metas_to_delete;
729
0
    for (auto& rs : to_delete) {
730
0
        rs_metas_to_delete.push_back(rs->rowset_meta());
731
0
        _row_binlog_rs_version_map.erase(rs->version());
732
0
        _row_binlog_version_tracker.delete_version(rs->version());
733
        // Delete binlog delvec immediately because input rowsets will be deleted.
734
0
        tablet_meta()->binlog_delvec().remove({rs->rowset_id(), 0, 0},
735
0
                                              {rs->rowset_id(), UINT32_MAX, UINT64_MAX});
736
0
        _engine.add_unused_rowset(rs);
737
0
    }
738
739
0
    std::vector<RowsetMetaSharedPtr> rs_metas_to_add;
740
0
    for (auto& rs : to_add) {
741
0
        rs_metas_to_add.push_back(rs->rowset_meta());
742
0
        _row_binlog_rs_version_map[rs->version()] = rs;
743
0
        _row_binlog_version_tracker.add_version(rs->version());
744
0
        ++_newly_created_rowset_num;
745
0
    }
746
747
0
    _tablet_meta->modify_row_binlog_rs_metas(rs_metas_to_add, rs_metas_to_delete);
748
0
    return Status::OK();
749
0
}
750
751
5
void Tablet::add_rowsets(const std::vector<RowsetSharedPtr>& to_add, bool copy_row_binlog) {
752
5
    if (to_add.empty()) {
753
0
        return;
754
0
    }
755
5
    std::vector<RowsetMetaSharedPtr> rs_metas;
756
5
    std::vector<RowsetMetaSharedPtr> row_binlog_rs_metas;
757
5
    rs_metas.reserve(to_add.size());
758
5
    row_binlog_rs_metas.reserve(to_add.size());
759
5
    for (auto& rs : to_add) {
760
5
        if (copy_row_binlog && rs->rowset_meta()->is_row_binlog()) {
761
0
            _row_binlog_rs_version_map.emplace(rs->version(), rs);
762
0
            _row_binlog_version_tracker.add_version(rs->version());
763
0
            row_binlog_rs_metas.push_back(rs->rowset_meta());
764
5
        } else {
765
5
            _rs_version_map.emplace(rs->version(), rs);
766
5
            _timestamped_version_tracker.add_version(rs->version());
767
5
            rs_metas.push_back(rs->rowset_meta());
768
5
        }
769
5
    }
770
5
    _tablet_meta->modify_rs_metas(rs_metas, {});
771
5
    if (copy_row_binlog) {
772
0
        _tablet_meta->modify_row_binlog_rs_metas(row_binlog_rs_metas, {});
773
0
    }
774
5
}
775
776
Status Tablet::delete_rowsets(const std::vector<RowsetSharedPtr>& to_delete, bool move_to_stale,
777
5
                              bool copy_row_binlog) {
778
5
    if (to_delete.empty()) {
779
0
        return Status::OK();
780
0
    }
781
5
    std::vector<RowsetMetaSharedPtr> rs_metas;
782
5
    std::vector<RowsetMetaSharedPtr> row_binlog_rs_metas;
783
5
    std::vector<RowsetSharedPtr> normal_rowsets;
784
5
    std::vector<RowsetSharedPtr> row_binlog_rowsets;
785
5
    rs_metas.reserve(to_delete.size());
786
5
    row_binlog_rs_metas.reserve(to_delete.size());
787
5
    int64_t now = ::time(nullptr);
788
5
    for (const auto& rs : to_delete) {
789
5
        if (copy_row_binlog && rs->rowset_meta()->is_row_binlog()) {
790
0
            row_binlog_rs_metas.push_back(rs->rowset_meta());
791
0
            _row_binlog_rs_version_map.erase(rs->version());
792
0
            row_binlog_rowsets.push_back(rs);
793
0
            continue;
794
0
        }
795
5
        if (move_to_stale) {
796
0
            rs->rowset_meta()->set_stale_at(now);
797
0
        }
798
5
        rs_metas.push_back(rs->rowset_meta());
799
5
        _rs_version_map.erase(rs->version());
800
5
        normal_rowsets.push_back(rs);
801
5
    }
802
5
    _tablet_meta->modify_rs_metas({}, rs_metas, !move_to_stale);
803
5
    if (copy_row_binlog) {
804
0
        _tablet_meta->modify_row_binlog_rs_metas({}, row_binlog_rs_metas);
805
0
    }
806
5
    if (move_to_stale) {
807
0
        for (const auto& rs : normal_rowsets) {
808
0
            _stale_rs_version_map[rs->version()] = rs;
809
0
        }
810
0
        _timestamped_version_tracker.add_stale_path_version(rs_metas);
811
5
    } else {
812
5
        for (const auto& rs : normal_rowsets) {
813
5
            _timestamped_version_tracker.delete_version(rs->version());
814
5
            if (rs->is_local()) {
815
5
                _engine.add_unused_rowset(rs);
816
5
                RETURN_IF_ERROR(RowsetMetaManager::remove(_data_dir->get_meta(), tablet_uid(),
817
5
                                                          rs->rowset_meta()->rowset_id()));
818
5
            }
819
5
        }
820
5
    }
821
5
    for (const auto& rs : row_binlog_rowsets) {
822
0
        _row_binlog_version_tracker.delete_version(rs->version());
823
0
        if (rs->is_local()) {
824
0
            _engine.add_unused_rowset(rs);
825
0
        }
826
0
    }
827
5
    return Status::OK();
828
5
}
829
830
0
RowsetSharedPtr Tablet::_rowset_with_largest_size() {
831
0
    RowsetSharedPtr largest_rowset = nullptr;
832
0
    for (auto& it : _rs_version_map) {
833
0
        if (it.second->empty() || it.second->zero_num_rows()) {
834
0
            continue;
835
0
        }
836
0
        if (largest_rowset == nullptr || it.second->rowset_meta()->index_disk_size() >
837
0
                                                 largest_rowset->rowset_meta()->index_disk_size()) {
838
0
            largest_rowset = it.second;
839
0
        }
840
0
    }
841
842
0
    return largest_rowset;
843
0
}
844
845
// add inc rowset should not persist tablet meta, because it will be persisted when publish txn.
846
Status Tablet::add_inc_rowset(const RowsetSharedPtr& rowset,
847
47.3k
                              const RowsetSharedPtr& row_binlog_rowset) {
848
47.3k
    DCHECK(rowset != nullptr);
849
47.3k
    std::lock_guard<std::shared_mutex> wrlock(_meta_lock);
850
47.3k
    SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
851
47.3k
    if (_contains_rowset(rowset->rowset_id())) {
852
        // Ensure binlog<row> is also added on retry.
853
0
        return _add_row_binlog_rowset_unlocked(rowset, row_binlog_rowset);
854
0
    }
855
47.3k
    RETURN_IF_ERROR(_contains_version(rowset->version()));
856
857
47.3k
    RETURN_IF_ERROR(_tablet_meta->add_rs_meta(rowset->rowset_meta()));
858
47.3k
    _rs_version_map[rowset->version()] = rowset;
859
860
47.3k
    _timestamped_version_tracker.add_version(rowset->version());
861
862
47.3k
    ++_newly_created_rowset_num;
863
864
47.3k
    add_compaction_score(rowset->rowset_meta()->get_compaction_score());
865
866
47.3k
    return _add_row_binlog_rowset_unlocked(rowset, row_binlog_rowset);
867
47.3k
}
868
869
31.3k
void Tablet::_delete_stale_rowset_by_version(const Version& version) {
870
31.3k
    RowsetMetaSharedPtr rowset_meta = _tablet_meta->acquire_stale_rs_meta_by_version(version);
871
31.3k
    if (rowset_meta == nullptr) {
872
8
        return;
873
8
    }
874
31.3k
    _tablet_meta->delete_stale_rs_meta_by_version(version);
875
31.3k
    VLOG_NOTICE << "delete stale rowset. tablet=" << tablet_id() << ", version=" << version;
876
31.3k
}
877
878
604k
void Tablet::delete_expired_stale_rowset() {
879
604k
    if (config::enable_mow_verbose_log) {
880
0
        LOG_INFO("begin delete_expired_stale_rowset for tablet={}", tablet_id());
881
0
    }
882
604k
    int64_t now = UnixSeconds();
883
604k
    std::vector<std::pair<Version, std::vector<RowsetId>>> deleted_stale_rowsets;
884
    // hold write lock while processing stable rowset
885
604k
    {
886
604k
        std::lock_guard<std::shared_mutex> wrlock(_meta_lock);
887
604k
        SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
888
        // Compute the end time to delete rowsets, when a expired rowset createtime less then this time, it will be deleted.
889
604k
        int64_t expired_stale_sweep_endtime =
890
604k
                static_cast<int64_t>(::difftime(now, config::tablet_rowset_stale_sweep_time_sec));
891
604k
        if (config::tablet_rowset_stale_sweep_by_size) {
892
0
            expired_stale_sweep_endtime = now;
893
0
        }
894
895
604k
        std::vector<int64_t> path_id_vec;
896
        // capture the path version to delete
897
604k
        _timestamped_version_tracker.capture_expired_paths(
898
604k
                static_cast<int64_t>(expired_stale_sweep_endtime), &path_id_vec);
899
900
604k
        if (path_id_vec.empty()) {
901
603k
            return;
902
603k
        }
903
904
1.06k
        const RowsetSharedPtr lastest_delta = get_rowset_with_max_version();
905
1.06k
        if (lastest_delta == nullptr) {
906
0
            LOG(WARNING) << "lastest_delta is null " << tablet_id();
907
0
            return;
908
0
        }
909
910
        // fetch missing version before delete
911
1.06k
        Versions missed_versions = get_missed_versions_unlocked(lastest_delta->end_version());
912
1.06k
        if (!missed_versions.empty()) {
913
0
            LOG(WARNING) << "tablet:" << tablet_id()
914
0
                         << ", missed version for version:" << lastest_delta->end_version();
915
0
            _print_missed_versions(missed_versions);
916
0
            return;
917
0
        }
918
919
        // do check consistent operation
920
1.06k
        auto path_id_iter = path_id_vec.begin();
921
922
1.06k
        std::map<int64_t, PathVersionListSharedPtr> stale_version_path_map;
923
4.54k
        while (path_id_iter != path_id_vec.end()) {
924
3.47k
            PathVersionListSharedPtr version_path =
925
3.47k
                    _timestamped_version_tracker.fetch_and_delete_path_by_id(*path_id_iter);
926
927
3.47k
            Version test_version = Version(0, lastest_delta->end_version());
928
3.47k
            stale_version_path_map[*path_id_iter] = version_path;
929
930
3.47k
            auto ret = capture_consistent_versions_unlocked(test_version, {});
931
            // 1. When there is no consistent versions, we must reconstruct the tracker.
932
3.47k
            if (!ret) {
933
                // 2. fetch missing version after delete
934
0
                Versions after_missed_versions =
935
0
                        get_missed_versions_unlocked(lastest_delta->end_version());
936
937
                // 2.1 check whether missed_versions and after_missed_versions are the same.
938
                // when they are the same, it means we can delete the path securely.
939
0
                bool is_missing = missed_versions.size() != after_missed_versions.size();
940
941
0
                if (!is_missing) {
942
0
                    for (int ver_index = 0; ver_index < missed_versions.size(); ver_index++) {
943
0
                        if (missed_versions[ver_index] != after_missed_versions[ver_index]) {
944
0
                            is_missing = true;
945
0
                            break;
946
0
                        }
947
0
                    }
948
0
                }
949
950
0
                if (is_missing) {
951
0
                    LOG(WARNING) << "The consistent version check fails, there are bugs. "
952
0
                                 << "Reconstruct the tracker to recover versions in tablet="
953
0
                                 << tablet_id();
954
955
                    // 3. try to recover
956
0
                    _timestamped_version_tracker.recover_versioned_tracker(stale_version_path_map);
957
958
                    // 4. double check the consistent versions
959
                    // fetch missing version after recover
960
0
                    Versions recover_missed_versions =
961
0
                            get_missed_versions_unlocked(lastest_delta->end_version());
962
963
                    // 4.1 check whether missed_versions and recover_missed_versions are the same.
964
                    // when they are the same, it means we recover successfully.
965
0
                    bool is_recover_missing =
966
0
                            missed_versions.size() != recover_missed_versions.size();
967
968
0
                    if (!is_recover_missing) {
969
0
                        for (int ver_index = 0; ver_index < missed_versions.size(); ver_index++) {
970
0
                            if (missed_versions[ver_index] != recover_missed_versions[ver_index]) {
971
0
                                is_recover_missing = true;
972
0
                                break;
973
0
                            }
974
0
                        }
975
0
                    }
976
977
                    // 5. check recover fail, version is mission
978
0
                    if (is_recover_missing) {
979
0
                        if (!config::ignore_rowset_stale_unconsistent_delete) {
980
0
                            LOG(FATAL)
981
0
                                    << "rowset stale unconsistent delete. tablet= " << tablet_id();
982
0
                        } else {
983
0
                            LOG(WARNING)
984
0
                                    << "rowset stale unconsistent delete. tablet= " << tablet_id();
985
0
                        }
986
0
                    }
987
0
                }
988
0
                return;
989
0
            }
990
3.47k
            path_id_iter++;
991
3.47k
        }
992
993
1.06k
        auto old_size = _stale_rs_version_map.size();
994
1.06k
        auto old_meta_size = _tablet_meta->all_stale_rs_metas().size();
995
996
        // do delete operation
997
1.06k
        auto to_delete_iter = stale_version_path_map.begin();
998
4.54k
        while (to_delete_iter != stale_version_path_map.end()) {
999
3.47k
            std::vector<TimestampedVersionSharedPtr>& to_delete_version =
1000
3.47k
                    to_delete_iter->second->timestamped_versions();
1001
3.47k
            int64_t start_version = -1;
1002
3.47k
            int64_t end_version = -1;
1003
3.47k
            std::vector<RowsetId> remove_rowset_ids;
1004
31.3k
            for (auto& timestampedVersion : to_delete_version) {
1005
31.3k
                auto it = _stale_rs_version_map.find(timestampedVersion->version());
1006
31.3k
                if (it != _stale_rs_version_map.end()) {
1007
31.3k
                    it->second->clear_cache();
1008
                    // delete rowset
1009
31.3k
                    if (it->second->is_local()) {
1010
31.3k
                        _engine.add_unused_rowset(it->second);
1011
31.3k
                        if (keys_type() == UNIQUE_KEYS && enable_unique_key_merge_on_write()) {
1012
                            // mow does not support cold data in object storage
1013
20.6k
                            remove_rowset_ids.emplace_back(it->second->rowset_id());
1014
20.6k
                        }
1015
31.3k
                    }
1016
31.3k
                    _stale_rs_version_map.erase(it);
1017
31.3k
                    VLOG_NOTICE << "delete stale rowset tablet=" << tablet_id() << " version["
1018
0
                                << timestampedVersion->version().first << ","
1019
0
                                << timestampedVersion->version().second
1020
0
                                << "] move to unused_rowset success " << std::fixed
1021
0
                                << expired_stale_sweep_endtime;
1022
31.3k
                } else {
1023
8
                    LOG(WARNING) << "delete stale rowset tablet=" << tablet_id() << " version["
1024
8
                                 << timestampedVersion->version().first << ","
1025
8
                                 << timestampedVersion->version().second
1026
8
                                 << "] not find in stale rs version map";
1027
8
                }
1028
31.3k
                if (start_version < 0) {
1029
3.47k
                    start_version = timestampedVersion->version().first;
1030
3.47k
                }
1031
31.3k
                end_version = timestampedVersion->version().second;
1032
31.3k
                _delete_stale_rowset_by_version(timestampedVersion->version());
1033
31.3k
            }
1034
3.47k
            Version version(start_version, end_version);
1035
3.47k
            to_delete_iter++;
1036
3.47k
            if (!remove_rowset_ids.empty()) {
1037
2.18k
                deleted_stale_rowsets.emplace_back(version, remove_rowset_ids);
1038
2.18k
            }
1039
3.47k
        }
1040
1041
1.06k
        bool reconstructed = _reconstruct_version_tracker_if_necessary();
1042
1043
1.06k
        VLOG_NOTICE << "delete stale rowset _stale_rs_version_map tablet=" << tablet_id()
1044
1
                    << " current_size=" << _stale_rs_version_map.size() << " old_size=" << old_size
1045
1
                    << " current_meta_size=" << _tablet_meta->all_stale_rs_metas().size()
1046
1
                    << " old_meta_size=" << old_meta_size << " sweep endtime " << std::fixed
1047
1
                    << expired_stale_sweep_endtime << ", reconstructed=" << reconstructed;
1048
1.06k
    }
1049
1.06k
    if (config::enable_agg_and_remove_pre_rowsets_delete_bitmap && !deleted_stale_rowsets.empty()) {
1050
        // agg delete bitmap for pre rowsets; record unused delete bitmap key ranges
1051
584
        OlapStopWatch watch;
1052
2.18k
        for (const auto& [version, remove_rowset_ids] : deleted_stale_rowsets) {
1053
            // agg delete bitmap for pre rowset
1054
2.18k
            DeleteBitmapKeyRanges remove_delete_bitmap_key_ranges;
1055
2.18k
            agg_delete_bitmap_for_stale_rowsets(version, remove_delete_bitmap_key_ranges);
1056
            // add remove delete bitmap
1057
2.18k
            if (!remove_delete_bitmap_key_ranges.empty()) {
1058
0
                _engine.add_unused_delete_bitmap_key_ranges(tablet_id(), remove_rowset_ids,
1059
0
                                                            remove_delete_bitmap_key_ranges);
1060
0
            }
1061
2.18k
        }
1062
584
        LOG(INFO) << "agg pre rowsets delete bitmap. tablet_id=" << tablet_id()
1063
584
                  << ", size=" << deleted_stale_rowsets.size()
1064
584
                  << ", cost(us)=" << watch.get_elapse_time_us();
1065
584
    }
1066
1.06k
#ifndef BE_TEST
1067
1.06k
    {
1068
1.06k
        std::shared_lock<std::shared_mutex> rlock(_meta_lock);
1069
1.06k
        save_meta();
1070
1.06k
    }
1071
1.06k
#endif
1072
1.06k
    if (config::enable_mow_verbose_log) {
1073
0
        LOG_INFO("finish delete_expired_stale_rowset for tablet={}", tablet_id());
1074
0
    }
1075
1.06k
    DBUG_EXECUTE_IF("Tablet.delete_expired_stale_rowset.start_delete_unused_rowset", {
1076
1.06k
        _engine.start_delete_unused_rowset();
1077
1.06k
        [[maybe_unused]] auto st = _engine.start_trash_sweep(nullptr);
1078
1.06k
    });
1079
1.06k
}
1080
1081
0
Status Tablet::check_version_integrity(const Version& version, bool quiet) {
1082
0
    std::shared_lock rdlock(_meta_lock);
1083
0
    [[maybe_unused]] auto _versions = DORIS_TRY(
1084
0
            capture_consistent_versions_unlocked(version, CaptureRowsetOps {.quiet = quiet}));
1085
0
    return Status::OK();
1086
0
}
1087
1088
36.8k
bool Tablet::exceed_version_limit(int32_t limit) {
1089
36.8k
    if (_tablet_meta->version_count() > limit) {
1090
0
        exceed_version_limit_counter << 1;
1091
0
        return true;
1092
0
    }
1093
36.8k
    return false;
1094
36.8k
}
1095
1096
// If any rowset contains the specific version, it means the version already exist
1097
36.8k
bool Tablet::check_version_exist(const Version& version) const {
1098
36.8k
    std::shared_lock rdlock(_meta_lock);
1099
61.3k
    for (auto& it : _rs_version_map) {
1100
61.3k
        if (it.first.contains(version)) {
1101
36.7k
            return true;
1102
36.7k
        }
1103
61.3k
    }
1104
13
    return false;
1105
36.8k
}
1106
1107
// The meta read lock should be held before calling
1108
void Tablet::acquire_version_and_rowsets(
1109
3.42k
        std::vector<std::pair<Version, RowsetSharedPtr>>* version_rowsets) const {
1110
9.34k
    for (const auto& it : _rs_version_map) {
1111
9.34k
        version_rowsets->emplace_back(it.first, it.second);
1112
9.34k
    }
1113
3.42k
}
1114
1115
Status Tablet::capture_rs_readers(const Version& spec_version, std::vector<RowSetSplits>* rs_splits,
1116
2
                                  const CaptureRowsetOps& opts) {
1117
2
    std::shared_lock rlock(_meta_lock);
1118
2
    std::vector<Version> version_path;
1119
2
    *rs_splits = DORIS_TRY(capture_rs_readers_unlocked(
1120
1
            spec_version, CaptureRowsetOps {.skip_missing_versions = opts.skip_missing_versions}));
1121
1
    return Status::OK();
1122
2
}
1123
1124
1.07k
Versions Tablet::calc_missed_versions(int64_t spec_version, Versions existing_versions) const {
1125
1.07k
    DCHECK(spec_version > 0) << "invalid spec_version: " << spec_version;
1126
1127
    // sort the existing versions in ascending order
1128
1.07k
    std::sort(existing_versions.begin(), existing_versions.end(),
1129
2.29k
              [](const Version& a, const Version& b) {
1130
                  // simple because 2 versions are certainly not overlapping
1131
2.29k
                  return a.first < b.first;
1132
2.29k
              });
1133
1134
    // From the first version(=0),  find the missing version until spec_version
1135
1.07k
    int64_t last_version = -1;
1136
1.07k
    Versions missed_versions;
1137
2.98k
    for (const Version& version : existing_versions) {
1138
2.98k
        if (version.first > last_version + 1) {
1139
4
            for (int64_t i = last_version + 1; i < version.first && i <= spec_version; ++i) {
1140
                // Don't merge missed_versions because clone & snapshot use single version.
1141
                // For example, if miss 4 ~ 6, clone need [4, 4], [5, 5], [6, 6], but not [4, 6].
1142
2
                missed_versions.emplace_back(i, i);
1143
2
            }
1144
2
        }
1145
2.98k
        last_version = version.second;
1146
2.98k
        if (last_version >= spec_version) {
1147
1.07k
            break;
1148
1.07k
        }
1149
2.98k
    }
1150
1.07k
    for (int64_t i = last_version + 1; i <= spec_version; ++i) {
1151
0
        missed_versions.emplace_back(i, i);
1152
0
    }
1153
1154
1.07k
    return missed_versions;
1155
1.07k
}
1156
1157
427M
bool Tablet::can_do_compaction(size_t path_hash, CompactionType compaction_type) {
1158
427M
    if (compaction_type == CompactionType::BASE_COMPACTION && tablet_state() != TABLET_RUNNING) {
1159
        // base compaction can only be done for tablet in TABLET_RUNNING state.
1160
        // but cumulative compaction can be done for TABLET_NOTREADY, such as tablet under alter process.
1161
2
        return false;
1162
2
    }
1163
1164
427M
    if (compaction_type == CompactionType::BINLOG_COMPACTION &&
1165
427M
        (!config::enable_feature_binlog || !enable_row_binlog())) {
1166
406M
        return false;
1167
406M
    }
1168
1169
21.3M
    if (data_dir()->path_hash() != path_hash || !is_used() || !init_succeeded()) {
1170
7.56M
        return false;
1171
7.56M
    }
1172
1173
    // In TABLET_NOTREADY, we keep last 10 versions in new tablet so base tablet max_version
1174
    // not merged in new tablet and then we can do compaction
1175
13.7M
    return tablet_state() == TABLET_RUNNING || tablet_state() == TABLET_NOTREADY;
1176
21.3M
}
1177
1178
uint32_t Tablet::calc_compaction_score(CompactionType compaction_type,
1179
14.2M
                                       int8_t* prefer_compaction_level) {
1180
14.2M
    if (compaction_type == CompactionType::BINLOG_COMPACTION) {
1181
504
        std::shared_lock rdlock(_meta_lock);
1182
504
        return _calc_binlog_compaction_score(prefer_compaction_level);
1183
504
    }
1184
1185
14.2M
    if (_score_check_cnt++ % config::check_score_rounds_num != 0) {
1186
13.9M
        std::shared_lock rdlock(_meta_lock);
1187
13.9M
        if (_compaction_score > 0) {
1188
13.9M
            return _compaction_score;
1189
13.9M
        }
1190
13.9M
    }
1191
1192
288k
    {
1193
        // Need meta lock, because it will iterator "all_rs_metas" of tablet meta.
1194
288k
        std::shared_lock rdlock(_meta_lock);
1195
288k
        int32_t score = get_real_compaction_score();
1196
288k
        if (_compaction_score > 0 && _compaction_score != score) {
1197
0
            LOG(WARNING) << "cumu cache score not equal real score, cache score; "
1198
0
                         << _compaction_score << ", real score: " << score
1199
0
                         << ", tablet: " << tablet_id();
1200
0
        }
1201
288k
        _compaction_score = score;
1202
288k
        return score;
1203
14.2M
    }
1204
14.2M
}
1205
1206
bool Tablet::suitable_for_compaction(
1207
        CompactionType compaction_type,
1208
1.39M
        std::shared_ptr<CumulativeCompactionPolicy> cumulative_compaction_policy) {
1209
    // Need meta lock, because it will iterator "all_rs_metas" of tablet meta.
1210
1.39M
    std::shared_lock rdlock(_meta_lock);
1211
1.39M
    int32_t score = -1;
1212
1.39M
    if (compaction_type == CompactionType::CUMULATIVE_COMPACTION) {
1213
297k
        score = _calc_cumulative_compaction_score(cumulative_compaction_policy);
1214
1.10M
    } else if (compaction_type == CompactionType::BASE_COMPACTION) {
1215
1.10M
        DCHECK_EQ(compaction_type, CompactionType::BASE_COMPACTION);
1216
1.10M
        score = _calc_base_compaction_score();
1217
1.10M
    } else if (compaction_type == CompactionType::BINLOG_COMPACTION) {
1218
492
        score = _calc_binlog_compaction_score();
1219
492
    } else {
1220
0
        DCHECK(false) << "Unknown compaction type: " << compaction_type;
1221
0
    }
1222
1.39M
    return score > 0;
1223
1.39M
}
1224
1225
7
uint32_t Tablet::calc_cold_data_compaction_score() const {
1226
7
    uint32_t score = 0;
1227
7
    std::vector<RowsetMetaSharedPtr> cooldowned_rowsets;
1228
7
    int64_t max_delete_version = 0;
1229
7
    {
1230
7
        std::shared_lock rlock(_meta_lock);
1231
28
        for (const auto& [_, rs_meta] : _tablet_meta->all_rs_metas()) {
1232
28
            if (!rs_meta->is_local()) {
1233
28
                cooldowned_rowsets.push_back(rs_meta);
1234
28
                if (rs_meta->has_delete_predicate() &&
1235
28
                    rs_meta->end_version() > max_delete_version) {
1236
0
                    max_delete_version = rs_meta->end_version();
1237
0
                }
1238
28
            }
1239
28
        }
1240
7
    }
1241
28
    for (auto& rs_meta : cooldowned_rowsets) {
1242
28
        if (rs_meta->end_version() < max_delete_version) {
1243
0
            score += rs_meta->num_segments();
1244
28
        } else {
1245
28
            score += rs_meta->get_compaction_score();
1246
28
        }
1247
28
    }
1248
7
    return (keys_type() != KeysType::DUP_KEYS) ? score * 2 : score;
1249
7
}
1250
1251
uint32_t Tablet::_calc_cumulative_compaction_score(
1252
297k
        std::shared_ptr<CumulativeCompactionPolicy> cumulative_compaction_policy) {
1253
297k
    if (cumulative_compaction_policy == nullptr) [[unlikely]] {
1254
0
        return 0;
1255
0
    }
1256
297k
#ifndef BE_TEST
1257
297k
    if (_cumulative_compaction_policy == nullptr ||
1258
297k
        _cumulative_compaction_policy->name() != cumulative_compaction_policy->name()) {
1259
9.81k
        _cumulative_compaction_policy = cumulative_compaction_policy;
1260
9.81k
    }
1261
297k
#endif
1262
297k
    DBUG_EXECUTE_IF("Tablet._calc_cumulative_compaction_score.return", {
1263
297k
        LOG_WARNING("Tablet._calc_cumulative_compaction_score.return")
1264
297k
                .tag("tablet id", tablet_id());
1265
297k
        return 0;
1266
297k
    });
1267
297k
    return _cumulative_compaction_policy->calc_cumulative_compaction_score(this);
1268
297k
}
1269
1270
1.10M
uint32_t Tablet::_calc_base_compaction_score() const {
1271
1.10M
    uint32_t score = 0;
1272
1.10M
    const int64_t point = cumulative_layer_point();
1273
1.10M
    bool base_rowset_exist = false;
1274
1.10M
    bool has_delete = false;
1275
2.00M
    for (const auto& [_, rs_meta] : _tablet_meta->all_rs_metas()) {
1276
2.00M
        if (rs_meta->start_version() == 0) {
1277
1.10M
            base_rowset_exist = true;
1278
1.10M
        }
1279
2.00M
        if (rs_meta->start_version() >= point || !rs_meta->is_local()) {
1280
            // all_rs_metas() is not sorted, so we use _continue_ other than _break_ here.
1281
1.92M
            continue;
1282
1.92M
        }
1283
87.8k
        if (rs_meta->has_delete_predicate()) {
1284
296
            has_delete = true;
1285
296
        }
1286
87.8k
        score += rs_meta->get_compaction_score();
1287
87.8k
    }
1288
1289
    // In the time series compaction policy, we want the base compaction to be triggered
1290
    // when there are delete versions present.
1291
1.10M
    if (_tablet_meta->compaction_policy() == CUMULATIVE_TIME_SERIES_POLICY) {
1292
88
        return (base_rowset_exist && has_delete) ? score : 0;
1293
88
    }
1294
1295
    // base不存在可能是tablet正在做alter table,先不选它,设score=0
1296
1.10M
    return base_rowset_exist ? score : 0;
1297
1.10M
}
1298
1299
0
void Tablet::max_continuous_version_from_beginning(Version* version, Version* max_version) {
1300
0
    bool has_version_cross;
1301
0
    std::shared_lock rdlock(_meta_lock);
1302
0
    _max_continuous_version_from_beginning_unlocked(version, max_version, &has_version_cross);
1303
0
}
1304
1305
void Tablet::_max_continuous_version_from_beginning_unlocked(Version* version, Version* max_version,
1306
1.26M
                                                             bool* has_version_cross) const {
1307
1.26M
    std::vector<Version> existing_versions;
1308
1.26M
    *has_version_cross = false;
1309
2.22M
    for (const auto& [ver, _] : _tablet_meta->all_rs_metas()) {
1310
2.22M
        existing_versions.emplace_back(ver);
1311
2.22M
    }
1312
1313
    // sort the existing versions in ascending order
1314
1.26M
    std::sort(existing_versions.begin(), existing_versions.end(),
1315
1.94M
              [](const Version& left, const Version& right) {
1316
                  // simple because 2 versions are certainly not overlapping
1317
1.94M
                  return left.first < right.first;
1318
1.94M
              });
1319
1320
1.26M
    Version max_continuous_version = {-1, -1};
1321
3.49M
    for (int i = 0; i < existing_versions.size(); ++i) {
1322
2.22M
        if (existing_versions[i].first > max_continuous_version.second + 1) {
1323
0
            break;
1324
2.22M
        } else if (existing_versions[i].first <= max_continuous_version.second) {
1325
0
            *has_version_cross = true;
1326
0
        }
1327
2.22M
        max_continuous_version = existing_versions[i];
1328
2.22M
    }
1329
1.26M
    *version = max_continuous_version;
1330
    // tablet may not has rowset, eg, tablet has just been clear for restore.
1331
1.26M
    if (max_version != nullptr && !existing_versions.empty()) {
1332
1.26M
        *max_version = existing_versions.back();
1333
1.26M
    }
1334
1.26M
}
1335
1336
178k
void Tablet::calculate_cumulative_point() {
1337
178k
    std::lock_guard<std::shared_mutex> wrlock(_meta_lock);
1338
178k
    SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
1339
178k
    int64_t ret_cumulative_point;
1340
178k
    _cumulative_compaction_policy->calculate_cumulative_point(
1341
178k
            this, _tablet_meta->all_rs_metas(), _cumulative_point, &ret_cumulative_point);
1342
1343
178k
    if (ret_cumulative_point == K_INVALID_CUMULATIVE_POINT) {
1344
176k
        return;
1345
176k
    }
1346
2.07k
    set_cumulative_layer_point(ret_cumulative_point);
1347
2.07k
}
1348
1349
// NOTE: only used when create_table, so it is sure that there is no concurrent reader and writer.
1350
0
void Tablet::delete_all_files() {
1351
    // Release resources like memory and disk space.
1352
0
    std::shared_lock rdlock(_meta_lock);
1353
0
    for (auto it : _rs_version_map) {
1354
0
        static_cast<void>(it.second->remove());
1355
0
    }
1356
0
    _rs_version_map.clear();
1357
1358
0
    for (auto it : _stale_rs_version_map) {
1359
0
        static_cast<void>(it.second->remove());
1360
0
    }
1361
0
    _stale_rs_version_map.clear();
1362
0
}
1363
1364
0
void Tablet::check_tablet_path_exists() {
1365
0
    if (!tablet_path().empty()) {
1366
0
        std::error_code ec;
1367
0
        if (std::filesystem::is_directory(tablet_path(), ec)) {
1368
0
            _is_tablet_path_exists.store(true, std::memory_order_relaxed);
1369
0
        } else if (ec.value() == ENOENT || ec.value() == 0) {
1370
0
            _is_tablet_path_exists.store(false, std::memory_order_relaxed);
1371
0
        }
1372
0
    }
1373
0
}
1374
1375
57.4k
Status Tablet::_contains_version(const Version& version) {
1376
    // check if there exist a rowset contains the added rowset
1377
555k
    for (auto& it : _rs_version_map) {
1378
555k
        if (it.first.contains(version)) {
1379
            // TODO(lingbin): Is this check unnecessary?
1380
            // because the value type is std::shared_ptr, when will it be nullptr?
1381
            // In addition, in this class, there are many places that do not make this judgment
1382
            // when access _rs_version_map's value.
1383
206
            CHECK(it.second != nullptr) << "there exist a version=" << it.first
1384
0
                                        << " contains the input rs with version=" << version
1385
0
                                        << ", but the related rs is null";
1386
206
            return Status::Error<PUSH_VERSION_ALREADY_EXIST>("Tablet push duplicate version {}",
1387
206
                                                             version.to_string());
1388
206
        }
1389
555k
    }
1390
1391
57.2k
    return Status::OK();
1392
57.4k
}
1393
1394
178k
std::vector<RowsetSharedPtr> Tablet::pick_candidate_rowsets_to_cumulative_compaction() {
1395
178k
    std::vector<RowsetSharedPtr> candidate_rowsets;
1396
178k
    if (_cumulative_point == K_INVALID_CUMULATIVE_POINT) {
1397
382
        return candidate_rowsets;
1398
382
    }
1399
178k
    return _pick_visible_rowsets_to_compaction(_cumulative_point,
1400
178k
                                               std::numeric_limits<int64_t>::max());
1401
178k
}
1402
1403
50.2k
std::vector<RowsetSharedPtr> Tablet::pick_candidate_rowsets_to_base_compaction() {
1404
50.2k
    return _pick_visible_rowsets_to_compaction(std::numeric_limits<int64_t>::min(),
1405
50.2k
                                               _cumulative_point - 1);
1406
50.2k
}
1407
1408
std::vector<RowsetSharedPtr> Tablet::_pick_visible_rowsets_to_compaction(
1409
228k
        int64_t min_start_version, int64_t max_start_version) {
1410
228k
    auto [visible_version, update_ts] = get_visible_version_and_time();
1411
228k
    bool update_time_long = MonotonicMillis() - update_ts >
1412
228k
                            config::compaction_keep_invisible_version_timeout_sec * 1000L;
1413
228k
    int32_t keep_invisible_version_limit =
1414
228k
            update_time_long ? config::compaction_keep_invisible_version_min_count
1415
228k
                             : config::compaction_keep_invisible_version_max_count;
1416
1417
228k
    std::vector<RowsetSharedPtr> candidate_rowsets;
1418
228k
    {
1419
228k
        std::shared_lock rlock(_meta_lock);
1420
1.00M
        for (const auto& [version, rs] : _rs_version_map) {
1421
1.00M
            int64_t version_start = version.first;
1422
            // rowset is remote or rowset is not in given range
1423
1.00M
            if (!rs->is_local() || version_start < min_start_version ||
1424
1.00M
                version_start > max_start_version) {
1425
365k
                continue;
1426
365k
            }
1427
1428
            // can compact, met one of the conditions:
1429
            // 1. had been visible;
1430
            // 2. exceeds the limit of keep invisible versions.
1431
640k
            int64_t version_end = version.second;
1432
640k
            if (version_end <= visible_version ||
1433
640k
                version_end > visible_version + keep_invisible_version_limit) {
1434
630k
                candidate_rowsets.push_back(rs);
1435
630k
            }
1436
640k
        }
1437
228k
    }
1438
228k
    std::sort(candidate_rowsets.begin(), candidate_rowsets.end(), Rowset::comparator);
1439
228k
    return candidate_rowsets;
1440
228k
}
1441
1442
0
std::vector<RowsetSharedPtr> Tablet::pick_candidate_rowsets_to_full_compaction() {
1443
0
    std::vector<RowsetSharedPtr> candidate_rowsets;
1444
0
    traverse_rowsets([&candidate_rowsets](const auto& rs) {
1445
        // Do full compaction on all local rowsets.
1446
0
        if (rs->is_local()) {
1447
0
            candidate_rowsets.emplace_back(rs);
1448
0
        }
1449
0
    });
1450
0
    std::sort(candidate_rowsets.begin(), candidate_rowsets.end(), Rowset::comparator);
1451
0
    return candidate_rowsets;
1452
0
}
1453
1454
492
std::vector<RowsetSharedPtr> Tablet::pick_candidate_rowsets_to_binlog_compaction() {
1455
492
    auto [visible_version, _] = get_visible_version_and_time();
1456
492
    int64_t now = UnixSeconds();
1457
492
    std::vector<RowsetSharedPtr> candidate_rowsets;
1458
492
    int64_t max_processable_old_version = 0;
1459
492
    int64_t max_version = 0;
1460
492
    {
1461
492
        std::shared_lock rlock(_meta_lock);
1462
492
        bool filter_new_rowset =
1463
492
                _row_binlog_rs_version_map.size() <= config::binlog_compaction_file_count_threshold;
1464
492
        for (const auto& [version, rs] : _row_binlog_rs_version_map) {
1465
492
            max_version = std::max(max_version, version.second);
1466
492
            if (filter_new_rowset && rs->rowset_meta()->is_singleton_delta() &&
1467
492
                rs->rowset_meta()->newest_write_timestamp() +
1468
0
                                config::binlog_compaction_wait_timesec_after_visible >
1469
0
                        now) {
1470
0
                continue;
1471
0
            }
1472
492
            max_processable_old_version = std::max(max_processable_old_version, version.second);
1473
492
        }
1474
1475
492
        if (max_version <= visible_version + config::compaction_keep_invisible_version_max_count) {
1476
492
            max_processable_old_version = std::min(max_processable_old_version, visible_version);
1477
492
        }
1478
1479
492
        for (const auto& [version, rs] : _row_binlog_rs_version_map) {
1480
492
            if (!rs->is_local() || version.first > max_processable_old_version) {
1481
0
                continue;
1482
0
            }
1483
492
            candidate_rowsets.emplace_back(rs);
1484
492
        }
1485
492
    }
1486
492
    std::sort(candidate_rowsets.begin(), candidate_rowsets.end(), Rowset::comparator);
1487
492
    return candidate_rowsets;
1488
492
}
1489
1490
996
uint32_t Tablet::_calc_binlog_compaction_score(int8_t* prefer_compaction_level) {
1491
996
    DCHECK(_binlog_compaction_policy != nullptr);
1492
996
    return _binlog_compaction_policy->calc_binlog_compaction_score(this, prefer_compaction_level);
1493
996
}
1494
1495
std::vector<RowsetSharedPtr> Tablet::pick_candidate_rowsets_to_build_inverted_index(
1496
21
        const std::set<int64_t>& alter_index_uids, bool is_drop_op) {
1497
21
    std::vector<RowsetSharedPtr> candidate_rowsets;
1498
21
    {
1499
21
        std::shared_lock rlock(_meta_lock);
1500
41
        auto has_alter_inverted_index = [&](RowsetSharedPtr rowset) -> bool {
1501
45
            for (const auto& index_id : alter_index_uids) {
1502
45
                if (rowset->tablet_schema()->has_inverted_index_with_index_id(index_id)) {
1503
8
                    return true;
1504
8
                }
1505
45
            }
1506
33
            return false;
1507
41
        };
1508
1509
21
        for (const auto& [version, rs] : _rs_version_map) {
1510
21
            if (!has_alter_inverted_index(rs) && is_drop_op) {
1511
1
                continue;
1512
1
            }
1513
20
            if (has_alter_inverted_index(rs) && !is_drop_op) {
1514
0
                continue;
1515
0
            }
1516
1517
20
            if (rs->is_local()) {
1518
20
                candidate_rowsets.push_back(rs);
1519
20
            }
1520
20
        }
1521
21
    }
1522
21
    std::sort(candidate_rowsets.begin(), candidate_rowsets.end(), Rowset::comparator);
1523
21
    return candidate_rowsets;
1524
21
}
1525
1526
1.49M
std::tuple<int64_t, int64_t> Tablet::get_visible_version_and_time() const {
1527
    // some old tablet has bug, its partition_id is 0, fe couldn't update its visible version.
1528
    // so let this tablet's visible version become int64 max.
1529
1.49M
    auto version_info = _visible_version.load();
1530
1.49M
    if (version_info != nullptr && partition_id() != 0) {
1531
1.49M
        return std::make_tuple(version_info->version.load(std::memory_order_relaxed),
1532
1.49M
                               version_info->update_ts);
1533
1.49M
    } else {
1534
70
        return std::make_tuple(std::numeric_limits<int64_t>::max(),
1535
70
                               std::numeric_limits<int64_t>::max());
1536
70
    }
1537
1.49M
}
1538
1539
// For http compaction action
1540
0
void Tablet::get_compaction_status(std::string* json_result) {
1541
0
    rapidjson::Document root;
1542
0
    root.SetObject();
1543
1544
0
    rapidjson::Document path_arr;
1545
0
    path_arr.SetArray();
1546
1547
0
    std::vector<RowsetSharedPtr> rowsets;
1548
0
    std::vector<RowsetSharedPtr> stale_rowsets;
1549
0
    std::vector<bool> delete_flags;
1550
0
    {
1551
0
        std::shared_lock rdlock(_meta_lock);
1552
0
        rowsets.reserve(_rs_version_map.size());
1553
0
        for (auto& it : _rs_version_map) {
1554
0
            rowsets.push_back(it.second);
1555
0
        }
1556
0
        std::sort(rowsets.begin(), rowsets.end(), Rowset::comparator);
1557
1558
0
        stale_rowsets.reserve(_stale_rs_version_map.size());
1559
0
        for (auto& it : _stale_rs_version_map) {
1560
0
            stale_rowsets.push_back(it.second);
1561
0
        }
1562
0
        std::sort(stale_rowsets.begin(), stale_rowsets.end(), Rowset::comparator);
1563
1564
0
        delete_flags.reserve(rowsets.size());
1565
0
        for (auto& rs : rowsets) {
1566
0
            delete_flags.push_back(rs->rowset_meta()->has_delete_predicate());
1567
0
        }
1568
        // get snapshot version path json_doc
1569
0
        _timestamped_version_tracker.get_stale_version_path_json_doc(path_arr);
1570
0
    }
1571
0
    rapidjson::Value cumulative_policy_type;
1572
0
    std::string policy_type_str = "cumulative compaction policy not initializied";
1573
0
    if (_cumulative_compaction_policy != nullptr) {
1574
0
        policy_type_str = _cumulative_compaction_policy->name();
1575
0
    }
1576
0
    cumulative_policy_type.SetString(policy_type_str.c_str(),
1577
0
                                     cast_set<uint32_t>(policy_type_str.length()),
1578
0
                                     root.GetAllocator());
1579
0
    root.AddMember("cumulative policy type", cumulative_policy_type, root.GetAllocator());
1580
0
    root.AddMember("cumulative point", _cumulative_point.load(), root.GetAllocator());
1581
1582
0
#define FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, key, unixmillis_value)                 \
1583
0
    {                                                                                \
1584
0
        rapidjson::Value value;                                                      \
1585
0
        std::string format_str = ToStringFromUnixMillis(unixmillis_value.load());    \
1586
0
        value.SetString(format_str.c_str(), cast_set<uint32_t>(format_str.length()), \
1587
0
                        root.GetAllocator());                                        \
1588
0
        root.AddMember(key, value, root.GetAllocator());                             \
1589
0
    }
1590
0
#define FORMAT_STRING_ADD_JSON_NODE(root, key, str_value)                          \
1591
0
    {                                                                              \
1592
0
        rapidjson::Value value;                                                    \
1593
0
        value.SetString(str_value.c_str(), cast_set<uint32_t>(str_value.length()), \
1594
0
                        root.GetAllocator());                                      \
1595
0
        root.AddMember(key, value, root.GetAllocator());                           \
1596
0
    }
1597
1598
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last cumulative failure time",
1599
0
                                    _last_cumu_compaction_failure_millis)
1600
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last base failure time",
1601
0
                                    _last_base_compaction_failure_millis)
1602
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last full failure time",
1603
0
                                    _last_full_compaction_failure_millis)
1604
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last binlog failure time",
1605
0
                                    _last_binlog_compaction_failure_millis)
1606
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last cumulative success time",
1607
0
                                    _last_cumu_compaction_success_millis)
1608
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last base success time",
1609
0
                                    _last_base_compaction_success_millis)
1610
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last full success time",
1611
0
                                    _last_full_compaction_success_millis) {
1612
0
        int64_t last_binlog_success_millis = 0;
1613
0
        for (int8_t level = 0; level < BinlogCompactionPolicy::kBinlogCompactionMaxLevel; ++level) {
1614
0
            last_binlog_success_millis =
1615
0
                    std::max(last_binlog_success_millis,
1616
0
                             _last_binlog_compaction_success_millis[level].load());
1617
0
        }
1618
0
        rapidjson::Value value;
1619
0
        std::string format_str = ToStringFromUnixMillis(last_binlog_success_millis);
1620
0
        value.SetString(format_str.c_str(), cast_set<uint32_t>(format_str.length()),
1621
0
                        root.GetAllocator());
1622
0
        root.AddMember("last binlog success time", value, root.GetAllocator());
1623
0
    }
1624
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last cumulative schedule time",
1625
0
                                    _last_cumu_compaction_schedule_millis)
1626
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last base schedule time",
1627
0
                                    _last_base_compaction_schedule_millis)
1628
0
    FORMAT_UNIXMILLIS_ADD_JSON_NODE(root, "last full schedule time",
1629
0
                                    _last_full_compaction_schedule_millis)
1630
0
    FORMAT_STRING_ADD_JSON_NODE(root, "last cumulative status", _last_cumu_compaction_status)
1631
0
    FORMAT_STRING_ADD_JSON_NODE(root, "last base status", _last_base_compaction_status)
1632
0
    FORMAT_STRING_ADD_JSON_NODE(root, "last full status", _last_full_compaction_status)
1633
0
    FORMAT_STRING_ADD_JSON_NODE(root, "last binlog status", _last_binlog_compaction_status)
1634
1635
    // last single replica compaction status
1636
    // "single replica compaction status": {
1637
    //     "remote peer": "172.100.1.0:10875",
1638
    //     "last failure status": "",
1639
    //     "last fetched rowset": "[8-10]"
1640
    // }
1641
0
    rapidjson::Document status;
1642
0
    status.SetObject();
1643
0
    TReplicaInfo replica_info;
1644
0
    std::string dummp_token;
1645
0
    if (tablet_meta()->tablet_schema()->enable_single_replica_compaction() &&
1646
0
        _engine.get_peer_replica_info(tablet_id(), &replica_info, &dummp_token)) {
1647
        // remote peer
1648
0
        rapidjson::Value peer_addr;
1649
0
        std::string addr = replica_info.host + ":" + std::to_string(replica_info.brpc_port);
1650
0
        peer_addr.SetString(addr.c_str(), cast_set<uint32_t>(addr.length()), status.GetAllocator());
1651
0
        status.AddMember("remote peer", peer_addr, status.GetAllocator());
1652
        // last failure status
1653
0
        rapidjson::Value compaction_status;
1654
0
        compaction_status.SetString(
1655
0
                _last_single_compaction_failure_status.c_str(),
1656
0
                cast_set<uint32_t>(_last_single_compaction_failure_status.length()),
1657
0
                status.GetAllocator());
1658
0
        status.AddMember("last failure status", compaction_status, status.GetAllocator());
1659
        // last fetched rowset
1660
0
        rapidjson::Value version;
1661
0
        std::string fetched_version = _last_fetched_version.to_string();
1662
0
        version.SetString(fetched_version.c_str(), cast_set<uint32_t>(fetched_version.length()),
1663
0
                          status.GetAllocator());
1664
0
        status.AddMember("last fetched rowset", version, status.GetAllocator());
1665
0
        root.AddMember("single replica compaction status", status, root.GetAllocator());
1666
0
    }
1667
1668
    // print all rowsets' version as an array
1669
0
    rapidjson::Document versions_arr;
1670
0
    rapidjson::Document missing_versions_arr;
1671
0
    versions_arr.SetArray();
1672
0
    missing_versions_arr.SetArray();
1673
0
    int64_t last_version = -1;
1674
0
    for (auto& rowset : rowsets) {
1675
0
        const Version& ver = rowset->version();
1676
0
        if (ver.first != last_version + 1) {
1677
0
            rapidjson::Value miss_value;
1678
0
            miss_value.SetString(fmt::format("[{}-{}]", last_version + 1, ver.first - 1).c_str(),
1679
0
                                 missing_versions_arr.GetAllocator());
1680
0
            missing_versions_arr.PushBack(miss_value, missing_versions_arr.GetAllocator());
1681
0
        }
1682
0
        rapidjson::Value value;
1683
0
        std::string version_str = rowset->get_rowset_info_str();
1684
0
        value.SetString(version_str.c_str(), cast_set<uint32_t>(version_str.length()),
1685
0
                        versions_arr.GetAllocator());
1686
0
        versions_arr.PushBack(value, versions_arr.GetAllocator());
1687
0
        last_version = ver.second;
1688
0
    }
1689
0
    root.AddMember("rowsets", versions_arr, root.GetAllocator());
1690
0
    root.AddMember("missing_rowsets", missing_versions_arr, root.GetAllocator());
1691
1692
    // print all stale rowsets' version as an array
1693
0
    rapidjson::Document stale_versions_arr;
1694
0
    stale_versions_arr.SetArray();
1695
0
    for (auto& rowset : stale_rowsets) {
1696
0
        rapidjson::Value value;
1697
0
        std::string version_str = rowset->get_rowset_info_str();
1698
0
        value.SetString(version_str.c_str(), cast_set<uint32_t>(version_str.length()),
1699
0
                        stale_versions_arr.GetAllocator());
1700
0
        stale_versions_arr.PushBack(value, stale_versions_arr.GetAllocator());
1701
0
    }
1702
0
    root.AddMember("stale_rowsets", stale_versions_arr, root.GetAllocator());
1703
1704
    // add stale version rowsets
1705
0
    root.AddMember("stale version path", path_arr, root.GetAllocator());
1706
1707
    // to json string
1708
0
    rapidjson::StringBuffer strbuf;
1709
0
    rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(strbuf);
1710
0
    root.Accept(writer);
1711
0
    *json_result = std::string(strbuf.GetString());
1712
0
}
1713
1714
294k
bool Tablet::do_tablet_meta_checkpoint() {
1715
294k
    std::lock_guard<std::shared_mutex> store_lock(_meta_store_lock);
1716
294k
    if (_newly_created_rowset_num == 0) {
1717
287k
        return false;
1718
287k
    }
1719
7.51k
    if (UnixMillis() - _last_checkpoint_time <
1720
7.51k
                config::tablet_meta_checkpoint_min_interval_secs * 1000 &&
1721
7.51k
        _newly_created_rowset_num < config::tablet_meta_checkpoint_min_new_rowsets_num) {
1722
60
        return false;
1723
60
    }
1724
    // hold read-lock other than write-lock, because it will not modify meta structure
1725
7.45k
    std::shared_lock rdlock(_meta_lock);
1726
7.45k
    if (tablet_state() != TABLET_RUNNING) {
1727
9
        LOG(INFO) << "tablet is under state=" << tablet_state()
1728
9
                  << ", not running, skip do checkpoint"
1729
9
                  << ", tablet=" << tablet_id();
1730
9
        return false;
1731
9
    }
1732
7.44k
    VLOG_NOTICE << "start to do tablet meta checkpoint, tablet=" << tablet_id();
1733
7.44k
    save_meta();
1734
    // if save meta successfully, then should remove the rowset meta existing in tablet
1735
    // meta from rowset meta store
1736
17.7k
    for (const auto& [_, rs_meta] : _tablet_meta->all_rs_metas()) {
1737
        // If we delete it from rowset manager's meta explicitly in previous checkpoint, just skip.
1738
17.7k
        if (rs_meta->is_remove_from_rowset_meta()) {
1739
369
            continue;
1740
369
        }
1741
17.3k
        if (RowsetMetaManager::check_rowset_meta(_data_dir->get_meta(), tablet_uid(),
1742
17.3k
                                                 rs_meta->rowset_id())) {
1743
12.1k
            RETURN_FALSE_IF_ERROR(RowsetMetaManager::remove(_data_dir->get_meta(), tablet_uid(),
1744
12.1k
                                                            rs_meta->rowset_id()));
1745
12.1k
            VLOG_NOTICE << "remove rowset id from meta store because it is already persistent with "
1746
0
                        << "tablet meta, rowset_id=" << rs_meta->rowset_id();
1747
12.1k
        }
1748
17.3k
        rs_meta->set_remove_from_rowset_meta();
1749
17.3k
    }
1750
1751
    // check _stale_rs_version_map to remove meta from rowset meta store
1752
31.8k
    for (const auto& [_, rs_meta] : _tablet_meta->all_stale_rs_metas()) {
1753
        // If we delete it from rowset manager's meta explicitly in previous checkpoint, just skip.
1754
31.8k
        if (rs_meta->is_remove_from_rowset_meta()) {
1755
2.30k
            continue;
1756
2.30k
        }
1757
29.5k
        if (RowsetMetaManager::check_rowset_meta(_data_dir->get_meta(), tablet_uid(),
1758
29.5k
                                                 rs_meta->rowset_id())) {
1759
28.4k
            RETURN_FALSE_IF_ERROR(RowsetMetaManager::remove(_data_dir->get_meta(), tablet_uid(),
1760
28.4k
                                                            rs_meta->rowset_id()));
1761
28.4k
            VLOG_NOTICE << "remove rowset id from meta store because it is already persistent with "
1762
0
                        << "tablet meta, rowset_id=" << rs_meta->rowset_id();
1763
28.4k
        }
1764
29.5k
        rs_meta->set_remove_from_rowset_meta();
1765
29.5k
    }
1766
1767
    // Remove row binlog metas from rowset meta store after tablet meta is checkpointed.
1768
    // Row binlog metas are stored in meta KV with key:
1769
    //   {kRowBinlogPrefix}{tablet_uid}_{base_rowset_id}_{row_binlog_rowset_id}
1770
    // Here we only have row binlog rowset metas, so locate base rowset by version.
1771
7.44k
    const auto& base_rs_metas = _tablet_meta->all_rs_metas();
1772
7.44k
    const auto& stale_rs_metas = _tablet_meta->all_stale_rs_metas();
1773
7.44k
    for (const auto& [_, rb_meta] : _tablet_meta->all_row_binlog_rs_metas()) {
1774
        // Reuse the same flag to avoid repeated removals across checkpoints.
1775
0
        if (rb_meta->is_remove_from_rowset_meta()) {
1776
0
            continue;
1777
0
        }
1778
1779
0
        RowsetMetaSharedPtr base_rs_meta;
1780
0
        if (auto base_it = base_rs_metas.find(rb_meta->version()); base_it != base_rs_metas.end()) {
1781
0
            base_rs_meta = base_it->second;
1782
0
        } else if (auto stale_it = stale_rs_metas.find(rb_meta->version());
1783
0
                   stale_it != stale_rs_metas.end()) {
1784
0
            base_rs_meta = stale_it->second;
1785
0
        }
1786
0
        if (base_rs_meta == nullptr) {
1787
0
            LOG(WARNING) << "failed to locate base rowset meta for binlog<row> by version, tablet="
1788
0
                         << tablet_id() << ", version=" << rb_meta->version().to_string()
1789
0
                         << ", binlog_rowset_id=" << rb_meta->rowset_id()
1790
0
                         << ", try to remove by scanning row-binlog meta store";
1791
0
            RETURN_FALSE_IF_ERROR(RowsetMetaManager::remove_row_binlog_metas(
1792
0
                    _data_dir->get_meta(), tablet_uid(), {rb_meta->rowset_id()}));
1793
0
            rb_meta->set_remove_from_rowset_meta();
1794
0
            continue;
1795
0
        }
1796
1797
0
        RETURN_FALSE_IF_ERROR(RowsetMetaManager::remove_row_binlog(
1798
0
                _data_dir->get_meta(), tablet_uid(), base_rs_meta->rowset_id(),
1799
0
                rb_meta->rowset_id()));
1800
0
        VLOG_NOTICE << "remove binlog<row> meta from meta store, base_rowset_id="
1801
0
                    << base_rs_meta->rowset_id() << ", binlog_rowset_id=" << rb_meta->rowset_id();
1802
0
        rb_meta->set_remove_from_rowset_meta();
1803
0
    }
1804
1805
7.44k
    if (keys_type() == UNIQUE_KEYS && enable_unique_key_merge_on_write()) {
1806
1.90k
        RETURN_FALSE_IF_ERROR(TabletMetaManager::remove_old_version_delete_bitmap(
1807
1.90k
                _data_dir, tablet_id(), max_version_unlocked()));
1808
1.90k
    }
1809
1810
7.44k
    _newly_created_rowset_num = 0;
1811
7.44k
    _last_checkpoint_time = UnixMillis();
1812
7.44k
    return true;
1813
7.44k
}
1814
1815
51.5k
bool Tablet::rowset_meta_is_useful(RowsetMetaSharedPtr rowset_meta) {
1816
51.5k
    std::shared_lock rdlock(_meta_lock);
1817
51.5k
    bool find_version = false;
1818
177k
    for (auto& version_rowset : _rs_version_map) {
1819
177k
        if (version_rowset.second->rowset_id() == rowset_meta->rowset_id()) {
1820
12.6k
            return true;
1821
12.6k
        }
1822
164k
        if (version_rowset.second->contains_version(rowset_meta->version())) {
1823
38.9k
            find_version = true;
1824
38.9k
        }
1825
164k
    }
1826
2.27M
    for (auto& stale_version_rowset : _stale_rs_version_map) {
1827
2.27M
        if (stale_version_rowset.second->rowset_id() == rowset_meta->rowset_id()) {
1828
38.2k
            return true;
1829
38.2k
        }
1830
2.23M
        if (stale_version_rowset.second->contains_version(rowset_meta->version())) {
1831
106k
            find_version = true;
1832
106k
        }
1833
2.23M
    }
1834
662
    if (rowset_meta->is_row_binlog()) {
1835
0
        for (auto& version_rowset : _row_binlog_rs_version_map) {
1836
0
            if (version_rowset.second->rowset_id() == rowset_meta->rowset_id()) {
1837
0
                return true;
1838
0
            }
1839
0
            if (version_rowset.second->contains_version(rowset_meta->version())) {
1840
0
                find_version = true;
1841
0
            }
1842
0
        }
1843
0
    }
1844
662
    return !find_version;
1845
662
}
1846
1847
58.1k
bool Tablet::_contains_rowset(const RowsetId rowset_id) {
1848
557k
    for (auto& version_rowset : _rs_version_map) {
1849
557k
        if (version_rowset.second->rowset_id() == rowset_id) {
1850
63
            return true;
1851
63
        }
1852
557k
    }
1853
1.79M
    for (auto& stale_version_rowset : _stale_rs_version_map) {
1854
1.79M
        if (stale_version_rowset.second->rowset_id() == rowset_id) {
1855
602
            return true;
1856
602
        }
1857
1.79M
    }
1858
57.4k
    return false;
1859
58.0k
}
1860
1861
// need check if consecutive version missing in full report
1862
// alter tablet will ignore this check
1863
void Tablet::build_tablet_report_info(TTabletInfo* tablet_info,
1864
                                      bool enable_consecutive_missing_check,
1865
1.26M
                                      bool enable_path_check) {
1866
1.26M
    std::shared_lock rdlock(_meta_lock);
1867
1.26M
    tablet_info->__set_tablet_id(_tablet_meta->tablet_id());
1868
1.26M
    tablet_info->__set_schema_hash(_tablet_meta->schema_hash());
1869
1.26M
    tablet_info->__set_row_count(_tablet_meta->num_rows());
1870
1.26M
    tablet_info->__set_data_size(_tablet_meta->tablet_local_size());
1871
1872
    // Here we need to report to FE if there are any missing versions of tablet.
1873
    // We start from the initial version and traverse backwards until we meet a discontinuous version.
1874
1.26M
    Version cversion;
1875
1.26M
    Version max_version;
1876
1.26M
    bool has_version_cross;
1877
1.26M
    _max_continuous_version_from_beginning_unlocked(&cversion, &max_version, &has_version_cross);
1878
    // cause publish version task runs concurrently, version may be flying
1879
    // so we add a consecutive miss check to solve this problem:
1880
    // if publish version 5 arrives but version 4 flying, we may judge replica miss version
1881
    // and set version miss in tablet_info, which makes fe treat this replica as unhealth
1882
    // and lead to other problems
1883
1.26M
    if (enable_consecutive_missing_check) {
1884
1.26M
        if (cversion.second < max_version.second) {
1885
0
            if (_last_missed_version == cversion.second + 1) {
1886
0
                if (MonotonicSeconds() - _last_missed_time_s >= 60) {
1887
                    // version missed for over 60 seconds
1888
0
                    tablet_info->__set_version_miss(true);
1889
0
                    _last_missed_version = -1;
1890
0
                    _last_missed_time_s = 0;
1891
0
                }
1892
0
            } else {
1893
0
                _last_missed_version = cversion.second + 1;
1894
0
                _last_missed_time_s = MonotonicSeconds();
1895
0
            }
1896
0
        }
1897
1.26M
    } else {
1898
22
        tablet_info->__set_version_miss(cversion.second < max_version.second);
1899
22
    }
1900
1901
1.26M
    DBUG_EXECUTE_IF("Tablet.build_tablet_report_info.version_miss", {
1902
1.26M
        auto tablet_id = dp->param<int64_t>("tablet_id", -1);
1903
1.26M
        if (tablet_id != -1 && tablet_id == _tablet_meta->tablet_id()) {
1904
1.26M
            auto miss = dp->param<bool>("version_miss", true);
1905
1.26M
            tablet_info->__set_version_miss(miss);
1906
1.26M
        }
1907
1.26M
    });
1908
1909
    // find rowset with max version
1910
1.26M
    auto iter = _rs_version_map.find(max_version);
1911
1.26M
    if (iter == _rs_version_map.end()) {
1912
        // If the tablet is in running state, it must not be doing schema-change. so if we can not
1913
        // access its rowsets, it means that the tablet is bad and needs to be reported to the FE
1914
        // for subsequent repairs (through the cloning task)
1915
0
        if (tablet_state() == TABLET_RUNNING) {
1916
0
            tablet_info->__set_used(false);
1917
0
        }
1918
        // For other states, FE knows that the tablet is in a certain change process, so here
1919
        // still sets the state to normal when reporting. Note that every task has an timeout,
1920
        // so if the task corresponding to this change hangs, when the task timeout, FE will know
1921
        // and perform state modification operations.
1922
0
    }
1923
1924
1.26M
    if (tablet_state() == TABLET_RUNNING) {
1925
1.26M
        if (has_version_cross || is_io_error_too_times() || !data_dir()->is_used()) {
1926
0
            LOG(INFO) << "report " << tablet_id() << " as bad, version_cross=" << has_version_cross
1927
0
                      << ", ioe times=" << get_io_error_times() << ", data_dir used "
1928
0
                      << data_dir()->is_used();
1929
0
            tablet_info->__set_used(false);
1930
0
        }
1931
1932
1.26M
        if (enable_path_check) {
1933
1.26M
            if (!_is_tablet_path_exists.exchange(true, std::memory_order_relaxed)) {
1934
0
                LOG(INFO) << "report " << tablet_id() << " as bad, tablet directory not found";
1935
0
                tablet_info->__set_used(false);
1936
0
            }
1937
1.26M
        }
1938
1.26M
    }
1939
1940
    // There are two cases when tablet state is TABLET_NOTREADY
1941
    // case 1: tablet is doing schema change. Fe knows it's state, doing nothing.
1942
    // case 2: tablet has finished schema change, but failed. Fe will perform recovery.
1943
1.26M
    if (tablet_state() == TABLET_NOTREADY && is_alter_failed()) {
1944
0
        tablet_info->__set_used(false);
1945
0
    }
1946
1947
1.26M
    if (tablet_state() == TABLET_SHUTDOWN) {
1948
0
        tablet_info->__set_used(false);
1949
0
    }
1950
1951
1.26M
    DBUG_EXECUTE_IF("Tablet.build_tablet_report_info.used", {
1952
1.26M
        auto tablet_id = dp->param<int64_t>("tablet_id", -1);
1953
1.26M
        if (tablet_id != -1 && tablet_id == _tablet_meta->tablet_id()) {
1954
1.26M
            auto used = dp->param<bool>("used", true);
1955
1.26M
            LOG_WARNING("Tablet.build_tablet_report_info.used")
1956
1.26M
                    .tag("tablet id", tablet_id)
1957
1.26M
                    .tag("used", used);
1958
1.26M
            tablet_info->__set_used(used);
1959
1.26M
        } else {
1960
1.26M
            LOG_WARNING("Tablet.build_tablet_report_info.used").tag("tablet id", tablet_id);
1961
1.26M
        }
1962
1.26M
    });
1963
1964
1.26M
    int64_t total_version_count = _tablet_meta->version_count();
1965
1966
    // For compatibility.
1967
    // For old fe, it wouldn't send visible version request to be, then be's visible version is always 0.
1968
    // Let visible_version_count set to total_version_count in be's report.
1969
1.26M
    int64_t visible_version_count = total_version_count;
1970
1.26M
    if (auto [visible_version, _] = get_visible_version_and_time(); visible_version > 0) {
1971
966k
        visible_version_count = _tablet_meta->version_count_cross_with_range({0, visible_version});
1972
966k
    }
1973
    // the report version is the largest continuous version, same logic as in FE side
1974
1.26M
    tablet_info->__set_version(cversion.second);
1975
    // Useless but it is a required filed in TTabletInfo
1976
1.26M
    tablet_info->__set_version_hash(0);
1977
1.26M
    tablet_info->__set_partition_id(_tablet_meta->partition_id());
1978
1.26M
    tablet_info->__set_storage_medium(_data_dir->storage_medium());
1979
1.26M
    tablet_info->__set_total_version_count(total_version_count);
1980
1.26M
    tablet_info->__set_visible_version_count(visible_version_count);
1981
1.26M
    tablet_info->__set_path_hash(_data_dir->path_hash());
1982
1.26M
    tablet_info->__set_is_in_memory(_tablet_meta->tablet_schema()->is_in_memory());
1983
1.26M
    tablet_info->__set_replica_id(replica_id());
1984
1.26M
    tablet_info->__set_remote_data_size(_tablet_meta->tablet_remote_size());
1985
1.26M
    if (_tablet_meta->cooldown_meta_id().initialized()) { // has cooldowned data
1986
8
        tablet_info->__set_cooldown_term(_cooldown_conf.term);
1987
8
        tablet_info->__set_cooldown_meta_id(_tablet_meta->cooldown_meta_id().to_thrift());
1988
8
    }
1989
1.26M
    if (tablet_state() == TABLET_RUNNING && _tablet_meta->storage_policy_id() > 0) {
1990
        // tablet may not have cooldowned data, but the storage policy is set
1991
1.30k
        tablet_info->__set_cooldown_term(_cooldown_conf.term);
1992
1.30k
    }
1993
1.26M
    tablet_info->__set_local_index_size(_tablet_meta->tablet_local_index_size());
1994
1.26M
    tablet_info->__set_local_segment_size(_tablet_meta->tablet_local_segment_size());
1995
1.26M
    tablet_info->__set_remote_index_size(_tablet_meta->tablet_remote_index_size());
1996
1.26M
    tablet_info->__set_remote_segment_size(_tablet_meta->tablet_remote_segment_size());
1997
1.26M
    if (enable_row_binlog()) {
1998
242
        int64_t total_binlog_size = _tablet_meta->binlog_size();
1999
242
        int64_t total_binlog_file_num = _tablet_meta->binlog_file_num();
2000
242
        tablet_info->__set_binlog_file_num(total_binlog_file_num);
2001
242
        tablet_info->__set_binlog_size(total_binlog_size);
2002
242
    }
2003
1.26M
}
2004
2005
2
void Tablet::report_error(const Status& st) {
2006
2
    if (st.is<ErrorCode::IO_ERROR>()) {
2007
0
        ++_io_error_times;
2008
2
    } else if (st.is<ErrorCode::CORRUPTION>()) {
2009
1
        _io_error_times = config::max_tablet_io_errors + 1;
2010
1
    } else if (st.is<ErrorCode::NOT_FOUND>()) {
2011
0
        check_tablet_path_exists();
2012
0
        if (!_is_tablet_path_exists.load(std::memory_order_relaxed)) {
2013
0
            _io_error_times = config::max_tablet_io_errors + 1;
2014
0
        }
2015
0
    }
2016
2
}
2017
2018
Status Tablet::prepare_compaction_and_calculate_permits(
2019
        CompactionType compaction_type, const TabletSharedPtr& tablet,
2020
        std::shared_ptr<CompactionMixin>& compaction, int64_t& permits,
2021
229k
        int8_t prefer_compaction_level) {
2022
229k
    if (compaction_type == CompactionType::CUMULATIVE_COMPACTION) {
2023
178k
        MonotonicStopWatch watch;
2024
178k
        watch.start();
2025
2026
178k
        compaction = std::make_shared<CumulativeCompaction>(tablet->_engine, tablet);
2027
178k
        DorisMetrics::instance()->cumulative_compaction_request_total->increment(1);
2028
178k
        Status res = compaction->prepare_compact();
2029
178k
        if (!config::disable_compaction_trace_log &&
2030
178k
            static_cast<double>(watch.elapsed_time()) / 1e9 >
2031
0
                    config::cumulative_compaction_trace_threshold) {
2032
0
            std::stringstream ss;
2033
0
            compaction->runtime_profile()->pretty_print(&ss);
2034
0
            LOG(WARNING) << "prepare cumulative compaction cost "
2035
0
                         << static_cast<double>(watch.elapsed_time()) / 1e9 << std::endl
2036
0
                         << ss.str();
2037
0
        }
2038
2039
178k
        if (!res.ok()) {
2040
174k
            permits = 0;
2041
            // if we meet a delete version, should increase the cumulative point to let base compaction handle the delete version.
2042
            // no need to wait 5s.
2043
174k
            if (!res.is<ErrorCode::CUMULATIVE_MEET_DELETE_VERSION>() ||
2044
174k
                config::enable_sleep_between_delete_cumu_compaction) {
2045
174k
                tablet->set_last_cumu_compaction_failure_time(UnixMillis());
2046
174k
            }
2047
174k
            if (!res.is<CUMULATIVE_NO_SUITABLE_VERSION>() &&
2048
174k
                !res.is<ErrorCode::CUMULATIVE_MEET_DELETE_VERSION>()) {
2049
0
                DorisMetrics::instance()->cumulative_compaction_request_failed->increment(1);
2050
0
                return Status::InternalError("prepare cumulative compaction with err: {}",
2051
0
                                             res.to_string());
2052
0
            }
2053
            // return OK if OLAP_ERR_CUMULATIVE_NO_SUITABLE_VERSION, so that we don't need to
2054
            // print too much useless logs.
2055
            // And because we set permits to 0, so even if we return OK here, nothing will be done.
2056
174k
            return Status::OK();
2057
174k
        }
2058
178k
    } else if (compaction_type == CompactionType::BASE_COMPACTION) {
2059
50.2k
        MonotonicStopWatch watch;
2060
50.2k
        watch.start();
2061
2062
50.2k
        compaction = std::make_shared<BaseCompaction>(tablet->_engine, tablet);
2063
50.2k
        DorisMetrics::instance()->base_compaction_request_total->increment(1);
2064
50.2k
        Status res = compaction->prepare_compact();
2065
50.2k
        if (!config::disable_compaction_trace_log &&
2066
50.2k
            static_cast<double>(watch.elapsed_time()) / 1e9 >
2067
0
                    config::base_compaction_trace_threshold) {
2068
0
            std::stringstream ss;
2069
0
            compaction->runtime_profile()->pretty_print(&ss);
2070
0
            LOG(WARNING) << "prepare base compaction cost "
2071
0
                         << static_cast<double>(watch.elapsed_time()) / 1e9 << std::endl
2072
0
                         << ss.str();
2073
0
        }
2074
2075
50.2k
        tablet->set_last_base_compaction_status(res.to_string());
2076
50.2k
        if (!res.ok()) {
2077
50.2k
            tablet->set_last_base_compaction_failure_time(UnixMillis());
2078
50.2k
            permits = 0;
2079
50.2k
            if (!res.is<BE_NO_SUITABLE_VERSION>()) {
2080
0
                DorisMetrics::instance()->base_compaction_request_failed->increment(1);
2081
0
                return Status::InternalError("prepare base compaction with err: {}",
2082
0
                                             res.to_string());
2083
0
            }
2084
            // return OK if OLAP_ERR_BE_NO_SUITABLE_VERSION, so that we don't need to
2085
            // print too much useless logs.
2086
            // And because we set permits to 0, so even if we return OK here, nothing will be done.
2087
50.2k
            return Status::OK();
2088
50.2k
        }
2089
50.2k
    } else if (compaction_type == CompactionType::FULL_COMPACTION) {
2090
0
        DCHECK_EQ(compaction_type, CompactionType::FULL_COMPACTION);
2091
2092
0
        compaction = std::make_shared<FullCompaction>(tablet->_engine, tablet);
2093
0
        Status res = compaction->prepare_compact();
2094
0
        if (!res.ok()) {
2095
0
            tablet->set_last_full_compaction_failure_time(UnixMillis());
2096
0
            permits = 0;
2097
0
            if (!res.is<FULL_NO_SUITABLE_VERSION>()) {
2098
0
                return Status::InternalError("prepare full compaction with err: {}",
2099
0
                                             res.to_string());
2100
0
            }
2101
            // return OK if OLAP_ERR_BE_NO_SUITABLE_VERSION, so that we don't need to
2102
            // print too much useless logs.
2103
            // And because we set permits to 0, so even if we return OK here, nothing will be done.
2104
0
            return Status::OK();
2105
0
        }
2106
492
    } else if (compaction_type == CompactionType::BINLOG_COMPACTION) {
2107
492
        MonotonicStopWatch watch;
2108
492
        watch.start();
2109
2110
492
        DCHECK_GE(prefer_compaction_level, 0);
2111
492
        compaction = std::make_shared<BinlogCompaction>(tablet->_engine, tablet,
2112
492
                                                        prefer_compaction_level);
2113
492
        Status res = compaction->prepare_compact();
2114
2115
492
        if (!res.ok()) {
2116
492
            tablet->set_last_binlog_compaction_failure_time(UnixMillis());
2117
492
            permits = 0;
2118
492
            if (!res.is<BINLOG_COMPACTION_NO_SUITABLE_VERSION>()) {
2119
0
                return Status::InternalError("prepare binlog compaction with err: {}",
2120
0
                                             res.to_string());
2121
0
            }
2122
            // return OK if OLAP_ERR_BINLOG_COMPACTION_NO_SUITABLE_VERSION, so that we don't need to
2123
            // print too much useless logs.
2124
            // And because we set permits to 0, so even if we return OK here, nothing will be done.
2125
492
            return Status::OK();
2126
492
        }
2127
492
    } else {
2128
0
        DCHECK(false) << "Unknown compaction type: " << compaction_type;
2129
0
    }
2130
2131
    // Time series policy does not rely on permits, it uses goal size to control memory
2132
3.51k
    if (tablet->tablet_meta()->compaction_policy() == CUMULATIVE_TIME_SERIES_POLICY) {
2133
        // permits = 0 means that prepare_compaction failed
2134
2
        permits = 1;
2135
3.51k
    } else {
2136
3.51k
        permits = compaction->get_compaction_permits();
2137
3.51k
    }
2138
3.51k
    return Status::OK();
2139
229k
}
2140
2141
0
void Tablet::execute_single_replica_compaction(SingleReplicaCompaction& compaction) {
2142
0
    Status res = compaction.execute_compact();
2143
0
    if (!res.ok()) {
2144
0
        set_last_failure_time(this, compaction, UnixMillis());
2145
0
        set_last_single_compaction_failure_status(res.to_string());
2146
0
        if (res.is<CANCELLED>()) {
2147
0
            DorisMetrics::instance()->single_compaction_request_cancelled->increment(1);
2148
            // "CANCELLED" indicates that the peer has not performed compaction,
2149
            // wait for the peer to perform compaction
2150
0
            set_skip_compaction(true, compaction.real_compact_type(), UnixSeconds());
2151
0
            VLOG_CRITICAL << "Cannel fetching from the remote peer. res=" << res
2152
0
                          << ", tablet=" << tablet_id();
2153
0
        } else {
2154
0
            DorisMetrics::instance()->single_compaction_request_failed->increment(1);
2155
0
            LOG(WARNING) << "failed to do single replica compaction. res=" << res
2156
0
                         << ", tablet=" << tablet_id();
2157
0
        }
2158
0
        return;
2159
0
    }
2160
0
    set_last_failure_time(this, compaction, 0);
2161
0
}
2162
2163
28.4M
bool Tablet::should_fetch_from_peer() {
2164
28.4M
    return tablet_meta()->tablet_schema()->enable_single_replica_compaction() &&
2165
28.4M
           _engine.should_fetch_from_peer(tablet_id());
2166
28.4M
}
2167
2168
3
std::vector<Version> Tablet::get_all_local_versions() {
2169
3
    std::vector<Version> local_versions;
2170
3
    {
2171
3
        std::shared_lock rlock(_meta_lock);
2172
63
        for (const auto& [version, rs] : _rs_version_map) {
2173
63
            if (rs->is_local()) {
2174
53
                local_versions.emplace_back(version);
2175
53
            }
2176
63
        }
2177
3
    }
2178
3
    std::sort(local_versions.begin(), local_versions.end(),
2179
324
              [](const Version& left, const Version& right) { return left.first < right.first; });
2180
3
    return local_versions;
2181
3
}
2182
2183
3.49k
void Tablet::execute_compaction(CompactionMixin& compaction) {
2184
3.49k
    signal::tablet_id = tablet_id();
2185
2186
3.49k
    MonotonicStopWatch watch;
2187
3.49k
    watch.start();
2188
2189
3.49k
    Status res = [&]() { RETURN_IF_CATCH_EXCEPTION({ return compaction.execute_compact(); }); }();
2190
2191
3.49k
    if (!res.ok()) [[unlikely]] {
2192
0
        set_last_failure_time(this, compaction, UnixMillis());
2193
0
        LOG(WARNING) << "failed to do " << compaction.compaction_name()
2194
0
                     << ", tablet=" << tablet_id() << " : " << res;
2195
3.49k
    } else {
2196
3.49k
        set_last_failure_time(this, compaction, 0);
2197
3.49k
    }
2198
2199
3.49k
    if (!config::disable_compaction_trace_log) {
2200
0
        auto need_trace = [&compaction, &watch] {
2201
0
            return compaction.compaction_type() == ReaderType::READER_CUMULATIVE_COMPACTION
2202
0
                           ? static_cast<double>(watch.elapsed_time()) / 1e9 >
2203
0
                                     config::cumulative_compaction_trace_threshold
2204
0
                   : compaction.compaction_type() == ReaderType::READER_BASE_COMPACTION
2205
0
                           ? static_cast<double>(watch.elapsed_time()) / 1e9 >
2206
0
                                     config::base_compaction_trace_threshold
2207
0
                           : false;
2208
0
        };
2209
0
        if (need_trace()) {
2210
0
            std::stringstream ss;
2211
0
            compaction.runtime_profile()->pretty_print(&ss);
2212
0
            LOG(WARNING) << "execute " << compaction.compaction_name() << " cost "
2213
0
                         << static_cast<double>(watch.elapsed_time()) / 1e9 << std::endl
2214
0
                         << ss.str();
2215
0
        }
2216
0
    }
2217
3.49k
}
2218
2219
7.72k
Status Tablet::create_initial_rowset(const int64_t req_version) {
2220
7.72k
    if (req_version < 1) {
2221
0
        return Status::Error<CE_CMD_PARAMS_ERROR>(
2222
0
                "init version of tablet should at least 1. req.ver={}", req_version);
2223
0
    }
2224
7.72k
    Version version(0, req_version);
2225
2226
7.72k
    auto get_rowset_writer_context = [&](RowsetWriterContext& context, TabletSchemaSPtr schema) {
2227
        // there is no data in init rowset, so overlapping info is unknown.
2228
7.70k
        context.version = version;
2229
7.70k
        context.rowset_state = VISIBLE;
2230
7.70k
        context.segments_overlap = OVERLAP_UNKNOWN;
2231
7.70k
        context.tablet_schema = schema;
2232
7.70k
        context.newest_write_timestamp = UnixSeconds();
2233
2234
7.70k
        return Status::OK();
2235
7.70k
    };
2236
2237
7.72k
    if (!enable_row_binlog()) {
2238
7.67k
        RowsetWriterContext context;
2239
7.67k
        RowsetSharedPtr new_rowset;
2240
7.67k
        RETURN_IF_ERROR(get_rowset_writer_context(context, tablet_schema()));
2241
7.67k
        auto rs_writer = DORIS_TRY(create_rowset_writer(context, false));
2242
2243
7.67k
        RETURN_IF_ERROR(rs_writer->flush());
2244
7.67k
        RETURN_IF_ERROR(rs_writer->build(new_rowset));
2245
7.67k
        RETURN_IF_ERROR(add_rowset(std::move(new_rowset), nullptr));
2246
7.67k
    } else {
2247
46
        std::unique_ptr<GroupRowsetWriter> group_rowset_writer;
2248
46
        RETURN_IF_ERROR(RowsetFactory::create_empty_group_rowset_writer(&group_rowset_writer));
2249
2250
46
        RowsetWriterContext data_context;
2251
46
        RETURN_IF_ERROR(get_rowset_writer_context(data_context, tablet_schema()));
2252
46
        auto data_writer = DORIS_TRY(create_rowset_writer(data_context, false));
2253
46
        group_rowset_writer->set_data_writer(std::move(data_writer));
2254
2255
46
        RowsetWriterContext row_binlog_context;
2256
46
        row_binlog_context.write_binlog_opt().enable = true;
2257
46
        RETURN_IF_ERROR(get_rowset_writer_context(row_binlog_context, row_binlog_tablet_schema()));
2258
46
        auto row_binlog_writer = DORIS_TRY(create_rowset_writer(row_binlog_context, false));
2259
46
        group_rowset_writer->set_row_binlog_writer(std::move(row_binlog_writer));
2260
2261
46
        RETURN_IF_ERROR(group_rowset_writer->flush_rowsets());
2262
2263
46
        std::vector<RowsetSharedPtr> waited_build_rowsets;
2264
2265
46
        RETURN_IF_ERROR(group_rowset_writer->build_rowsets(waited_build_rowsets));
2266
46
        RETURN_IF_ERROR(add_rowset(waited_build_rowsets.at(0), waited_build_rowsets.at(1)));
2267
46
    }
2268
2269
7.72k
    set_cumulative_layer_point(req_version + 1);
2270
7.72k
    return Status::OK();
2271
7.72k
}
2272
2273
Result<std::unique_ptr<RowsetWriter>> Tablet::create_rowset_writer(RowsetWriterContext& context,
2274
48.4k
                                                                   bool vertical) {
2275
48.4k
    context.rowset_id = _engine.next_rowset_id();
2276
48.4k
    _init_context_common_fields(context);
2277
48.4k
    return RowsetFactory::create_rowset_writer(_engine, context, vertical);
2278
48.4k
}
2279
2280
// create a rowset writer with rowset_id and seg_id
2281
// after writer, merge this transient rowset with original rowset
2282
Result<std::unique_ptr<RowsetWriter>> Tablet::create_transient_rowset_writer(
2283
        const Rowset& rowset, std::shared_ptr<PartialUpdateInfo> partial_update_info,
2284
420
        int64_t txn_expiration) {
2285
420
    RowsetWriterContext context;
2286
420
    context.rowset_state = PREPARED;
2287
420
    context.segments_overlap = OVERLAPPING;
2288
420
    context.db_id = rowset.rowset_meta()->db_id();
2289
420
    context.table_id = rowset.rowset_meta()->table_id();
2290
420
    context.tablet_schema = std::make_shared<TabletSchema>();
2291
    // During a partial update, the extracted columns of a variant should not be included in the tablet schema.
2292
    // This is because the partial update for a variant needs to ignore the extracted columns.
2293
    // Otherwise, the schema types in different rowsets might be inconsistent. When performing a partial update,
2294
    // the complete variant is constructed by reading all the sub-columns of the variant.
2295
420
    context.tablet_schema = rowset.tablet_schema()->copy_without_variant_extracted_columns();
2296
420
    context.newest_write_timestamp = UnixSeconds();
2297
420
    context.tablet_id = table_id();
2298
420
    context.enable_segcompaction = false;
2299
    // ATTN: context.tablet is a shared_ptr, can't simply set it's value to `this`. We should
2300
    // get the shared_ptr from tablet_manager.
2301
420
    auto tablet = _engine.tablet_manager()->get_tablet(tablet_id());
2302
420
    if (!tablet) {
2303
0
        LOG(WARNING) << "cant find tablet by tablet_id=" << tablet_id();
2304
0
        return ResultError(Status::NotFound("cant find tablet by tablet_id={}", tablet_id()));
2305
0
    }
2306
420
    context.tablet = tablet;
2307
420
    context.write_type = DataWriteType::TYPE_DIRECT;
2308
420
    context.partial_update_info = std::move(partial_update_info);
2309
420
    context.is_transient_rowset_writer = true;
2310
2311
420
    if (rowset.rowset_meta() != nullptr && rowset.rowset_meta()->is_row_binlog()) {
2312
0
        context.write_binlog_opt().enable = true;
2313
0
    }
2314
2315
420
    return create_transient_rowset_writer(context, rowset.rowset_id())
2316
420
            .transform([&](auto&& writer) {
2317
420
                writer->set_segment_start_id(cast_set<int32_t>(rowset.num_segments()));
2318
420
                return writer;
2319
420
            });
2320
420
}
2321
2322
Result<std::unique_ptr<RowsetWriter>> Tablet::create_transient_rowset_writer(
2323
422
        RowsetWriterContext& context, const RowsetId& rowset_id) {
2324
422
    context.rowset_id = rowset_id;
2325
422
    _init_context_common_fields(context);
2326
422
    return RowsetFactory::create_rowset_writer(_engine, context, false);
2327
422
}
2328
2329
48.7k
void Tablet::_init_context_common_fields(RowsetWriterContext& context) {
2330
48.7k
    if (context.db_id <= 0) {
2331
11.1k
        context.db_id = tablet_meta()->tablet_schema()->db_id();
2332
11.1k
    }
2333
48.7k
    if (context.table_id <= 0) {
2334
11.2k
        context.table_id = tablet_meta()->tablet_schema()->table_id();
2335
11.2k
    }
2336
2337
48.7k
    context.tablet_uid = tablet_uid();
2338
48.7k
    context.tablet_id = tablet_id();
2339
48.7k
    context.partition_id = partition_id();
2340
48.7k
    context.tablet_schema_hash = schema_hash();
2341
48.7k
    context.rowset_type = tablet_meta()->preferred_rowset_type();
2342
    // Alpha Rowset will be removed in the future, so that if the tablet's default rowset type is
2343
    // alpha rowset, then set the newly created rowset to storage engine's default rowset.
2344
48.7k
    if (context.rowset_type == ALPHA_ROWSET) {
2345
0
        context.rowset_type = _engine.default_rowset_type();
2346
0
    }
2347
2348
48.8k
    if (context.is_local_rowset()) {
2349
48.8k
        context.tablet_path = _tablet_path;
2350
48.8k
    }
2351
2352
48.7k
    context.data_dir = data_dir();
2353
48.7k
    context.enable_unique_key_merge_on_write = enable_unique_key_merge_on_write();
2354
2355
48.7k
    context.encrypt_algorithm = tablet_meta()->encryption_algorithm();
2356
2357
48.7k
    if (context.write_binlog_opt().enable) {
2358
9
        context.tablet_schema_hash = row_binlog_schema_hash();
2359
9
        bool need_before = tablet_meta()->binlog_config().need_historical_value();
2360
9
        context.write_binlog_opt().set_need_before(need_before);
2361
9
        context.tablet_path = row_binlog_path();
2362
9
    }
2363
48.7k
}
2364
2365
485k
Status Tablet::create_rowset(const RowsetMetaSharedPtr& rowset_meta, RowsetSharedPtr* rowset) {
2366
485k
    return RowsetFactory::create_rowset(_tablet_meta->tablet_schema(), get_rowset_path(rowset_meta),
2367
485k
                                        rowset_meta, rowset);
2368
485k
}
2369
2370
7
Status Tablet::cooldown(RowsetSharedPtr rowset) {
2371
7
    std::unique_lock schema_change_lock(_schema_change_lock, std::try_to_lock);
2372
7
    if (!schema_change_lock.owns_lock()) {
2373
0
        return Status::Error<TRY_LOCK_FAILED>(
2374
0
                "try schema_change_lock failed, schema change running or inverted index built "
2375
0
                "on "
2376
0
                "this tablet={}",
2377
0
                tablet_id());
2378
0
    }
2379
    // Check executing serially with compaction task.
2380
7
    std::unique_lock base_compaction_lock(_base_compaction_lock, std::try_to_lock);
2381
7
    if (!base_compaction_lock.owns_lock()) {
2382
0
        return Status::Error<TRY_LOCK_FAILED>("try base_compaction_lock failed");
2383
0
    }
2384
7
    std::unique_lock cumu_compaction_lock(_cumulative_compaction_lock, std::try_to_lock);
2385
7
    if (!cumu_compaction_lock.owns_lock()) {
2386
0
        return Status::Error<TRY_LOCK_FAILED>("try cumu_compaction_lock failed");
2387
0
    }
2388
7
    std::shared_lock cooldown_conf_rlock(_cooldown_conf_lock);
2389
7
    if (_cooldown_conf.cooldown_replica_id <= 0) { // wait for FE to push cooldown conf
2390
2
        return Status::InternalError("invalid cooldown_replica_id");
2391
2
    }
2392
2393
5
    auto mem_tracker = MemTrackerLimiter::create_shared(
2394
5
            MemTrackerLimiter::Type::OTHER,
2395
5
            fmt::format("Tablet::cooldown#tableId={}:replicaId={}", std::to_string(tablet_id()),
2396
5
                        std::to_string(replica_id())));
2397
5
    SCOPED_ATTACH_TASK(mem_tracker);
2398
2399
5
    if (_cooldown_conf.cooldown_replica_id == replica_id()) {
2400
        // this replica is cooldown replica
2401
5
        RETURN_IF_ERROR(_cooldown_data(std::move(rowset)));
2402
5
    } else {
2403
0
        Status st = _follow_cooldowned_data();
2404
0
        if (UNLIKELY(!st.ok())) {
2405
0
            _last_failed_follow_cooldown_time = time(nullptr);
2406
0
            return st;
2407
0
        }
2408
0
        _last_failed_follow_cooldown_time = 0;
2409
0
    }
2410
5
    return Status::OK();
2411
5
}
2412
2413
// hold SHARED `cooldown_conf_lock`
2414
5
Status Tablet::_cooldown_data(RowsetSharedPtr rowset) {
2415
5
    DCHECK(_cooldown_conf.cooldown_replica_id == replica_id());
2416
2417
5
    auto storage_resource = DORIS_TRY(get_resource_by_storage_policy_id(storage_policy_id()));
2418
5
    RowsetSharedPtr old_rowset = nullptr;
2419
2420
5
    if (rowset) {
2421
0
        const auto& rowset_id = rowset->rowset_id();
2422
0
        const auto& rowset_version = rowset->version();
2423
0
        std::shared_lock meta_rlock(_meta_lock);
2424
0
        auto iter = _rs_version_map.find(rowset_version);
2425
0
        if (iter != _rs_version_map.end() && iter->second->rowset_id() == rowset_id) {
2426
0
            old_rowset = rowset;
2427
0
        }
2428
0
    }
2429
2430
5
    if (!old_rowset) {
2431
5
        old_rowset = pick_cooldown_rowset();
2432
5
    }
2433
2434
5
    if (!old_rowset) {
2435
0
        LOG(INFO) << "cannot pick cooldown rowset in tablet " << tablet_id();
2436
0
        return Status::OK();
2437
0
    }
2438
2439
5
    RowsetId new_rowset_id = _engine.next_rowset_id();
2440
5
    auto pending_rs_guard = _engine.pending_remote_rowsets().add(new_rowset_id);
2441
5
    Status st;
2442
5
    Defer defer {[&] {
2443
5
        if (!st.ok()) {
2444
            // reclaim the incomplete rowset data in remote storage
2445
0
            record_unused_remote_rowset(new_rowset_id, storage_resource.fs->id(),
2446
0
                                        old_rowset->num_segments());
2447
0
        }
2448
5
    }};
2449
5
    auto start = std::chrono::steady_clock::now();
2450
5
    if (st = old_rowset->upload_to(storage_resource, new_rowset_id); !st.ok()) {
2451
0
        return st;
2452
0
    }
2453
2454
5
    auto duration = std::chrono::duration<double>(std::chrono::steady_clock::now() - start);
2455
5
    LOG(INFO) << "Upload rowset " << old_rowset->version() << " " << new_rowset_id.to_string()
2456
5
              << " to " << storage_resource.fs->root_path().native()
2457
5
              << ", tablet_id=" << tablet_id() << ", duration=" << duration.count()
2458
5
              << ", capacity=" << old_rowset->total_disk_size()
2459
5
              << ", tp=" << static_cast<double>(old_rowset->total_disk_size()) / duration.count()
2460
5
              << ", old rowset_id=" << old_rowset->rowset_id().to_string();
2461
2462
    // gen a new rowset
2463
5
    auto new_rowset_meta = std::make_shared<RowsetMeta>();
2464
5
    new_rowset_meta->init(old_rowset->rowset_meta().get());
2465
5
    new_rowset_meta->set_rowset_id(new_rowset_id);
2466
5
    new_rowset_meta->set_remote_storage_resource(std::move(storage_resource));
2467
5
    new_rowset_meta->set_creation_time(time(nullptr));
2468
5
    UniqueId cooldown_meta_id = UniqueId::gen_uid();
2469
5
    RowsetSharedPtr new_rowset;
2470
5
    RETURN_IF_ERROR(RowsetFactory::create_rowset(_tablet_meta->tablet_schema(), "", new_rowset_meta,
2471
5
                                                 &new_rowset));
2472
2473
5
    {
2474
5
        std::unique_lock meta_wlock(_meta_lock);
2475
5
        SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
2476
5
        if (tablet_state() == TABLET_RUNNING) {
2477
5
            RETURN_IF_ERROR(delete_rowsets({std::move(old_rowset)}, false));
2478
5
            add_rowsets({std::move(new_rowset)});
2479
            // TODO(plat1ko): process primary key
2480
5
            _tablet_meta->set_cooldown_meta_id(cooldown_meta_id);
2481
5
        }
2482
5
    }
2483
5
    {
2484
5
        std::shared_lock meta_rlock(_meta_lock);
2485
5
        SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
2486
5
        save_meta();
2487
5
    }
2488
    // Upload cooldowned rowset meta to remote fs
2489
    // ATTN: Even if it is an empty rowset, in order for the followers to synchronize, the coolown meta must be
2490
    // uploaded, otherwise followers may never completely cooldown.
2491
5
    if (auto t = _engine.tablet_manager()->get_tablet(tablet_id());
2492
5
        t != nullptr) { // `t` can be nullptr if it has been dropped
2493
5
        async_write_cooldown_meta(std::move(t));
2494
5
    }
2495
5
    return Status::OK();
2496
5
}
2497
2498
// hold SHARED `cooldown_conf_lock`
2499
Status Tablet::_read_cooldown_meta(const StorageResource& storage_resource,
2500
0
                                   TabletMetaPB* tablet_meta_pb) {
2501
0
    std::string remote_meta_path = storage_resource.cooldown_tablet_meta_path(
2502
0
            tablet_id(), _cooldown_conf.cooldown_replica_id, _cooldown_conf.term);
2503
0
    io::FileReaderSPtr tablet_meta_reader;
2504
0
    RETURN_IF_ERROR(storage_resource.fs->open_file(remote_meta_path, &tablet_meta_reader));
2505
0
    auto file_size = tablet_meta_reader->size();
2506
0
    size_t bytes_read;
2507
0
    auto buf = make_unique_buffer<uint8_t>(file_size);
2508
0
    RETURN_IF_ERROR(tablet_meta_reader->read_at(0, {buf.get(), file_size}, &bytes_read));
2509
0
    RETURN_IF_ERROR(tablet_meta_reader->close());
2510
0
    if (!tablet_meta_pb->ParseFromArray(buf.get(), cast_set<int>(file_size))) {
2511
0
        return Status::InternalError("malformed tablet meta, path={}/{}",
2512
0
                                     storage_resource.fs->root_path().native(), remote_meta_path);
2513
0
    }
2514
0
    return Status::OK();
2515
0
}
2516
2517
// `rs_metas` MUST already be sorted by `RowsetMeta::comparator`
2518
7
Status check_version_continuity(const std::vector<RowsetMetaSharedPtr>& rs_metas) {
2519
7
    if (rs_metas.size() < 2) {
2520
2
        return Status::OK();
2521
2
    }
2522
5
    auto prev = rs_metas.begin();
2523
15
    for (auto it = rs_metas.begin() + 1; it != rs_metas.end(); ++it) {
2524
10
        if ((*prev)->end_version() + 1 != (*it)->start_version()) {
2525
0
            return Status::InternalError("versions are not continuity: prev={} cur={}",
2526
0
                                         (*prev)->version().to_string(),
2527
0
                                         (*it)->version().to_string());
2528
0
        }
2529
10
        prev = it;
2530
10
    }
2531
5
    return Status::OK();
2532
5
}
2533
2534
// It's guaranteed the write cooldown meta task would be invoked at the end unless BE crashes
2535
// one tablet would at most have one async task to be done
2536
7
void Tablet::async_write_cooldown_meta(TabletSharedPtr tablet) {
2537
7
    ExecEnv::GetInstance()->write_cooldown_meta_executors()->submit(std::move(tablet));
2538
7
}
2539
2540
328
bool Tablet::update_cooldown_conf(int64_t cooldown_term, int64_t cooldown_replica_id) {
2541
328
    std::unique_lock wlock(_cooldown_conf_lock, std::try_to_lock);
2542
328
    if (!wlock.owns_lock()) {
2543
0
        LOG(INFO) << "try cooldown_conf_lock failed, tablet_id=" << tablet_id();
2544
0
        return false;
2545
0
    }
2546
328
    if (cooldown_term <= _cooldown_conf.term) {
2547
0
        return false;
2548
0
    }
2549
328
    LOG(INFO) << "update cooldown conf. tablet_id=" << tablet_id()
2550
328
              << " cooldown_replica_id: " << _cooldown_conf.cooldown_replica_id << " -> "
2551
328
              << cooldown_replica_id << ", cooldown_term: " << _cooldown_conf.term << " -> "
2552
328
              << cooldown_term;
2553
328
    _cooldown_conf.cooldown_replica_id = cooldown_replica_id;
2554
328
    _cooldown_conf.term = cooldown_term;
2555
328
    return true;
2556
328
}
2557
2558
124k
Status Tablet::write_cooldown_meta() {
2559
124k
    std::shared_lock rlock(_cooldown_conf_lock);
2560
124k
    if (_cooldown_conf.cooldown_replica_id != _tablet_meta->replica_id()) {
2561
0
        return Status::Aborted<false>("not cooldown replica({} vs {}) tablet_id={}",
2562
0
                                      _tablet_meta->replica_id(),
2563
0
                                      _cooldown_conf.cooldown_replica_id, tablet_id());
2564
0
    }
2565
2566
124k
    auto storage_resource = DORIS_TRY(get_resource_by_storage_policy_id(storage_policy_id()));
2567
2568
7
    std::vector<RowsetMetaSharedPtr> cooldowned_rs_metas;
2569
7
    UniqueId cooldown_meta_id;
2570
7
    {
2571
7
        std::shared_lock meta_rlock(_meta_lock);
2572
19
        for (const auto& [_, rs_meta] : _tablet_meta->all_rs_metas()) {
2573
19
            if (!rs_meta->is_local()) {
2574
17
                cooldowned_rs_metas.push_back(rs_meta);
2575
17
            }
2576
19
        }
2577
7
        cooldown_meta_id = _tablet_meta->cooldown_meta_id();
2578
7
    }
2579
7
    if (cooldowned_rs_metas.empty()) {
2580
0
        LOG(INFO) << "no cooldown meta to write, tablet_id=" << tablet_id();
2581
0
        return Status::OK();
2582
0
    }
2583
7
    std::sort(cooldowned_rs_metas.begin(), cooldowned_rs_metas.end(), RowsetMeta::comparator);
2584
7
    DCHECK(cooldowned_rs_metas.front()->start_version() == 0);
2585
    // If version not continuous, it must be a bug
2586
7
    if (auto st = check_version_continuity(cooldowned_rs_metas); !st.ok()) {
2587
0
        DCHECK(st.ok()) << st << " tablet_id=" << tablet_id();
2588
0
        st.set_code(ABORTED);
2589
0
        return st;
2590
0
    }
2591
2592
7
    TabletMetaPB tablet_meta_pb;
2593
7
    auto* rs_metas = tablet_meta_pb.mutable_rs_metas();
2594
7
    rs_metas->Reserve(cast_set<int>(cooldowned_rs_metas.size()));
2595
17
    for (auto& rs_meta : cooldowned_rs_metas) {
2596
17
        rs_metas->Add(rs_meta->get_rowset_pb());
2597
17
    }
2598
7
    tablet_meta_pb.mutable_cooldown_meta_id()->set_hi(cooldown_meta_id.hi);
2599
7
    tablet_meta_pb.mutable_cooldown_meta_id()->set_lo(cooldown_meta_id.lo);
2600
2601
7
    std::string remote_meta_path = storage_resource.cooldown_tablet_meta_path(
2602
7
            tablet_id(), _cooldown_conf.cooldown_replica_id, _cooldown_conf.term);
2603
7
    io::FileWriterPtr tablet_meta_writer;
2604
    // FIXME(plat1ko): What if object store permanently unavailable?
2605
7
    RETURN_IF_ERROR(storage_resource.fs->create_file(remote_meta_path, &tablet_meta_writer));
2606
7
    auto val = tablet_meta_pb.SerializeAsString();
2607
7
    RETURN_IF_ERROR(tablet_meta_writer->append({val.data(), val.size()}));
2608
7
    return tablet_meta_writer->close();
2609
7
}
2610
2611
// hold SHARED `cooldown_conf_lock`
2612
0
Status Tablet::_follow_cooldowned_data() {
2613
0
    DCHECK(_cooldown_conf.cooldown_replica_id != replica_id());
2614
0
    LOG(INFO) << "try to follow cooldowned data. tablet_id=" << tablet_id()
2615
0
              << " cooldown_replica_id=" << _cooldown_conf.cooldown_replica_id
2616
0
              << " local replica=" << replica_id();
2617
2618
0
    auto storage_resource = DORIS_TRY(get_resource_by_storage_policy_id(storage_policy_id()));
2619
    // MUST executing serially with cold data compaction, because compaction input rowsets may be deleted by this function
2620
0
    std::unique_lock cold_compaction_lock(_cold_compaction_lock, std::try_to_lock);
2621
0
    if (!cold_compaction_lock.owns_lock()) {
2622
0
        return Status::Error<TRY_LOCK_FAILED>("try cold_compaction_lock failed");
2623
0
    }
2624
2625
0
    TabletMetaPB cooldown_meta_pb;
2626
0
    auto st = _read_cooldown_meta(storage_resource, &cooldown_meta_pb);
2627
0
    if (!st.ok()) {
2628
0
        LOG(INFO) << "cannot read cooldown meta: " << st;
2629
0
        return Status::InternalError<false>("cannot read cooldown meta");
2630
0
    }
2631
2632
0
    if (cooldown_meta_pb.rs_metas_size() <= 0) {
2633
0
        LOG(WARNING)
2634
0
                << "Cooldown meta file exists but rs_metas is empty for tablet " << tablet_id()
2635
0
                << ". Cooldown meta id: " << cooldown_meta_pb.cooldown_meta_id()
2636
0
                << ". This may indicate a cooldown meta synchronization issue or an invalid file.";
2637
0
        return Status::InternalError<false>("Cooldown meta rs_metas is empty");
2638
0
    }
2639
2640
0
    if (_tablet_meta->cooldown_meta_id() == cooldown_meta_pb.cooldown_meta_id()) {
2641
        // cooldowned rowsets are same, no need to follow
2642
0
        return Status::OK();
2643
0
    }
2644
2645
0
    int64_t cooldowned_version = cooldown_meta_pb.rs_metas().rbegin()->end_version();
2646
2647
0
    std::vector<RowsetSharedPtr> overlap_rowsets;
2648
0
    bool version_aligned = false;
2649
2650
    // Holding these to delete rowsets' shared ptr until save meta can avoid trash sweeping thread
2651
    // deleting these rowsets' files before rowset meta has been removed from disk, which may cause
2652
    // data loss when BE reboot before save meta to disk.
2653
0
    std::vector<RowsetSharedPtr> to_delete;
2654
0
    std::vector<RowsetSharedPtr> to_add;
2655
2656
0
    {
2657
0
        std::lock_guard wlock(_meta_lock);
2658
0
        SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
2659
0
        if (tablet_state() != TABLET_RUNNING) {
2660
0
            return Status::InternalError<false>("tablet not running");
2661
0
        }
2662
2663
0
        for (auto& [v, rs] : _rs_version_map) {
2664
0
            if (v.second <= cooldowned_version) {
2665
0
                overlap_rowsets.push_back(rs);
2666
0
                if (!version_aligned && v.second == cooldowned_version) {
2667
0
                    version_aligned = true;
2668
0
                }
2669
0
            } else if (!rs->is_local()) {
2670
0
                return Status::InternalError<false>(
2671
0
                        "cooldowned version larger than that to follow with cooldown version "
2672
0
                        "{}",
2673
0
                        cooldowned_version);
2674
0
            }
2675
0
        }
2676
2677
0
        if (!version_aligned) {
2678
0
            return Status::InternalError<false>("cooldowned version is not aligned with version {}",
2679
0
                                                cooldowned_version);
2680
0
        }
2681
2682
0
        std::sort(overlap_rowsets.begin(), overlap_rowsets.end(), Rowset::comparator);
2683
2684
        // Find different rowset in `overlap_rowsets` and `cooldown_meta_pb.rs_metas`
2685
0
        auto rs_pb_it = cooldown_meta_pb.rs_metas().begin();
2686
0
        auto rs_it = overlap_rowsets.begin();
2687
0
        for (; rs_pb_it != cooldown_meta_pb.rs_metas().end() && rs_it != overlap_rowsets.end();
2688
0
             ++rs_pb_it, ++rs_it) {
2689
0
            if (rs_pb_it->rowset_id_v2() != (*rs_it)->rowset_id().to_string()) {
2690
0
                break;
2691
0
            }
2692
0
        }
2693
2694
0
        to_delete.assign(rs_it, overlap_rowsets.end());
2695
0
        to_add.reserve(cooldown_meta_pb.rs_metas().end() - rs_pb_it);
2696
0
        for (; rs_pb_it != cooldown_meta_pb.rs_metas().end(); ++rs_pb_it) {
2697
0
            auto rs_meta = std::make_shared<RowsetMeta>();
2698
0
            rs_meta->init_from_pb(*rs_pb_it);
2699
0
            RowsetSharedPtr rs;
2700
0
            RETURN_IF_ERROR(
2701
0
                    RowsetFactory::create_rowset(_tablet_meta->tablet_schema(), "", rs_meta, &rs));
2702
0
            to_add.push_back(std::move(rs));
2703
0
        }
2704
        // Note: We CANNOT call `modify_rowsets` here because `modify_rowsets` cannot process version graph correctly.
2705
0
        RETURN_IF_ERROR(delete_rowsets(to_delete, false));
2706
0
        add_rowsets(to_add);
2707
        // TODO(plat1ko): process primary key
2708
0
        _tablet_meta->set_cooldown_meta_id(cooldown_meta_pb.cooldown_meta_id());
2709
0
    }
2710
2711
0
    {
2712
0
        std::lock_guard rlock(_meta_lock);
2713
0
        SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
2714
0
        save_meta();
2715
0
    }
2716
2717
0
    if (!to_add.empty()) {
2718
0
        LOG(INFO) << "modify rowsets when follow cooldowned data, tablet_id=" << tablet_id()
2719
0
                  << [&]() {
2720
0
                         std::stringstream ss;
2721
0
                         ss << " delete rowsets:\n";
2722
0
                         for (auto&& rs : to_delete) {
2723
0
                             ss << rs->version() << ' ' << rs->rowset_id() << '\n';
2724
0
                         }
2725
0
                         ss << "add rowsets:\n";
2726
0
                         for (auto&& rs : to_add) {
2727
0
                             ss << rs->version() << ' ' << rs->rowset_id() << '\n';
2728
0
                         }
2729
0
                         return ss.str();
2730
0
                     }();
2731
0
    }
2732
2733
0
    return Status::OK();
2734
0
}
2735
2736
3.93M
bool Tablet::_has_data_to_cooldown() {
2737
3.93M
    int64_t min_local_version = std::numeric_limits<int64_t>::max();
2738
3.93M
    RowsetSharedPtr rowset;
2739
3.93M
    std::shared_lock meta_rlock(_meta_lock);
2740
    // Ususally once the tablet has done cooldown successfully then the first
2741
    // rowset would always be remote rowset
2742
3.93M
    bool has_cooldowned = false;
2743
6.91M
    for (const auto& [_, rs] : _rs_version_map) {
2744
6.91M
        if (!rs->is_local()) {
2745
28
            has_cooldowned = true;
2746
28
            break;
2747
28
        }
2748
6.91M
    }
2749
6.91M
    for (auto& [v, rs] : _rs_version_map) {
2750
6.91M
        auto predicate = rs->is_local() && v.first < min_local_version;
2751
6.91M
        if (!has_cooldowned) {
2752
6.91M
            predicate = predicate && (rs->data_disk_size() > 0);
2753
6.91M
        }
2754
6.91M
        if (predicate) {
2755
            // this is a local rowset and has data
2756
919k
            min_local_version = v.first;
2757
919k
            rowset = rs;
2758
919k
        }
2759
6.91M
    }
2760
2761
3.93M
    int64_t newest_cooldown_time = 0;
2762
3.93M
    if (rowset != nullptr) {
2763
820k
        newest_cooldown_time = _get_newest_cooldown_time(rowset);
2764
820k
    }
2765
2766
3.93M
    return (newest_cooldown_time != 0) && (newest_cooldown_time < UnixSeconds());
2767
3.93M
}
2768
2769
3.93M
RowsetSharedPtr Tablet::pick_cooldown_rowset() {
2770
3.93M
    RowsetSharedPtr rowset;
2771
2772
3.93M
    if (!_has_data_to_cooldown()) {
2773
3.93M
        return nullptr;
2774
3.93M
    }
2775
2776
    // TODO(plat1ko): should we maintain `cooldowned_version` in `Tablet`?
2777
8
    int64_t cooldowned_version = -1;
2778
    // We pick the rowset with smallest start version in local.
2779
8
    int64_t min_local_version = std::numeric_limits<int64_t>::max();
2780
8
    {
2781
8
        std::shared_lock meta_rlock(_meta_lock);
2782
26
        for (auto& [v, rs] : _rs_version_map) {
2783
26
            if (!rs->is_local()) {
2784
4
                cooldowned_version = std::max(cooldowned_version, v.second);
2785
22
            } else if (v.first < min_local_version) { // this is a local rowset
2786
10
                min_local_version = v.first;
2787
10
                rowset = rs;
2788
10
            }
2789
26
        }
2790
8
    }
2791
8
    if (!rowset) {
2792
0
        return nullptr;
2793
0
    }
2794
8
    if (tablet_footprint() == 0) {
2795
0
        VLOG_DEBUG << "skip cooldown due to empty tablet_id = " << tablet_id();
2796
0
        return nullptr;
2797
0
    }
2798
8
    if (min_local_version != cooldowned_version + 1) { // ensure version continuity
2799
0
        if (UNLIKELY(cooldowned_version != -1)) {
2800
0
            LOG(WARNING) << "version not continuous. tablet_id=" << tablet_id()
2801
0
                         << " cooldowned_version=" << cooldowned_version
2802
0
                         << " min_local_version=" << min_local_version;
2803
0
        }
2804
0
        return nullptr;
2805
0
    }
2806
8
    return rowset;
2807
8
}
2808
2809
820k
int64_t Tablet::_get_newest_cooldown_time(const RowsetSharedPtr& rowset) {
2810
820k
    int64_t id = storage_policy_id();
2811
820k
    if (id <= 0) {
2812
820k
        VLOG_DEBUG << "tablet does not need cooldown, tablet id: " << tablet_id();
2813
820k
        return 0;
2814
820k
    }
2815
12
    auto storage_policy = get_storage_policy(id);
2816
12
    if (!storage_policy) {
2817
0
        LOG(WARNING) << "Cannot get storage policy: " << id;
2818
0
        return 0;
2819
0
    }
2820
12
    auto cooldown_ttl_sec = storage_policy->cooldown_ttl;
2821
12
    auto cooldown_datetime = storage_policy->cooldown_datetime;
2822
12
    int64_t newest_cooldown_time = std::numeric_limits<int64_t>::max();
2823
2824
12
    if (cooldown_ttl_sec >= 0) {
2825
9
        newest_cooldown_time = rowset->newest_write_timestamp() + cooldown_ttl_sec;
2826
9
    }
2827
12
    if (cooldown_datetime > 0) {
2828
10
        newest_cooldown_time = std::min(newest_cooldown_time, cooldown_datetime);
2829
10
    }
2830
2831
12
    return newest_cooldown_time;
2832
12
}
2833
2834
3.93M
RowsetSharedPtr Tablet::need_cooldown(int64_t* cooldown_timestamp, size_t* file_size) {
2835
3.93M
    RowsetSharedPtr rowset = pick_cooldown_rowset();
2836
3.93M
    if (!rowset) {
2837
3.93M
        VLOG_DEBUG << "pick cooldown rowset, get null, tablet id: " << tablet_id();
2838
3.93M
        return nullptr;
2839
3.93M
    }
2840
2841
3
    auto newest_cooldown_time = _get_newest_cooldown_time(rowset);
2842
2843
    // the rowset should do cooldown job only if it's cooldown ttl plus newest write time is less than
2844
    // current time or it's datatime is less than current time
2845
3
    if (newest_cooldown_time != 0 && newest_cooldown_time < UnixSeconds()) {
2846
3
        *cooldown_timestamp = newest_cooldown_time;
2847
3
        *file_size = rowset->total_disk_size();
2848
3
        VLOG_DEBUG << "tablet need cooldown, tablet id: " << tablet_id()
2849
0
                   << " file_size: " << *file_size;
2850
3
        return rowset;
2851
3
    }
2852
2853
0
    VLOG_DEBUG << "tablet does not need cooldown, tablet id: " << tablet_id()
2854
0
               << " newest write time: " << rowset->newest_write_timestamp();
2855
0
    return nullptr;
2856
3
}
2857
2858
void Tablet::record_unused_remote_rowset(const RowsetId& rowset_id, const std::string& resource,
2859
0
                                         int64_t num_segments) {
2860
0
    auto gc_key = REMOTE_ROWSET_GC_PREFIX + rowset_id.to_string();
2861
0
    RemoteRowsetGcPB gc_pb;
2862
0
    gc_pb.set_resource_id(resource);
2863
0
    gc_pb.set_tablet_id(tablet_id());
2864
0
    gc_pb.set_num_segments(num_segments);
2865
0
    auto st =
2866
0
            _data_dir->get_meta()->put(META_COLUMN_FAMILY_INDEX, gc_key, gc_pb.SerializeAsString());
2867
0
    if (!st.ok()) {
2868
0
        LOG(WARNING) << "failed to record unused remote rowset. tablet_id=" << tablet_id()
2869
0
                     << " rowset_id=" << rowset_id << " resource_id=" << resource;
2870
0
    }
2871
0
    unused_remote_rowset_num << 1;
2872
0
}
2873
2874
4.70k
Status Tablet::remove_all_remote_rowsets() {
2875
4.70k
    DCHECK(tablet_state() == TABLET_SHUTDOWN);
2876
4.70k
    std::set<std::string> resource_ids;
2877
9.34k
    for (const auto& [_, rs_meta] : _tablet_meta->all_rs_metas()) {
2878
9.34k
        if (!rs_meta->is_local()) {
2879
0
            resource_ids.insert(rs_meta->resource_id());
2880
0
        }
2881
9.34k
    }
2882
4.70k
    if (resource_ids.empty()) {
2883
4.70k
        return Status::OK();
2884
4.70k
    }
2885
0
    auto tablet_gc_key = REMOTE_TABLET_GC_PREFIX + std::to_string(tablet_id());
2886
0
    RemoteTabletGcPB gc_pb;
2887
0
    for (auto& resource_id : resource_ids) {
2888
0
        gc_pb.add_resource_ids(resource_id);
2889
0
    }
2890
0
    return _data_dir->get_meta()->put(META_COLUMN_FAMILY_INDEX, tablet_gc_key,
2891
0
                                      gc_pb.SerializeAsString());
2892
4.70k
}
2893
2894
0
void Tablet::update_max_version_schema(const TabletSchemaSPtr& tablet_schema) {
2895
0
    std::lock_guard wrlock(_meta_lock);
2896
0
    SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
2897
    // Double Check for concurrent update
2898
0
    if (!_max_version_schema ||
2899
0
        tablet_schema->schema_version() > _max_version_schema->schema_version()) {
2900
0
        _max_version_schema = tablet_schema;
2901
0
    }
2902
0
}
2903
2904
0
CalcDeleteBitmapExecutor* Tablet::calc_delete_bitmap_executor() {
2905
0
    return _engine.calc_delete_bitmap_executor();
2906
0
}
2907
2908
Status Tablet::save_delete_bitmap(const TabletTxnInfo* txn_info, int64_t txn_id,
2909
                                  DeleteBitmapPtr delete_bitmap, RowsetWriter* rowset_writer,
2910
                                  const RowsetIdUnorderedSet& cur_rowset_ids, int64_t lock_id,
2911
23.0k
                                  int64_t next_visible_version) {
2912
23.0k
    RowsetSharedPtr rowset = txn_info->rowset;
2913
23.0k
    int64_t cur_version = rowset->start_version();
2914
2915
    // For binlog<Row> publish, sync current rowset delete bitmap deltas to `binlog_delvec`
2916
    // so row binlog reads can skip rows deleted by MOW conflict resolution.
2917
23.0k
    const bool build_row_binlog = !txn_info->attach_rowsets.empty();
2918
23.0k
    const RowsetId cur_build_rid = txn_info->rowset->rowset_id();
2919
23.0k
    const RowsetId binlog_rid =
2920
23.0k
            build_row_binlog ? txn_info->attach_rowsets[0]->rowset_id() : cur_build_rid;
2921
23.0k
    auto* binlog_delvec = txn_info->binlog_delvec.get();
2922
23.0k
    if (build_row_binlog) {
2923
0
        DCHECK(txn_info->attach_rowsets[0] != nullptr);
2924
0
        DCHECK(txn_info->attach_rowsets[0]->rowset_meta() != nullptr);
2925
0
        DCHECK(txn_info->attach_rowsets[0]->rowset_meta()->is_row_binlog());
2926
0
        DCHECK(binlog_delvec != nullptr);
2927
0
    }
2928
2929
    // update version without write lock, compaction and publish_txn
2930
    // will update delete bitmap, handle compaction with _rowset_update_lock
2931
    // and publish_txn runs sequential so no need to lock here
2932
60.7k
    for (auto& [key, bitmap] : delete_bitmap->delete_bitmap) {
2933
        // skip sentinel mark, which is used for delete bitmap correctness check
2934
60.7k
        if (std::get<1>(key) != DeleteBitmap::INVALID_SEGMENT_ID) {
2935
936
            _tablet_meta->delete_bitmap().merge({std::get<0>(key), std::get<1>(key), cur_version},
2936
936
                                                bitmap);
2937
2938
936
            if (build_row_binlog && std::get<0>(key) == cur_build_rid) {
2939
0
                const DeleteBitmap::SegmentId& sid = std::get<1>(key);
2940
                // Merge current delete bitmap deltas to binlog delvec because publish-phase
2941
                // partial-update and seq replace will influence the binlog file.
2942
0
                binlog_delvec->merge({binlog_rid, sid, cast_set<uint64_t>(cur_version)}, bitmap);
2943
0
                _tablet_meta->binlog_delvec().merge(
2944
0
                        {binlog_rid, sid, cast_set<uint64_t>(cur_version)}, bitmap);
2945
0
            }
2946
936
        }
2947
60.7k
    }
2948
2949
23.0k
    return Status::OK();
2950
23.0k
}
2951
2952
2.37k
void Tablet::merge_delete_bitmap(const DeleteBitmap& delete_bitmap) {
2953
2.37k
    _tablet_meta->delete_bitmap().merge(delete_bitmap);
2954
2.37k
}
2955
2956
0
bool Tablet::check_all_rowset_segment() {
2957
0
    std::shared_lock rdlock(_meta_lock);
2958
0
    for (auto& version_rowset : _rs_version_map) {
2959
0
        RowsetSharedPtr rowset = version_rowset.second;
2960
0
        if (!rowset->check_rowset_segment()) {
2961
0
            LOG(WARNING) << "Tablet Segment Check. find a bad tablet, tablet_id=" << tablet_id();
2962
0
            return false;
2963
0
        }
2964
0
    }
2965
0
    return true;
2966
0
}
2967
2968
13.8M
void Tablet::set_skip_compaction(bool skip, CompactionType compaction_type, int64_t start) {
2969
13.8M
    if (!skip) {
2970
0
        _skip_cumu_compaction = false;
2971
0
        _skip_base_compaction = false;
2972
0
        _skip_binlog_compaction = false;
2973
0
        return;
2974
0
    }
2975
13.8M
    if (compaction_type == CompactionType::CUMULATIVE_COMPACTION) {
2976
7.68M
        _skip_cumu_compaction = true;
2977
7.68M
        _skip_cumu_compaction_ts = start;
2978
7.68M
    } else if (compaction_type == CompactionType::BASE_COMPACTION) {
2979
6.15M
        _skip_base_compaction = true;
2980
6.15M
        _skip_base_compaction_ts = start;
2981
6.15M
    } else {
2982
504
        DCHECK(compaction_type == CompactionType::BINLOG_COMPACTION);
2983
504
        _skip_binlog_compaction = true;
2984
504
        _skip_binlog_compaction_ts = start;
2985
504
    }
2986
13.8M
}
2987
2988
687M
bool Tablet::should_skip_compaction(CompactionType compaction_type, int64_t now) {
2989
687M
    if (compaction_type == CompactionType::CUMULATIVE_COMPACTION && _skip_cumu_compaction &&
2990
687M
        now < _skip_cumu_compaction_ts + config::skip_tablet_compaction_second) {
2991
245M
        return true;
2992
441M
    } else if (compaction_type == CompactionType::BASE_COMPACTION && _skip_base_compaction &&
2993
441M
               now < _skip_base_compaction_ts + config::skip_tablet_compaction_second) {
2994
19.2M
        return true;
2995
422M
    } else if (compaction_type == CompactionType::BINLOG_COMPACTION && _skip_binlog_compaction &&
2996
422M
               now < _skip_binlog_compaction_ts + config::skip_tablet_compaction_second) {
2997
90.5k
        return true;
2998
90.5k
    }
2999
422M
    return false;
3000
687M
}
3001
3002
0
std::pair<std::string, int64_t> Tablet::get_binlog_info(std::string_view binlog_version) const {
3003
0
    return RowsetMetaManager::get_binlog_info(_data_dir->get_meta(), tablet_uid(), binlog_version);
3004
0
}
3005
3006
std::string Tablet::get_rowset_binlog_meta(std::string_view binlog_version,
3007
0
                                           std::string_view rowset_id) const {
3008
0
    return RowsetMetaManager::get_rowset_binlog_meta(_data_dir->get_meta(), tablet_uid(),
3009
0
                                                     binlog_version, rowset_id);
3010
0
}
3011
3012
Status Tablet::get_rowset_binlog_metas(const std::vector<int64_t>& binlog_versions,
3013
0
                                       RowsetBinlogMetasPB* metas_pb) {
3014
0
    return RowsetMetaManager::get_rowset_binlog_metas(_data_dir->get_meta(), tablet_uid(),
3015
0
                                                      binlog_versions, metas_pb);
3016
0
}
3017
3018
4
Status Tablet::get_rowset_binlog_metas(Version binlog_versions, RowsetBinlogMetasPB* metas_pb) {
3019
4
    return RowsetMetaManager::get_rowset_binlog_metas(_data_dir->get_meta(), tablet_uid(),
3020
4
                                                      binlog_versions, metas_pb);
3021
4
}
3022
3023
std::string Tablet::get_segment_filepath(std::string_view rowset_id,
3024
0
                                         std::string_view segment_index) const {
3025
0
    return fmt::format("{}/_binlog/{}_{}.dat", _tablet_path, rowset_id, segment_index);
3026
0
}
3027
3028
1
std::string Tablet::get_segment_filepath(std::string_view rowset_id, int64_t segment_index) const {
3029
1
    return fmt::format("{}/_binlog/{}_{}.dat", _tablet_path, rowset_id, segment_index);
3030
1
}
3031
3032
0
std::vector<std::string> Tablet::get_binlog_filepath(std::string_view binlog_version) const {
3033
0
    const auto& [rowset_id, num_segments] = get_binlog_info(binlog_version);
3034
0
    std::vector<std::string> binlog_filepath;
3035
0
    for (int i = 0; i < num_segments; ++i) {
3036
        // TODO(Drogon): rewrite by filesystem path
3037
0
        auto segment_file = fmt::format("{}_{}.dat", rowset_id, i);
3038
0
        binlog_filepath.emplace_back(fmt::format("{}/_binlog/{}", _tablet_path, segment_file));
3039
0
    }
3040
0
    return binlog_filepath;
3041
0
}
3042
3043
0
bool Tablet::can_add_binlog(uint64_t total_binlog_size) const {
3044
0
    return !_data_dir->reach_capacity_limit(total_binlog_size);
3045
0
}
3046
3047
0
void Tablet::set_binlog_config(BinlogConfig binlog_config) {
3048
0
    tablet_meta()->set_binlog_config(binlog_config);
3049
0
}
3050
3051
2
void Tablet::gc_binlogs(int64_t version) {
3052
2
    auto meta = _data_dir->get_meta();
3053
2
    DCHECK(meta != nullptr);
3054
3055
2
    const auto& tablet_uid = this->tablet_uid();
3056
2
    const auto tablet_id = this->tablet_id();
3057
2
    std::string begin_key = make_binlog_meta_key_prefix(tablet_uid);
3058
2
    std::string end_key = make_binlog_meta_key_prefix(tablet_uid, version + 1);
3059
2
    LOG(INFO) << fmt::format("gc binlog meta, tablet_id:{}, begin_key:{}, end_key:{}", tablet_id,
3060
2
                             begin_key, end_key);
3061
3062
2
    std::vector<std::string> wait_for_deleted_binlog_keys;
3063
2
    std::vector<std::string> wait_for_deleted_binlog_files;
3064
2
    auto add_to_wait_for_deleted = [&](std::string_view key, std::string_view rowset_id,
3065
2
                                       int64_t num_segments) {
3066
        // add binlog meta key and binlog data key
3067
1
        wait_for_deleted_binlog_keys.emplace_back(key);
3068
1
        wait_for_deleted_binlog_keys.push_back(get_binlog_data_key_from_meta_key(key));
3069
3070
        // add binlog segment files and index files
3071
2
        for (int64_t i = 0; i < num_segments; ++i) {
3072
1
            auto segment_file_path = get_segment_filepath(rowset_id, i);
3073
1
            wait_for_deleted_binlog_files.emplace_back(segment_file_path);
3074
3075
            // index files
3076
1
            if (tablet_schema()->has_inverted_index()) {
3077
1
                if (tablet_schema()->get_inverted_index_storage_format() ==
3078
1
                    InvertedIndexStorageFormatPB::V1) {
3079
0
                    for (const auto& index : tablet_schema()->inverted_indexes()) {
3080
0
                        auto index_file = InvertedIndexDescriptor::get_index_file_path_v1(
3081
0
                                InvertedIndexDescriptor::get_index_file_path_prefix(
3082
0
                                        segment_file_path),
3083
0
                                index->index_id(), index->get_index_suffix());
3084
0
                        wait_for_deleted_binlog_files.emplace_back(index_file);
3085
0
                    }
3086
1
                } else {
3087
1
                    auto index_file = InvertedIndexDescriptor::get_index_file_path_v2(
3088
1
                            InvertedIndexDescriptor::get_index_file_path_prefix(segment_file_path));
3089
1
                    wait_for_deleted_binlog_files.emplace_back(index_file);
3090
1
                }
3091
1
            }
3092
1
        }
3093
1
    };
3094
3095
2
    auto check_binlog_ttl = [&](std::string_view key, std::string_view value) mutable -> bool {
3096
2
        if (key >= end_key) {
3097
1
            return false;
3098
1
        }
3099
3100
1
        BinlogMetaEntryPB binlog_meta_entry_pb;
3101
1
        if (!binlog_meta_entry_pb.ParseFromArray(value.data(), cast_set<int>(value.size()))) {
3102
0
            LOG(WARNING) << "failed to parse binlog meta entry, key:" << key;
3103
0
            return true;
3104
0
        }
3105
3106
1
        auto num_segments = binlog_meta_entry_pb.num_segments();
3107
1
        std::string rowset_id;
3108
1
        if (binlog_meta_entry_pb.has_rowset_id_v2()) {
3109
1
            rowset_id = binlog_meta_entry_pb.rowset_id_v2();
3110
1
        } else {
3111
            // key is 'binlog_meta_6943f1585fe834b5-e542c2b83a21d0b7_00000000000000000069_020000000000000135449d7cd7eadfe672aa0f928fa99593', extract last part '020000000000000135449d7cd7eadfe672aa0f928fa99593'
3112
0
            auto pos = key.rfind('_');
3113
0
            if (pos == std::string::npos) {
3114
0
                LOG(WARNING) << fmt::format("invalid binlog meta key:{}", key);
3115
0
                return false;
3116
0
            }
3117
0
            rowset_id = key.substr(pos + 1);
3118
0
        }
3119
1
        add_to_wait_for_deleted(key, rowset_id, num_segments);
3120
3121
1
        return true;
3122
1
    };
3123
3124
2
    auto status = meta->iterate(META_COLUMN_FAMILY_INDEX, begin_key, check_binlog_ttl);
3125
2
    if (!status.ok()) {
3126
0
        LOG(WARNING) << "failed to iterate binlog meta, status:" << status;
3127
0
        return;
3128
0
    }
3129
3130
    // first remove binlog files, if failed, just break, then retry next time
3131
    // this keep binlog meta in meta store, so that binlog can be removed next time
3132
2
    bool remove_binlog_files_failed = false;
3133
2
    for (auto& file : wait_for_deleted_binlog_files) {
3134
2
        if (unlink(file.c_str()) != 0) {
3135
            // file not exist, continue
3136
0
            if (errno == ENOENT) {
3137
0
                continue;
3138
0
            }
3139
3140
0
            remove_binlog_files_failed = true;
3141
0
            LOG(WARNING) << "failed to remove binlog file:" << file << ", errno:" << errno;
3142
0
            break;
3143
0
        }
3144
2
    }
3145
2
    if (!remove_binlog_files_failed) {
3146
2
        static_cast<void>(meta->remove(META_COLUMN_FAMILY_INDEX, wait_for_deleted_binlog_keys));
3147
2
    }
3148
2
}
3149
3150
0
Status Tablet::ingest_binlog_metas(RowsetBinlogMetasPB* metas_pb) {
3151
0
    return RowsetMetaManager::ingest_binlog_metas(_data_dir->get_meta(), tablet_uid(), metas_pb);
3152
0
}
3153
3154
9.98k
void Tablet::clear_cache() {
3155
9.98k
    std::vector<RowsetSharedPtr> rowsets;
3156
9.98k
    {
3157
9.98k
        std::shared_lock rlock(get_header_lock());
3158
9.98k
        SCOPED_SIMPLE_TRACE_IF_TIMEOUT(TRACE_TABLET_LOCK_THRESHOLD);
3159
3160
40.0k
        for (auto& [_, rowset] : rowset_map()) {
3161
40.0k
            rowsets.push_back(rowset);
3162
40.0k
        }
3163
9.98k
        for (auto& [_, rowset] : stale_rowset_map()) {
3164
5.49k
            rowsets.push_back(rowset);
3165
5.49k
        }
3166
9.98k
    }
3167
45.5k
    for (auto& rowset : rowsets) {
3168
45.5k
        rowset->clear_cache();
3169
45.5k
    }
3170
9.98k
}
3171
3172
25.3k
void Tablet::check_table_size_correctness() {
3173
25.4k
    if (!config::enable_table_size_correctness_check) {
3174
25.4k
        return;
3175
25.4k
    }
3176
18.4E
    for (const auto& [_, rs_meta] : _tablet_meta->all_rs_metas()) {
3177
0
        int64_t total_segment_size = get_segment_file_size(rs_meta);
3178
0
        int64_t total_inverted_index_size = get_inverted_index_file_size(rs_meta);
3179
0
        if (rs_meta->data_disk_size() != total_segment_size ||
3180
0
            rs_meta->index_disk_size() != total_inverted_index_size ||
3181
0
            rs_meta->data_disk_size() + rs_meta->index_disk_size() != rs_meta->total_disk_size()) {
3182
0
            LOG(WARNING) << "[Local table table size check failed]:"
3183
0
                         << " tablet id: " << rs_meta->tablet_id()
3184
0
                         << ", rowset id:" << rs_meta->rowset_id()
3185
0
                         << ", rowset data disk size:" << rs_meta->data_disk_size()
3186
0
                         << ", rowset real data disk size:" << total_segment_size
3187
0
                         << ", rowset index disk size:" << rs_meta->index_disk_size()
3188
0
                         << ", rowset real index disk size:" << total_inverted_index_size
3189
0
                         << ", rowset total disk size:" << rs_meta->total_disk_size()
3190
0
                         << ", rowset segment path:"
3191
0
                         << StorageResource().remote_segment_path(
3192
0
                                    rs_meta->tablet_id(), rs_meta->rowset_id().to_string(), 0);
3193
0
            DCHECK(false);
3194
0
        }
3195
0
    }
3196
18.4E
}
3197
3198
0
std::string Tablet::get_segment_path(const RowsetMetaSharedPtr& rs_meta, int64_t seg_id) {
3199
0
    std::string segment_path;
3200
0
    if (rs_meta->is_local()) {
3201
0
        segment_path = local_segment_path(_tablet_path, rs_meta->rowset_id().to_string(), seg_id);
3202
0
    } else {
3203
0
        segment_path = rs_meta->remote_storage_resource().value()->remote_segment_path(
3204
0
                rs_meta->tablet_id(), rs_meta->rowset_id().to_string(), seg_id);
3205
0
    }
3206
0
    return segment_path;
3207
0
}
3208
3209
0
int64_t Tablet::get_segment_file_size(const RowsetMetaSharedPtr& rs_meta) {
3210
0
    const auto& fs = rs_meta->fs();
3211
0
    if (!fs) {
3212
0
        LOG(WARNING) << "get fs failed, resource_id={}" << rs_meta->resource_id();
3213
0
    }
3214
0
    int64_t total_segment_size = 0;
3215
0
    for (int64_t seg_id = 0; seg_id < rs_meta->num_segments(); seg_id++) {
3216
0
        std::string segment_path = get_segment_path(rs_meta, seg_id);
3217
0
        int64_t segment_file_size = 0;
3218
0
        auto st = fs->file_size(segment_path, &segment_file_size);
3219
0
        if (!st.ok()) {
3220
0
            segment_file_size = 0;
3221
0
            LOG(WARNING) << "table size correctness check get segment size failed! msg:"
3222
0
                         << st.to_string() << ", segment path:" << segment_path;
3223
0
        }
3224
0
        total_segment_size += segment_file_size;
3225
0
    }
3226
0
    return total_segment_size;
3227
0
}
3228
3229
0
int64_t Tablet::get_inverted_index_file_size(const RowsetMetaSharedPtr& rs_meta) {
3230
0
    const auto& fs = rs_meta->fs();
3231
0
    if (!fs) {
3232
0
        LOG(WARNING) << "get fs failed, resource_id={}" << rs_meta->resource_id();
3233
0
    }
3234
0
    int64_t total_inverted_index_size = 0;
3235
3236
0
    if (rs_meta->tablet_schema()->get_inverted_index_storage_format() ==
3237
0
        InvertedIndexStorageFormatPB::V1) {
3238
0
        const auto& indices = rs_meta->tablet_schema()->inverted_indexes();
3239
0
        for (auto& index : indices) {
3240
0
            for (int seg_id = 0; seg_id < rs_meta->num_segments(); ++seg_id) {
3241
0
                std::string segment_path = get_segment_path(rs_meta, seg_id);
3242
0
                int64_t file_size = 0;
3243
3244
0
                std::string inverted_index_file_path =
3245
0
                        InvertedIndexDescriptor::get_index_file_path_v1(
3246
0
                                InvertedIndexDescriptor::get_index_file_path_prefix(segment_path),
3247
0
                                index->index_id(), index->get_index_suffix());
3248
0
                auto st = fs->file_size(inverted_index_file_path, &file_size);
3249
0
                if (!st.ok()) {
3250
0
                    file_size = 0;
3251
0
                    LOG(WARNING) << " tablet id: " << get_tablet_info().tablet_id
3252
0
                                 << ", rowset id:" << rs_meta->rowset_id()
3253
0
                                 << ", table size correctness check get inverted index v1 "
3254
0
                                    "size failed! msg:"
3255
0
                                 << st.to_string()
3256
0
                                 << ", inverted index path:" << inverted_index_file_path;
3257
0
                }
3258
0
                total_inverted_index_size += file_size;
3259
0
            }
3260
0
        }
3261
0
    } else {
3262
0
        for (int seg_id = 0; seg_id < rs_meta->num_segments(); ++seg_id) {
3263
0
            int64_t file_size = 0;
3264
0
            std::string segment_path = get_segment_path(rs_meta, seg_id);
3265
0
            std::string inverted_index_file_path = InvertedIndexDescriptor::get_index_file_path_v2(
3266
0
                    InvertedIndexDescriptor::get_index_file_path_prefix(segment_path));
3267
0
            auto st = fs->file_size(inverted_index_file_path, &file_size);
3268
0
            if (!st.ok()) {
3269
0
                file_size = 0;
3270
0
                if (st.is<NOT_FOUND>()) {
3271
0
                    LOG(INFO) << " tablet id: " << get_tablet_info().tablet_id
3272
0
                              << ", rowset id:" << rs_meta->rowset_id()
3273
0
                              << ", table size correctness check get inverted index v2 failed "
3274
0
                                 "because file not exist:"
3275
0
                              << inverted_index_file_path;
3276
0
                } else {
3277
0
                    LOG(WARNING) << " tablet id: " << get_tablet_info().tablet_id
3278
0
                                 << ", rowset id:" << rs_meta->rowset_id()
3279
0
                                 << ", table size correctness check get inverted index v2 "
3280
0
                                    "size failed! msg:"
3281
0
                                 << st.to_string()
3282
0
                                 << ", inverted index path:" << inverted_index_file_path;
3283
0
                }
3284
0
            }
3285
0
            total_inverted_index_size += file_size;
3286
0
        }
3287
0
    }
3288
0
    return total_inverted_index_size;
3289
0
}
3290
3291
Status Tablet::prepare_txn(TPartitionId partition_id, TTransactionId transaction_id,
3292
37.1k
                           const PUniqueId& load_id, bool ingest) {
3293
37.1k
    std::shared_lock base_migration_lock(get_migration_lock(), std::defer_lock);
3294
3295
    // TODO: rename debugpoint.
3296
37.1k
    DBUG_EXECUTE_IF("PushHandler::_do_streaming_ingestion.try_lock_fail", {
3297
37.1k
        return Status::ObtainLockFailed("PushHandler::_do_streaming_ingestion get lock failed");
3298
37.1k
    })
3299
3300
37.1k
    if (!base_migration_lock.try_lock_for(
3301
37.1k
                std::chrono::milliseconds(config::migration_lock_timeout_ms))) {
3302
0
        return Status::ObtainLockFailed("try_lock migration lock failed after {}ms",
3303
0
                                        config::migration_lock_timeout_ms);
3304
0
    }
3305
3306
37.1k
    std::lock_guard<std::mutex> push_lock(get_push_lock());
3307
37.1k
    return _engine.txn_manager()->prepare_txn(partition_id, transaction_id, tablet_id(),
3308
37.1k
                                              tablet_uid(), load_id, ingest);
3309
37.1k
}
3310
3311
} // namespace doris