// 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 "exec/connector/vjdbc_connector.h"

#include <gen_cpp/Types_types.h>

#include <algorithm>

// IWYU pragma: no_include <bits/std_abs.h>

#include <cmath> // IWYU pragma: keep

#include <memory>

#include <ostream>

#include <utility>

#include "absl/strings/substitute.h"

#include "cloud/config.h"

#include "common/logging.h"

#include "common/status.h"

#include "core/block/block.h"

#include "core/column/column_nullable.h"

#include "core/data_type/data_type_nullable.h"

#include "core/data_type/data_type_string.h"

#include "exec/connector/jni_connector.h"

#include "exec/table_connector.h"

#include "exprs/function/simple_function_factory.h"

#include "exprs/vexpr.h"

#include "jni.h"

#include "runtime/descriptors.h"

#include "runtime/plugin/cloud_plugin_downloader.h"

#include "runtime/runtime_profile.h"

#include "runtime/runtime_state.h"

#include "runtime/user_function_cache.h"

#include "util/jni-util.h"

namespace doris {

#include "common/compile_check_begin.h"

const char* JDBC_EXECUTOR_FACTORY_CLASS = "org/apache/doris/jdbc/JdbcExecutorFactory";

const char* JDBC_EXECUTOR_CTOR_SIGNATURE = "([B)V";

const char* JDBC_EXECUTOR_STMT_WRITE_SIGNATURE = "(Ljava/util/Map;)I";

const char* JDBC_EXECUTOR_HAS_NEXT_SIGNATURE = "()Z";

const char* JDBC_EXECUTOR_CLOSE_SIGNATURE = "()V";

const char* JDBC_EXECUTOR_TRANSACTION_SIGNATURE = "()V";

JdbcConnector::JdbcConnector(const JdbcConnectorParam& param)

        : TableConnector(param.tuple_desc, param.use_transaction, param.table_name,

                         param.query_string),

          _conn_param(param),

          _closed(false) {}

JdbcConnector::~JdbcConnector() {

    if (!_closed) {

        static_cast<void>(close());

    }

}

Status JdbcConnector::close(Status /*unused*/) {

    SCOPED_RAW_TIMER(&_jdbc_statistic._connector_close_timer);

    if (_closed) {

        return Status::OK();

    }

    if (!_is_open) {

        _closed = true;

        return Status::OK();

    }

    // Try to abort transaction and call Java close(), but don't block cleanup

    if (_is_in_transaction) {

        Status abort_status = abort_trans();

        if (!abort_status.ok()) {

            LOG(WARNING) << "Failed to abort transaction: " << abort_status.to_string();

        }

    }

    JNIEnv* env = nullptr;

    RETURN_IF_ERROR(Jni::Env::Get(&env));

    RETURN_IF_ERROR(

            _executor_obj.call_nonvirtual_void_method(env, _executor_clazz, _executor_close_id)

                    .call());

    _closed = true;

    return Status::OK();

}

Status JdbcConnector::open(RuntimeState* state, bool read) {

    if (_is_open) {

        LOG(INFO) << "this scanner of jdbc already opened";

        return Status::OK();

    }

    JNIEnv* env = nullptr;

    RETURN_IF_ERROR(Jni::Env::Get(&env));

    RETURN_IF_ERROR(Jni::Util::get_jni_scanner_class(env, JDBC_EXECUTOR_FACTORY_CLASS,

                                                     &_executor_factory_clazz));

    RETURN_IF_ERROR(_executor_factory_clazz.get_static_method(

            env, "getExecutorClass", "(Lorg/apache/doris/thrift/TOdbcTableType;)Ljava/lang/String;",

            &_executor_factory_ctor_id));

    Jni::LocalObject jtable_type;

    RETURN_IF_ERROR(_get_java_table_type(env, _conn_param.table_type, &jtable_type));

    Jni::LocalString executor_name;

    RETURN_IF_ERROR(

            _executor_factory_clazz.call_static_object_method(env, _executor_factory_ctor_id)

                    .with_arg(jtable_type)

                    .call(&executor_name));

    Jni::LocalStringBufferGuard executor_name_str;

    RETURN_IF_ERROR(executor_name.get_string_chars(env, &executor_name_str));

    RETURN_IF_ERROR(

            Jni::Util::get_jni_scanner_class(env, executor_name_str.get(), &_executor_clazz));

#undef GET_BASIC_JAVA_CLAZZ

    RETURN_IF_ERROR(_register_func_id(env));

    std::string driver_path;

    RETURN_IF_ERROR(_get_real_url(_conn_param.driver_path, &driver_path));

    TJdbcExecutorCtorParams ctor_params;

    ctor_params.__set_statement(_sql_str);

    ctor_params.__set_catalog_id(_conn_param.catalog_id);

    ctor_params.__set_jdbc_url(_conn_param.jdbc_url);

    ctor_params.__set_jdbc_user(_conn_param.user);

    ctor_params.__set_jdbc_password(_conn_param.passwd);

    ctor_params.__set_jdbc_driver_class(_conn_param.driver_class);

    ctor_params.__set_driver_path(driver_path);

    ctor_params.__set_jdbc_driver_checksum(_conn_param.driver_checksum);

    if (state == nullptr) {

        ctor_params.__set_batch_size(read ? 1 : 0);

    } else {

        ctor_params.__set_batch_size(read ? state->batch_size() : 0);

    }

    ctor_params.__set_op(read ? TJdbcOperation::READ : TJdbcOperation::WRITE);

    ctor_params.__set_table_type(_conn_param.table_type);

    ctor_params.__set_connection_pool_min_size(_conn_param.connection_pool_min_size);

    ctor_params.__set_connection_pool_max_size(_conn_param.connection_pool_max_size);

    ctor_params.__set_connection_pool_max_wait_time(_conn_param.connection_pool_max_wait_time);

    ctor_params.__set_connection_pool_max_life_time(_conn_param.connection_pool_max_life_time);

    ctor_params.__set_connection_pool_cache_clear_time(

            config::jdbc_connection_pool_cache_clear_time_sec);

    ctor_params.__set_connection_pool_keep_alive(_conn_param.connection_pool_keep_alive);

    ctor_params.__set_is_tvf(_conn_param.is_tvf);

    Jni::LocalArray ctor_params_bytes;

    RETURN_IF_ERROR(Jni::Util::SerializeThriftMsg(env, &ctor_params, &ctor_params_bytes));

    {

        SCOPED_RAW_TIMER(&_jdbc_statistic._init_connector_timer);

        RETURN_IF_ERROR(_executor_clazz.new_object(env, _executor_ctor_id)

                                .with_arg(ctor_params_bytes)

                                .call(&_executor_obj));

    }

    _is_open = true;

    RETURN_IF_ERROR(begin_trans());

    return Status::OK();

}

Status JdbcConnector::test_connection() {

    RETURN_IF_ERROR(open(nullptr, true));

    JNIEnv* env = nullptr;

    RETURN_IF_ERROR(Jni::Env::Get(&env));

    return _executor_obj

            .call_nonvirtual_void_method(env, _executor_clazz, _executor_test_connection_id)

            .call();

}

Status JdbcConnector::clean_datasource() {

    if (!_is_open) {

        return Status::OK();

    }

    JNIEnv* env = nullptr;

    RETURN_IF_ERROR(Jni::Env::Get(&env));

    return _executor_obj

            .call_nonvirtual_void_method(env, _executor_clazz, _executor_clean_datasource_id)

            .call();

}

Status JdbcConnector::query() {

    if (!_is_open) {

        return Status::InternalError("Query before open of JdbcConnector.");

    }

    // check materialize num equal

    auto materialize_num = _tuple_desc->slots().size();

    JNIEnv* env = nullptr;

    RETURN_IF_ERROR(Jni::Env::Get(&env));

    {

        SCOPED_RAW_TIMER(&_jdbc_statistic._execte_read_timer);

        jint colunm_count = 0;

        auto st = _executor_obj.call_nonvirtual_int_method(env, _executor_clazz, _executor_read_id)

                          .call(&colunm_count);

        if (!st.ok()) {

            return Status::InternalError("GetJniExceptionMsg meet error, query={}, msg={}",

                                         _conn_param.query_string, st.to_string());

        }

        if (colunm_count < materialize_num) {

            return Status::InternalError(

                    "JDBC query returned fewer columns ({}) than required ({}).", colunm_count,

                    materialize_num);

        }

    }

    LOG(INFO) << "JdbcConnector::query has exec success: " << _sql_str;

    return Status::OK();

}

Status JdbcConnector::get_next(bool* eos, Block* block, int batch_size) {

    SCOPED_RAW_TIMER(&_jdbc_statistic._get_data_timer); // Timer for the entire method

    if (!_is_open) {

        return Status::InternalError("get_next before open of jdbc connector.");

    }

    JNIEnv* env = nullptr;

    {

        SCOPED_RAW_TIMER(&_jdbc_statistic._jni_setup_timer); // Timer for setting up JNI environment

        RETURN_IF_ERROR(Jni::Env::Get(&env));

    } // _jni_setup_timer stops when going out of this scope

    jboolean has_next = JNI_FALSE;

    {

        SCOPED_RAW_TIMER(&_jdbc_statistic._has_next_timer); // Timer for hasNext check

        RETURN_IF_ERROR(

                _executor_obj

                        .call_nonvirtual_boolean_method(env, _executor_clazz, _executor_has_next_id)

                        .call(&has_next));

    } // _has_next_timer stops here

    if (has_next != JNI_TRUE) {

        *eos = true;

        return Status::OK();

    }

    auto column_size = _tuple_desc->slots().size();

    auto slots = _tuple_desc->slots();

    Jni::LocalObject map;

    {

        SCOPED_RAW_TIMER(&_jdbc_statistic._prepare_params_timer); // Timer for preparing params

        RETURN_IF_ERROR(_get_reader_params(block, env, column_size, &map));

    } // _prepare_params_timer stops here

    long address = 0;

    {

        SCOPED_RAW_TIMER(

                &_jdbc_statistic

                         ._read_and_fill_vector_table_timer); // Timer for getBlockAddress call

        RETURN_IF_ERROR(_executor_obj.call_long_method(env, _executor_get_block_address_id)

                                .with_arg(batch_size)

                                .with_arg(map)

                                .call(&address));

    } // _get_block_address_timer stops here

    std::vector<uint32_t> all_columns;

    for (uint32_t i = 0; i < column_size; ++i) {

        all_columns.push_back(i);

    }

    Status fill_block_status;

    {

        SCOPED_RAW_TIMER(&_jdbc_statistic._fill_block_timer); // Timer for fill_block

        fill_block_status = JniConnector::fill_block(block, all_columns, address);

    } // _fill_block_timer stops here

    if (!fill_block_status) {

        return fill_block_status;

    }

    Status cast_status;

    {

        SCOPED_RAW_TIMER(&_jdbc_statistic._cast_timer); // Timer for casting process

        cast_status = _cast_string_to_special(block, env, column_size);

    } // _cast_timer stops here

    return Status::OK();

}

Status JdbcConnector::append(Block* block, const VExprContextSPtrs& output_vexpr_ctxs,

                             uint32_t start_send_row, uint32_t* num_rows_sent,

                             TOdbcTableType::type table_type) {

    RETURN_IF_ERROR(exec_stmt_write(block, output_vexpr_ctxs, num_rows_sent));

    COUNTER_UPDATE(_sent_rows_counter, *num_rows_sent);

    return Status::OK();

}

Status JdbcConnector::exec_stmt_write(Block* block, const VExprContextSPtrs& output_vexpr_ctxs,

                                      uint32_t* num_rows_sent) {

    SCOPED_TIMER(_result_send_timer);

    JNIEnv* env = nullptr;

    RETURN_IF_ERROR(Jni::Env::Get(&env));

    // prepare table meta information

    std::unique_ptr<long[]> meta_data;

    RETURN_IF_ERROR(JniConnector::to_java_table(block, meta_data));

    long meta_address = (long)meta_data.get();

    auto table_schema = JniConnector::parse_table_schema(block);

    // prepare constructor parameters

    std::map<String, String> write_params = {{"meta_address", std::to_string(meta_address)},

                                             {"required_fields", table_schema.first},

                                             {"columns_types", table_schema.second}};

    Jni::LocalObject hashmap_object;

    RETURN_IF_ERROR(Jni::Util::convert_to_java_map(env, write_params, &hashmap_object));

    RETURN_IF_ERROR(

            _executor_obj.call_nonvirtual_int_method(env, _executor_clazz, _executor_stmt_write_id)

                    .with_arg(hashmap_object)

                    .call());

    *num_rows_sent = static_cast<uint32_t>(block->rows());

    return Status::OK();

}

Status JdbcConnector::begin_trans() {

    if (_use_tranaction) {

        JNIEnv* env = nullptr;

        RETURN_IF_ERROR(Jni::Env::Get(&env));

        RETURN_IF_ERROR(

                _executor_obj

                        .call_nonvirtual_void_method(env, _executor_clazz, _executor_begin_trans_id)

                        .call());

        _is_in_transaction = true;

    }

    return Status::OK();

}

Status JdbcConnector::abort_trans() {

    if (!_is_in_transaction) {

        return Status::InternalError("Abort transaction before begin trans.");

    }

    JNIEnv* env = nullptr;

    RETURN_IF_ERROR(Jni::Env::Get(&env));

    RETURN_IF_ERROR(

            _executor_obj

                    .call_nonvirtual_void_method(env, _executor_clazz, _executor_abort_trans_id)

                    .call());

    return Status::OK();

}

Status JdbcConnector::finish_trans() {

    if (_use_tranaction && _is_in_transaction) {

        JNIEnv* env = nullptr;

        RETURN_IF_ERROR(Jni::Env::Get(&env));

        RETURN_IF_ERROR(_executor_obj

                                .call_nonvirtual_void_method(env, _executor_clazz,

                                                             _executor_finish_trans_id)

                                .call());

        _is_in_transaction = false;

    }

    return Status::OK();

}

Status JdbcConnector::_register_func_id(JNIEnv* env) {

    RETURN_IF_ERROR(_executor_clazz.get_method(env, "<init>", JDBC_EXECUTOR_CTOR_SIGNATURE,

                                               &_executor_ctor_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(env, "write", JDBC_EXECUTOR_STMT_WRITE_SIGNATURE,

                                               &_executor_stmt_write_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(env, "read", "()I", &_executor_read_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(env, "close", JDBC_EXECUTOR_CLOSE_SIGNATURE,

                                               &_executor_close_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(env, "hasNext", JDBC_EXECUTOR_HAS_NEXT_SIGNATURE,

                                               &_executor_has_next_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(env, "getBlockAddress", "(ILjava/util/Map;)J",

                                               &_executor_get_block_address_id));

    RETURN_IF_ERROR(

            _executor_clazz.get_method(env, "getCurBlockRows", "()I", &_executor_block_rows_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(

            env, "openTrans", JDBC_EXECUTOR_TRANSACTION_SIGNATURE, &_executor_begin_trans_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(

            env, "commitTrans", JDBC_EXECUTOR_TRANSACTION_SIGNATURE, &_executor_finish_trans_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(

            env, "rollbackTrans", JDBC_EXECUTOR_TRANSACTION_SIGNATURE, &_executor_abort_trans_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(env, "testConnection", "()V",

                                               &_executor_test_connection_id));

    RETURN_IF_ERROR(_executor_clazz.get_method(env, "cleanDataSource", "()V",

                                               &_executor_clean_datasource_id));

    return Status::OK();

}

Status JdbcConnector::_get_reader_params(Block* block, JNIEnv* env, size_t column_size,

                                         Jni::LocalObject* ans) {

    std::ostringstream columns_nullable;

    std::ostringstream columns_replace_string;

    std::ostringstream required_fields;

    std::ostringstream columns_types;

    for (int i = 0; i < column_size; ++i) {

        auto* slot = _tuple_desc->slots()[i];

        auto type = slot->type();

        // Record if column is nullable

        columns_nullable << (slot->is_nullable() ? "true" : "false") << ",";

        // Check column type and replace accordingly

        std::string replace_type = "not_replace";

        if (type->get_primitive_type() == PrimitiveType::TYPE_BITMAP) {

            replace_type = "bitmap";

        } else if (type->get_primitive_type() == PrimitiveType::TYPE_HLL) {

            replace_type = "hll";

        } else if (type->get_primitive_type() == PrimitiveType::TYPE_JSONB) {

            replace_type = "jsonb";

        }

        columns_replace_string << replace_type << ",";

        if (replace_type != "not_replace") {

            block->get_by_position(i).column = std::make_shared<DataTypeString>()

                                                       ->create_column()

                                                       ->convert_to_full_column_if_const();

            block->get_by_position(i).type = std::make_shared<DataTypeString>();

            if (slot->is_nullable()) {

                block->get_by_position(i).column = make_nullable(block->get_by_position(i).column);

                block->get_by_position(i).type = make_nullable(block->get_by_position(i).type);

            }

        }

        // Record required fields and column types

        std::string field = slot->col_name();

        std::string jni_type;

        if (slot->type()->get_primitive_type() == PrimitiveType::TYPE_BITMAP ||

            slot->type()->get_primitive_type() == PrimitiveType::TYPE_HLL ||

            slot->type()->get_primitive_type() == PrimitiveType::TYPE_JSONB) {

            jni_type = "string";

        } else {

            jni_type = JniConnector::get_jni_type_with_different_string(slot->type());

        }

        required_fields << (i != 0 ? "," : "") << field;

        columns_types << (i != 0 ? "#" : "") << jni_type;

    }

    std::map<String, String> reader_params = {{"is_nullable", columns_nullable.str()},

                                              {"replace_string", columns_replace_string.str()},

                                              {"required_fields", required_fields.str()},

                                              {"columns_types", columns_types.str()}};

    return Jni::Util::convert_to_java_map(env, reader_params, ans);

}

Status JdbcConnector::_cast_string_to_special(Block* block, JNIEnv* env, size_t column_size) {

    for (size_t column_index = 0; column_index < column_size; ++column_index) {

        auto* slot_desc = _tuple_desc->slots()[column_index];

        jint num_rows = 0;

        RETURN_IF_ERROR(

                _executor_obj

                        .call_nonvirtual_int_method(env, _executor_clazz, _executor_block_rows_id)

                        .call(&num_rows));

        if (slot_desc->type()->get_primitive_type() == PrimitiveType::TYPE_HLL) {

            RETURN_IF_ERROR(_cast_string_to_hll(slot_desc, block, static_cast<int>(column_index),

                                                static_cast<int>(num_rows)));

        } else if (slot_desc->type()->get_primitive_type() == PrimitiveType::TYPE_JSONB) {

            RETURN_IF_ERROR(_cast_string_to_json(slot_desc, block, static_cast<int>(column_index),

                                                 static_cast<int>(num_rows)));

        } else if (slot_desc->type()->get_primitive_type() == PrimitiveType::TYPE_BITMAP) {

            RETURN_IF_ERROR(_cast_string_to_bitmap(slot_desc, block, static_cast<int>(column_index),

                                                   static_cast<int>(num_rows)));

        }

    }

    return Status::OK();

}

Status JdbcConnector::_cast_string_to_hll(const SlotDescriptor* slot_desc, Block* block,

                                          int column_index, int rows) {

    _map_column_idx_to_cast_idx_hll[column_index] =

            static_cast<int>(_input_hll_string_types.size());

    if (slot_desc->is_nullable()) {

        _input_hll_string_types.push_back(make_nullable(std::make_shared<DataTypeString>()));

    } else {

        _input_hll_string_types.push_back(std::make_shared<DataTypeString>());

    }

    DataTypePtr _target_data_type = slot_desc->get_data_type_ptr();

    std::string _target_data_type_name = _target_data_type->get_name();

    DataTypePtr _cast_param_data_type = _target_data_type;

    ColumnPtr _cast_param = _cast_param_data_type->create_column_const_with_default_value(1);

    auto& input_col = block->get_by_position(column_index).column;

    ColumnsWithTypeAndName argument_template;

    argument_template.reserve(2);

    argument_template.emplace_back(

            std::move(input_col),

            _input_hll_string_types[_map_column_idx_to_cast_idx_hll[column_index]],

            "java.sql.String");

    argument_template.emplace_back(_cast_param, _cast_param_data_type, _target_data_type_name);

    FunctionBasePtr func_cast = SimpleFunctionFactory::instance().get_function(

            "CAST", argument_template, make_nullable(_target_data_type));

    Block cast_block(argument_template);

    int result_idx = cast_block.columns();

    cast_block.insert({nullptr, make_nullable(_target_data_type), "cast_result"});

    RETURN_IF_ERROR(func_cast->execute(nullptr, cast_block, {0}, result_idx, rows));

    auto res_col = cast_block.get_by_position(result_idx).column;

    block->get_by_position(column_index).type = _target_data_type;

    if (_target_data_type->is_nullable()) {

        block->replace_by_position(column_index, res_col);

    } else {

        auto nested_ptr =

                reinterpret_cast<const ColumnNullable*>(res_col.get())->get_nested_column_ptr();

        block->replace_by_position(column_index, nested_ptr);

    }

    return Status::OK();

}

Status JdbcConnector::_cast_string_to_bitmap(const SlotDescriptor* slot_desc, Block* block,

                                             int column_index, int rows) {

    _map_column_idx_to_cast_idx_bitmap[column_index] =

            static_cast<int>(_input_bitmap_string_types.size());

    if (slot_desc->is_nullable()) {

        _input_bitmap_string_types.push_back(make_nullable(std::make_shared<DataTypeString>()));

    } else {

        _input_bitmap_string_types.push_back(std::make_shared<DataTypeString>());

    }

    DataTypePtr _target_data_type = slot_desc->get_data_type_ptr();

    std::string _target_data_type_name = _target_data_type->get_name();

    DataTypePtr _cast_param_data_type = _target_data_type;

    ColumnPtr _cast_param = _cast_param_data_type->create_column_const_with_default_value(1);

    auto& input_col = block->get_by_position(column_index).column;

    ColumnsWithTypeAndName argument_template;

    argument_template.reserve(2);

    argument_template.emplace_back(

            std::move(input_col),

            _input_bitmap_string_types[_map_column_idx_to_cast_idx_bitmap[column_index]],

            "java.sql.String");

    argument_template.emplace_back(_cast_param, _cast_param_data_type, _target_data_type_name);

    FunctionBasePtr func_cast = SimpleFunctionFactory::instance().get_function(

            "CAST", argument_template, make_nullable(_target_data_type));

    Block cast_block(argument_template);

    int result_idx = cast_block.columns();

    cast_block.insert({nullptr, make_nullable(_target_data_type), "cast_result"});

    RETURN_IF_ERROR(func_cast->execute(nullptr, cast_block, {0}, result_idx, rows));

    auto res_col = cast_block.get_by_position(result_idx).column;

    block->get_by_position(column_index).type = _target_data_type;

    if (_target_data_type->is_nullable()) {

        block->replace_by_position(column_index, res_col);

    } else {

        auto nested_ptr =

                reinterpret_cast<const ColumnNullable*>(res_col.get())->get_nested_column_ptr();

        block->replace_by_position(column_index, nested_ptr);

    }

    return Status::OK();

}

// Deprecated, this code is retained only for compatibility with query problems that may be encountered when upgrading the version that maps JSON to JSONB to this version, and will be deleted in subsequent versions.

Status JdbcConnector::_cast_string_to_json(const SlotDescriptor* slot_desc, Block* block,

                                           int column_index, int rows) {

    _map_column_idx_to_cast_idx_json[column_index] =

            static_cast<int>(_input_json_string_types.size());

    if (slot_desc->is_nullable()) {

        _input_json_string_types.push_back(make_nullable(std::make_shared<DataTypeString>()));

    } else {

        _input_json_string_types.push_back(std::make_shared<DataTypeString>());

    }

    DataTypePtr _target_data_type = slot_desc->get_data_type_ptr();

    std::string _target_data_type_name = _target_data_type->get_name();

    DataTypePtr _cast_param_data_type = _target_data_type;

    ColumnPtr _cast_param =

            _cast_param_data_type->create_column_const(1, Field::create_field<TYPE_STRING>("{}"));

    auto& input_col = block->get_by_position(column_index).column;

    ColumnsWithTypeAndName argument_template;

    argument_template.reserve(2);

    argument_template.emplace_back(

            std::move(input_col),

            _input_json_string_types[_map_column_idx_to_cast_idx_json[column_index]],

            "java.sql.String");

    argument_template.emplace_back(_cast_param, _cast_param_data_type, _target_data_type_name);

    FunctionBasePtr func_cast = SimpleFunctionFactory::instance().get_function(

            "CAST", argument_template, make_nullable(_target_data_type));

    Block cast_block(argument_template);

    int result_idx = cast_block.columns();

    cast_block.insert({nullptr, make_nullable(_target_data_type), "cast_result"});

    RETURN_IF_ERROR(func_cast->execute(nullptr, cast_block, {0}, result_idx, rows));

    auto res_col = cast_block.get_by_position(result_idx).column;

    block->get_by_position(column_index).type = _target_data_type;

    if (_target_data_type->is_nullable()) {

        block->replace_by_position(column_index, res_col);

    } else {

        auto nested_ptr =

                reinterpret_cast<const ColumnNullable*>(res_col.get())->get_nested_column_ptr();

        block->replace_by_position(column_index, nested_ptr);

    }

    return Status::OK();

}

Status JdbcConnector::_get_java_table_type(JNIEnv* env, TOdbcTableType::type table_type,

                                           Jni::LocalObject* java_enum_obj) {

    Jni::LocalClass enum_class;

    RETURN_IF_ERROR(

            Jni::Util::find_class(env, "org/apache/doris/thrift/TOdbcTableType", &enum_class));

    Jni::MethodId find_by_value_method;

    RETURN_IF_ERROR(enum_class.get_static_method(env, "findByValue",

                                                 "(I)Lorg/apache/doris/thrift/TOdbcTableType;",

                                                 &find_by_value_method));

    return enum_class.call_static_object_method(env, find_by_value_method)

            .with_arg(static_cast<jint>(table_type))

            .call(java_enum_obj);

}

Status JdbcConnector::_get_real_url(const std::string& url, std::string* result_url) {

    if (url.find(":/") == std::string::npos) {

        return _check_and_return_default_driver_url(url, result_url);

    }

    *result_url = url;

    return Status::OK();

}

Status JdbcConnector::_check_and_return_default_driver_url(const std::string& url,

                                                           std::string* result_url) {

    const char* doris_home = std::getenv("DORIS_HOME");

    std::string default_url = std::string(doris_home) + "/plugins/jdbc_drivers";

    std::string default_old_url = std::string(doris_home) + "/jdbc_drivers";

    if (config::jdbc_drivers_dir == default_url) {

        // If true, which means user does not set `jdbc_drivers_dir` and use the default one.

        // Because in 2.1.8, we change the default value of `jdbc_drivers_dir`

        // from `DORIS_HOME/jdbc_drivers` to `DORIS_HOME/plugins/jdbc_drivers`,

        // so we need to check the old default dir for compatibility.

        std::string target_path = default_url + "/" + url;

        std::string old_target_path = default_old_url + "/" + url;

        if (std::filesystem::exists(target_path)) {

            // File exists in new default directory

            *result_url = "file://" + target_path;

            return Status::OK();

        } else if (std::filesystem::exists(old_target_path)) {

            // File exists in old default directory

            *result_url = "file://" + old_target_path;

            return Status::OK();

        } else if (config::is_cloud_mode()) {

            // Cloud mode: try to download from cloud to new default directory

            std::string downloaded_path;

            Status status = CloudPluginDownloader::download_from_cloud(

                    CloudPluginDownloader::PluginType::JDBC_DRIVERS, url, target_path,

                    &downloaded_path);

            if (status.ok() && !downloaded_path.empty()) {

                *result_url = "file://" + downloaded_path;

                return Status::OK();

            }

            // Download failed, log warning but continue to fallback

            LOG(WARNING) << "Failed to download JDBC driver from cloud: " << status.to_string()

                         << ", fallback to old directory";

        } else {

            return Status::InternalError("JDBC driver file does not exist: " + url);

        }

    } else {

        // User specified custom directory - use directly

        *result_url = "file://" + config::jdbc_drivers_dir + "/" + url;

    }

    return Status::OK();

}

#include "common/compile_check_end.h"

} // namespace doris