// 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.

#pragma once

#include <butil/macros.h>

#include <fmt/format.h>

#include <gen_cpp/Descriptors_types.h>

#include <glog/logging.h>

#include <stddef.h>

#include <stdint.h>

#include <memory>

#include <ostream>

#include <set>

#include <shared_mutex>

#include <string>

#include <unordered_map>

#include <unordered_set>

#include <utility>

#include <vector>

#include "common/config.h"

#include "common/object_pool.h"

#include "common/status.h"

#include "olap/column_mapping.h"

#include "olap/olap_common.h"

#include "olap/rowset/pending_rowset_helper.h"

#include "olap/rowset/rowset.h"

#include "olap/rowset/rowset_reader.h"

#include "olap/rowset/rowset_writer.h"

#include "olap/rowset/segment_v2/inverted_index_writer.h"

#include "olap/storage_engine.h"

#include "olap/tablet.h"

#include "olap/tablet_fwd.h"

#include "olap/tablet_schema.h"

#include "runtime/descriptors.h"

#include "runtime/memory/mem_tracker.h"

#include "vec/data_types/data_type.h"

namespace doris {

class DeleteHandler;

class Field;

class TAlterInvertedIndexReq;

class TAlterTabletReqV2;

class TExpr;

enum AlterTabletType : int;

enum RowsetTypePB : int;

enum SegmentsOverlapPB : int;

namespace vectorized {

class Block;

class OlapBlockDataConvertor;

} // namespace vectorized

class BlockChanger {

public:

    BlockChanger(TabletSchemaSPtr tablet_schema, DescriptorTbl desc_tbl);

    ~BlockChanger();

    ColumnMapping* get_mutable_column_mapping(size_t column_index);

    Status change_block(vectorized::Block* ref_block, vectorized::Block* new_block) const;

    void set_where_expr(const std::shared_ptr<TExpr>& where_expr) { _where_expr = where_expr; }

    void set_type(AlterTabletType type) { _type = type; }

    void set_compatible_version(int32_t version) noexcept { _fe_compatible_version = version; }

    bool has_where() const { return _where_expr != nullptr; }

private:

    static Status _check_cast_valid(vectorized::ColumnPtr ref_column,

                                    vectorized::ColumnPtr new_column);

    // @brief column-mapping specification of new schema

    SchemaMapping _schema_mapping;

    DescriptorTbl _desc_tbl;

    std::shared_ptr<TExpr> _where_expr;

    AlterTabletType _type;

    int32_t _fe_compatible_version = -1;

};

class SchemaChange {

public:

    SchemaChange() = default;

    virtual ~SchemaChange() = default;

    virtual Status process(RowsetReaderSharedPtr rowset_reader, RowsetWriter* rowset_writer,

                           BaseTabletSPtr new_tablet, BaseTabletSPtr base_tablet,

                           TabletSchemaSPtr base_tablet_schema,

                           TabletSchemaSPtr new_tablet_schema) {

        if (rowset_reader->rowset()->empty() || rowset_reader->rowset()->num_rows() == 0) {

            RETURN_IF_ERROR(rowset_writer->flush());

            return Status::OK();

        }

        _filtered_rows = 0;

        _merged_rows = 0;

        RETURN_IF_ERROR_OR_CATCH_EXCEPTION(_inner_process(rowset_reader, rowset_writer, new_tablet,

                                                          base_tablet_schema, new_tablet_schema));

        // Check row num changes

        if (!_check_row_nums(rowset_reader, *rowset_writer)) {

            return Status::Error<ErrorCode::ALTER_STATUS_ERR>("SchemaChange check row nums failed");

        }

        LOG(INFO) << "all row nums. source_rows=" << rowset_reader->rowset()->num_rows()

                  << ", source_filtered_rows=" << rowset_reader->filtered_rows()

                  << ", source_merged_rows=" << rowset_reader->merged_rows()

                  << ", merged_rows=" << merged_rows() << ", filtered_rows=" << filtered_rows()

                  << ", new_index_rows=" << rowset_writer->num_rows()

                  << ", writer_filtered_rows=" << rowset_writer->num_rows_filtered();

        return Status::OK();

    }

    uint64_t filtered_rows() const { return _filtered_rows; }

    uint64_t merged_rows() const { return _merged_rows; }

protected:

    void _add_filtered_rows(uint64_t filtered_rows) { _filtered_rows += filtered_rows; }

    void _add_merged_rows(uint64_t merged_rows) { _merged_rows += merged_rows; }

    virtual Status _inner_process(RowsetReaderSharedPtr rowset_reader, RowsetWriter* rowset_writer,

                                  BaseTabletSPtr new_tablet, TabletSchemaSPtr base_tablet_schema,

                                  TabletSchemaSPtr new_tablet_schema) {

        return Status::NotSupported("inner process unsupported.");

    }

    virtual bool _check_row_nums(RowsetReaderSharedPtr reader, const RowsetWriter& writer) const {

        if (reader->rowset()->num_rows() - reader->filtered_rows() - reader->merged_rows() !=

            writer.num_rows() + writer.num_rows_filtered() + _merged_rows + _filtered_rows) {

            LOG(WARNING) << "fail to check row num! "

                         << "source_rows=" << reader->rowset()->num_rows()

                         << ", source_filtered_rows=" << reader->filtered_rows()

                         << ", source_merged_rows=" << reader->merged_rows()

                         << ", written_rows=" << writer.num_rows()

                         << ", writer_filtered_rows=" << writer.num_rows_filtered()

                         << ", merged_rows=" << merged_rows()

                         << ", filtered_rows=" << filtered_rows();

            if (!config::ignore_schema_change_check) {

                return false;

            }

        }

        return true;

    }

private:

    uint64_t _filtered_rows {};

    uint64_t _merged_rows {};

};

class LinkedSchemaChange : public SchemaChange {

public:

    LinkedSchemaChange() = default;

    ~LinkedSchemaChange() override = default;

    Status process(RowsetReaderSharedPtr rowset_reader, RowsetWriter* rowset_writer,

                   BaseTabletSPtr new_tablet, BaseTabletSPtr base_tablet,

                   TabletSchemaSPtr base_tablet_schema,

                   TabletSchemaSPtr new_tablet_schema) override;

private:

    DISALLOW_COPY_AND_ASSIGN(LinkedSchemaChange);

};

class VSchemaChangeDirectly : public SchemaChange {

public:

    VSchemaChangeDirectly(const BlockChanger& changer) : _changer(changer) {}

private:

    Status _inner_process(RowsetReaderSharedPtr rowset_reader, RowsetWriter* rowset_writer,

                          BaseTabletSPtr new_tablet, TabletSchemaSPtr base_tablet_schema,

                          TabletSchemaSPtr new_tablet_schema) override;

    bool _check_row_nums(RowsetReaderSharedPtr reader, const RowsetWriter& writer) const override {

        return _changer.has_where() || SchemaChange::_check_row_nums(reader, writer);

    }

    const BlockChanger& _changer;

};

class VBaseSchemaChangeWithSorting : public SchemaChange {

public:

    VBaseSchemaChangeWithSorting(const BlockChanger& changer, size_t memory_limitation);

    ~VBaseSchemaChangeWithSorting() override = default;

    Status _inner_process(RowsetReaderSharedPtr rowset_reader, RowsetWriter* rowset_writer,

                          BaseTabletSPtr new_tablet, TabletSchemaSPtr base_tablet_schema,

                          TabletSchemaSPtr new_tablet_schema) override;

    virtual Result<RowsetSharedPtr> _internal_sorting(

            const std::vector<std::unique_ptr<vectorized::Block>>& blocks,

            const Version& temp_delta_versions, int64_t newest_write_timestamp,

            BaseTabletSPtr new_tablet, RowsetTypePB new_rowset_type,

            SegmentsOverlapPB segments_overlap, TabletSchemaSPtr new_tablet_schema);

    Status _external_sorting(std::vector<RowsetSharedPtr>& src_rowsets, RowsetWriter* rowset_writer,

                             BaseTabletSPtr new_tablet, TabletSchemaSPtr new_tablet_schema);

protected:

    // for external sorting

    // src_rowsets to store the rowset generated by internal sorting

    std::vector<RowsetSharedPtr> _src_rowsets;

private:

    bool _check_row_nums(RowsetReaderSharedPtr reader, const RowsetWriter& writer) const override {

        return _changer.has_where() || SchemaChange::_check_row_nums(reader, writer);

    }

    const BlockChanger& _changer;

    size_t _memory_limitation;

    Version _temp_delta_versions;

    std::unique_ptr<MemTracker> _mem_tracker;

};

// @breif schema change with sorting

// Mixin for local StorageEngine

class VLocalSchemaChangeWithSorting final : public VBaseSchemaChangeWithSorting {

public:

    VLocalSchemaChangeWithSorting(const BlockChanger& changer, size_t memory_limitation,

                                  StorageEngine& local_storage_engine)

            : VBaseSchemaChangeWithSorting(changer, memory_limitation),

              _local_storage_engine(local_storage_engine) {}

    ~VLocalSchemaChangeWithSorting() override = default;

    Status _inner_process(RowsetReaderSharedPtr rowset_reader, RowsetWriter* rowset_writer,

                          BaseTabletSPtr new_tablet, TabletSchemaSPtr base_tablet_schema,

                          TabletSchemaSPtr new_tablet_schema) override;

    Result<RowsetSharedPtr> _internal_sorting(

            const std::vector<std::unique_ptr<vectorized::Block>>& blocks,

            const Version& temp_delta_versions, int64_t newest_write_timestamp,

            BaseTabletSPtr new_tablet, RowsetTypePB new_rowset_type,

            SegmentsOverlapPB segments_overlap, TabletSchemaSPtr new_tablet_schema) override;

private:

    StorageEngine& _local_storage_engine;

    std::vector<PendingRowsetGuard> _pending_rs_guards;

};

struct AlterMaterializedViewParam {

    std::string column_name;

    std::string origin_column_name;

    std::shared_ptr<TExpr> expr;

};

struct SchemaChangeParams {

    AlterTabletType alter_tablet_type;

    bool enable_unique_key_merge_on_write = false;

    std::vector<RowsetReaderSharedPtr> ref_rowset_readers;

    DeleteHandler* delete_handler = nullptr;

    std::unordered_map<std::string, AlterMaterializedViewParam> materialized_params_map;

    DescriptorTbl* desc_tbl = nullptr;

    ObjectPool pool;

    int32_t be_exec_version;

    std::string vault_id;

};

class SchemaChangeJob {

public:

    SchemaChangeJob(StorageEngine& local_storage_engine, const TAlterTabletReqV2& request,

                    const std::string& job_id);

    Status process_alter_tablet(const TAlterTabletReqV2& request);

    bool tablet_in_converting(int64_t tablet_id);

    static Status parse_request(const SchemaChangeParams& sc_params,

                                TabletSchema* base_tablet_schema, TabletSchema* new_tablet_schema,

                                BlockChanger* changer, bool* sc_sorting, bool* sc_directly);

private:

    std::unique_ptr<SchemaChange> _get_sc_procedure(const BlockChanger& changer, bool sc_sorting,

                                                    bool sc_directly, int64_t mem_limit) {

        if (sc_sorting) {

            return std::make_unique<VLocalSchemaChangeWithSorting>(changer, mem_limit,

                                                                   _local_storage_engine);

        }

        if (sc_directly) {

            return std::make_unique<VSchemaChangeDirectly>(changer);

        }

        return std::make_unique<LinkedSchemaChange>();

    }

    Status _get_versions_to_be_changed(std::vector<Version>* versions_to_be_changed,

                                       RowsetSharedPtr* max_rowset);

    Status _do_process_alter_tablet(const TAlterTabletReqV2& request);

    Status _validate_alter_result(const TAlterTabletReqV2& request);

    Status _convert_historical_rowsets(const SchemaChangeParams& sc_params,

                                       int64_t* real_alter_version);

    // Initialization Settings for creating a default value

    static Status _init_column_mapping(ColumnMapping* column_mapping,

                                       const TabletColumn& column_schema, const std::string& value);

    Status _calc_delete_bitmap_for_mow_table(int64_t alter_version);

    StorageEngine& _local_storage_engine;

    TabletSharedPtr _base_tablet;

    TabletSharedPtr _new_tablet;

    TabletSchemaSPtr _base_tablet_schema;

    TabletSchemaSPtr _new_tablet_schema;

    std::shared_mutex _mutex;

    std::unordered_set<int64_t> _tablet_ids_in_converting;

    std::set<std::string> _supported_functions;

    std::string _job_id;

};

} // namespace doris