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

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

// distributed with this work for additional information

// regarding copyright ownership.  The ASF licenses this file

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

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

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

//

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

//

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

// software distributed under the License is distributed on an

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

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

// specific language governing permissions and limitations

// under the License.

#include "format/table/hudi_jni_reader.h"

#include <map>

#include "core/types.h"

#include "runtime/descriptors.h"

#include "runtime/runtime_state.h"

namespace doris {

class RuntimeProfile;

class RuntimeState;

class Block;

} // namespace doris

namespace doris {

#include "common/compile_check_begin.h"

const std::string HudiJniReader::HOODIE_CONF_PREFIX = "hoodie.";

const std::string HudiJniReader::HADOOP_CONF_PREFIX = "hadoop_conf.";

HudiJniReader::HudiJniReader(const TFileScanRangeParams& scan_params,

                             const THudiFileDesc& hudi_params,

                             const std::vector<SlotDescriptor*>& file_slot_descs,

                             RuntimeState* state, RuntimeProfile* profile)

        : JniReader(file_slot_descs, state, profile),

          _scan_params(scan_params),

          _hudi_params(hudi_params) {

    std::vector<std::string> required_fields;

    for (const auto& desc : _file_slot_descs) {

        required_fields.emplace_back(desc->col_name());

    }

    std::map<String, String> params = {

            {"query_id", print_id(_state->query_id())},

            {"base_path", _hudi_params.base_path},

            {"data_file_path", _hudi_params.data_file_path},

            {"data_file_length", std::to_string(_hudi_params.data_file_length)},

            {"delta_file_paths", join(_hudi_params.delta_logs, ",")},

            {"hudi_column_names", join(_hudi_params.column_names, ",")},

            {"hudi_column_types", join(_hudi_params.column_types, "#")},

            {"required_fields", join(required_fields, ",")},

            {"instant_time", _hudi_params.instant_time},

            {"serde", _hudi_params.serde},

            {"input_format", _hudi_params.input_format},

            {"time_zone", state->timezone_obj().name()}};

    // Use compatible hadoop client to read data

    for (const auto& kv : _scan_params.properties) {

        if (kv.first.starts_with(HOODIE_CONF_PREFIX)) {

            params[kv.first] = kv.second;

        } else {

            params[HADOOP_CONF_PREFIX + kv.first] = kv.second;

        }

    }

    _jni_connector = std::make_unique<JniConnector>("org/apache/doris/hudi/HadoopHudiJniScanner",

                                                    params, required_fields);

}

Status HudiJniReader::init_reader() {

    RETURN_IF_ERROR(_jni_connector->init());

    return _jni_connector->open(_state, _profile);

}

#include "common/compile_check_end.h"

} // namespace doris