Use ast-paths for subquery planning to improve performance (vitessio#…

…17738)

Signed-off-by: Andres Taylor <andres@planetscale.com>
Signed-off-by: Manan Gupta <manan@planetscale.com>
Signed-off-by: Vicent Marti <vmg@strn.cat>
Signed-off-by: Harshit Gangal <harshit@planetscale.com>
Co-authored-by: Manan Gupta <manan@planetscale.com>
Co-authored-by: Vicent Marti <vmg@strn.cat>
Co-authored-by: Harshit Gangal <harshit@planetscale.com>

go/tools/asthelpergen/asthelpergen.go

@@ -32,7 +32,7 @@ import (
 	"vitess.io/vitess/go/tools/codegen"
 )
 
-const licenseFileHeader = `Copyright 2023 The Vitess Authors.
+const licenseFileHeader = `Copyright 2025 The Vitess Authors.
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
@@ -51,10 +51,10 @@ type (
 		addType(t types.Type)
 		scope() *types.Scope
 		findImplementations(iff *types.Interface, impl func(types.Type) error) error
-		iface() *types.Interface
+		iface() *types.Interface // the root interface that all nodes are expected to implement
 	}
 	generator interface {
-		genFile() (string, *jen.File)
+		genFile(generatorSPI) (string, *jen.File)
 		interfaceMethod(t types.Type, iface *types.Interface, spi generatorSPI) error
 		structMethod(t types.Type, strct *types.Struct, spi generatorSPI) error
 		ptrToStructMethod(t types.Type, strct *types.Struct, spi generatorSPI) error
@@ -206,9 +206,11 @@ func GenerateASTHelpers(options *Options) (map[string]*jen.File, error) {
 
 	nt := tt.Type().(*types.Named)
 	pName := nt.Obj().Pkg().Name()
+	ifaceName := types.TypeString(nt, noQualifier)
 	generator := newGenerator(loaded[0].Module, loaded[0].TypesSizes, nt,
 		newEqualsGen(pName, &options.Equals),
 		newCloneGen(pName, &options.Clone),
+		newPathGen(pName, ifaceName),
 		newVisitGen(pName),
 		newRewriterGen(pName, types.TypeString(nt, noQualifier)),
 		newCOWGen(pName, nt),
@@ -298,7 +300,7 @@ func (gen *astHelperGen) createFiles() map[string]*jen.File {
 
 	result := map[string]*jen.File{}
 	for _, g := range gen.gens {
-		fName, jenFile := g.genFile()
+		fName, jenFile := g.genFile(gen)
 		result[fName] = jenFile
 	}
 	return result

go/tools/asthelpergen/clone_gen.go

@@ -56,7 +56,7 @@ func (c *cloneGen) addFunc(name string, code *jen.Statement) {
 	c.file.Add(code)
 }
 
-func (c *cloneGen) genFile() (string, *jen.File) {
+func (c *cloneGen) genFile(generatorSPI) (string, *jen.File) {
 	return "ast_clone.go", c.file
 }
 

go/tools/asthelpergen/copy_on_rewrite_gen.go

@@ -44,7 +44,7 @@ func (c *cowGen) addFunc(code *jen.Statement) {
 	c.file.Add(code)
 }
 
-func (c *cowGen) genFile() (string, *jen.File) {
+func (c *cowGen) genFile(generatorSPI) (string, *jen.File) {
 	return "ast_copy_on_rewrite.go", c.file
 }
 

go/tools/asthelpergen/equals_gen.go

@@ -62,7 +62,7 @@ func (e *equalsGen) customComparatorField(t types.Type) string {
 	return printableTypeName(t) + "_"
 }
 
-func (e *equalsGen) genFile() (string, *jen.File) {
+func (e *equalsGen) genFile(generatorSPI) (string, *jen.File) {
 	e.file.Type().Id(Comparator).StructFunc(func(g *jen.Group) {
 		for tname, t := range e.comparators {
 			if t == nil {
@@ -303,6 +303,4 @@ func (e *equalsGen) sliceMethod(t types.Type, slice *types.Slice, spi generatorS
 	return nil
 }
 
-func (e *equalsGen) basicMethod(types.Type, *types.Basic, generatorSPI) error {
-	return nil
-}
+func (*equalsGen) basicMethod(types.Type, *types.Basic, generatorSPI) error { return nil }