[PoC] Implementation of distinct in Window Aggregates

Started by Ankit Kumar Pandeyover 3 years ago7 messageshackers
Jump to latest
#1Ankit Kumar Pandey
itsankitkp@gmail.com

Hi,

This is a PoC patch which implements distinct operation in window
aggregates (without order by and for single column aggregation, final
version may vary wrt these limitations). Purpose of this PoC is to get
feedback on the approach used and corresponding implementation, any
nitpicking as deemed reasonable.

Distinct operation is mirrored from implementation in nodeAgg. Existing
partitioning logic determines if row is in partition and when distinct is
required, all tuples for the aggregate column are stored in tuplesort. When
finalize_windowaggregate gets called, tuples are sorted and duplicates are
removed, followed by calling the transition function on each tuple.
When distinct is not required, the above process is skipped and the
transition function gets called directly and nothing gets inserted into
tuplesort.
Note: For each partition, in tuplesort_begin and tuplesort_end is involved
to rinse tuplesort, so at any time, max tuples in tuplesort is equal to
tuples in a particular partition.

I have verified it for interger and interval column aggregates (to rule out
obvious issues related to data types).

Sample cases:

create table mytable(id int, name text);
insert into mytable values(1, 'A');
insert into mytable values(1, 'A');
insert into mytable values(5, 'B');
insert into mytable values(3, 'A');
insert into mytable values(1, 'A');

select avg(distinct id) over (partition by name) from mytable;
avg
--------------------
2.0000000000000000
2.0000000000000000
2.0000000000000000
2.0000000000000000
5.0000000000000000

select avg(id) over (partition by name) from mytable;
avg
--------------------
1.5000000000000000
1.5000000000000000
1.5000000000000000
1.5000000000000000
5.0000000000000000

select avg(distinct id) over () from mytable;
avg
--------------------
3.0000000000000000
3.0000000000000000
3.0000000000000000
3.0000000000000000
3.0000000000000000

select avg(distinct id) from mytable;
avg
--------------------
3.0000000000000000

This is my first-time contribution. Please let me know if anything can be
improved as I`m eager to learn.

Regards,
Ankit Kumar Pandey

Attachments:

v1-0001-Implement-distinct-in-Window-Aggregates.patchtext/x-patch; charset=US-ASCII; name=v1-0001-Implement-distinct-in-Window-Aggregates.patchDownload+258-41
#2Ankit Kumar Pandey
itsankitkp@gmail.com
In reply to: Ankit Kumar Pandey (#1)
Re: [PoC] Implementation of distinct in Window Aggregates

On 24/12/22 18:22, Ankit Pandey wrote:

Hi,

This is a PoC patch which implements distinct operation in window
aggregates (without order by and for single column aggregation, final
version may vary wrt these limitations). Purpose of this PoC is to get
feedback on the approach used and corresponding implementation, any
nitpicking as deemed reasonable.

Distinct operation is mirrored from implementation in nodeAgg.
Existing partitioning logic determines if row is in partition and when
distinct is required, all tuples for the aggregate column are stored
in tuplesort. When finalize_windowaggregate gets called, tuples are
sorted and duplicates are removed, followed by calling the transition
function on each tuple.
When distinct is not required, the above process is skipped and the
transition function gets called directly and nothing gets inserted
into tuplesort.
Note: For each partition, in tuplesort_begin and tuplesort_end is
involved to rinse tuplesort, so at any time, max tuples in tuplesort
is equal to tuples in a particular partition.

I have verified it for interger and interval column aggregates (to
rule out obvious issues related to data types).

Sample cases:

create table mytable(id int, name text);
insert into mytable values(1, 'A');
insert into mytable values(1, 'A');
insert into mytable values(5, 'B');
insert into mytable values(3, 'A');
insert into mytable values(1, 'A');

select avg(distinct id) over (partition by name) from mytable;
        avg
--------------------
2.0000000000000000
2.0000000000000000
2.0000000000000000
2.0000000000000000
5.0000000000000000

select avg(id) over (partition by name) from mytable;
        avg
--------------------
 1.5000000000000000
 1.5000000000000000
 1.5000000000000000
 1.5000000000000000
 5.0000000000000000

select avg(distinct id) over () from mytable;
        avg
--------------------
 3.0000000000000000
 3.0000000000000000
 3.0000000000000000
 3.0000000000000000
 3.0000000000000000

select avg(distinct id)  from mytable;
        avg
--------------------
 3.0000000000000000

This is my first-time contribution. Please let me know if anything can be
improved as I`m eager to learn.

Regards,
Ankit Kumar Pandey

Hi all,

I know everyone is busy with holidays (well, Happy Holidays!) but I will
be glad if someone can take a quick look at this PoC and share thoughts.

This is my first time contribution so I am pretty sure there will be
some very obvious feedbacks (which will help me to move forward with
this change).

--
Regards,
Ankit Kumar Pandey

#3Ankit Kumar Pandey
itsankitkp@gmail.com
In reply to: Ankit Kumar Pandey (#2)
Re: [PoC] Implementation of distinct in Window Aggregates

On 29/12/22 20:58, Ankit Kumar Pandey wrote:

On 24/12/22 18:22, Ankit Pandey wrote:

Hi,

This is a PoC patch which implements distinct operation in window
aggregates (without order by and for single column aggregation, final
version may vary wrt these limitations). Purpose of this PoC is to
get feedback on the approach used and corresponding implementation,
any nitpicking as deemed reasonable.

Distinct operation is mirrored from implementation in nodeAgg.
Existing partitioning logic determines if row is in partition and
when distinct is required, all tuples for the aggregate column are
stored in tuplesort. When finalize_windowaggregate gets called,
tuples are sorted and duplicates are removed, followed by calling the
transition function on each tuple.
When distinct is not required, the above process is skipped and the
transition function gets called directly and nothing gets inserted
into tuplesort.
Note: For each partition, in tuplesort_begin and tuplesort_end is
involved to rinse tuplesort, so at any time, max tuples in tuplesort
is equal to tuples in a particular partition.

I have verified it for interger and interval column aggregates (to
rule out obvious issues related to data types).

Sample cases:

create table mytable(id int, name text);
insert into mytable values(1, 'A');
insert into mytable values(1, 'A');
insert into mytable values(5, 'B');
insert into mytable values(3, 'A');
insert into mytable values(1, 'A');

select avg(distinct id) over (partition by name) from mytable;
        avg
--------------------
2.0000000000000000
2.0000000000000000
2.0000000000000000
2.0000000000000000
5.0000000000000000

select avg(id) over (partition by name) from mytable;
        avg
--------------------
 1.5000000000000000
 1.5000000000000000
 1.5000000000000000
 1.5000000000000000
 5.0000000000000000

select avg(distinct id) over () from mytable;
        avg
--------------------
 3.0000000000000000
 3.0000000000000000
 3.0000000000000000
 3.0000000000000000
 3.0000000000000000

select avg(distinct id)  from mytable;
        avg
--------------------
 3.0000000000000000

This is my first-time contribution. Please let me know if anything
can be
improved as I`m eager to learn.

Regards,
Ankit Kumar Pandey

Hi all,

I know everyone is busy with holidays (well, Happy Holidays!) but I
will be glad if someone can take a quick look at this PoC and share
thoughts.

This is my first time contribution so I am pretty sure there will be
some very obvious feedbacks (which will help me to move forward with
this change).

Updated patch with latest master. Last patch was an year old.

--
Regards,
Ankit Kumar Pandey

Attachments:

v1-0002-Implement-distinct-in-Window-Aggregates.patchtext/x-patch; charset=UTF-8; name=v1-0002-Implement-distinct-in-Window-Aggregates.patchDownload+257-40
#4Ankit Kumar Pandey
itsankitkp@gmail.com
In reply to: Ankit Kumar Pandey (#3)
Re: [PoC] Implementation of distinct in Window Aggregates

On 04/01/23 18:10, Ankit Kumar Pandey wrote:

On 29/12/22 20:58, Ankit Kumar Pandey wrote:

On 24/12/22 18:22, Ankit Pandey wrote:

Hi,

This is a PoC patch which implements distinct operation in window
aggregates (without order by and for single column aggregation, final
version may vary wrt these limitations). Purpose of this PoC is to
get feedback on the approach used and corresponding implementation,
any nitpicking as deemed reasonable.

Distinct operation is mirrored from implementation in nodeAgg.
Existing partitioning logic determines if row is in partition and
when distinct is required, all tuples for the aggregate column are
stored in tuplesort. When finalize_windowaggregate gets called,
tuples are sorted and duplicates are removed, followed by calling the
transition function on each tuple.
When distinct is not required, the above process is skipped and the
transition function gets called directly and nothing gets inserted
into tuplesort.
Note: For each partition, in tuplesort_begin and tuplesort_end is
involved to rinse tuplesort, so at any time, max tuples in tuplesort
is equal to tuples in a particular partition.

I have verified it for interger and interval column aggregates (to
rule out obvious issues related to data types).

Sample cases:

create table mytable(id int, name text);
insert into mytable values(1, 'A');
insert into mytable values(1, 'A');
insert into mytable values(5, 'B');
insert into mytable values(3, 'A');
insert into mytable values(1, 'A');

select avg(distinct id) over (partition by name) from mytable;
        avg
--------------------
2.0000000000000000
2.0000000000000000
2.0000000000000000
2.0000000000000000
5.0000000000000000

select avg(id) over (partition by name) from mytable;
        avg
--------------------
 1.5000000000000000
 1.5000000000000000
 1.5000000000000000
 1.5000000000000000
 5.0000000000000000

select avg(distinct id) over () from mytable;
        avg
--------------------
 3.0000000000000000
 3.0000000000000000
 3.0000000000000000
 3.0000000000000000
 3.0000000000000000

select avg(distinct id)  from mytable;
        avg
--------------------
 3.0000000000000000

This is my first-time contribution. Please let me know if anything
can be
improved as I`m eager to learn.

Regards,
Ankit Kumar Pandey

Hi all,

I know everyone is busy with holidays (well, Happy Holidays!) but I
will be glad if someone can take a quick look at this PoC and share
thoughts.

This is my first time contribution so I am pretty sure there will be
some very obvious feedbacks (which will help me to move forward with
this change).

Updated patch with latest master. Last patch was an year old.

Attaching patch with rebase from latest HEAD

Thanks,

Ankit

Attachments:

distinct_windows.patchtext/x-patch; charset=UTF-8; name=distinct_windows.patchDownload+40-257
#5Ankit Kumar Pandey
itsankitkp@gmail.com
In reply to: Ankit Kumar Pandey (#4)
Re: [PoC] Implementation of distinct in Window Aggregates

Attaching updated patch with a fix for an issue in window function.

I have also fixed naming convention of patch as last patch had
incompatible name.

Note:

1. Pending: Investigation of test cases failures.

Regards,

Ankit

Attachments:

v1-0003-Implement-distinct-in-Window-Aggregates.patchtext/x-patch; charset=UTF-8; name=v1-0003-Implement-distinct-in-Window-Aggregates.patchDownload+131-303
#6Andreas Karlsson
andreas.karlsson@percona.com
In reply to: Ankit Kumar Pandey (#5)
Re: [PoC] Implementation of distinct in Window Aggregates

On 3/12/23 09:17, Ankit Kumar Pandey wrote:

Attaching updated patch with a fix for an issue in window function.

I have also fixed naming convention of patch as last patch had
incompatible name.

Hi,

This patch does not apply to master. Could you rebase it and submit it
as one patch which applies directly to master? Maybe I am wrong but the
latest version looks like it only applies on top of one of your previous
patches which makes it hard for the reviewer.

Andreas

#7Daniel Gustafsson
daniel@yesql.se
In reply to: Andreas Karlsson (#6)
Re: [PoC] Implementation of distinct in Window Aggregates

On 11 Jul 2023, at 01:06, Andreas Karlsson <andreas@proxel.se> wrote:

On 3/12/23 09:17, Ankit Kumar Pandey wrote:

Attaching updated patch with a fix for an issue in window function.
I have also fixed naming convention of patch as last patch had incompatible name.

Hi,

This patch does not apply to master. Could you rebase it and submit it as one patch which applies directly to master? Maybe I am wrong but the latest version looks like it only applies on top of one of your previous patches which makes it hard for the reviewer.

Since no update was posted, the patch was considered a PoC and the thread has
stalled, I will mark this returned with feedback. Please feel free to reopen a
new CF entry when there is a new patch available which addresses Andreas'
feedback on patch structure.

--
Daniel Gustafsson