Method randomAggregateType

src/Client/BuzzHouse/Generator/SQLTypes.cpp:1783–1831

}

std::unique_ptr<SQLType> StatementGenerator::randomAggregateType(RandomGenerator & rg, const bool simple, BottomTypeName * tp)
{
    uint32_t col_counter2 = 0;
    std::vector<AggregateParam> params;
    std::vector<std::unique_ptr<SQLType>> subtypes;
    AggregateFunction * af = tp ? tp->mutable_aggr() : nullptr;
    const CHAggregate & agg = rg.pickRandomly(simple ? simple_det_aggrs : det_aggrs);
    std::string aggr = agg.fname;
    uint32_t nargs = agg.min_args == agg.max_args ? agg.min_args : rg.randomInt<uint32_t>(agg.min_args, agg.max_args);

    if (agg.min_params > 0)
    {
        const uint32_t nparams = agg.min_params == agg.max_params
            ? agg.min_params
            : rg.randomInt<uint32_t>(agg.min_params, std::min(agg.max_params, UINT32_C(5)));
        for (uint32_t i = 0; i < nparams; i++)
        {
            AggregateParam p;
            if (rg.nextBool())
                p.set_float_param(rg.randomInt<uint32_t>(0, 100) / 100.0);
            else
                p.set_int_param(rg.randomInt<uint64_t>(0, 100));
            params.push_back(p);
            if (af)
                *af->add_params() = p;
        }
    }
    if (nargs == 0 && (simple || this->depth >= this->fc.max_depth))
    {
        aggr = "any";
        nargs = 1;
        params.clear();
        if (af)
            af->clear_params();
    }
    for (uint32_t i = 0; i < nargs; i++)
    {
        this->depth++;
        subtypes.emplace_back(
            this->randomNextType(rg, this->next_type_mask & ~(allow_nested), col_counter2, tp ? af->add_types() : nullptr));
        this->depth--;
    }
    if (tp)
    {
        af->set_simple(simple);
        af->set_aggr(aggr);
    }
    return std::make_unique<AggregateFunctionType>(simple, std::move(aggr), std::move(params), std::move(subtypes));
}

std::unique_ptr<SQLType>
StatementGenerator::bottomType(RandomGenerator & rg, const uint64_t allowed_types, const bool low_card, BottomTypeName * tp)