Coverage Report

Created: 2026-03-23 15:14

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/root/doris/be/src/olap/tablet_meta.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 "olap/tablet_meta.h"
19
20
#include <bvar/bvar.h>
21
#include <gen_cpp/Descriptors_types.h>
22
#include <gen_cpp/FrontendService_types.h>
23
#include <gen_cpp/Types_types.h>
24
#include <gen_cpp/olap_common.pb.h>
25
#include <gen_cpp/olap_file.pb.h>
26
#include <gen_cpp/segment_v2.pb.h>
27
#include <gen_cpp/types.pb.h>
28
#include <json2pb/pb_to_json.h>
29
#include <time.h>
30
31
#include <cstdint>
32
#include <memory>
33
#include <random>
34
#include <set>
35
#include <utility>
36
37
#include "cloud/cloud_meta_mgr.h"
38
#include "cloud/cloud_storage_engine.h"
39
#include "cloud/config.h"
40
#include "common/config.h"
41
#include "io/fs/file_writer.h"
42
#include "io/fs/local_file_system.h"
43
#include "olap/data_dir.h"
44
#include "olap/file_header.h"
45
#include "olap/lru_cache.h"
46
#include "olap/olap_common.h"
47
#include "olap/olap_define.h"
48
#include "olap/rowset/rowset.h"
49
#include "olap/rowset/rowset_meta_manager.h"
50
#include "olap/tablet_fwd.h"
51
#include "olap/tablet_meta_manager.h"
52
#include "olap/tablet_schema_cache.h"
53
#include "olap/utils.h"
54
#include "util/debug_points.h"
55
#include "util/mem_info.h"
56
#include "util/parse_util.h"
57
#include "util/string_util.h"
58
#include "util/time.h"
59
#include "util/uid_util.h"
60
61
using std::string;
62
using std::unordered_map;
63
using std::vector;
64
65
namespace doris {
66
#include "common/compile_check_begin.h"
67
using namespace ErrorCode;
68
69
bvar::Adder<uint64_t> g_contains_agg_with_cache_if_eligible_total(
70
        "g_contains_agg_with_cache_if_eligible_total");
71
bvar::Adder<uint64_t> g_contains_agg_with_cache_if_eligible_partial_hit(
72
        "g_contains_agg_with_cache_if_eligible_partial_hit");
73
bvar::Adder<uint64_t> g_contains_agg_with_cache_if_eligible_full_hit(
74
        "g_contains_agg_with_cache_if_eligible_full_hit");
75
bvar::Window<bvar::Adder<uint64_t>> g_contains_agg_with_cache_if_eligible_total_minute(
76
        "g_contains_agg_with_cache_if_eligible_total_1m",
77
        &g_contains_agg_with_cache_if_eligible_total, 60);
78
bvar::Window<bvar::Adder<uint64_t>> g_contains_agg_with_cache_if_eligible_partial_hit_minute(
79
        "g_contains_agg_with_cache_if_eligible_partial_hit_1m",
80
        &g_contains_agg_with_cache_if_eligible_partial_hit, 60);
81
bvar::Window<bvar::Adder<uint64_t>> g_contains_agg_with_cache_if_eligible_full_hit_minute(
82
        "g_contains_agg_with_cache_if_eligible_full_hit_1m",
83
        &g_contains_agg_with_cache_if_eligible_full_hit, 60);
84
85
TabletMetaSharedPtr TabletMeta::create(
86
        const TCreateTabletReq& request, const TabletUid& tablet_uid, uint64_t shard_id,
87
        uint32_t next_unique_id,
88
304
        const unordered_map<uint32_t, uint32_t>& col_ordinal_to_unique_id) {
89
304
    std::optional<TBinlogConfig> binlog_config;
90
304
    if (request.__isset.binlog_config) {
91
0
        binlog_config = request.binlog_config;
92
0
    }
93
304
    TInvertedIndexFileStorageFormat::type inverted_index_file_storage_format =
94
304
            request.inverted_index_file_storage_format;
95
96
    // We will discard this format. Don't make any further changes here.
97
304
    if (request.__isset.inverted_index_storage_format) {
98
304
        switch (request.inverted_index_storage_format) {
99
0
        case TInvertedIndexStorageFormat::V1:
100
0
            inverted_index_file_storage_format = TInvertedIndexFileStorageFormat::V1;
101
0
            break;
102
0
        case TInvertedIndexStorageFormat::V2:
103
0
            inverted_index_file_storage_format = TInvertedIndexFileStorageFormat::V2;
104
0
            break;
105
304
        default:
106
304
            break;
107
304
        }
108
304
    }
109
    // Decide storage format for this tablet. DEFAULT / not-set fall back to V2 on BE side.
110
304
    TStorageFormat::type storage_format =
111
304
            request.__isset.storage_format ? request.storage_format : TStorageFormat::V2;
112
304
    return std::make_shared<TabletMeta>(
113
304
            request.table_id, request.partition_id, request.tablet_id, request.replica_id,
114
304
            request.tablet_schema.schema_hash, shard_id, request.tablet_schema, next_unique_id,
115
304
            col_ordinal_to_unique_id, tablet_uid,
116
304
            request.__isset.tablet_type ? request.tablet_type : TTabletType::TABLET_TYPE_DISK,
117
304
            request.__isset.compression_type ? request.compression_type : TCompressionType::LZ4F,
118
304
            request.__isset.storage_policy_id ? request.storage_policy_id : -1,
119
304
            request.__isset.enable_unique_key_merge_on_write
120
304
                    ? request.enable_unique_key_merge_on_write
121
304
                    : false,
122
304
            std::move(binlog_config), request.compaction_policy,
123
304
            request.time_series_compaction_goal_size_mbytes,
124
304
            request.time_series_compaction_file_count_threshold,
125
304
            request.time_series_compaction_time_threshold_seconds,
126
304
            request.time_series_compaction_empty_rowsets_threshold,
127
304
            request.time_series_compaction_level_threshold, inverted_index_file_storage_format,
128
304
            request.tde_algorithm, storage_format,
129
304
            request.__isset.vertical_compaction_num_columns_per_group
130
304
                    ? request.vertical_compaction_num_columns_per_group
131
304
                    : 5);
132
304
}
133
134
1.32k
TabletMeta::~TabletMeta() {
135
1.32k
    if (_handle) {
136
1.13k
        TabletSchemaCache::instance()->release(_handle);
137
1.13k
    }
138
1.32k
}
139
140
TabletMeta::TabletMeta()
141
557
        : _tablet_uid(0, 0),
142
557
          _schema(new TabletSchema),
143
557
          _delete_bitmap(new DeleteBitmap(_tablet_id)) {}
144
145
TabletMeta::TabletMeta(int64_t table_id, int64_t partition_id, int64_t tablet_id,
146
                       int64_t replica_id, int32_t schema_hash, int32_t shard_id,
147
                       const TTabletSchema& tablet_schema, uint32_t next_unique_id,
148
                       const std::unordered_map<uint32_t, uint32_t>& col_ordinal_to_unique_id,
149
                       TabletUid tablet_uid, TTabletType::type tabletType,
150
                       TCompressionType::type compression_type, int64_t storage_policy_id,
151
                       bool enable_unique_key_merge_on_write,
152
                       std::optional<TBinlogConfig> binlog_config, std::string compaction_policy,
153
                       int64_t time_series_compaction_goal_size_mbytes,
154
                       int64_t time_series_compaction_file_count_threshold,
155
                       int64_t time_series_compaction_time_threshold_seconds,
156
                       int64_t time_series_compaction_empty_rowsets_threshold,
157
                       int64_t time_series_compaction_level_threshold,
158
                       TInvertedIndexFileStorageFormat::type inverted_index_file_storage_format,
159
                       TEncryptionAlgorithm::type tde_algorithm,
160
                       TStorageFormat::type storage_format,
161
                       int32_t vertical_compaction_num_columns_per_group)
162
630
        : _tablet_uid(0, 0),
163
630
          _schema(new TabletSchema),
164
630
          _delete_bitmap(new DeleteBitmap(tablet_id)),
165
630
          _storage_format(storage_format) {
166
630
    TabletMetaPB tablet_meta_pb;
167
630
    tablet_meta_pb.set_table_id(table_id);
168
630
    tablet_meta_pb.set_partition_id(partition_id);
169
630
    tablet_meta_pb.set_tablet_id(tablet_id);
170
630
    tablet_meta_pb.set_replica_id(replica_id);
171
630
    tablet_meta_pb.set_schema_hash(schema_hash);
172
630
    tablet_meta_pb.set_shard_id(shard_id);
173
    // Persist the creation time, but it is not used
174
630
    tablet_meta_pb.set_creation_time(time(nullptr));
175
630
    tablet_meta_pb.set_cumulative_layer_point(-1);
176
630
    tablet_meta_pb.set_tablet_state(PB_RUNNING);
177
630
    *(tablet_meta_pb.mutable_tablet_uid()) = tablet_uid.to_proto();
178
630
    tablet_meta_pb.set_tablet_type(tabletType == TTabletType::TABLET_TYPE_DISK
179
630
                                           ? TabletTypePB::TABLET_TYPE_DISK
180
630
                                           : TabletTypePB::TABLET_TYPE_MEMORY);
181
630
    tablet_meta_pb.set_enable_unique_key_merge_on_write(enable_unique_key_merge_on_write);
182
630
    tablet_meta_pb.set_storage_policy_id(storage_policy_id);
183
630
    tablet_meta_pb.set_compaction_policy(compaction_policy);
184
630
    tablet_meta_pb.set_time_series_compaction_goal_size_mbytes(
185
630
            time_series_compaction_goal_size_mbytes);
186
630
    tablet_meta_pb.set_time_series_compaction_file_count_threshold(
187
630
            time_series_compaction_file_count_threshold);
188
630
    tablet_meta_pb.set_time_series_compaction_time_threshold_seconds(
189
630
            time_series_compaction_time_threshold_seconds);
190
630
    tablet_meta_pb.set_time_series_compaction_empty_rowsets_threshold(
191
630
            time_series_compaction_empty_rowsets_threshold);
192
630
    tablet_meta_pb.set_time_series_compaction_level_threshold(
193
630
            time_series_compaction_level_threshold);
194
630
    tablet_meta_pb.set_vertical_compaction_num_columns_per_group(
195
630
            vertical_compaction_num_columns_per_group);
196
630
    TabletSchemaPB* schema = tablet_meta_pb.mutable_schema();
197
630
    schema->set_num_short_key_columns(tablet_schema.short_key_column_count);
198
630
    schema->set_num_rows_per_row_block(config::default_num_rows_per_column_file_block);
199
630
    schema->set_sequence_col_idx(tablet_schema.sequence_col_idx);
200
630
    switch (tablet_schema.keys_type) {
201
50
    case TKeysType::DUP_KEYS:
202
50
        schema->set_keys_type(KeysType::DUP_KEYS);
203
50
        break;
204
303
    case TKeysType::UNIQUE_KEYS:
205
303
        schema->set_keys_type(KeysType::UNIQUE_KEYS);
206
303
        break;
207
68
    case TKeysType::AGG_KEYS:
208
68
        schema->set_keys_type(KeysType::AGG_KEYS);
209
68
        break;
210
209
    default:
211
209
        LOG(WARNING) << "unknown tablet keys type";
212
209
        break;
213
630
    }
214
    // compress_kind used to compress segment files
215
630
    schema->set_compress_kind(COMPRESS_LZ4);
216
217
    // compression_type used to compress segment page
218
630
    switch (compression_type) {
219
0
    case TCompressionType::NO_COMPRESSION:
220
0
        schema->set_compression_type(segment_v2::NO_COMPRESSION);
221
0
        break;
222
0
    case TCompressionType::SNAPPY:
223
0
        schema->set_compression_type(segment_v2::SNAPPY);
224
0
        break;
225
0
    case TCompressionType::LZ4:
226
0
        schema->set_compression_type(segment_v2::LZ4);
227
0
        break;
228
630
    case TCompressionType::LZ4F:
229
630
        schema->set_compression_type(segment_v2::LZ4F);
230
630
        break;
231
0
    case TCompressionType::ZLIB:
232
0
        schema->set_compression_type(segment_v2::ZLIB);
233
0
        break;
234
0
    case TCompressionType::ZSTD:
235
0
        schema->set_compression_type(segment_v2::ZSTD);
236
0
        break;
237
0
    default:
238
0
        schema->set_compression_type(segment_v2::LZ4F);
239
0
        break;
240
630
    }
241
242
630
    switch (inverted_index_file_storage_format) {
243
0
    case TInvertedIndexFileStorageFormat::V1:
244
0
        schema->set_inverted_index_storage_format(InvertedIndexStorageFormatPB::V1);
245
0
        break;
246
630
    case TInvertedIndexFileStorageFormat::V2:
247
630
        schema->set_inverted_index_storage_format(InvertedIndexStorageFormatPB::V2);
248
630
        break;
249
0
    case TInvertedIndexFileStorageFormat::V3:
250
0
        schema->set_inverted_index_storage_format(InvertedIndexStorageFormatPB::V3);
251
0
        break;
252
0
    default:
253
0
        schema->set_inverted_index_storage_format(InvertedIndexStorageFormatPB::V3);
254
0
        break;
255
630
    }
256
257
630
    switch (tablet_schema.sort_type) {
258
0
    case TSortType::type::ZORDER:
259
0
        schema->set_sort_type(SortType::ZORDER);
260
0
        break;
261
630
    default:
262
630
        schema->set_sort_type(SortType::LEXICAL);
263
630
    }
264
630
    schema->set_sort_col_num(tablet_schema.sort_col_num);
265
630
    for (const auto& i : tablet_schema.cluster_key_uids) {
266
2
        schema->add_cluster_key_uids(i);
267
2
    }
268
630
    tablet_meta_pb.set_in_restore_mode(false);
269
270
    // set column information
271
630
    uint32_t col_ordinal = 0;
272
630
    bool has_bf_columns = false;
273
2.22k
    for (TColumn tcolumn : tablet_schema.columns) {
274
2.22k
        ColumnPB* column = schema->add_column();
275
2.22k
        uint32_t unique_id = -1;
276
2.22k
        if (tcolumn.col_unique_id >= 0) {
277
11
            unique_id = tcolumn.col_unique_id;
278
2.21k
        } else {
279
2.21k
            unique_id = col_ordinal_to_unique_id.at(col_ordinal);
280
2.21k
        }
281
2.22k
        col_ordinal++;
282
2.22k
        init_column_from_tcolumn(unique_id, tcolumn, column);
283
284
2.22k
        if (column->is_bf_column()) {
285
0
            has_bf_columns = true;
286
0
        }
287
288
2.22k
        if (tablet_schema.__isset.indexes) {
289
2
            for (auto& index : tablet_schema.indexes) {
290
2
                if (index.index_type == TIndexType::type::BLOOMFILTER ||
291
2
                    index.index_type == TIndexType::type::NGRAM_BF) {
292
0
                    DCHECK_EQ(index.columns.size(), 1);
293
0
                    if (iequal(tcolumn.column_name, index.columns[0])) {
294
0
                        column->set_is_bf_column(true);
295
0
                        break;
296
0
                    }
297
0
                }
298
2
            }
299
2
        }
300
2.22k
    }
301
302
    // copy index meta
303
630
    if (tablet_schema.__isset.indexes) {
304
1
        for (auto& index : tablet_schema.indexes) {
305
1
            TabletIndexPB* index_pb = schema->add_index();
306
1
            index_pb->set_index_id(index.index_id);
307
1
            index_pb->set_index_name(index.index_name);
308
            // init col_unique_id in index at be side, since col_unique_id may be -1 at fe side
309
            // get column unique id by name
310
1
            for (auto column_name : index.columns) {
311
2
                for (auto column : schema->column()) {
312
2
                    if (iequal(column.name(), column_name)) {
313
1
                        index_pb->add_col_unique_id(column.unique_id());
314
1
                    }
315
2
                }
316
1
            }
317
1
            switch (index.index_type) {
318
1
            case TIndexType::BITMAP:
319
1
                index_pb->set_index_type(IndexType::BITMAP);
320
1
                break;
321
0
            case TIndexType::INVERTED:
322
0
                index_pb->set_index_type(IndexType::INVERTED);
323
0
                break;
324
0
            case TIndexType::ANN:
325
0
                index_pb->set_index_type(IndexType::ANN);
326
0
                break;
327
0
            case TIndexType::BLOOMFILTER:
328
0
                index_pb->set_index_type(IndexType::BLOOMFILTER);
329
0
                break;
330
0
            case TIndexType::NGRAM_BF:
331
0
                index_pb->set_index_type(IndexType::NGRAM_BF);
332
0
                break;
333
1
            }
334
335
1
            if (index.__isset.properties) {
336
0
                auto properties = index_pb->mutable_properties();
337
0
                for (auto kv : index.properties) {
338
0
                    (*properties)[kv.first] = kv.second;
339
0
                }
340
0
            }
341
1
        }
342
1
    }
343
344
630
    schema->set_next_column_unique_id(next_unique_id);
345
630
    if (has_bf_columns && tablet_schema.__isset.bloom_filter_fpp) {
346
0
        schema->set_bf_fpp(tablet_schema.bloom_filter_fpp);
347
0
    }
348
349
630
    if (tablet_schema.__isset.is_in_memory) {
350
0
        schema->set_is_in_memory(tablet_schema.is_in_memory);
351
0
    }
352
353
630
    if (tablet_schema.__isset.disable_auto_compaction) {
354
10
        schema->set_disable_auto_compaction(tablet_schema.disable_auto_compaction);
355
10
    }
356
357
630
    if (tablet_schema.__isset.variant_enable_flatten_nested) {
358
630
        schema->set_enable_variant_flatten_nested(tablet_schema.variant_enable_flatten_nested);
359
630
    }
360
361
630
    if (tablet_schema.__isset.enable_single_replica_compaction) {
362
630
        schema->set_enable_single_replica_compaction(
363
630
                tablet_schema.enable_single_replica_compaction);
364
630
    }
365
366
630
    if (tablet_schema.__isset.delete_sign_idx) {
367
630
        schema->set_delete_sign_idx(tablet_schema.delete_sign_idx);
368
630
    }
369
630
    if (tablet_schema.__isset.store_row_column) {
370
630
        schema->set_store_row_column(tablet_schema.store_row_column);
371
630
    }
372
630
    if (tablet_schema.__isset.row_store_page_size) {
373
630
        schema->set_row_store_page_size(tablet_schema.row_store_page_size);
374
630
    }
375
630
    if (tablet_schema.__isset.storage_page_size) {
376
630
        schema->set_storage_page_size(tablet_schema.storage_page_size);
377
630
    }
378
630
    if (tablet_schema.__isset.storage_dict_page_size) {
379
630
        schema->set_storage_dict_page_size(tablet_schema.storage_dict_page_size);
380
630
    }
381
630
    if (tablet_schema.__isset.skip_write_index_on_load) {
382
630
        schema->set_skip_write_index_on_load(tablet_schema.skip_write_index_on_load);
383
630
    }
384
630
    if (tablet_schema.__isset.row_store_col_cids) {
385
0
        schema->mutable_row_store_column_unique_ids()->Add(tablet_schema.row_store_col_cids.begin(),
386
0
                                                           tablet_schema.row_store_col_cids.end());
387
0
    }
388
630
    if (binlog_config.has_value()) {
389
0
        BinlogConfig tmp_binlog_config;
390
0
        tmp_binlog_config = binlog_config.value();
391
0
        tmp_binlog_config.to_pb(tablet_meta_pb.mutable_binlog_config());
392
0
    }
393
394
630
    switch (tde_algorithm) {
395
0
    case doris::TEncryptionAlgorithm::AES256:
396
0
        tablet_meta_pb.set_encryption_algorithm(EncryptionAlgorithmPB::AES_256_CTR);
397
0
        break;
398
0
    case doris::TEncryptionAlgorithm::SM4:
399
0
        tablet_meta_pb.set_encryption_algorithm(EncryptionAlgorithmPB::SM4_128_CTR);
400
0
        break;
401
630
    default:
402
630
        tablet_meta_pb.set_encryption_algorithm(EncryptionAlgorithmPB::PLAINTEXT);
403
630
    }
404
405
    // Initialize default external ColumnMeta usage according to storage format.
406
    // V2: legacy behavior, inline ColumnMetaPB only.
407
    // V3: V2 + external ColumnMetaPB (CMO) enabled by default.
408
630
    switch (_storage_format) {
409
630
    case TStorageFormat::V2:
410
630
    case TStorageFormat::DEFAULT:
411
630
    case TStorageFormat::V1:
412
630
        break;
413
0
    case TStorageFormat::V3:
414
0
        schema->set_is_external_segment_column_meta_used(true);
415
0
        _schema->set_external_segment_meta_used_default(true);
416
417
0
        schema->set_integer_type_default_use_plain_encoding(true);
418
0
        _schema->set_integer_type_default_use_plain_encoding(true);
419
0
        schema->set_binary_plain_encoding_default_impl(
420
0
                BinaryPlainEncodingTypePB::BINARY_PLAIN_ENCODING_V2);
421
0
        _schema->set_binary_plain_encoding_default_impl(
422
0
                BinaryPlainEncodingTypePB::BINARY_PLAIN_ENCODING_V2);
423
0
        break;
424
0
    default:
425
0
        break;
426
630
    }
427
428
630
    init_from_pb(tablet_meta_pb);
429
630
}
430
431
TabletMeta::TabletMeta(const TabletMeta& b)
432
81
        : MetadataAdder(b),
433
81
          _table_id(b._table_id),
434
81
          _index_id(b._index_id),
435
81
          _partition_id(b._partition_id),
436
81
          _tablet_id(b._tablet_id),
437
81
          _replica_id(b._replica_id),
438
81
          _schema_hash(b._schema_hash),
439
81
          _shard_id(b._shard_id),
440
81
          _creation_time(b._creation_time),
441
81
          _cumulative_layer_point(b._cumulative_layer_point),
442
81
          _tablet_uid(b._tablet_uid),
443
81
          _tablet_type(b._tablet_type),
444
81
          _tablet_state(b._tablet_state),
445
81
          _schema(b._schema),
446
81
          _rs_metas(b._rs_metas),
447
81
          _stale_rs_metas(b._stale_rs_metas),
448
81
          _in_restore_mode(b._in_restore_mode),
449
81
          _preferred_rowset_type(b._preferred_rowset_type),
450
81
          _storage_policy_id(b._storage_policy_id),
451
81
          _cooldown_meta_id(b._cooldown_meta_id),
452
81
          _enable_unique_key_merge_on_write(b._enable_unique_key_merge_on_write),
453
81
          _delete_bitmap(b._delete_bitmap),
454
81
          _binlog_config(b._binlog_config),
455
81
          _compaction_policy(b._compaction_policy),
456
81
          _time_series_compaction_goal_size_mbytes(b._time_series_compaction_goal_size_mbytes),
457
          _time_series_compaction_file_count_threshold(
458
81
                  b._time_series_compaction_file_count_threshold),
459
          _time_series_compaction_time_threshold_seconds(
460
81
                  b._time_series_compaction_time_threshold_seconds),
461
          _time_series_compaction_empty_rowsets_threshold(
462
81
                  b._time_series_compaction_empty_rowsets_threshold),
463
81
          _time_series_compaction_level_threshold(b._time_series_compaction_level_threshold),
464
          _vertical_compaction_num_columns_per_group(
465
81
                  b._vertical_compaction_num_columns_per_group) {};
466
467
void TabletMeta::init_column_from_tcolumn(uint32_t unique_id, const TColumn& tcolumn,
468
2.23k
                                          ColumnPB* column) {
469
2.23k
    column->set_unique_id(unique_id);
470
2.23k
    column->set_name(tcolumn.column_name);
471
2.23k
    column->set_is_auto_increment(tcolumn.is_auto_increment);
472
2.23k
    if (tcolumn.__isset.is_on_update_current_timestamp) {
473
2.23k
        column->set_is_on_update_current_timestamp(tcolumn.is_on_update_current_timestamp);
474
2.23k
    }
475
2.23k
    string data_type;
476
2.23k
    EnumToString(TPrimitiveType, tcolumn.column_type.type, data_type);
477
2.23k
    column->set_type(data_type);
478
479
2.23k
    uint32_t length = TabletColumn::get_field_length_by_type(tcolumn.column_type.type,
480
2.23k
                                                             tcolumn.column_type.len);
481
2.23k
    column->set_length(length);
482
2.23k
    column->set_index_length(length);
483
2.23k
    column->set_precision(tcolumn.column_type.precision);
484
2.23k
    column->set_frac(tcolumn.column_type.scale);
485
486
2.23k
    if (tcolumn.__isset.result_is_nullable) {
487
0
        column->set_result_is_nullable(tcolumn.result_is_nullable);
488
0
    }
489
490
2.23k
    if (tcolumn.__isset.be_exec_version) {
491
2.23k
        column->set_be_exec_version(tcolumn.be_exec_version);
492
2.23k
    }
493
494
2.23k
    if (tcolumn.column_type.type == TPrimitiveType::VARCHAR ||
495
2.23k
        tcolumn.column_type.type == TPrimitiveType::STRING) {
496
108
        if (!tcolumn.column_type.__isset.index_len) {
497
108
            column->set_index_length(10);
498
108
        } else {
499
0
            column->set_index_length(tcolumn.column_type.index_len);
500
0
        }
501
108
    }
502
2.23k
    if (!tcolumn.is_key) {
503
1.17k
        column->set_is_key(false);
504
1.17k
        if (tcolumn.__isset.aggregation) {
505
0
            column->set_aggregation(tcolumn.aggregation);
506
1.17k
        } else {
507
1.17k
            string aggregation_type;
508
1.17k
            EnumToString(TAggregationType, tcolumn.aggregation_type, aggregation_type);
509
1.17k
            column->set_aggregation(aggregation_type);
510
1.17k
        }
511
1.17k
    } else {
512
1.06k
        column->set_is_key(true);
513
1.06k
        column->set_aggregation("NONE");
514
1.06k
    }
515
2.23k
    column->set_is_nullable(tcolumn.is_allow_null);
516
2.23k
    if (tcolumn.__isset.default_value) {
517
1
        column->set_default_value(tcolumn.default_value);
518
1
    }
519
2.23k
    if (tcolumn.__isset.is_bloom_filter_column) {
520
1
        column->set_is_bf_column(tcolumn.is_bloom_filter_column);
521
1
    }
522
2.23k
    if (tcolumn.__isset.visible) {
523
2.23k
        column->set_visible(tcolumn.visible);
524
2.23k
    }
525
2.23k
    for (size_t i = 0; i < tcolumn.children_column.size(); i++) {
526
0
        ColumnPB* children_column = column->add_children_columns();
527
0
        init_column_from_tcolumn(tcolumn.children_column[i].col_unique_id,
528
0
                                 tcolumn.children_column[i], children_column);
529
0
    }
530
2.23k
    if (tcolumn.column_type.__isset.variant_max_subcolumns_count) {
531
2.23k
        column->set_variant_max_subcolumns_count(tcolumn.column_type.variant_max_subcolumns_count);
532
2.23k
    }
533
2.23k
    if (tcolumn.__isset.pattern_type) {
534
1
        switch (tcolumn.pattern_type) {
535
0
        case TPatternType::MATCH_NAME:
536
0
            column->set_pattern_type(PatternTypePB::MATCH_NAME);
537
0
            break;
538
1
        case TPatternType::MATCH_NAME_GLOB:
539
1
            column->set_pattern_type(PatternTypePB::MATCH_NAME_GLOB);
540
1
        }
541
1
    }
542
2.23k
    if (tcolumn.__isset.variant_enable_typed_paths_to_sparse) {
543
2.23k
        column->set_variant_enable_typed_paths_to_sparse(
544
2.23k
                tcolumn.variant_enable_typed_paths_to_sparse);
545
2.23k
    }
546
2.23k
    if (tcolumn.__isset.variant_max_sparse_column_statistics_size) {
547
2.23k
        column->set_variant_max_sparse_column_statistics_size(
548
2.23k
                tcolumn.variant_max_sparse_column_statistics_size);
549
2.23k
    }
550
2.23k
    if (tcolumn.__isset.variant_sparse_hash_shard_count) {
551
0
        column->set_variant_sparse_hash_shard_count(tcolumn.variant_sparse_hash_shard_count);
552
0
    }
553
2.23k
    if (tcolumn.__isset.variant_enable_doc_mode) {
554
0
        column->set_variant_enable_doc_mode(tcolumn.variant_enable_doc_mode);
555
0
    }
556
2.23k
    if (tcolumn.__isset.variant_doc_materialization_min_rows) {
557
0
        column->set_variant_doc_materialization_min_rows(
558
0
                tcolumn.variant_doc_materialization_min_rows);
559
0
    }
560
2.23k
    if (tcolumn.__isset.variant_doc_hash_shard_count) {
561
0
        column->set_variant_doc_hash_shard_count(tcolumn.variant_doc_hash_shard_count);
562
0
    }
563
2.23k
    if (tcolumn.__isset.variant_enable_nested_group) {
564
0
        column->set_variant_enable_nested_group(tcolumn.variant_enable_nested_group);
565
0
    }
566
2.23k
}
567
568
0
void TabletMeta::remove_rowset_delete_bitmap(const RowsetId& rowset_id, const Version& version) {
569
0
    if (_enable_unique_key_merge_on_write) {
570
0
        delete_bitmap().remove({rowset_id, 0, 0}, {rowset_id, UINT32_MAX, 0});
571
0
        if (config::enable_mow_verbose_log) {
572
0
            LOG_INFO("delete rowset delete bitmap. tablet={}, rowset={}, version={}", tablet_id(),
573
0
                     rowset_id.to_string(), version.to_string());
574
0
        }
575
0
        size_t rowset_cache_version_size = delete_bitmap().remove_rowset_cache_version(rowset_id);
576
0
        _check_mow_rowset_cache_version_size(rowset_cache_version_size);
577
0
    }
578
0
}
579
580
4
Status TabletMeta::create_from_file(const string& file_path) {
581
4
    TabletMetaPB tablet_meta_pb;
582
4
    RETURN_IF_ERROR(load_from_file(file_path, &tablet_meta_pb));
583
4
    init_from_pb(tablet_meta_pb);
584
4
    return Status::OK();
585
4
}
586
587
12
Status TabletMeta::load_from_file(const string& file_path, TabletMetaPB* tablet_meta_pb) {
588
12
    FileHeader<TabletMetaPB> file_header(file_path);
589
    // In file_header.deserialize(), it validates file length, signature, checksum of protobuf.
590
12
    RETURN_IF_ERROR(file_header.deserialize());
591
12
    try {
592
12
        tablet_meta_pb->CopyFrom(file_header.message());
593
12
    } catch (const std::exception& e) {
594
0
        LOG(WARNING) << "Failed to copy protocol buffer object: " << e.what()
595
0
                     << ", file=" << file_path;
596
0
        return Status::Error<PARSE_PROTOBUF_ERROR>(
597
0
                "fail to copy protocol buffer object. file={}, error={}", file_path, e.what());
598
0
    }
599
12
    return Status::OK();
600
12
}
601
602
6
Status TabletMeta::create_from_buffer(const uint8_t* buffer, size_t buffer_size) {
603
6
    FileHeader<TabletMetaPB> file_header(""); // empty file path
604
6
    RETURN_IF_ERROR(file_header.deserialize_from_memory(buffer, buffer_size));
605
606
2
    TabletMetaPB tablet_meta_pb;
607
2
    try {
608
2
        tablet_meta_pb.CopyFrom(file_header.message());
609
2
    } catch (const std::exception& e) {
610
0
        LOG(WARNING) << "Failed to copy protocol buffer object from buffer: " << e.what();
611
0
        return Status::Error<ErrorCode::PARSE_PROTOBUF_ERROR>(
612
0
                "fail to copy protocol buffer object from buffer. error={}", e.what());
613
0
    }
614
615
2
    init_from_pb(tablet_meta_pb);
616
2
    return Status::OK();
617
2
}
618
619
std::string TabletMeta::construct_header_file_path(const string& schema_hash_path,
620
3
                                                   int64_t tablet_id) {
621
3
    std::stringstream header_name_stream;
622
3
    header_name_stream << schema_hash_path << "/" << tablet_id << ".hdr";
623
3
    return header_name_stream.str();
624
3
}
625
626
0
Status TabletMeta::save_as_json(const string& file_path) {
627
0
    std::string json_meta;
628
0
    json2pb::Pb2JsonOptions json_options;
629
0
    json_options.pretty_json = true;
630
0
    json_options.bytes_to_base64 = true;
631
0
    to_json(&json_meta, json_options);
632
    // save to file
633
0
    io::FileWriterPtr file_writer;
634
0
    RETURN_IF_ERROR(io::global_local_filesystem()->create_file(file_path, &file_writer));
635
0
    RETURN_IF_ERROR(file_writer->append(json_meta));
636
0
    RETURN_IF_ERROR(file_writer->close());
637
0
    return Status::OK();
638
0
}
639
640
230
Status TabletMeta::save(const string& file_path) {
641
230
    TabletMetaPB tablet_meta_pb;
642
230
    to_meta_pb(&tablet_meta_pb, false);
643
230
    return TabletMeta::save(file_path, tablet_meta_pb);
644
230
}
645
646
236
Status TabletMeta::save(const string& file_path, const TabletMetaPB& tablet_meta_pb) {
647
236
    DCHECK(!file_path.empty());
648
236
    FileHeader<TabletMetaPB> file_header(file_path);
649
236
    try {
650
236
        file_header.mutable_message()->CopyFrom(tablet_meta_pb);
651
236
    } catch (...) {
652
0
        LOG(WARNING) << "fail to copy protocol buffer object. file='" << file_path;
653
0
        return Status::Error<ErrorCode::INTERNAL_ERROR>(
654
0
                "fail to copy protocol buffer object. file={}", file_path);
655
0
    }
656
236
    RETURN_IF_ERROR(file_header.prepare());
657
236
    RETURN_IF_ERROR(file_header.serialize());
658
236
    return Status::OK();
659
236
}
660
661
572
Status TabletMeta::save_meta(DataDir* data_dir) {
662
572
    std::lock_guard<std::shared_mutex> wrlock(_meta_lock);
663
572
    return _save_meta(data_dir);
664
572
}
665
666
572
Status TabletMeta::_save_meta(DataDir* data_dir) {
667
    // check if tablet uid is valid
668
572
    if (_tablet_uid.hi == 0 && _tablet_uid.lo == 0) {
669
0
        LOG(FATAL) << "tablet_uid is invalid"
670
0
                   << " tablet=" << tablet_id() << " _tablet_uid=" << _tablet_uid.to_string();
671
0
    }
672
572
    string meta_binary;
673
674
572
    auto t1 = MonotonicMicros();
675
572
    serialize(&meta_binary);
676
572
    auto t2 = MonotonicMicros();
677
572
    Status status = TabletMetaManager::save(data_dir, tablet_id(), schema_hash(), meta_binary);
678
572
    if (!status.ok()) {
679
0
        LOG(FATAL) << "fail to save tablet_meta. status=" << status << ", tablet_id=" << tablet_id()
680
0
                   << ", schema_hash=" << schema_hash();
681
0
    }
682
572
    auto t3 = MonotonicMicros();
683
572
    auto cost = t3 - t1;
684
572
    if (cost > 1 * 1000 * 1000) {
685
0
        LOG(INFO) << "save tablet(" << tablet_id() << ") meta too slow. serialize cost " << t2 - t1
686
0
                  << "(us), serialized binary size: " << meta_binary.length()
687
0
                  << "(bytes), write rocksdb cost " << t3 - t2 << "(us)";
688
0
    }
689
572
    return status;
690
572
}
691
692
577
void TabletMeta::serialize(string* meta_binary) {
693
577
    TabletMetaPB tablet_meta_pb;
694
577
    to_meta_pb(&tablet_meta_pb, false);
695
577
    if (tablet_meta_pb.partition_id() <= 0) {
696
468
        LOG(WARNING) << "invalid partition id " << tablet_meta_pb.partition_id() << " tablet "
697
468
                     << tablet_meta_pb.tablet_id();
698
468
    }
699
577
    DBUG_EXECUTE_IF("TabletMeta::serialize::zero_partition_id", {
700
577
        long partition_id = tablet_meta_pb.partition_id();
701
577
        tablet_meta_pb.set_partition_id(0);
702
577
        LOG(WARNING) << "set debug point TabletMeta::serialize::zero_partition_id old="
703
577
                     << partition_id << " new=" << tablet_meta_pb.DebugString();
704
577
    });
705
577
    bool serialize_success = tablet_meta_pb.SerializeToString(meta_binary);
706
577
    if (!_rs_metas.empty() || !_stale_rs_metas.empty()) {
707
577
        _avg_rs_meta_serialize_size =
708
577
                meta_binary->length() / (_rs_metas.size() + _stale_rs_metas.size());
709
577
        if (meta_binary->length() > config::tablet_meta_serialize_size_limit ||
710
577
            !serialize_success) {
711
0
            int64_t origin_meta_size = meta_binary->length();
712
0
            int64_t stale_rowsets_num = tablet_meta_pb.stale_rs_metas().size();
713
0
            tablet_meta_pb.clear_stale_rs_metas();
714
0
            meta_binary->clear();
715
0
            serialize_success = tablet_meta_pb.SerializeToString(meta_binary);
716
0
            LOG(WARNING) << "tablet meta serialization size exceeds limit: "
717
0
                         << config::tablet_meta_serialize_size_limit
718
0
                         << " clean up stale rowsets, tablet id: " << tablet_id()
719
0
                         << " stale rowset num: " << stale_rowsets_num
720
0
                         << " serialization size before clean " << origin_meta_size
721
0
                         << " serialization size after clean " << meta_binary->length();
722
0
        }
723
577
    }
724
725
577
    if (!serialize_success) {
726
0
        LOG(FATAL) << "failed to serialize meta " << tablet_id();
727
0
    }
728
577
}
729
730
461
Status TabletMeta::deserialize(std::string_view meta_binary) {
731
461
    TabletMetaPB tablet_meta_pb;
732
461
    bool parsed = tablet_meta_pb.ParseFromArray(meta_binary.data(),
733
461
                                                static_cast<int32_t>(meta_binary.size()));
734
461
    if (!parsed) {
735
0
        return Status::Error<INIT_FAILED>("parse tablet meta failed");
736
0
    }
737
461
    init_from_pb(tablet_meta_pb);
738
461
    return Status::OK();
739
461
}
740
741
1.13k
void TabletMeta::init_from_pb(const TabletMetaPB& tablet_meta_pb) {
742
1.13k
    _table_id = tablet_meta_pb.table_id();
743
1.13k
    _index_id = tablet_meta_pb.index_id();
744
1.13k
    _partition_id = tablet_meta_pb.partition_id();
745
1.13k
    _tablet_id = tablet_meta_pb.tablet_id();
746
1.13k
    _replica_id = tablet_meta_pb.replica_id();
747
1.13k
    _schema_hash = tablet_meta_pb.schema_hash();
748
1.13k
    _shard_id = tablet_meta_pb.shard_id();
749
1.13k
    _creation_time = tablet_meta_pb.creation_time();
750
1.13k
    _cumulative_layer_point = tablet_meta_pb.cumulative_layer_point();
751
1.13k
    _tablet_uid = TabletUid(tablet_meta_pb.tablet_uid());
752
1.13k
    _ttl_seconds = tablet_meta_pb.ttl_seconds();
753
1.13k
    if (tablet_meta_pb.has_tablet_type()) {
754
1.10k
        _tablet_type = tablet_meta_pb.tablet_type();
755
1.10k
    } else {
756
29
        _tablet_type = TabletTypePB::TABLET_TYPE_DISK;
757
29
    }
758
759
    // init _tablet_state
760
1.13k
    switch (tablet_meta_pb.tablet_state()) {
761
32
    case PB_NOTREADY:
762
32
        _tablet_state = TabletState::TABLET_NOTREADY;
763
32
        break;
764
877
    case PB_RUNNING:
765
877
        _tablet_state = TabletState::TABLET_RUNNING;
766
877
        break;
767
0
    case PB_TOMBSTONED:
768
0
        _tablet_state = TabletState::TABLET_TOMBSTONED;
769
0
        break;
770
0
    case PB_STOPPED:
771
0
        _tablet_state = TabletState::TABLET_STOPPED;
772
0
        break;
773
225
    case PB_SHUTDOWN:
774
225
        _tablet_state = TabletState::TABLET_SHUTDOWN;
775
225
        break;
776
0
    default:
777
0
        LOG(WARNING) << "tablet has no state. tablet=" << tablet_id()
778
0
                     << ", schema_hash=" << schema_hash();
779
1.13k
    }
780
781
    // init _schema
782
1.13k
    TabletSchemaSPtr schema = std::make_shared<TabletSchema>();
783
1.13k
    schema->init_from_pb(tablet_meta_pb.schema());
784
1.13k
    if (_handle) {
785
4
        TabletSchemaCache::instance()->release(_handle);
786
4
    }
787
1.13k
    auto pair = TabletSchemaCache::instance()->insert(schema->to_key());
788
1.13k
    _handle = pair.first;
789
1.13k
    _schema = pair.second;
790
791
1.13k
    if (tablet_meta_pb.has_enable_unique_key_merge_on_write()) {
792
1.10k
        _enable_unique_key_merge_on_write = tablet_meta_pb.enable_unique_key_merge_on_write();
793
1.10k
        _delete_bitmap->set_tablet_id(_tablet_id);
794
1.10k
    }
795
796
    // init _rs_metas
797
10.8k
    for (auto& it : tablet_meta_pb.rs_metas()) {
798
10.8k
        RowsetMetaSharedPtr rs_meta(new RowsetMeta());
799
10.8k
        rs_meta->init_from_pb(it);
800
10.8k
        _rs_metas.emplace(rs_meta->version(), rs_meta);
801
10.8k
    }
802
803
    // For mow table, delete bitmap of stale rowsets has not been persisted.
804
    // When be restart, query should not read the stale rowset, otherwise duplicate keys
805
    // will be read out. Therefore, we don't add them to _stale_rs_meta for mow table.
806
1.13k
    if (!config::skip_loading_stale_rowset_meta && !_enable_unique_key_merge_on_write) {
807
1.09k
        for (auto& it : tablet_meta_pb.stale_rs_metas()) {
808
0
            RowsetMetaSharedPtr rs_meta(new RowsetMeta());
809
0
            rs_meta->init_from_pb(it);
810
0
            _stale_rs_metas.emplace(rs_meta->version(), rs_meta);
811
0
        }
812
1.09k
    }
813
814
1.13k
    if (tablet_meta_pb.has_in_restore_mode()) {
815
1.10k
        _in_restore_mode = tablet_meta_pb.in_restore_mode();
816
1.10k
    }
817
818
1.13k
    if (tablet_meta_pb.has_preferred_rowset_type()) {
819
475
        _preferred_rowset_type = tablet_meta_pb.preferred_rowset_type();
820
475
    }
821
822
1.13k
    _storage_policy_id = tablet_meta_pb.storage_policy_id();
823
1.13k
    if (tablet_meta_pb.has_cooldown_meta_id()) {
824
0
        _cooldown_meta_id = tablet_meta_pb.cooldown_meta_id();
825
0
    }
826
827
1.13k
    if (tablet_meta_pb.has_delete_bitmap()) {
828
0
        int rst_ids_size = tablet_meta_pb.delete_bitmap().rowset_ids_size();
829
0
        int seg_ids_size = tablet_meta_pb.delete_bitmap().segment_ids_size();
830
0
        int versions_size = tablet_meta_pb.delete_bitmap().versions_size();
831
0
        int seg_maps_size = tablet_meta_pb.delete_bitmap().segment_delete_bitmaps_size();
832
0
        CHECK(rst_ids_size == seg_ids_size && seg_ids_size == seg_maps_size &&
833
0
              seg_maps_size == versions_size);
834
0
        for (int i = 0; i < rst_ids_size; ++i) {
835
0
            RowsetId rst_id;
836
0
            rst_id.init(tablet_meta_pb.delete_bitmap().rowset_ids(i));
837
0
            auto seg_id = tablet_meta_pb.delete_bitmap().segment_ids(i);
838
0
            auto ver = tablet_meta_pb.delete_bitmap().versions(i);
839
0
            auto bitmap = tablet_meta_pb.delete_bitmap().segment_delete_bitmaps(i).data();
840
0
            delete_bitmap().delete_bitmap[{rst_id, seg_id, ver}] = roaring::Roaring::read(bitmap);
841
0
        }
842
0
    }
843
844
1.13k
    if (tablet_meta_pb.has_binlog_config()) {
845
473
        _binlog_config = tablet_meta_pb.binlog_config();
846
473
    }
847
1.13k
    _compaction_policy = tablet_meta_pb.compaction_policy();
848
1.13k
    _time_series_compaction_goal_size_mbytes =
849
1.13k
            tablet_meta_pb.time_series_compaction_goal_size_mbytes();
850
1.13k
    _time_series_compaction_file_count_threshold =
851
1.13k
            tablet_meta_pb.time_series_compaction_file_count_threshold();
852
1.13k
    _time_series_compaction_time_threshold_seconds =
853
1.13k
            tablet_meta_pb.time_series_compaction_time_threshold_seconds();
854
1.13k
    _time_series_compaction_empty_rowsets_threshold =
855
1.13k
            tablet_meta_pb.time_series_compaction_empty_rowsets_threshold();
856
1.13k
    _time_series_compaction_level_threshold =
857
1.13k
            tablet_meta_pb.time_series_compaction_level_threshold();
858
1.13k
    _vertical_compaction_num_columns_per_group =
859
1.13k
            tablet_meta_pb.vertical_compaction_num_columns_per_group();
860
861
1.13k
    if (tablet_meta_pb.has_encryption_algorithm()) {
862
1.10k
        _encryption_algorithm = tablet_meta_pb.encryption_algorithm();
863
1.10k
    }
864
1.13k
}
865
866
822
void TabletMeta::to_meta_pb(TabletMetaPB* tablet_meta_pb, bool cloud_get_rowset_meta) {
867
822
    tablet_meta_pb->set_table_id(table_id());
868
822
    tablet_meta_pb->set_index_id(index_id());
869
822
    tablet_meta_pb->set_partition_id(partition_id());
870
822
    tablet_meta_pb->set_tablet_id(tablet_id());
871
822
    tablet_meta_pb->set_replica_id(replica_id());
872
822
    tablet_meta_pb->set_schema_hash(schema_hash());
873
822
    tablet_meta_pb->set_shard_id(shard_id());
874
822
    tablet_meta_pb->set_creation_time(creation_time());
875
822
    tablet_meta_pb->set_cumulative_layer_point(cumulative_layer_point());
876
822
    *(tablet_meta_pb->mutable_tablet_uid()) = tablet_uid().to_proto();
877
822
    tablet_meta_pb->set_tablet_type(_tablet_type);
878
822
    tablet_meta_pb->set_ttl_seconds(_ttl_seconds);
879
822
    switch (tablet_state()) {
880
9
    case TABLET_NOTREADY:
881
9
        tablet_meta_pb->set_tablet_state(PB_NOTREADY);
882
9
        break;
883
332
    case TABLET_RUNNING:
884
332
        tablet_meta_pb->set_tablet_state(PB_RUNNING);
885
332
        break;
886
0
    case TABLET_TOMBSTONED:
887
0
        tablet_meta_pb->set_tablet_state(PB_TOMBSTONED);
888
0
        break;
889
0
    case TABLET_STOPPED:
890
0
        tablet_meta_pb->set_tablet_state(PB_STOPPED);
891
0
        break;
892
481
    case TABLET_SHUTDOWN:
893
481
        tablet_meta_pb->set_tablet_state(PB_SHUTDOWN);
894
481
        break;
895
822
    }
896
897
    // RowsetMetaPB is separated from TabletMetaPB
898
822
    if (!config::is_cloud_mode() || cloud_get_rowset_meta) {
899
21.6k
        for (const auto& [_, rs] : _rs_metas) {
900
21.6k
            rs->to_rowset_pb(tablet_meta_pb->add_rs_metas());
901
21.6k
        }
902
822
        for (const auto& [_, rs] : _stale_rs_metas) {
903
0
            rs->to_rowset_pb(tablet_meta_pb->add_stale_rs_metas());
904
0
        }
905
822
    }
906
907
822
    _schema->to_schema_pb(tablet_meta_pb->mutable_schema());
908
909
822
    tablet_meta_pb->set_in_restore_mode(in_restore_mode());
910
911
    // to avoid modify tablet meta to the greatest extend
912
822
    if (_preferred_rowset_type == BETA_ROWSET) {
913
822
        tablet_meta_pb->set_preferred_rowset_type(_preferred_rowset_type);
914
822
    }
915
822
    if (_storage_policy_id > 0) {
916
5
        tablet_meta_pb->set_storage_policy_id(_storage_policy_id);
917
5
    }
918
822
    if (_cooldown_meta_id.initialized()) {
919
5
        tablet_meta_pb->mutable_cooldown_meta_id()->CopyFrom(_cooldown_meta_id.to_proto());
920
5
    }
921
922
822
    tablet_meta_pb->set_enable_unique_key_merge_on_write(_enable_unique_key_merge_on_write);
923
924
822
    if (_enable_unique_key_merge_on_write) {
925
4
        std::set<RowsetId> stale_rs_ids;
926
4
        for (const auto& [_, rowset] : _stale_rs_metas) {
927
0
            stale_rs_ids.insert(rowset->rowset_id());
928
0
        }
929
4
        DeleteBitmapPB* delete_bitmap_pb = tablet_meta_pb->mutable_delete_bitmap();
930
4
        for (auto& [id, bitmap] : delete_bitmap().snapshot().delete_bitmap) {
931
2
            auto& [rowset_id, segment_id, ver] = id;
932
2
            if (stale_rs_ids.count(rowset_id) != 0) {
933
0
                continue;
934
0
            }
935
2
            delete_bitmap_pb->add_rowset_ids(rowset_id.to_string());
936
2
            delete_bitmap_pb->add_segment_ids(segment_id);
937
2
            delete_bitmap_pb->add_versions(ver);
938
2
            std::string bitmap_data(bitmap.getSizeInBytes(), '\0');
939
2
            bitmap.write(bitmap_data.data());
940
2
            *(delete_bitmap_pb->add_segment_delete_bitmaps()) = std::move(bitmap_data);
941
2
        }
942
4
    }
943
822
    _binlog_config.to_pb(tablet_meta_pb->mutable_binlog_config());
944
822
    tablet_meta_pb->set_compaction_policy(compaction_policy());
945
822
    tablet_meta_pb->set_time_series_compaction_goal_size_mbytes(
946
822
            time_series_compaction_goal_size_mbytes());
947
822
    tablet_meta_pb->set_time_series_compaction_file_count_threshold(
948
822
            time_series_compaction_file_count_threshold());
949
822
    tablet_meta_pb->set_time_series_compaction_time_threshold_seconds(
950
822
            time_series_compaction_time_threshold_seconds());
951
822
    tablet_meta_pb->set_time_series_compaction_empty_rowsets_threshold(
952
822
            time_series_compaction_empty_rowsets_threshold());
953
822
    tablet_meta_pb->set_time_series_compaction_level_threshold(
954
822
            time_series_compaction_level_threshold());
955
822
    tablet_meta_pb->set_vertical_compaction_num_columns_per_group(
956
822
            vertical_compaction_num_columns_per_group());
957
958
822
    tablet_meta_pb->set_encryption_algorithm(_encryption_algorithm);
959
822
}
960
961
2
void TabletMeta::to_json(string* json_string, json2pb::Pb2JsonOptions& options) {
962
2
    TabletMetaPB tablet_meta_pb;
963
2
    to_meta_pb(&tablet_meta_pb, false);
964
2
    json2pb::ProtoMessageToJson(tablet_meta_pb, json_string, options);
965
2
}
966
967
165
Version TabletMeta::max_version() const {
968
165
    Version max_version = {-1, 0};
969
3.16k
    for (const auto& [_, rs_meta] : _rs_metas) {
970
3.16k
        if (rs_meta->end_version() > max_version.second) {
971
153
            max_version = rs_meta->version();
972
153
        }
973
3.16k
    }
974
165
    return max_version;
975
165
}
976
977
0
size_t TabletMeta::version_count_cross_with_range(const Version& range) const {
978
0
    size_t count = 0;
979
0
    for (const auto& [_, rs_meta] : _rs_metas) {
980
0
        if (!(range.first > rs_meta->version().second || range.second < rs_meta->version().first)) {
981
0
            count++;
982
0
        }
983
0
    }
984
0
    return count;
985
0
}
986
987
14.1k
Status TabletMeta::add_rs_meta(const RowsetMetaSharedPtr& rs_meta) {
988
    // check RowsetMeta is valid
989
690k
    for (const auto& [_, rs] : _rs_metas) {
990
690k
        if (rs->version() == rs_meta->version()) {
991
0
            if (rs->rowset_id() != rs_meta->rowset_id()) {
992
0
                return Status::Error<PUSH_VERSION_ALREADY_EXIST>(
993
0
                        "version already exist. rowset_id={}, version={}, tablet={}",
994
0
                        rs->rowset_id().to_string(), rs->version().to_string(), tablet_id());
995
0
            } else {
996
                // rowsetid,version is equal, it is a duplicate req, skip it
997
0
                return Status::OK();
998
0
            }
999
0
        }
1000
690k
    }
1001
14.1k
    _rs_metas.emplace(rs_meta->version(), rs_meta);
1002
14.1k
    return Status::OK();
1003
14.1k
}
1004
1005
247
void TabletMeta::add_rowsets_unchecked(const std::vector<RowsetSharedPtr>& to_add) {
1006
732
    for (const auto& rs : to_add) {
1007
732
        _rs_metas.emplace(rs->rowset_meta()->version(), rs->rowset_meta());
1008
732
    }
1009
247
}
1010
1011
void TabletMeta::delete_rs_meta_by_version(const Version& version,
1012
0
                                           std::vector<RowsetMetaSharedPtr>* deleted_rs_metas) {
1013
0
    size_t rowset_cache_version_size = 0;
1014
0
    if (auto it = _rs_metas.find(version); it != _rs_metas.end()) {
1015
0
        if (deleted_rs_metas != nullptr) {
1016
0
            deleted_rs_metas->push_back(it->second);
1017
0
        }
1018
0
        auto rowset_id = it->second->rowset_id();
1019
0
        _rs_metas.erase(it);
1020
0
        if (_enable_unique_key_merge_on_write) {
1021
0
            rowset_cache_version_size = _delete_bitmap->remove_rowset_cache_version(rowset_id);
1022
0
        }
1023
0
        return;
1024
0
    }
1025
0
    _check_mow_rowset_cache_version_size(rowset_cache_version_size);
1026
0
}
1027
1028
void TabletMeta::modify_rs_metas(const std::vector<RowsetMetaSharedPtr>& to_add,
1029
                                 const std::vector<RowsetMetaSharedPtr>& to_delete,
1030
122
                                 bool same_version) {
1031
122
    size_t rowset_cache_version_size = 0;
1032
    // Remove to_delete rowsets from _rs_metas
1033
581
    for (auto rs_to_del : to_delete) {
1034
581
        if (auto it = _rs_metas.find(rs_to_del->version()); it != _rs_metas.end()) {
1035
581
            auto rowset_id = it->second->rowset_id();
1036
581
            _rs_metas.erase(it);
1037
581
            if (_enable_unique_key_merge_on_write) {
1038
180
                rowset_cache_version_size = _delete_bitmap->remove_rowset_cache_version(rowset_id);
1039
180
            }
1040
581
        }
1041
581
    }
1042
122
    if (!same_version) {
1043
        // put to_delete rowsets in _stale_rs_metas.
1044
560
        for (auto rs_to_del : to_delete) {
1045
560
            _stale_rs_metas.emplace(rs_to_del->version(), rs_to_del);
1046
560
        }
1047
101
    }
1048
1049
    // put to_add rowsets in _rs_metas.
1050
122
    for (auto rs_to_add : to_add) {
1051
23
        _rs_metas.emplace(rs_to_add->version(), rs_to_add);
1052
23
    }
1053
122
    _check_mow_rowset_cache_version_size(rowset_cache_version_size);
1054
122
}
1055
1056
// Use the passing "rs_metas" to replace the rs meta in this tablet meta
1057
// Also clear the _stale_rs_metas because this tablet meta maybe copyied from
1058
// an existing tablet before. Add after revise, only the passing "rs_metas"
1059
// is needed.
1060
5
void TabletMeta::revise_rs_metas(std::vector<RowsetMetaSharedPtr>&& rs_metas) {
1061
5
    {
1062
5
        std::lock_guard<std::shared_mutex> wrlock(_meta_lock);
1063
5
        _rs_metas.clear();
1064
10
        for (auto& rs_meta : rs_metas) {
1065
10
            _rs_metas.emplace(rs_meta->version(), rs_meta);
1066
10
        }
1067
5
        _stale_rs_metas.clear();
1068
5
    }
1069
5
    if (_enable_unique_key_merge_on_write) {
1070
0
        _delete_bitmap->clear_rowset_cache_version();
1071
0
    }
1072
5
}
1073
1074
// This method should call after revise_rs_metas, since new rs_metas might be a subset
1075
// of original tablet, we should revise the delete_bitmap according to current rowset.
1076
//
1077
// Delete bitmap is protected by Tablet::_meta_lock, we don't need to acquire the
1078
// TabletMeta's _meta_lock
1079
1
void TabletMeta::revise_delete_bitmap_unlocked(const DeleteBitmap& delete_bitmap) {
1080
1
    _delete_bitmap = std::make_unique<DeleteBitmap>(tablet_id());
1081
2
    for (const auto& [_, rs] : _rs_metas) {
1082
2
        DeleteBitmap rs_bm(tablet_id());
1083
2
        delete_bitmap.subset({rs->rowset_id(), 0, 0}, {rs->rowset_id(), UINT32_MAX, INT64_MAX},
1084
2
                             &rs_bm);
1085
2
        _delete_bitmap->merge(rs_bm);
1086
2
    }
1087
1
    for (const auto& [_, rs] : _stale_rs_metas) {
1088
0
        DeleteBitmap rs_bm(tablet_id());
1089
0
        delete_bitmap.subset({rs->rowset_id(), 0, 0}, {rs->rowset_id(), UINT32_MAX, INT64_MAX},
1090
0
                             &rs_bm);
1091
0
        _delete_bitmap->merge(rs_bm);
1092
0
    }
1093
1
}
1094
1095
0
void TabletMeta::delete_stale_rs_meta_by_version(const Version& version) {
1096
0
    _stale_rs_metas.erase(version);
1097
0
}
1098
1099
0
RowsetMetaSharedPtr TabletMeta::acquire_rs_meta_by_version(const Version& version) const {
1100
0
    if (auto it = _rs_metas.find(version); it != _rs_metas.end()) {
1101
0
        return it->second;
1102
0
    }
1103
0
    return nullptr;
1104
0
}
1105
1106
8
RowsetMetaSharedPtr TabletMeta::acquire_stale_rs_meta_by_version(const Version& version) const {
1107
8
    if (auto it = _stale_rs_metas.find(version); it != _stale_rs_metas.end()) {
1108
0
        return it->second;
1109
0
    }
1110
8
    return nullptr;
1111
8
}
1112
1113
23
Status TabletMeta::set_partition_id(int64_t partition_id) {
1114
23
    if ((_partition_id > 0 && _partition_id != partition_id) || partition_id < 1) {
1115
0
        LOG(WARNING) << "cur partition id=" << _partition_id << " new partition id=" << partition_id
1116
0
                     << " not equal";
1117
0
    }
1118
23
    _partition_id = partition_id;
1119
23
    return Status::OK();
1120
23
}
1121
1122
0
void TabletMeta::clear_stale_rowset() {
1123
0
    _stale_rs_metas.clear();
1124
0
    if (_enable_unique_key_merge_on_write) {
1125
0
        _delete_bitmap->clear_rowset_cache_version();
1126
0
    }
1127
0
}
1128
1129
0
void TabletMeta::clear_rowsets() {
1130
0
    _rs_metas.clear();
1131
0
    if (_enable_unique_key_merge_on_write) {
1132
0
        _delete_bitmap->clear_rowset_cache_version();
1133
0
    }
1134
0
}
1135
1136
122
void TabletMeta::_check_mow_rowset_cache_version_size(size_t rowset_cache_version_size) {
1137
122
    if (_enable_unique_key_merge_on_write && config::enable_mow_verbose_log &&
1138
122
        rowset_cache_version_size > _rs_metas.size() + _stale_rs_metas.size()) {
1139
0
        std::stringstream ss;
1140
0
        auto rowset_ids = _delete_bitmap->get_rowset_cache_version();
1141
0
        std::set<std::string> tablet_rowset_ids;
1142
0
        {
1143
0
            std::shared_lock rlock(_meta_lock);
1144
0
            for (const auto& [_, rs_meta] : _rs_metas) {
1145
0
                tablet_rowset_ids.emplace(rs_meta->rowset_id().to_string());
1146
0
            }
1147
0
            for (const auto& [_, rs_meta] : _stale_rs_metas) {
1148
0
                tablet_rowset_ids.emplace(rs_meta->rowset_id().to_string());
1149
0
            }
1150
0
        }
1151
0
        for (const auto& rowset_id : rowset_ids) {
1152
0
            if (tablet_rowset_ids.find(rowset_id) == tablet_rowset_ids.end()) {
1153
0
                ss << rowset_id << ", ";
1154
0
            }
1155
0
        }
1156
        // size(rowset_cache_version) <= size(_rs_metas) + size(_stale_rs_metas) + size(_unused_rs)
1157
0
        std::string msg = fmt::format(
1158
0
                "tablet: {}, rowset_cache_version size: {}, "
1159
0
                "_rs_metas size: {}, _stale_rs_metas size: {}, delta: {}. rowset only in cache: {}",
1160
0
                _tablet_id, rowset_cache_version_size, _rs_metas.size(), _stale_rs_metas.size(),
1161
0
                rowset_cache_version_size - _rs_metas.size() - _stale_rs_metas.size(), ss.str());
1162
0
        LOG(INFO) << msg;
1163
0
    }
1164
122
}
1165
1166
3
bool operator==(const TabletMeta& a, const TabletMeta& b) {
1167
3
    if (a._table_id != b._table_id) return false;
1168
3
    if (a._index_id != b._index_id) return false;
1169
3
    if (a._partition_id != b._partition_id) return false;
1170
3
    if (a._tablet_id != b._tablet_id) return false;
1171
3
    if (a._replica_id != b._replica_id) return false;
1172
3
    if (a._schema_hash != b._schema_hash) return false;
1173
3
    if (a._shard_id != b._shard_id) return false;
1174
3
    if (a._creation_time != b._creation_time) return false;
1175
3
    if (a._cumulative_layer_point != b._cumulative_layer_point) return false;
1176
3
    if (a._tablet_uid != b._tablet_uid) return false;
1177
3
    if (a._tablet_type != b._tablet_type) return false;
1178
3
    if (a._tablet_state != b._tablet_state) return false;
1179
3
    if (*a._schema != *b._schema) return false;
1180
3
    if (a._rs_metas != b._rs_metas) return false;
1181
3
    if (a._in_restore_mode != b._in_restore_mode) return false;
1182
3
    if (a._preferred_rowset_type != b._preferred_rowset_type) return false;
1183
3
    if (a._storage_policy_id != b._storage_policy_id) return false;
1184
3
    if (a._compaction_policy != b._compaction_policy) return false;
1185
3
    if (a._time_series_compaction_goal_size_mbytes != b._time_series_compaction_goal_size_mbytes)
1186
0
        return false;
1187
3
    if (a._time_series_compaction_file_count_threshold !=
1188
3
        b._time_series_compaction_file_count_threshold)
1189
0
        return false;
1190
3
    if (a._time_series_compaction_time_threshold_seconds !=
1191
3
        b._time_series_compaction_time_threshold_seconds)
1192
0
        return false;
1193
3
    if (a._time_series_compaction_empty_rowsets_threshold !=
1194
3
        b._time_series_compaction_empty_rowsets_threshold)
1195
0
        return false;
1196
3
    if (a._time_series_compaction_level_threshold != b._time_series_compaction_level_threshold)
1197
0
        return false;
1198
3
    return true;
1199
3
}
1200
1201
0
bool operator!=(const TabletMeta& a, const TabletMeta& b) {
1202
0
    return !(a == b);
1203
0
}
1204
1205
// We cannot just copy the underlying memory to construct a string
1206
// due to equivalent objects may have different padding bytes.
1207
// Reading padding bytes is undefined behavior, neither copy nor
1208
// placement new will help simplify the code.
1209
// Refer to C11 standards §6.2.6.1/6 and §6.7.9/21 for more info.
1210
64
static std::string agg_cache_key(int64_t tablet_id, const DeleteBitmap::BitmapKey& bmk) {
1211
64
    std::string ret(sizeof(tablet_id) + sizeof(bmk), '\0');
1212
64
    *reinterpret_cast<int64_t*>(ret.data()) = tablet_id;
1213
64
    auto t = reinterpret_cast<DeleteBitmap::BitmapKey*>(ret.data() + sizeof(tablet_id));
1214
64
    std::get<RowsetId>(*t).version = std::get<RowsetId>(bmk).version;
1215
64
    std::get<RowsetId>(*t).hi = std::get<RowsetId>(bmk).hi;
1216
64
    std::get<RowsetId>(*t).mi = std::get<RowsetId>(bmk).mi;
1217
64
    std::get<RowsetId>(*t).lo = std::get<RowsetId>(bmk).lo;
1218
64
    std::get<1>(*t) = std::get<1>(bmk);
1219
64
    std::get<2>(*t) = std::get<2>(bmk);
1220
64
    return ret;
1221
64
}
1222
1223
// decode cache key info from a agg_cache_key
1224
static void decode_agg_cache_key(const std::string& key_str, int64_t& tablet_id,
1225
0
                                 DeleteBitmap::BitmapKey& bmk) {
1226
0
    const char* ptr = key_str.data();
1227
0
    tablet_id = *reinterpret_cast<const int64_t*>(ptr);
1228
0
    ptr += sizeof(tablet_id);
1229
0
    const auto* t = reinterpret_cast<const DeleteBitmap::BitmapKey*>(ptr);
1230
0
    std::get<RowsetId>(bmk).version = std::get<RowsetId>(*t).version;
1231
0
    std::get<RowsetId>(bmk).hi = std::get<RowsetId>(*t).hi;
1232
0
    std::get<RowsetId>(bmk).mi = std::get<RowsetId>(*t).mi;
1233
0
    std::get<RowsetId>(bmk).lo = std::get<RowsetId>(*t).lo;
1234
0
    std::get<1>(bmk) = std::get<1>(*t);
1235
0
    std::get<2>(bmk) = std::get<2>(*t);
1236
0
}
1237
1238
DeleteBitmapAggCache::DeleteBitmapAggCache(size_t capacity)
1239
1
        : LRUCachePolicy(CachePolicy::CacheType::DELETE_BITMAP_AGG_CACHE, capacity,
1240
1
                         LRUCacheType::SIZE, config::delete_bitmap_agg_cache_stale_sweep_time_sec,
1241
1
                         /*num_shards*/ 256,
1242
1
                         /*element_count_capacity*/ 0, /*enable_prune*/ true,
1243
1
                         /*is_lru_k*/ false) {}
1244
1245
190
DeleteBitmapAggCache* DeleteBitmapAggCache::instance() {
1246
190
    return ExecEnv::GetInstance()->delete_bitmap_agg_cache();
1247
190
}
1248
1249
1
DeleteBitmapAggCache* DeleteBitmapAggCache::create_instance(size_t capacity) {
1250
1
    return new DeleteBitmapAggCache(capacity);
1251
1
}
1252
1253
0
DeleteBitmap DeleteBitmapAggCache::snapshot(int64_t tablet_id) {
1254
0
    DeleteBitmap ret(tablet_id);
1255
0
    auto collector = [&](const LRUHandle* handle) {
1256
0
        auto key = handle->key().to_string();
1257
0
        int64_t key_tablet_id;
1258
0
        DeleteBitmap::BitmapKey bmk;
1259
0
        decode_agg_cache_key(key, key_tablet_id, bmk);
1260
0
        if (key_tablet_id == tablet_id) {
1261
0
            const auto& dbm = reinterpret_cast<DeleteBitmapAggCache::Value*>(handle->value)->bitmap;
1262
0
            ret.set(bmk, dbm);
1263
0
        }
1264
0
    };
1265
0
    DeleteBitmapAggCache::instance()->for_each_entry(collector);
1266
0
    return ret;
1267
0
}
1268
1269
1.22k
DeleteBitmap::DeleteBitmap(int64_t tablet_id) : _tablet_id(tablet_id) {}
1270
1271
7
DeleteBitmap::DeleteBitmap(const DeleteBitmap& o) {
1272
7
    std::shared_lock l1(o.lock);
1273
7
    delete_bitmap = o.delete_bitmap;
1274
7
    _tablet_id = o._tablet_id;
1275
7
}
1276
1277
0
DeleteBitmap& DeleteBitmap::operator=(const DeleteBitmap& o) {
1278
0
    if (this == &o) return *this;
1279
0
    if (this < &o) {
1280
0
        std::unique_lock l1(lock);
1281
0
        std::shared_lock l2(o.lock);
1282
0
        delete_bitmap = o.delete_bitmap;
1283
0
        _tablet_id = o._tablet_id;
1284
0
    } else {
1285
0
        std::shared_lock l2(o.lock);
1286
0
        std::unique_lock l1(lock);
1287
0
        delete_bitmap = o.delete_bitmap;
1288
0
        _tablet_id = o._tablet_id;
1289
0
    }
1290
0
    return *this;
1291
0
}
1292
1293
0
DeleteBitmap::DeleteBitmap(DeleteBitmap&& o) noexcept {
1294
0
    std::scoped_lock l(o.lock, o._rowset_cache_version_lock);
1295
0
    delete_bitmap = std::move(o.delete_bitmap);
1296
0
    _tablet_id = std::move(o._tablet_id);
1297
0
    o._rowset_cache_version.clear();
1298
0
}
1299
1300
0
DeleteBitmap& DeleteBitmap::operator=(DeleteBitmap&& o) noexcept {
1301
0
    if (this == &o) return *this;
1302
0
    std::scoped_lock l(lock, o.lock, o._rowset_cache_version_lock);
1303
0
    delete_bitmap = std::move(o.delete_bitmap);
1304
0
    _tablet_id = std::move(o._tablet_id);
1305
0
    o._rowset_cache_version.clear();
1306
0
    return *this;
1307
0
}
1308
1309
0
DeleteBitmap DeleteBitmap::from_pb(const DeleteBitmapPB& pb, int64_t tablet_id) {
1310
0
    size_t len = pb.rowset_ids().size();
1311
0
    DCHECK_EQ(len, pb.segment_ids().size());
1312
0
    DCHECK_EQ(len, pb.versions().size());
1313
0
    DeleteBitmap delete_bitmap(tablet_id);
1314
0
    for (int32_t i = 0; i < len; ++i) {
1315
0
        RowsetId rs_id;
1316
0
        rs_id.init(pb.rowset_ids(i));
1317
0
        BitmapKey key = {rs_id, pb.segment_ids(i), pb.versions(i)};
1318
0
        delete_bitmap.delete_bitmap[key] =
1319
0
                roaring::Roaring::read(pb.segment_delete_bitmaps(i).data());
1320
0
    }
1321
0
    return delete_bitmap;
1322
0
}
1323
1324
0
DeleteBitmapPB DeleteBitmap::to_pb() {
1325
0
    std::shared_lock l(lock);
1326
0
    DeleteBitmapPB ret;
1327
0
    for (const auto& [k, v] : delete_bitmap) {
1328
0
        ret.mutable_rowset_ids()->Add(std::get<0>(k).to_string());
1329
0
        ret.mutable_segment_ids()->Add(std::get<1>(k));
1330
0
        ret.mutable_versions()->Add(std::get<2>(k));
1331
0
        std::string bitmap_data(v.getSizeInBytes(), '\0');
1332
0
        v.write(bitmap_data.data());
1333
0
        ret.mutable_segment_delete_bitmaps()->Add(std::move(bitmap_data));
1334
0
    }
1335
0
    return ret;
1336
0
}
1337
1338
7
DeleteBitmap DeleteBitmap::snapshot() const {
1339
7
    std::shared_lock l(lock);
1340
7
    return DeleteBitmap(*this);
1341
7
}
1342
1343
3
DeleteBitmap DeleteBitmap::snapshot(Version version) const {
1344
    // Take snapshot first, then remove keys greater than given version.
1345
3
    DeleteBitmap snapshot = this->snapshot();
1346
3
    auto it = snapshot.delete_bitmap.begin();
1347
412
    while (it != snapshot.delete_bitmap.end()) {
1348
409
        if (std::get<2>(it->first) > version) {
1349
4
            it = snapshot.delete_bitmap.erase(it);
1350
405
        } else {
1351
405
            it++;
1352
405
        }
1353
409
    }
1354
3
    return snapshot;
1355
3
}
1356
1357
463k
void DeleteBitmap::add(const BitmapKey& bmk, uint32_t row_id) {
1358
463k
    std::lock_guard l(lock);
1359
463k
    delete_bitmap[bmk].add(row_id);
1360
463k
}
1361
1362
0
int DeleteBitmap::remove(const BitmapKey& bmk, uint32_t row_id) {
1363
0
    std::lock_guard l(lock);
1364
0
    auto it = delete_bitmap.find(bmk);
1365
0
    if (it == delete_bitmap.end()) return -1;
1366
0
    it->second.remove(row_id);
1367
0
    return 0;
1368
0
}
1369
1370
8
void DeleteBitmap::remove(const BitmapKey& start, const BitmapKey& end) {
1371
8
    std::lock_guard l(lock);
1372
107
    for (auto it = delete_bitmap.lower_bound(start); it != delete_bitmap.end();) {
1373
101
        auto& [k, _] = *it;
1374
101
        if (k >= end) {
1375
2
            break;
1376
2
        }
1377
99
        it = delete_bitmap.erase(it);
1378
99
    }
1379
8
}
1380
1381
0
void DeleteBitmap::remove(const std::vector<std::tuple<BitmapKey, BitmapKey>>& key_ranges) {
1382
0
    std::lock_guard l(lock);
1383
0
    for (auto& [start, end] : key_ranges) {
1384
0
        for (auto it = delete_bitmap.lower_bound(start); it != delete_bitmap.end();) {
1385
0
            auto& [k, _] = *it;
1386
0
            if (k >= end) {
1387
0
                break;
1388
0
            }
1389
0
            it = delete_bitmap.erase(it);
1390
0
        }
1391
0
    }
1392
0
}
1393
1394
6
bool DeleteBitmap::contains(const BitmapKey& bmk, uint32_t row_id) const {
1395
6
    std::shared_lock l(lock);
1396
6
    auto it = delete_bitmap.find(bmk);
1397
6
    return it != delete_bitmap.end() && it->second.contains(row_id);
1398
6
}
1399
1400
2
bool DeleteBitmap::contains_agg(const BitmapKey& bmk, uint32_t row_id) const {
1401
2
    return get_agg(bmk)->contains(row_id);
1402
2
}
1403
1404
0
bool DeleteBitmap::empty() const {
1405
0
    std::shared_lock l(lock);
1406
0
    return delete_bitmap.empty();
1407
0
}
1408
1409
63
uint64_t DeleteBitmap::cardinality() const {
1410
63
    std::shared_lock l(lock);
1411
63
    uint64_t res = 0;
1412
320
    for (auto entry : delete_bitmap) {
1413
320
        if (std::get<1>(entry.first) != DeleteBitmap::INVALID_SEGMENT_ID) {
1414
320
            res += entry.second.cardinality();
1415
320
        }
1416
320
    }
1417
63
    return res;
1418
63
}
1419
1420
0
uint64_t DeleteBitmap::get_size() const {
1421
0
    std::shared_lock l(lock);
1422
0
    uint64_t charge = 0;
1423
0
    for (auto& [k, v] : delete_bitmap) {
1424
0
        if (std::get<1>(k) != DeleteBitmap::INVALID_SEGMENT_ID) {
1425
0
            charge += v.getSizeInBytes();
1426
0
        }
1427
0
    }
1428
0
    return charge;
1429
0
}
1430
1431
bool DeleteBitmap::contains_agg_with_cache_if_eligible(const BitmapKey& bmk,
1432
1
                                                       uint32_t row_id) const {
1433
1
    g_contains_agg_with_cache_if_eligible_total << 1;
1434
1
    int64_t start_version {0};
1435
1
    if (config::enable_mow_get_agg_by_cache) {
1436
1
        auto deleter = [&](Cache::Handle* handle) {
1437
0
            DeleteBitmapAggCache::instance()->release(handle);
1438
0
        };
1439
1
        std::unique_ptr<Cache::Handle, decltype(deleter)> dbm_handle(nullptr, deleter);
1440
1
        int64_t cached_version = 0;
1441
        // 1. try to lookup the desired key directly
1442
1
        dbm_handle.reset(DeleteBitmapAggCache::instance()->lookup(agg_cache_key(_tablet_id, bmk)));
1443
1
        if (dbm_handle != nullptr) {
1444
0
            cached_version = std::get<2>(bmk);
1445
1
        } else {
1446
            // 2. if not found, try to lookup with cached version
1447
1
            cached_version = _get_rowset_cache_version(bmk);
1448
1
            if (cached_version > 0) {
1449
0
                if (cached_version > std::get<2>(bmk)) {
1450
0
                    cached_version = 0;
1451
0
                } else {
1452
0
                    dbm_handle.reset(DeleteBitmapAggCache::instance()->lookup(agg_cache_key(
1453
0
                            _tablet_id, {std::get<0>(bmk), std::get<1>(bmk), cached_version})));
1454
0
                }
1455
0
            }
1456
1
        }
1457
1
        if (dbm_handle != nullptr) {
1458
0
            const auto& cached_dbm =
1459
0
                    reinterpret_cast<DeleteBitmapAggCache::Value*>(
1460
0
                            DeleteBitmapAggCache::instance()->value(dbm_handle.get()))
1461
0
                            ->bitmap;
1462
0
            if (cached_version == std::get<2>(bmk)) {
1463
0
                g_contains_agg_with_cache_if_eligible_full_hit << 1;
1464
0
            } else {
1465
0
                g_contains_agg_with_cache_if_eligible_partial_hit << 1;
1466
0
            }
1467
0
            if (cached_dbm.contains(row_id)) {
1468
0
                return true;
1469
0
            }
1470
0
            if (cached_version == std::get<2>(bmk)) {
1471
0
                return false;
1472
0
            }
1473
0
            start_version = cached_version + 1;
1474
0
        }
1475
1
    }
1476
1
    DeleteBitmap::BitmapKey start {std::get<0>(bmk), std::get<1>(bmk), start_version};
1477
1
    std::shared_lock l(lock);
1478
1
    for (auto it = delete_bitmap.lower_bound(start); it != delete_bitmap.end(); ++it) {
1479
0
        auto& [k, bm] = *it;
1480
0
        if (std::get<0>(k) != std::get<0>(bmk) || std::get<1>(k) != std::get<1>(bmk) ||
1481
0
            std::get<2>(k) > std::get<2>(bmk)) {
1482
0
            break;
1483
0
        }
1484
0
        if (bm.contains(row_id)) {
1485
0
            return true;
1486
0
        }
1487
0
    }
1488
1
    return false;
1489
1
}
1490
1491
0
void DeleteBitmap::remove_sentinel_marks() {
1492
0
    std::lock_guard l(lock);
1493
0
    for (auto it = delete_bitmap.begin(), end = delete_bitmap.end(); it != end;) {
1494
0
        if (std::get<1>(it->first) == DeleteBitmap::INVALID_SEGMENT_ID) {
1495
0
            it = delete_bitmap.erase(it);
1496
0
        } else {
1497
0
            ++it;
1498
0
        }
1499
0
    }
1500
0
}
1501
1502
38
int DeleteBitmap::set(const BitmapKey& bmk, const roaring::Roaring& segment_delete_bitmap) {
1503
38
    std::lock_guard l(lock);
1504
38
    auto [_, inserted] = delete_bitmap.insert_or_assign(bmk, segment_delete_bitmap);
1505
38
    return inserted;
1506
38
}
1507
1508
7
int DeleteBitmap::get(const BitmapKey& bmk, roaring::Roaring* segment_delete_bitmap) const {
1509
7
    std::shared_lock l(lock);
1510
7
    auto it = delete_bitmap.find(bmk);
1511
7
    if (it == delete_bitmap.end()) return -1;
1512
7
    *segment_delete_bitmap = it->second; // copy
1513
7
    return 0;
1514
7
}
1515
1516
54
const roaring::Roaring* DeleteBitmap::get(const BitmapKey& bmk) const {
1517
54
    std::shared_lock l(lock);
1518
54
    auto it = delete_bitmap.find(bmk);
1519
54
    if (it == delete_bitmap.end()) return nullptr;
1520
41
    return &(it->second); // get address
1521
54
}
1522
1523
void DeleteBitmap::subset(const BitmapKey& start, const BitmapKey& end,
1524
3
                          DeleteBitmap* subset_rowset_map) const {
1525
3
    DCHECK(start < end);
1526
3
    std::shared_lock l(lock);
1527
26
    for (auto it = delete_bitmap.lower_bound(start); it != delete_bitmap.end(); ++it) {
1528
25
        auto& [k, bm] = *it;
1529
25
        if (k >= end) {
1530
2
            break;
1531
2
        }
1532
23
        subset_rowset_map->set(k, bm);
1533
23
    }
1534
3
}
1535
1536
void DeleteBitmap::subset(std::vector<std::pair<RowsetId, int64_t>>& rowset_ids,
1537
                          int64_t start_version, int64_t end_version,
1538
0
                          DeleteBitmap* subset_delete_map) const {
1539
0
    DCHECK(start_version <= end_version);
1540
0
    for (auto& [rowset_id, _] : rowset_ids) {
1541
0
        BitmapKey start {rowset_id, 0, 0};
1542
0
        BitmapKey end {rowset_id, UINT32_MAX, end_version + 1};
1543
0
        std::shared_lock l(lock);
1544
0
        for (auto it = delete_bitmap.lower_bound(start); it != delete_bitmap.end(); ++it) {
1545
0
            auto& [k, bm] = *it;
1546
0
            if (k >= end) {
1547
0
                break;
1548
0
            }
1549
0
            auto version = std::get<2>(k);
1550
0
            if (version >= start_version && version <= end_version) {
1551
0
                subset_delete_map->merge(k, bm);
1552
0
                VLOG_DEBUG << "subset delete bitmap, tablet=" << _tablet_id << ", version=["
1553
0
                           << start_version << ", " << end_version
1554
0
                           << "]. rowset=" << std::get<0>(k).to_string()
1555
0
                           << ", segment=" << std::get<1>(k) << ", version=" << version
1556
0
                           << ", cardinality=" << bm.cardinality();
1557
0
            }
1558
0
        }
1559
0
    }
1560
0
}
1561
1562
void DeleteBitmap::subset_and_agg(std::vector<std::pair<RowsetId, int64_t>>& rowset_ids,
1563
                                  int64_t start_version, int64_t end_version,
1564
1
                                  DeleteBitmap* subset_delete_map) const {
1565
1
    DCHECK(start_version <= end_version);
1566
2
    for (auto& [rowset_id, segment_num] : rowset_ids) {
1567
6
        for (int64_t seg_id = 0; seg_id < segment_num; ++seg_id) {
1568
4
            BitmapKey end {rowset_id, seg_id, end_version};
1569
4
            auto bm = get_agg_without_cache(end, start_version);
1570
4
            VLOG_DEBUG << "subset delete bitmap, tablet=" << _tablet_id << ", rowset=" << rowset_id
1571
0
                       << ", segment=" << seg_id << ", version=[" << start_version << "-"
1572
0
                       << end_version << "], cardinality=" << bm->cardinality();
1573
4
            if (bm->isEmpty()) {
1574
0
                continue;
1575
0
            }
1576
4
            subset_delete_map->merge(end, *bm);
1577
4
        }
1578
2
    }
1579
1
}
1580
1581
0
size_t DeleteBitmap::get_count_with_range(const BitmapKey& start, const BitmapKey& end) const {
1582
0
    DCHECK(start < end);
1583
0
    size_t count = 0;
1584
0
    std::shared_lock l(lock);
1585
0
    for (auto it = delete_bitmap.lower_bound(start); it != delete_bitmap.end(); ++it) {
1586
0
        auto& [k, bm] = *it;
1587
0
        if (k >= end) {
1588
0
            break;
1589
0
        }
1590
0
        count++;
1591
0
    }
1592
0
    return count;
1593
0
}
1594
1595
6
void DeleteBitmap::merge(const BitmapKey& bmk, const roaring::Roaring& segment_delete_bitmap) {
1596
6
    std::lock_guard l(lock);
1597
6
    auto [iter, succ] = delete_bitmap.emplace(bmk, segment_delete_bitmap);
1598
6
    if (!succ) {
1599
0
        iter->second |= segment_delete_bitmap;
1600
0
    }
1601
6
}
1602
1603
9
void DeleteBitmap::merge(const DeleteBitmap& other) {
1604
9
    std::lock_guard l(lock);
1605
29
    for (auto& i : other.delete_bitmap) {
1606
29
        auto [j, succ] = this->delete_bitmap.insert(i);
1607
29
        if (!succ) j->second |= i.second;
1608
29
    }
1609
9
}
1610
1611
63
uint64_t DeleteBitmap::get_delete_bitmap_count() {
1612
63
    std::shared_lock l(lock);
1613
63
    uint64_t count = 0;
1614
383
    for (auto it = delete_bitmap.begin(); it != delete_bitmap.end(); it++) {
1615
320
        if (std::get<1>(it->first) != DeleteBitmap::INVALID_SEGMENT_ID) {
1616
320
            count++;
1617
320
        }
1618
320
    }
1619
63
    return count;
1620
63
}
1621
1622
void DeleteBitmap::traverse_rowset_and_version(
1623
0
        const std::function<int(const RowsetId& rowsetId, int64_t version)>& func) const {
1624
0
    std::shared_lock l(lock);
1625
0
    auto it = delete_bitmap.cbegin();
1626
0
    while (it != delete_bitmap.cend()) {
1627
0
        RowsetId rowset_id = std::get<0>(it->first);
1628
0
        int64_t version = std::get<2>(it->first);
1629
0
        int result = func(rowset_id, version);
1630
0
        if (result == -2) {
1631
            // find next <rowset, version>
1632
0
            it++;
1633
0
        } else {
1634
            // find next <rowset>
1635
0
            it = delete_bitmap.upper_bound({rowset_id, std::numeric_limits<SegmentId>::max(),
1636
0
                                            std::numeric_limits<Version>::max()});
1637
0
        }
1638
0
    }
1639
0
}
1640
1641
0
bool DeleteBitmap::has_calculated_for_multi_segments(const RowsetId& rowset_id) const {
1642
0
    return contains({rowset_id, INVALID_SEGMENT_ID, TEMP_VERSION_COMMON}, ROWSET_SENTINEL_MARK);
1643
0
}
1644
1645
180
size_t DeleteBitmap::remove_rowset_cache_version(const RowsetId& rowset_id) {
1646
180
    std::lock_guard l(_rowset_cache_version_lock);
1647
180
    _rowset_cache_version.erase(rowset_id);
1648
180
    VLOG_DEBUG << "remove agg cache version for tablet=" << _tablet_id
1649
0
               << ", rowset=" << rowset_id.to_string();
1650
180
    return _rowset_cache_version.size();
1651
180
}
1652
1653
0
void DeleteBitmap::clear_rowset_cache_version() {
1654
0
    std::lock_guard l(_rowset_cache_version_lock);
1655
0
    _rowset_cache_version.clear();
1656
0
    VLOG_DEBUG << "clear agg cache version for tablet=" << _tablet_id;
1657
0
}
1658
1659
0
std::set<std::string> DeleteBitmap::get_rowset_cache_version() {
1660
0
    std::set<std::string> set;
1661
0
    std::shared_lock l(_rowset_cache_version_lock);
1662
0
    for (auto& [k, _] : _rowset_cache_version) {
1663
0
        set.insert(k.to_string());
1664
0
    }
1665
0
    return set;
1666
0
}
1667
1668
49
DeleteBitmap::Version DeleteBitmap::_get_rowset_cache_version(const BitmapKey& bmk) const {
1669
49
    std::shared_lock l(_rowset_cache_version_lock);
1670
49
    if (auto it = _rowset_cache_version.find(std::get<0>(bmk)); it != _rowset_cache_version.end()) {
1671
39
        auto& segment_cache_version = it->second;
1672
39
        if (auto it1 = segment_cache_version.find(std::get<1>(bmk));
1673
39
            it1 != segment_cache_version.end()) {
1674
10
            return it1->second;
1675
10
        }
1676
39
    }
1677
39
    return 0;
1678
49
}
1679
1680
0
DeleteBitmap DeleteBitmap::agg_cache_snapshot() {
1681
0
    return DeleteBitmapAggCache::instance()->snapshot(_tablet_id);
1682
0
}
1683
1684
1.10k
void DeleteBitmap::set_tablet_id(int64_t tablet_id) {
1685
1.10k
    _tablet_id = tablet_id;
1686
1.10k
}
1687
1688
54
std::shared_ptr<roaring::Roaring> DeleteBitmap::get_agg(const BitmapKey& bmk) const {
1689
54
    std::string key_str = agg_cache_key(_tablet_id, bmk); // Cache key container
1690
54
    CacheKey key(key_str);
1691
54
    Cache::Handle* handle = DeleteBitmapAggCache::instance()->lookup(key);
1692
1693
54
    DeleteBitmapAggCache::Value* val =
1694
54
            handle == nullptr ? nullptr
1695
54
                              : reinterpret_cast<DeleteBitmapAggCache::Value*>(
1696
6
                                        DeleteBitmapAggCache::instance()->value(handle));
1697
    // FIXME: do we need a mutex here to get rid of duplicated initializations
1698
    //        of cache entries in some cases?
1699
54
    if (val == nullptr) { // Renew if needed, put a new Value to cache
1700
48
        val = new DeleteBitmapAggCache::Value();
1701
48
        Version start_version =
1702
48
                config::enable_mow_get_agg_by_cache ? _get_rowset_cache_version(bmk) : 0;
1703
48
        if (start_version > 0) {
1704
9
            Cache::Handle* handle2 = DeleteBitmapAggCache::instance()->lookup(
1705
9
                    agg_cache_key(_tablet_id, {std::get<0>(bmk), std::get<1>(bmk), start_version}));
1706
1707
9
            DBUG_EXECUTE_IF("DeleteBitmap::get_agg.cache_miss", {
1708
9
                if (handle2 != nullptr) {
1709
9
                    auto p = dp->param("percent", 0.3);
1710
9
                    std::mt19937 gen {std::random_device {}()};
1711
9
                    std::bernoulli_distribution inject_fault {p};
1712
9
                    if (inject_fault(gen)) {
1713
9
                        LOG_INFO("injection DeleteBitmap::get_agg.cache_miss, tablet_id={}",
1714
9
                                 _tablet_id);
1715
9
                        handle2 = nullptr;
1716
9
                    }
1717
9
                }
1718
9
            });
1719
9
            if (handle2 == nullptr || start_version > std::get<2>(bmk)) {
1720
0
                start_version = 0;
1721
9
            } else {
1722
9
                val->bitmap |= reinterpret_cast<DeleteBitmapAggCache::Value*>(
1723
9
                                       DeleteBitmapAggCache::instance()->value(handle2))
1724
9
                                       ->bitmap;
1725
9
                VLOG_DEBUG << "get agg cache version=" << start_version
1726
0
                           << " for tablet=" << _tablet_id
1727
0
                           << ", rowset=" << std::get<0>(bmk).to_string()
1728
0
                           << ", segment=" << std::get<1>(bmk);
1729
9
                start_version += 1;
1730
9
            }
1731
9
            if (handle2 != nullptr) {
1732
9
                DeleteBitmapAggCache::instance()->release(handle2);
1733
9
            }
1734
9
        }
1735
48
        {
1736
48
            std::shared_lock l(lock);
1737
48
            DeleteBitmap::BitmapKey start {std::get<0>(bmk), std::get<1>(bmk), start_version};
1738
87
            for (auto it = delete_bitmap.lower_bound(start); it != delete_bitmap.end(); ++it) {
1739
84
                auto& [k, bm] = *it;
1740
84
                if (std::get<0>(k) != std::get<0>(bmk) || std::get<1>(k) != std::get<1>(bmk) ||
1741
84
                    std::get<2>(k) > std::get<2>(bmk)) {
1742
45
                    break;
1743
45
                }
1744
39
                val->bitmap |= bm;
1745
39
            }
1746
48
        }
1747
48
        size_t charge = val->bitmap.getSizeInBytes() + sizeof(DeleteBitmapAggCache::Value);
1748
48
        handle = DeleteBitmapAggCache::instance()->insert(key, val, charge, charge,
1749
48
                                                          CachePriority::NORMAL);
1750
48
        if (config::enable_mow_get_agg_by_cache && !val->bitmap.isEmpty()) {
1751
37
            std::lock_guard l(_rowset_cache_version_lock);
1752
            // this version is already agg
1753
37
            _rowset_cache_version[std::get<0>(bmk)][std::get<1>(bmk)] = std::get<2>(bmk);
1754
37
            VLOG_DEBUG << "set agg cache version=" << std::get<2>(bmk)
1755
0
                       << " for tablet=" << _tablet_id
1756
0
                       << ", rowset=" << std::get<0>(bmk).to_string()
1757
0
                       << ", segment=" << std::get<1>(bmk);
1758
37
        }
1759
48
        if (start_version > 0 && config::enable_mow_get_agg_correctness_check_core) {
1760
0
            std::shared_ptr<roaring::Roaring> bitmap = get_agg_without_cache(bmk);
1761
0
            if (val->bitmap != *bitmap) {
1762
0
                CHECK(false) << ". get agg correctness check failed for tablet=" << _tablet_id
1763
0
                             << ", rowset=" << std::get<0>(bmk).to_string()
1764
0
                             << ", segment=" << std::get<1>(bmk) << ", version=" << std::get<2>(bmk)
1765
0
                             << ". start_version from cache=" << start_version
1766
0
                             << ", delete_bitmap cardinality with cache="
1767
0
                             << val->bitmap.cardinality()
1768
0
                             << ", delete_bitmap cardinality without cache="
1769
0
                             << bitmap->cardinality();
1770
0
            }
1771
0
        }
1772
48
    }
1773
1774
    // It is natural for the cache to reclaim the underlying memory
1775
54
    return std::shared_ptr<roaring::Roaring>(
1776
54
            &val->bitmap, [handle](...) { DeleteBitmapAggCache::instance()->release(handle); });
1777
54
}
1778
1779
std::shared_ptr<roaring::Roaring> DeleteBitmap::get_agg_without_cache(
1780
4
        const BitmapKey& bmk, const int64_t start_version) const {
1781
4
    std::shared_ptr<roaring::Roaring> bitmap = std::make_shared<roaring::Roaring>();
1782
4
    std::shared_lock l(lock);
1783
4
    DeleteBitmap::BitmapKey start {std::get<0>(bmk), std::get<1>(bmk), start_version};
1784
24
    for (auto it = delete_bitmap.lower_bound(start); it != delete_bitmap.end(); ++it) {
1785
23
        auto& [k, bm] = *it;
1786
23
        if (std::get<0>(k) != std::get<0>(bmk) || std::get<1>(k) != std::get<1>(bmk) ||
1787
23
            std::get<2>(k) > std::get<2>(bmk)) {
1788
3
            break;
1789
3
        }
1790
20
        *bitmap |= bm;
1791
20
    }
1792
4
    return bitmap;
1793
4
}
1794
1795
0
DeleteBitmap DeleteBitmap::diffset(const std::set<BitmapKey>& key_set) const {
1796
0
    std::shared_lock l(lock);
1797
0
    auto diff_key_set_view =
1798
0
            delete_bitmap | std::ranges::views::transform([](const auto& kv) { return kv.first; }) |
1799
0
            std::ranges::views::filter(
1800
0
                    [&key_set](const auto& key) { return !key_set.contains(key); });
1801
1802
0
    DeleteBitmap dbm(_tablet_id);
1803
0
    for (const auto& key : diff_key_set_view) {
1804
0
        const auto* bitmap = get(key);
1805
0
        DCHECK_NE(bitmap, nullptr);
1806
0
        dbm.delete_bitmap[key] = *bitmap;
1807
0
    }
1808
0
    return dbm;
1809
0
}
1810
1811
0
std::string tablet_state_name(TabletState state) {
1812
0
    switch (state) {
1813
0
    case TABLET_NOTREADY:
1814
0
        return "TABLET_NOTREADY";
1815
1816
0
    case TABLET_RUNNING:
1817
0
        return "TABLET_RUNNING";
1818
1819
0
    case TABLET_TOMBSTONED:
1820
0
        return "TABLET_TOMBSTONED";
1821
1822
0
    case TABLET_STOPPED:
1823
0
        return "TABLET_STOPPED";
1824
1825
0
    case TABLET_SHUTDOWN:
1826
0
        return "TABLET_SHUTDOWN";
1827
1828
0
    default:
1829
0
        return "TabletState(" + std::to_string(state) + ")";
1830
0
    }
1831
0
}
1832
1833
#include "common/compile_check_end.h"
1834
} // namespace doris