Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Pushdown simplest sorting for aggregate command #2530

Merged
merged 26 commits into from
May 3, 2023
Merged

Pushdown simplest sorting for aggregate command #2530

Changes from 1 commit
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
e13337b
wip
noisersup Apr 24, 2023
12d2d3d
wip
noisersup Apr 25, 2023
a31ebe7
wip
noisersup Apr 25, 2023
787b05a
wip
noisersup Apr 26, 2023
00d97d9
wip
noisersup Apr 26, 2023
78605d0
wip
noisersup Apr 27, 2023
5b97891
wip
noisersup Apr 27, 2023
281bed0
wip
noisersup Apr 28, 2023
8770a28
Merge remote-tracking branch 'upstream/main' into pushdown-sort-2438
noisersup Apr 28, 2023
e3a2333
wip
noisersup Apr 28, 2023
256ee3b
wip
noisersup Apr 28, 2023
47baa37
wip
noisersup Apr 28, 2023
773fb94
Merge branch 'main' into pushdown-sort-2438
noisersup Apr 28, 2023
0883777
Merge branch 'main' into pushdown-sort-2438
mergify[bot] Apr 28, 2023
d34e056
wip
noisersup Apr 28, 2023
e5ffaea
wip
noisersup Apr 28, 2023
5c93458
wip
noisersup Apr 28, 2023
ae05028
wip
noisersup Apr 28, 2023
50497e1
wip
rumyantseva May 2, 2023
ccfbe61
wip
rumyantseva May 2, 2023
79c9dd1
wip
rumyantseva May 2, 2023
40c99dd
wip
rumyantseva May 2, 2023
ebbe58f
wip
rumyantseva May 2, 2023
2013ab4
wip
rumyantseva May 3, 2023
329306e
wip
rumyantseva May 3, 2023
a81592e
Merge branch 'main' into pushdown-aggregation-sort-2438
May 3, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
wip
  • Loading branch information
@rumyantseva
rumyantseva committed May 2, 2023
commit ccfbe614e5fb09e8716379b318908625e22861e7
36 changes: 20 additions & 16 deletions internal/handlers/common/aggregations/stages/pushdown.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,23 +19,26 @@ import (
"github.com/FerretDB/FerretDB/internal/util/must"
)

// GetPushdownQuery gets pushdown query for aggregation.
// When the first two aggregation stages are $match or $sort they are
// used for pushdown, and returned in order: $match, $sort, otherwise nil is return.
// If any of those stages repeat in first two stages, it also returns nil.
func GetPushdownQuery(stagesDocs []any) (*types.Document, *types.Document) {
// GetPushdownQuery gets pushdown query ($match and $sort) for aggregation.
//
// If the first two stages are either $match, $sort, or a combination of them, we can push them down.
// In this case, we return the first match and sort statements to pushdown.
// If $match stage is not present, match is returned as nil.
// If $sort stage is not present, sort is returned as nil.
func GetPushdownQuery(stagesDocs []any) (match *types.Document, sort *types.Document) {
if len(stagesDocs) == 0 {
return nil, nil
return
}

var match, sort *types.Document
stagesToPushdown := []any{stagesDocs[0]}

for i, doc := range stagesDocs {
if i > 2 {
break
}
if len(stagesDocs) > 1 {
stagesToPushdown = append(stagesToPushdown, stagesDocs[1])
}

for _, s := range stagesToPushdown {
stage, isDoc := s.(*types.Document)

stage, isDoc := doc.(*types.Document)
if !isDoc {
return nil, nil
}
Expand All @@ -46,7 +49,7 @@ func GetPushdownQuery(stagesDocs []any) (*types.Document, *types.Document) {
query, isDoc := matchQuery.(*types.Document)

if !isDoc || match != nil {
return nil, nil
continue
}

match = query
Expand All @@ -56,15 +59,16 @@ func GetPushdownQuery(stagesDocs []any) (*types.Document, *types.Document) {
query, isDoc := sortQuery.(*types.Document)

if !isDoc || sort != nil {
return nil, nil
continue
}

sort = query

default:
return nil, nil
// not $match nor $sort, we shouldn't continue pushdown
return
}
}

return match, sort
return
}