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.

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

frontend: update query pruning #10026

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Conversation

krajorama
Copy link
Contributor

@krajorama krajorama commented Nov 26, 2024

What this PR does

The current method of excluding/including sub query results in PromQL by comparing to -Inf or +Inf is no longer valid after prometheus/prometheus#15245 due to comparison of native histograms to a float with < or > result in Jeanette's warning, not empty set.

To ease migration away from the old wrong logic, I'm adding the new logic first.

The new method uses logical AND operation to intersect the sub query with either a const vector() or an empty vector(). E.g.

subquery and on() (vector(1)==1)
subquery and on() (vector(-1)==1)

which become:

subquery
(vector(-1)==1)

Which issue(s) this PR fixes or relates to

Fixes N/A

Checklist

  • Tests updated.
  • N/A Documentation added.
  • CHANGELOG.md updated - the order of entries should be [CHANGE], [FEATURE], [ENHANCEMENT], [BUGFIX].
  • N/A about-versioning.md updated with experimental features.

@krajorama krajorama marked this pull request as ready for review November 26, 2024 10:35
@krajorama krajorama requested a review from a team as a code owner November 26, 2024 10:35
@krajorama krajorama marked this pull request as draft November 26, 2024 12:11
The current method of excluding/including sub query results in PromQL
by comparing to -Inf or +Inf is no longer valid after
prometheus/prometheus#15245
due to comparison of native histograms to a float with < or > result
in Jeanette's warning, not empty set.

To ease migrating to the correct version, I'm not removing the old
prune code yet, just adding the new method.

The new method uses logical AND operation to intersect the sub query with
either a const vector() or an empty vector(). E.g.

subquery and on() (vector(1)==1)
subquery and on() (vector(-1)==1)

which become:

subquery
(vector(-1)==1)

Signed-off-by: György Krajcsovits <[email protected]>
@krajorama krajorama marked this pull request as ready for review November 26, 2024 12:46
Copy link
Contributor

@charleskorn charleskorn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think we can further prune the expression in some cases:

{
// "and on()" is not on top level, "or" has lower precedence.
`(avg(rate(foo[1m]))) and on() (vector(0) == 1) or avg(rate(bar[1m]))`,
`(vector(0) == 1) or avg(rate(bar[1m]))`,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Couldn't we simplify this further, to avg(rate(bar[1m]))?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

That's more complicated. The main point is to avoid loading chunks and (vector(0)==1) doesn't load chunks, so should be very efficient.
We can add that logic in a new PR, seems to be independent from this optimization, wdyt?
But having two algorithms raises the question of whether they need to be run repeatedly to optimize away everything, so it gets even more complicated.

{
// "and on()" is not on top level, due to left-right associativity.
`(avg(rate(foo[1m]))) and on() (vector(0) == 1) and avg(rate(bar[1m]))`,
`(vector(0) == 1) and avg(rate(bar[1m]))`,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Couldn't we simplify this further, to vector(0) == 1?

},
{ // The const expression is on the wrong side.
`(vector(0) == 1) and on() (avg(rate(foo[1m])))`,
`(vector(0) == 1) and on() (avg(rate(foo[1m])))`,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same as below - I think this can be simplified to vector(0) == 1.

},
{ // Matching on labels.
`(avg(rate(foo[1m]))) and on(id) (vector(0) == 1)`,
`(avg(rate(foo[1m]))) and on(id) (vector(0) == 1)`,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Couldn't this be simplified to vector(0) == 1? The right side will never produce a result, so we'll never return anything from the left side.

},
{ // Not "on" expression.
`(avg(rate(foo[1m]))) and ignoring() (vector(0) == 1)`,
`(avg(rate(foo[1m]))) and ignoring() (vector(0) == 1)`,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same as above - the right side will never produce a result.

Copy link
Contributor

@beorn7 beorn7 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good at first glance. (I assume you don't want a code-level review from me, just that the idea is correct.)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants