drop table if exists test.summing_merge_tree_aggregate_function;
drop table if exists test.summing_merge_tree_null;

---- partition merge
create table test.summing_merge_tree_aggregate_function (
	d Date,
	k UInt64,
	u AggregateFunction(uniq, UInt64)
) engine=SummingMergeTree(d, k, 1);

insert into test.summing_merge_tree_aggregate_function
select today() as d,
       number as k,
       uniqState(toUInt64(number % 500))
from numbers(5000)
group by d, k;

insert into test.summing_merge_tree_aggregate_function
select today() as d,
       number + 5000 as k,
       uniqState(toUInt64(number % 500))
from numbers(5000)
group by d, k;

select count() from test.summing_merge_tree_aggregate_function;
optimize table test.summing_merge_tree_aggregate_function;
select count() from test.summing_merge_tree_aggregate_function;

drop table test.summing_merge_tree_aggregate_function;

---- sum + uniq + uniqExact
create table test.summing_merge_tree_aggregate_function (
	d materialized today(),
	k UInt64,
	c UInt64,
	u AggregateFunction(uniq, UInt8),
	ue AggregateFunction(uniqExact, UInt8)
) engine=SummingMergeTree(d, k, 8192);

insert into test.summing_merge_tree_aggregate_function select 1, 1, uniqState(1), uniqExactState(1);
insert into test.summing_merge_tree_aggregate_function select 1, 1, uniqState(2), uniqExactState(2);
insert into test.summing_merge_tree_aggregate_function select 1, 1, uniqState(3), uniqExactState(2);
insert into test.summing_merge_tree_aggregate_function select 1, 1, uniqState(1), uniqExactState(1);
insert into test.summing_merge_tree_aggregate_function select 1, 1, uniqState(2), uniqExactState(2);
insert into test.summing_merge_tree_aggregate_function select 1, 1, uniqState(3), uniqExactState(3);

select
	k, sum(c),
	uniqMerge(u), uniqExactMerge(ue)
from test.summing_merge_tree_aggregate_function group by k;

optimize table test.summing_merge_tree_aggregate_function;

select
	k, sum(c),
	uniqMerge(u), uniqExactMerge(ue)
from test.summing_merge_tree_aggregate_function group by k;

drop table test.summing_merge_tree_aggregate_function;

---- sum + topK
create table test.summing_merge_tree_aggregate_function (d materialized today(), k UInt64, c UInt64, x AggregateFunction(topK(2), UInt8)) engine=SummingMergeTree(d, k, 8192);

insert into test.summing_merge_tree_aggregate_function select 1, 1, topKState(2)(1);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKState(2)(2);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKState(2)(2);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKState(2)(3);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKState(2)(3);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKState(2)(3);
select k, sum(c), topKMerge(2)(x) from test.summing_merge_tree_aggregate_function group by k;
optimize table test.summing_merge_tree_aggregate_function;
select k, sum(c), topKMerge(2)(x) from test.summing_merge_tree_aggregate_function group by k;

drop table test.summing_merge_tree_aggregate_function;

---- sum + topKWeighed
create table test.summing_merge_tree_aggregate_function (d materialized today(), k UInt64, c UInt64, x AggregateFunction(topKWeighed(2), UInt8, UInt8)) engine=SummingMergeTree(d, k, 8192);

insert into test.summing_merge_tree_aggregate_function select 1, 1, topKWeighedState(2)(1, 1);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKWeighedState(2)(1, 1);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKWeighedState(2)(1, 1);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKWeighedState(2)(2, 2);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKWeighedState(2)(2, 2);
insert into test.summing_merge_tree_aggregate_function select 1, 1, topKWeighedState(2)(3, 5);
select k, sum(c), topKWeighedMerge(2)(x) from test.summing_merge_tree_aggregate_function group by k;
optimize table test.summing_merge_tree_aggregate_function;
select k, sum(c), topKWeighedMerge(2)(x) from test.summing_merge_tree_aggregate_function group by k;

drop table test.summing_merge_tree_aggregate_function;

---- avg
create table test.summing_merge_tree_aggregate_function (d materialized today(), k UInt64, x AggregateFunction(avg, Float64)) engine=SummingMergeTree(d, k, 8192);

insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.0);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.1);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.2);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.3);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.4);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.5);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.6);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.7);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.8);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(0.9);
insert into test.summing_merge_tree_aggregate_function select 1, avgState(1.0);
select k, avgMerge(x) from test.summing_merge_tree_aggregate_function group by k;
optimize table test.summing_merge_tree_aggregate_function;
select k, avgMerge(x) from test.summing_merge_tree_aggregate_function group by k;

drop table test.summing_merge_tree_aggregate_function;

---- quantile
create table test.summing_merge_tree_aggregate_function (d materialized today(), k UInt64, x AggregateFunction(quantile(0.1), Float64)) engine=SummingMergeTree(d, k, 8192);

insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.0);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.1);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.2);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.3);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.4);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.5);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.6);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.7);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.8);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(0.9);
insert into test.summing_merge_tree_aggregate_function select 1, quantileState(0.1)(1.0);
select k, quantileMerge(0.1)(x) from test.summing_merge_tree_aggregate_function group by k;
optimize table test.summing_merge_tree_aggregate_function;
select k, quantileMerge(0.1)(x) from test.summing_merge_tree_aggregate_function group by k;

drop table test.summing_merge_tree_aggregate_function;

---- sum + uniq with more data
create table test.summing_merge_tree_null (
    d materialized today(),
    k UInt64,
    c UInt64,
    u UInt64
) engine=Null;

create materialized view test.summing_merge_tree_aggregate_function (
	d materialized today(),
	k UInt64,
	c UInt64,
	u AggregateFunction(uniq, UInt64)
) engine=SummingMergeTree(d, k, 8192)
as select d, k, sum(c) as c, uniqState(u) as u
from test.summing_merge_tree_null
group by d, k;

-- prime number 53 to avoid resonanse between %3 and %53
insert into test.summing_merge_tree_null select number % 3, 1, number % 53 from numbers(999999);

select k, sum(c), uniqMerge(u) from test.summing_merge_tree_aggregate_function group by k order by k;
optimize table test.summing_merge_tree_aggregate_function;
select k, sum(c), uniqMerge(u) from test.summing_merge_tree_aggregate_function group by k order by k;

drop table test.summing_merge_tree_aggregate_function;
drop table test.summing_merge_tree_null;