put pg_catalog implicitly on the search path

Author: zachmu

core/context.go

@@ -199,7 +199,7 @@ func GetSqlTableFromContext(ctx *sql.Context, databaseName string, tableName dol
 	var searchPath []string
 	if len(tableName.Schema) == 0 {
 		// If a schema was not provided, then we'll use the search path
-		searchPath, err = resolve.SearchPath(ctx)
+		searchPath, err = SearchPath(ctx)
 		if err != nil {
 			return nil, err
 		}
@@ -228,6 +228,30 @@ func GetSqlTableFromContext(ctx *sql.Context, databaseName string, tableName dol
 	return nil, nil
 }
 
+// SearchPath returns the effective schema search path for the current session
+func SearchPath(ctx *sql.Context) ([]string, error) {
+	path, err := resolve.SearchPath(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	// pg_catalog is *always* implicitly part of the search path as the first element, unless it's specifically
+	// included later. This allows users to override built-in names with user-defined names, but they have to
+	// opt in to that behavior.
+	hasPgCatalog := false
+	for _, schema := range path {
+		if schema == "pg_catalog" {
+			hasPgCatalog = true
+			break
+		}
+	}
+
+	if !hasPgCatalog {
+		path = append([]string{"pg_catalog"}, path...)
+	}
+	return path, nil
+}
+
 // GetExtensionsCollectionFromContext returns the extensions collection from the given context. Will always return a
 // collection if no error is returned.
 func GetExtensionsCollectionFromContext(ctx *sql.Context, database string) (*extensions.Collection, error) {

server/functions/pg_table_is_visible.go

@@ -15,7 +15,7 @@
 package functions
 
 import (
-	"github.com/dolthub/dolt/go/libraries/doltcore/sqle/resolve"
+	"github.com/dolthub/doltgresql/core"
 	"github.com/dolthub/go-mysql-server/sql"
 
 	"github.com/dolthub/doltgresql/core/id"
@@ -38,7 +38,7 @@ var pg_table_is_visible_oid = framework.Function1{
 	Strict:             true,
 	Callable: func(ctx *sql.Context, _ [2]*pgtypes.DoltgresType, val any) (any, error) {
 		oidVal := val.(id.Id)
-		paths, err := resolve.SearchPath(ctx)
+		paths, err := core.SearchPath(ctx)
 		if err != nil {
 			return false, err
 		}

server/functions/pg_type_is_visible.go

@@ -15,7 +15,7 @@
 package functions
 
 import (
-	"github.com/dolthub/dolt/go/libraries/doltcore/sqle/resolve"
+	"github.com/dolthub/doltgresql/core"
 	"github.com/dolthub/go-mysql-server/sql"
 
 	"github.com/dolthub/doltgresql/core/id"
@@ -48,7 +48,7 @@ var pg_type_is_visible = framework.Function1{
 		schemaName := oidVal.Segment(0)
 
 		// Get the current search path
-		searchPath, err := resolve.SearchPath(ctx)
+		searchPath, err := core.SearchPath(ctx)
 		if err != nil {
 			return false, err
 		}

server/functions/regclass.go

@@ -19,7 +19,7 @@ import (
 	"strconv"
 
 	"github.com/cockroachdb/errors"
-	"github.com/dolthub/dolt/go/libraries/doltcore/sqle/resolve"
+	"github.com/dolthub/doltgresql/core"
 	"github.com/dolthub/go-mysql-server/sql"
 
 	"github.com/dolthub/doltgresql/core/id"
@@ -65,7 +65,7 @@ var regclassin = framework.Function1{
 		switch len(sections) {
 		case 1:
 			database = ctx.GetCurrentDatabase()
-			searchSchemas, err = resolve.SearchPath(ctx)
+			searchSchemas, err = core.SearchPath(ctx)
 			if err != nil {
 				return id.Null, err
 			}
@@ -79,7 +79,7 @@ var regclassin = framework.Function1{
 			searchSchemas = []string{sections[2]}
 			relationName = sections[4]
 		default:
-			return id.Null, errors.Errorf("regclass failed validation")
+			return id.Null, errors.Errorf("unexpected input for regclass: %s", input)
 		}
 
 		// Iterate over all of the items to find which relation matches.

server/node/create_trigger.go

@@ -16,7 +16,6 @@ package node
 
 import (
 	"github.com/cockroachdb/errors"
-	"github.com/dolthub/dolt/go/libraries/doltcore/sqle/resolve"
 	"github.com/dolthub/go-mysql-server/sql"
 	"github.com/dolthub/go-mysql-server/sql/plan"
 	vitess "github.com/dolthub/vitess/go/vt/sqlparser"
@@ -180,11 +179,10 @@ func loadFunction(ctx *sql.Context, funcCollection *functions.Collection, funcID
 			return functions.Function{}, err
 		}
 	} else {
-		searchPaths, err := resolve.SearchPath(ctx)
+		searchPaths, err := core.SearchPath(ctx)
 		if err != nil {
 			return functions.Function{}, err
 		}
-		searchPaths = append(searchPaths, "pg_catalog") // This isn't included in the search path but functions use it
 		for _, searchPath := range searchPaths {
 			function, err = funcCollection.GetFunction(ctx, id.NewFunction(searchPath, funcID.FunctionName()))
 			if err != nil {

testing/go/prepared_statement_test.go

@@ -1334,15 +1334,14 @@ var pgCatalogTests = []ScriptTest{
 		},
 	},
 	{
-		Name: "pg_class",
+		Name:  "pg_class",
+		Focus: true,
 		SetUpScript: []string{
 			`CREATE SCHEMA testschema;`,
 			`CREATE TABLE testschema.testtable (id int primary key, v1 text)`,
 		},
 		Assertions: []ScriptTestAssertion{
 			{
-				// https://github.com/dolthub/doltgresql/issues/2217
-				Skip: true,
 				Query: `SELECT c.oid,d.description,pg_catalog.pg_get_expr(c.relpartbound, c.oid) as partition_expr,  pg_catalog.pg_get_partkeydef(c.oid) as partition_key 
 FROM pg_catalog.pg_class c
 LEFT OUTER JOIN pg_catalog.pg_description d ON d.objoid=c.oid AND d.objsubid=0 AND d.classoid='pg_class'::regclass
@@ -1351,8 +1350,6 @@ WHERE c.relnamespace=$1 AND c.relkind not in ('i','I','c') and c.oid not in (sel
 				Expected: []sql.Row{{1712283605, nil, nil, ""}},
 			},
 			{
-				// https://github.com/dolthub/doltgresql/issues/2217
-				Skip:  true,
 				Query: `SELECT d.description from pg_catalog.pg_description d WHERE d.classoid='pg_class'::regclass`,
 				// TODO: add expected values
 			},