// Licensed to the Apache Software Foundation (ASF) under one

// or more contributor license agreements.  See the NOTICE file

// distributed with this work for additional information

// regarding copyright ownership.  The ASF licenses this file

// to you under the Apache License, Version 2.0 (the

// "License"); you may not use this file except in compliance

// with the License.  You may obtain a copy of the License at

//

//   http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing,

// software distributed under the License is distributed on an

// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

// KIND, either express or implied.  See the License for the

// specific language governing permissions and limitations

// under the License.

#ifndef DORIS_BE_SRC_OLAP_ROWSET_ROWSET_META_H

#define DORIS_BE_SRC_OLAP_ROWSET_ROWSET_META_H

#include <gen_cpp/olap_file.pb.h>

#include <glog/logging.h>

#include <atomic>

#include <chrono>

#include <cstddef>

#include <cstdint>

#include <iterator>

#include <memory>

#include <string>

#include <vector>

#include "common/cast_set.h"

#include "common/check.h"

#include "common/config.h"

#include "common/status.h"

#include "io/fs/encrypted_fs_factory.h"

#include "io/fs/file_system.h"

#include "runtime/memory/lru_cache_policy.h"

#include "storage/metadata_adder.h"

#include "storage/olap_common.h"

#include "storage/rowset/rowset_fwd.h"

#include "storage/rowset/rowset_segment_id.h"

#include "storage/storage_policy.h"

#include "storage/tablet/tablet_fwd.h"

#include "util/once.h"

namespace doris {

class RowsetSegmentMetaView;

class RowsetSegmentMetaRange;

class RowsetMeta : public MetadataAdder<RowsetMeta> {

public:

    RowsetMeta() = default;

    ~RowsetMeta();

    bool init(std::string_view pb_rowset_meta);

    bool init(const RowsetMeta* rowset_meta);

    bool init_from_pb(const RowsetMetaPB& rowset_meta_pb);

    bool init_from_json(const std::string& json_rowset_meta);

    bool serialize(std::string* value) { return _serialize_to_pb(value); }

    bool json_rowset_meta(std::string* json_rowset_meta);

    // If the rowset is a local rowset, return the global local file system.

    // Otherwise, return the remote file system corresponding to rowset's resource id.

    // Note that if the resource id cannot be found for the corresponding remote file system, nullptr will be returned.

    MOCK_FUNCTION io::FileSystemSPtr fs();

    // The bare file system holding this rowset's files. It resolves neither packed files nor

    // encryption, so it cannot open a segment whose bytes live inside a packed object. Prefer

    // `fs()`, or `packed_physical_fs()` when the raw bytes are what you are after.

    io::FileSystemSPtr physical_fs();

    // Same as `physical_fs()`, but additionally wrapped with `PackedFileSystem` when this

    // rowset's files are packed into shared objects, so that segment/index paths still

    // resolve. Unlike `fs()`, no encryption layer is applied, i.e. reads return the raw

    // on-disk bytes. Callers that inspect the physical layout of a file (encryption footer,

    // magic code, ...) must use this instead of `physical_fs()`, otherwise packed files

    // cannot be opened at all.

    io::FileSystemSPtr packed_physical_fs();

    Result<const StorageResource*> remote_storage_resource();

    void set_remote_storage_resource(StorageResource resource);

    const std::string& resource_id() const { return _rowset_meta_pb.resource_id(); }

    void set_resource_id(const std::string& resource_id) {

        _rowset_meta_pb.set_resource_id(resource_id);

    }

    bool is_local() const { return !_rowset_meta_pb.has_resource_id(); }

    bool has_variant_type_in_schema() const;

    RowsetId rowset_id() const { return _rowset_id; }

    void set_rowset_id(const RowsetId& rowset_id) {

        // rowset id is a required field, just set it to 0

        _rowset_meta_pb.set_rowset_id(0);

        _rowset_id = rowset_id;

        _rowset_meta_pb.set_rowset_id_v2(rowset_id.to_string());

    }

    int64_t tablet_id() const { return _rowset_meta_pb.tablet_id(); }

    void set_tablet_id(int64_t tablet_id) { _rowset_meta_pb.set_tablet_id(tablet_id); }

    int64_t db_id() const { return _rowset_meta_pb.db_id(); }

    void set_db_id(int64_t db_id) { _rowset_meta_pb.set_db_id(db_id); }

    int64_t table_id() const { return _rowset_meta_pb.table_id(); }

    void set_table_id(int64_t table_id) { _rowset_meta_pb.set_table_id(table_id); }

    int64_t index_id() const { return _rowset_meta_pb.index_id(); }

    void set_index_id(int64_t index_id) { _rowset_meta_pb.set_index_id(index_id); }

    bool has_inverted_index_storage_format() const {

        return _rowset_meta_pb.has_inverted_index_storage_format();

    }

    InvertedIndexStorageFormatPB inverted_index_storage_format() const {

        return _rowset_meta_pb.inverted_index_storage_format();

    }

    void set_inverted_index_storage_format(InvertedIndexStorageFormatPB format);

    TabletUid tablet_uid() const { return _rowset_meta_pb.tablet_uid(); }

    void set_tablet_uid(TabletUid tablet_uid) {

        *(_rowset_meta_pb.mutable_tablet_uid()) = tablet_uid.to_proto();

    }

    int64_t txn_id() const { return _rowset_meta_pb.txn_id(); }

    void set_txn_id(int64_t txn_id) { _rowset_meta_pb.set_txn_id(txn_id); }

    int32_t tablet_schema_hash() const { return _rowset_meta_pb.tablet_schema_hash(); }

    void set_tablet_schema_hash(int32_t tablet_schema_hash) {

        _rowset_meta_pb.set_tablet_schema_hash(tablet_schema_hash);

    }

    void mark_row_binlog() { _rowset_meta_pb.set_is_row_binlog(true); }

    bool is_row_binlog() const {

        return _rowset_meta_pb.has_is_row_binlog() && _rowset_meta_pb.is_row_binlog();

    }

    RowsetTypePB rowset_type() const { return _rowset_meta_pb.rowset_type(); }

    void set_rowset_type(RowsetTypePB rowset_type) { _rowset_meta_pb.set_rowset_type(rowset_type); }

    RowsetStatePB rowset_state() const { return _rowset_meta_pb.rowset_state(); }

    void set_rowset_state(RowsetStatePB rowset_state) {

        _rowset_meta_pb.set_rowset_state(rowset_state);

    }

    Version version() const {

        return {_rowset_meta_pb.start_version(), _rowset_meta_pb.end_version()};

    }

    void set_version(Version version) {

        _rowset_meta_pb.set_start_version(version.first);

        _rowset_meta_pb.set_end_version(version.second);

    }

    bool has_version() const {

        return _rowset_meta_pb.has_start_version() && _rowset_meta_pb.has_end_version();

    }

    int64_t start_version() const { return _rowset_meta_pb.start_version(); }

    int64_t end_version() const { return _rowset_meta_pb.end_version(); }

    int64_t num_rows() const { return _rowset_meta_pb.num_rows(); }

    void set_num_rows(int64_t num_rows) { _rowset_meta_pb.set_num_rows(num_rows); }

    void set_num_segment_rows(const std::vector<uint32_t>& num_segment_rows) {

        _rowset_meta_pb.mutable_num_segment_rows()->Assign(num_segment_rows.cbegin(),

                                                           num_segment_rows.cend());

    }

    void get_num_segment_rows(std::vector<uint32_t>* num_segment_rows) const {

        num_segment_rows->assign(_rowset_meta_pb.num_segment_rows().cbegin(),

                                 _rowset_meta_pb.num_segment_rows().cend());

    }

    auto& get_num_segment_rows() const { return _rowset_meta_pb.num_segment_rows(); }

    void set_segment_group_sizes(const std::vector<int32_t>& segment_group_sizes) {

        DORIS_CHECK_GT(segment_group_sizes.size(), 1);

        int64_t segment_count = 0;

        for (const auto group_size : segment_group_sizes) {

            DORIS_CHECK_GT(group_size, 0);

            segment_count += group_size;

        }

        DORIS_CHECK_EQ(segment_count, num_segments());

        _rowset_meta_pb.mutable_segment_group_sizes()->Assign(segment_group_sizes.cbegin(),

                                                              segment_group_sizes.cend());

    }

    void clear_segment_group_sizes() { _rowset_meta_pb.clear_segment_group_sizes(); }

    const auto& segment_group_sizes() const { return _rowset_meta_pb.segment_group_sizes(); }

    int64_t total_disk_size() const { return _rowset_meta_pb.total_disk_size(); }

    void set_total_disk_size(int64_t total_disk_size) {

        _rowset_meta_pb.set_total_disk_size(total_disk_size);

    }

    int64_t data_disk_size() const { return _rowset_meta_pb.data_disk_size(); }

    void set_data_disk_size(int64_t data_disk_size) {

        _rowset_meta_pb.set_data_disk_size(data_disk_size);

    }

    int64_t index_disk_size() const { return _rowset_meta_pb.index_disk_size(); }

    void set_index_disk_size(int64_t index_disk_size) {

        _rowset_meta_pb.set_index_disk_size(index_disk_size);

    }

    void zone_maps(std::vector<::doris::ZoneMap>* zone_maps) {

        for (const ::doris::ZoneMap& zone_map : _rowset_meta_pb.zone_maps()) {

            zone_maps->push_back(zone_map);

        }

    }

    void set_zone_maps(const std::vector<::doris::ZoneMap>& zone_maps) {

        for (const ::doris::ZoneMap& zone_map : zone_maps) {

            ::doris::ZoneMap* new_zone_map = _rowset_meta_pb.add_zone_maps();

            *new_zone_map = zone_map;

        }

    }

    void add_zone_map(const ::doris::ZoneMap& zone_map) {

        ::doris::ZoneMap* new_zone_map = _rowset_meta_pb.add_zone_maps();

        *new_zone_map = zone_map;

    }

    bool has_delete_predicate() const { return _rowset_meta_pb.has_delete_predicate(); }

    const DeletePredicatePB& delete_predicate() const { return _rowset_meta_pb.delete_predicate(); }

    DeletePredicatePB* mutable_delete_predicate() {

        return _rowset_meta_pb.mutable_delete_predicate();

    }

    void set_delete_predicate(DeletePredicatePB delete_predicate) {

        DeletePredicatePB* new_delete_condition = _rowset_meta_pb.mutable_delete_predicate();

        *new_delete_condition = std::move(delete_predicate);

    }

    bool empty() const { return _rowset_meta_pb.empty(); }

    void set_empty(bool empty) { _rowset_meta_pb.set_empty(empty); }

    PUniqueId load_id() const { return _rowset_meta_pb.load_id(); }

    void set_load_id(PUniqueId load_id) {

        PUniqueId* new_load_id = _rowset_meta_pb.mutable_load_id();

        new_load_id->set_hi(load_id.hi());

        new_load_id->set_lo(load_id.lo());

    }

    void set_job_id(const std::string& job_id) { _rowset_meta_pb.set_job_id(job_id); }

    const std::string& job_id() const { return _rowset_meta_pb.job_id(); }

    bool delete_flag() const { return _rowset_meta_pb.delete_flag(); }

    int64_t creation_time() const { return _rowset_meta_pb.creation_time(); }

    void set_creation_time(int64_t creation_time) {

        return _rowset_meta_pb.set_creation_time(creation_time);

    }

    int64_t stale_at() const {

        int64_t stale_time = _stale_at_s.load();

        return stale_time > 0 ? stale_time : _rowset_meta_pb.creation_time();

    }

    bool has_stale_at() const { return _stale_at_s.load() > 0; }

    void set_stale_at(int64_t stale_at) { _stale_at_s.store(stale_at); }

    int64_t partition_id() const { return _rowset_meta_pb.partition_id(); }

    void set_partition_id(int64_t partition_id) {

        return _rowset_meta_pb.set_partition_id(partition_id);

    }

    int64_t num_segments() const {

        DCHECK(_rowset_meta_pb.segment_ids_size() == 0 ||

               _rowset_meta_pb.segment_ids_size() == _rowset_meta_pb.num_segments());

        return _rowset_meta_pb.num_segments();

    }

    void set_num_segments(int64_t num_segments) { _rowset_meta_pb.set_num_segments(num_segments); }

    bool has_segment_ids() const { return _rowset_meta_pb.segment_ids_size() > 0; }

    const auto& segment_ids() const { return _rowset_meta_pb.segment_ids(); }

    void set_segment_ids(const std::vector<int64_t>& segment_ids);

    int64_t segment_id(size_t pos) const {

        DORIS_CHECK_LT(pos, cast_set<size_t>(num_segments()));

        return has_segment_ids() ? _rowset_meta_pb.segment_ids(cast_set<int>(pos))

                                 : cast_set<int64_t>(pos);

    }

    RowsetSegmentRef segment_ref(size_t pos) const { return {pos, segment_id(pos)}; }

    RowsetSegmentMetaView segment(size_t pos) const;

    RowsetSegmentMetaRange segments() const;

    size_t position_of(int64_t seg_id) const;

    // Convert to RowsetMetaPB, skip_schema is only used by cloud to separate schema from rowset meta.

    void to_rowset_pb(RowsetMetaPB* rs_meta_pb, bool skip_schema = false) const;

    // Convert to RowsetMetaPB, skip_schema is only used by cloud to separate schema from rowset meta.

    RowsetMetaPB get_rowset_pb(bool skip_schema = false) const;

    inline DeletePredicatePB* mutable_delete_pred_pb() {

        return _rowset_meta_pb.mutable_delete_predicate();

    }

    bool is_singleton_delta() const {

        return has_version() && _rowset_meta_pb.start_version() == _rowset_meta_pb.end_version();

    }

    // Some time, we may check if this rowset is in rowset meta manager's meta by using RowsetMetaManager::check_rowset_meta.

    // But, this check behavior may cost a lot of time when it is frequent.

    // If we explicitly remove this rowset from rowset meta manager's meta, we can set _is_removed_from_rowset_meta to true,

    // And next time when we want to check if this rowset is in rowset mata manager's meta, we can

    // check is_remove_from_rowset_meta() first.

    void set_remove_from_rowset_meta() { _is_removed_from_rowset_meta = true; }

    bool is_remove_from_rowset_meta() const { return _is_removed_from_rowset_meta; }

    SegmentsOverlapPB segments_overlap() const { return _rowset_meta_pb.segments_overlap_pb(); }

    void set_segments_overlap(SegmentsOverlapPB segments_overlap) {

        _rowset_meta_pb.set_segments_overlap_pb(segments_overlap);

    }

    static bool comparator(const RowsetMetaSharedPtr& left, const RowsetMetaSharedPtr& right) {

        return left->end_version() < right->end_version();

    }

    // return true if segments in this rowset has overlapping data.

    // this is not same as `segments_overlap()` method.

    // `segments_overlap()` only return the value of "segments_overlap" field in rowset meta,

    // but "segments_overlap" may be UNKNOWN.

    //

    // Returns true if all of the following conditions are met:

    // 1. The rowset contains more than one segment.

    // 2. segments_overlap() is not NONOVERLAPPING (OVERLAP_UNKNOWN, OVERLAPPING, and

    //    NONOVERLAPPING_WITHIN_GROUP are considered overlapping).

    // 3. The rowset has a singleton version, except row-binlog LMax quick merge rowsets that

    //    explicitly set segments_overlap() to OVERLAPPING.

    bool is_segments_overlapping() const {

        return num_segments() > 1 && segments_overlap() != NONOVERLAPPING &&

               (is_singleton_delta() || (is_row_binlog() && segments_overlap() == OVERLAPPING));

    }

    bool produced_by_compaction() const {

        return has_version() && (start_version() < end_version() ||

                                 (start_version() == end_version() &&

                                  (segments_overlap() == NONOVERLAPPING ||

                                   segments_overlap() == NONOVERLAPPING_WITHIN_GROUP)));

    }

    // get the compaction score of this rowset.

    // if segments are overlapping, the score equals to the number of segments,

    // otherwise, score is 1.

    uint32_t get_compaction_score() const {

        uint32_t score = 0;

        if (!is_segments_overlapping()) {

            score = 1;

        } else {

            auto num_seg = num_segments();

            DCHECK_GT(num_seg, 0);

            score = cast_set<uint32_t>(num_seg);

            CHECK(score > 0);

        }

        return score;

    }

    uint32_t get_merge_way_num() const {

        uint32_t way_num = 0;

        if (!is_segments_overlapping()) {

            if (num_segments() == 0) {

                way_num = 0;

            } else {

                way_num = 1;

            }

        } else {

            auto num_seg = num_segments();

            DCHECK_GT(num_seg, 0);

            way_num = cast_set<uint32_t>(num_seg);

            CHECK(way_num > 0);

        }

        return way_num;

    }

    void get_segments_key_bounds(std::vector<KeyBoundsPB>* segments_key_bounds) const {

        for (const KeyBoundsPB& key_range : _rowset_meta_pb.segments_key_bounds()) {

            segments_key_bounds->push_back(key_range);

        }

    }

    auto& get_segments_key_bounds() const { return _rowset_meta_pb.segments_key_bounds(); }

    bool is_segments_key_bounds_truncated() const {

        return _rowset_meta_pb.has_segments_key_bounds_truncated() &&

               _rowset_meta_pb.segments_key_bounds_truncated();

    }

    void set_segments_key_bounds_truncated(bool truncated) {

        _rowset_meta_pb.set_segments_key_bounds_truncated(truncated);

    }

    // When true, `segments_key_bounds` holds a single aggregated

    // [rowset_min, rowset_max] entry instead of per-segment bounds.

    bool is_segments_key_bounds_aggregated() const {

        return _rowset_meta_pb.has_segments_key_bounds_aggregated() &&

               _rowset_meta_pb.segments_key_bounds_aggregated();

    }

    void set_segments_key_bounds_aggregated(bool aggregated) {

        _rowset_meta_pb.set_segments_key_bounds_aggregated(aggregated);

    }

    bool get_first_segment_key_bound(KeyBoundsPB* key_bounds) {

        // for compatibility, old version has not segment key bounds

        if (_rowset_meta_pb.segments_key_bounds_size() == 0) {

            return false;

        }

        *key_bounds = *_rowset_meta_pb.segments_key_bounds().begin();

        return true;

    }

    bool get_last_segment_key_bound(KeyBoundsPB* key_bounds) {

        if (_rowset_meta_pb.segments_key_bounds_size() == 0) {

            return false;

        }

        *key_bounds = *_rowset_meta_pb.segments_key_bounds().rbegin();

        return true;

    }

    // If `aggregate_into_single` is true, collapse per-segment bounds into a single

    // [rowset_min, rowset_max] entry and mark this rowset as aggregated.

    void set_segments_key_bounds(const std::vector<KeyBoundsPB>& segments_key_bounds,

                                 bool aggregate_into_single = false);

    void add_segment_key_bounds(KeyBoundsPB segments_key_bounds) {

        *_rowset_meta_pb.add_segments_key_bounds() = std::move(segments_key_bounds);

        set_segments_overlap(OVERLAPPING);

    }

    void set_newest_write_timestamp(int64_t timestamp) {

        _rowset_meta_pb.set_newest_write_timestamp(timestamp);

    }

    int64_t newest_write_timestamp() const { return _rowset_meta_pb.newest_write_timestamp(); }

    // for cloud only

    bool has_visible_ts_ms() const { return _rowset_meta_pb.has_visible_ts_ms(); }

    int64_t visible_ts_ms() const { return _rowset_meta_pb.visible_ts_ms(); }

    std::chrono::time_point<std::chrono::system_clock> visible_timestamp() const {

        using namespace std::chrono;

        if (has_visible_ts_ms()) {

            return time_point<system_clock>(milliseconds(visible_ts_ms()));

        }

        return system_clock::from_time_t(newest_write_timestamp());

    }

    void set_visible_ts_ms(int64_t visible_ts_ms) {

        _rowset_meta_pb.set_visible_ts_ms(visible_ts_ms);

    }

    void set_tablet_schema(const TabletSchemaSPtr& tablet_schema);

    void set_tablet_schema(const TabletSchemaPB& tablet_schema);

    const TabletSchemaSPtr& tablet_schema() const { return _schema; }

    void set_txn_expiration(int64_t expiration) { _rowset_meta_pb.set_txn_expiration(expiration); }

    void set_compaction_level(int64_t compaction_level) {

        _rowset_meta_pb.set_compaction_level(compaction_level);

    }

    int64_t compaction_level() { return _rowset_meta_pb.compaction_level(); }

    // `seg_file_size` MUST be ordered by rowset segment position.

    void add_segments_file_size(const std::vector<size_t>& seg_file_size);

    // Return -1 if segment file size is unknown

    int64_t segment_file_size_by_pos(size_t pos) const;

    const auto& segments_file_size() const { return _rowset_meta_pb.segments_file_size(); }

    // Used for partial update, when publish, partial update may add a new rowset and we should update rowset meta

    void merge_rowset_meta(const RowsetMeta& other);

    InvertedIndexFileInfo inverted_index_file_info_by_pos(size_t pos) const;

    const auto& inverted_index_file_info() const {

        return _rowset_meta_pb.inverted_index_file_info();

    }

    void add_inverted_index_files_info(

            const std::vector<const InvertedIndexFileInfo*>& idx_file_info);

    int64_t get_metadata_size() const override;

    // Because the member field '_handle' is a raw pointer, use member func 'init' to replace copy ctor

    RowsetMeta(const RowsetMeta&) = delete;

    RowsetMeta operator=(const RowsetMeta&) = delete;

    void add_packed_slice_location(const std::string& segment_path,

                                   const std::string& packed_file_path, int64_t offset,

                                   int64_t size, int64_t packed_file_size) {

        auto* index_map = _rowset_meta_pb.mutable_packed_slice_locations();

        auto& index_pb = (*index_map)[segment_path];

        index_pb.set_packed_file_path(packed_file_path);

        index_pb.set_offset(offset);

        index_pb.set_size(size);

        index_pb.set_packed_file_size(packed_file_size);

    }

    int32_t schema_version() const { return _rowset_meta_pb.schema_version(); }

    std::string debug_string() const { return _rowset_meta_pb.ShortDebugString(); }

    // Pre-set the encryption algorithm to avoid re-entrant get_tablet calls

    // that can cause SingleFlight deadlock during tablet loading.

    void set_encryption_algorithm(EncryptionAlgorithmPB algorithm) {

        _determine_encryption_once.call(

                [algorithm]() -> Result<EncryptionAlgorithmPB> { return algorithm; });

    }

    TsoRange commit_tso() const {

        const auto& commit_tso_pb = _rowset_meta_pb.commit_tso();

        return {commit_tso_pb.start_tso(), commit_tso_pb.end_tso()};

    }

    bool has_commit_tso() const { return _rowset_meta_pb.has_commit_tso(); }

    void set_commit_tso(const TsoRange& commit_tso) {

        auto* commit_tso_pb = _rowset_meta_pb.mutable_commit_tso();

        commit_tso_pb->set_start_tso(commit_tso.start_tso());

        commit_tso_pb->set_end_tso(commit_tso.end_tso());

    }

    void set_commit_tso(int64_t commit_tso) { set_commit_tso({commit_tso, commit_tso}); }

    void set_cloud_fields_after_visible(int64_t visible_version, int64_t version_update_time_ms) {

        // Update rowset meta with correct version and visible_ts

        // !!ATTENTION!!: this code should be updated if there are more fields

        // in rowset meta which will be modified in meta-service when commit_txn in the future

        set_version({visible_version, visible_version});

        if (version_update_time_ms > 0) {

            set_visible_ts_ms(version_update_time_ms);

        }

    }

private:

    // Wraps `fs` with `PackedFileSystem` if this rowset has packed slice locations,

    // otherwise returns `fs` unchanged.

    io::FileSystemSPtr _wrap_packed_fs(io::FileSystemSPtr fs);

    bool _deserialize_from_pb(std::string_view value);

    bool _serialize_to_pb(std::string* value);

    void _init();

    void _validate_segment_ids() const;

    friend bool operator==(const RowsetMeta& a, const RowsetMeta& b);

    friend bool operator!=(const RowsetMeta& a, const RowsetMeta& b) { return !(a == b); }

private:

    RowsetMetaPB _rowset_meta_pb;

    TabletSchemaSPtr _schema;

    Cache::Handle* _handle = nullptr;

    RowsetId _rowset_id;

    StorageResource _storage_resource;

    bool _is_removed_from_rowset_meta = false;

    DorisCallOnce<Result<EncryptionAlgorithmPB>> _determine_encryption_once;

    std::atomic<int64_t> _stale_at_s {0};

};

class RowsetSegmentMetaView {

public:

    RowsetSegmentMetaView(const RowsetMeta* meta, size_t pos)

            : _meta(meta), _ref(meta->segment_ref(pos)) {}

    size_t pos() const { return _ref.pos; }

    int64_t id() const { return _ref.id; }

    RowsetSegmentRef ref() const { return _ref; }

    int64_t file_size() const { return _meta->segment_file_size_by_pos(pos()); }

    InvertedIndexFileInfo inverted_index_file_info() const {

        return _meta->inverted_index_file_info_by_pos(pos());

    }

    bool has_num_rows() const {

        return cast_set<size_t>(_meta->get_num_segment_rows().size()) > pos();

    }

    int64_t num_rows() const {

        DORIS_CHECK(has_num_rows());

        return _meta->get_num_segment_rows().Get(cast_set<int>(pos()));

    }

    bool has_position_key_bounds() const {

        return !_meta->is_segments_key_bounds_aggregated() &&

               cast_set<size_t>(_meta->get_segments_key_bounds().size()) > pos();

    }

    const KeyBoundsPB& key_bounds() const {

        DORIS_CHECK(has_position_key_bounds());

        return _meta->get_segments_key_bounds().Get(cast_set<int>(pos()));

    }

private:

    const RowsetMeta* _meta;

    RowsetSegmentRef _ref;

};

class RowsetSegmentMetaRange {

public:

    class Iterator {

    public:

        using iterator_category = std::forward_iterator_tag;

        using value_type = RowsetSegmentMetaView;

        using difference_type = std::ptrdiff_t;

        Iterator(const RowsetMeta* meta, size_t pos) : _meta(meta), _pos(pos) {}

        RowsetSegmentMetaView operator*() const { return {_meta, _pos}; }

        Iterator& operator++() {

            ++_pos;

            return *this;

        }

        bool operator==(const Iterator& other) const {

            return _meta == other._meta && _pos == other._pos;

        }

        bool operator!=(const Iterator& other) const { return !(*this == other); }

    private:

        const RowsetMeta* _meta;

        size_t _pos;

    };

    explicit RowsetSegmentMetaRange(const RowsetMeta* meta) : _meta(meta) {}

    Iterator begin() const { return {_meta, 0}; }

    Iterator end() const { return {_meta, cast_set<size_t>(_meta->num_segments())}; }

private:

    const RowsetMeta* _meta;

};

inline RowsetSegmentMetaView RowsetMeta::segment(size_t pos) const {

    return {this, pos};

}

inline RowsetSegmentMetaRange RowsetMeta::segments() const {

    return RowsetSegmentMetaRange(this);

}

} // namespace doris

#endif // DORIS_BE_SRC_OLAP_ROWSET_ROWSET_META_H