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

Improve typing during query planning #16310

Merged
merged 5 commits into from
Jul 4, 2024
Merged

Conversation

systay
Copy link
Collaborator

@systay systay commented Jul 2, 2024

Description

We've long had pretty bad typing in the semantic analysis, and pretty great typing in the evalengine. This PR tries to connect the two.

The only typing during semantic analysis that we'll do is to copy information over from the VSchema, and fill the types of column names known to the vschema.

Actual typing will happen when the planner asks for type info, and not before that. The typing will be more expensive, but will only be done when it's really needed.

The PR contains TODOs about enabling the type logic for aggregate functions as well. At the moment we don't calculate these types correctly, so let's revisit once that is done.

Related Issue(s)

Checklist

  • "Backport to:" labels have been added if this change should be back-ported to release branches
  • If this change is to be back-ported to previous releases, a justification is included in the PR description
  • Tests were added or are not required
  • Did the new or modified tests pass consistently locally and on CI?
  • Documentation was added or is not required

Deployment Notes

Signed-off-by: Andres Taylor <andres@planetscale.com>
Copy link
Contributor

vitess-bot bot commented Jul 2, 2024

Review Checklist

Hello reviewers! 👋 Please follow this checklist when reviewing this Pull Request.

General

  • Ensure that the Pull Request has a descriptive title.
  • Ensure there is a link to an issue (except for internal cleanup and flaky test fixes), new features should have an RFC that documents use cases and test cases.

Tests

  • Bug fixes should have at least one unit or end-to-end test, enhancement and new features should have a sufficient number of tests.

Documentation

  • Apply the release notes (needs details) label if users need to know about this change.
  • New features should be documented.
  • There should be some code comments as to why things are implemented the way they are.
  • There should be a comment at the top of each new or modified test to explain what the test does.

New flags

  • Is this flag really necessary?
  • Flag names must be clear and intuitive, use dashes (-), and have a clear help text.

If a workflow is added or modified:

  • Each item in Jobs should be named in order to mark it as required.
  • If the workflow needs to be marked as required, the maintainer team must be notified.

Backward compatibility

  • Protobuf changes should be wire-compatible.
  • Changes to _vt tables and RPCs need to be backward compatible.
  • RPC changes should be compatible with vitess-operator
  • If a flag is removed, then it should also be removed from vitess-operator and arewefastyet, if used there.
  • vtctl command output order should be stable and awk-able.

@vitess-bot vitess-bot bot added NeedsBackportReason If backport labels have been applied to a PR, a justification is required NeedsDescriptionUpdate The description is not clear or comprehensive enough, and needs work NeedsIssue A linked issue is missing for this Pull Request NeedsWebsiteDocsUpdate What it says labels Jul 2, 2024
@systay systay added Type: Enhancement Logical improvement (somewhere between a bug and feature) Component: Query Serving and removed NeedsDescriptionUpdate The description is not clear or comprehensive enough, and needs work NeedsWebsiteDocsUpdate What it says NeedsIssue A linked issue is missing for this Pull Request NeedsBackportReason If backport labels have been applied to a PR, a justification is required labels Jul 2, 2024
@systay systay changed the title feat: use the evalengine typer to get types for most expressions Improve typing during query planning Jul 2, 2024
@github-actions github-actions bot added this to the v21.0.0 milestone Jul 2, 2024
Copy link

codecov bot commented Jul 2, 2024

Codecov Report

Attention: Patch coverage is 54.28571% with 32 lines in your changes missing coverage. Please review.

Project coverage is 68.71%. Comparing base (694a0cf) to head (78d9067).

Files Patch % Lines
...vtgate/planbuilder/plancontext/planning_context.go 50.00% 32 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main   #16310      +/-   ##
==========================================
- Coverage   68.71%   68.71%   -0.01%     
==========================================
  Files        1547     1547              
  Lines      198286   198338      +52     
==========================================
+ Hits       136257   136281      +24     
- Misses      62029    62057      +28     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

Signed-off-by: Andres Taylor <andres@planetscale.com>
@systay systay marked this pull request as draft July 2, 2024 10:29
systay added 2 commits July 3, 2024 10:10
Signed-off-by: Andres Taylor <andres@planetscale.com>
Signed-off-by: Andres Taylor <andres@planetscale.com>
@systay systay marked this pull request as ready for review July 3, 2024 09:36
Comment on lines +273 to +278
// replaceAggrWithArg replaces aggregate functions with Arguments in the given expression.
// this is to prepare for sending the expression to the evalengine compiler to figure out the type
func (ctx *PlanningContext) replaceAggrWithArg(e sqlparser.Expr, cfg *evalengine.Config, env *evalengine.ExpressionEnv) (expr sqlparser.Expr, unknown bool) {
expr = sqlparser.CopyOnRewrite(e, nil, func(cursor *sqlparser.CopyOnWriteCursor) {
agg, ok := cursor.Node().(sqlparser.AggrFunc)
if !ok {
Copy link
Member

Choose a reason for hiding this comment

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

I see the TODO, do you want to keep this unused method before removing the TODO?

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

yeah, that was my thought, but it's not important. if you think it should go, I can delete it

Signed-off-by: Andres Taylor <andres@planetscale.com>
Copy link
Member

@GuptaManan100 GuptaManan100 left a comment

Choose a reason for hiding this comment

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

LGTM!

@systay systay merged commit d303601 into vitessio:main Jul 4, 2024
94 checks passed
@systay systay deleted the more-typing branch July 4, 2024 21:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Component: Query Serving Type: Enhancement Logical improvement (somewhere between a bug and feature)
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants