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

// This file is copied from

// https://github.com/ClickHouse/ClickHouse/blob/master/src/AggregateFunctions/AggregateFunctionUniq.cpp

// and modified by Doris

#include "exprs/aggregate/aggregate_function_uniq.h"

#include <string>

#include "common/exception.h"

#include "core/data_type/data_type.h"

#include "core/data_type/data_type_variant_v2.h"

#include "core/data_type/define_primitive_type.h"

#include "exec/common/hash_table/hash.h" // IWYU pragma: keep

#include "exprs/aggregate/aggregate_function_simple_factory.h"

#include "exprs/aggregate/helpers.h"

namespace doris {

template <template <PrimitiveType> class Data>

AggregateFunctionPtr create_aggregate_function_uniq(const std::string& name,

                                                    const DataTypes& argument_types,

                                                    const DataTypePtr& result_type,

                                                    const bool result_is_nullable,

                                                    const AggregateFunctionAttr& attr) {

    if (argument_types.size() == 1) {

        const DataTypePtr argument_type = remove_nullable(argument_types[0]);

        if (argument_type->get_primitive_type() == TYPE_VARIANT) {

            DORIS_CHECK(dynamic_cast<const DataTypeVariantV2*>(argument_type.get()) != nullptr);

            return creator_without_type::create<AggregateFunctionUniqVariant>(

                    argument_types, result_is_nullable, attr);

        }

    }

    return creator_with_type_list<

            TYPE_BOOLEAN, TYPE_TINYINT, TYPE_SMALLINT, TYPE_INT, TYPE_BIGINT, TYPE_LARGEINT,

            TYPE_DECIMAL32, TYPE_DECIMAL64, TYPE_DECIMAL128I, TYPE_DECIMAL256, TYPE_VARCHAR,

            TYPE_ARRAY, TYPE_FLOAT, TYPE_DOUBLE, TYPE_DATEV2, TYPE_DATETIMEV2, TYPE_TIMESTAMP_NS,

            TYPE_TIMESTAMPTZ, TYPE_VARBINARY,

            TYPE_UUID>::create<AggregateFunctionUniq, Data>(argument_types, result_is_nullable,

                                                            attr);

}

void register_aggregate_function_uniq(AggregateFunctionSimpleFactory& factory) {

    AggregateFunctionCreator creator =

            create_aggregate_function_uniq<AggregateFunctionUniqExactData>;

    factory.register_function_both("multi_distinct_count", creator);

}

} // namespace doris