Add support for pg_relation_filenode(), upgrade compress

Author: exAspArk

server/config.go

@@ -9,7 +9,7 @@ import (
 )
 
 const (
-	VERSION = "2.0.1"
+	VERSION = "2.1.0"
 
 	ENV_LOG_LEVEL     = "LOG_LEVEL"
 	ENV_AUDIT_ENABLED = "AUDIT_ENABLED"

server/go.mod

@@ -46,7 +46,7 @@ require (
 	github.com/google/flatbuffers v25.12.19+incompatible // indirect
 	github.com/jackc/pgpassfile v1.0.0 // indirect
 	github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
-	github.com/klauspost/compress v1.18.3 // indirect
+	github.com/klauspost/compress v1.18.7 // indirect
 	github.com/klauspost/cpuid/v2 v2.3.0 // indirect
 	github.com/pierrec/lz4/v4 v4.1.25 // indirect
 	github.com/xdg-go/pbkdf2 v1.0.0 // indirect

server/go.sum

@@ -81,8 +81,8 @@ github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo
 github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4=
 github.com/klauspost/asmfmt v1.3.2 h1:4Ri7ox3EwapiOjCki+hw14RyKk201CN4rzyCJRFLpK4=
 github.com/klauspost/asmfmt v1.3.2/go.mod h1:AG8TuvYojzulgDAMCnYn50l/5QV3Bs/tp6j0HLHbNSE=
-github.com/klauspost/compress v1.18.3 h1:9PJRvfbmTabkOX8moIpXPbMMbYN60bWImDDU7L+/6zw=
-github.com/klauspost/compress v1.18.3/go.mod h1:R0h/fSBs8DE4ENlcrlib3PsXS61voFxhIs2DeRhCvJ4=
+github.com/klauspost/compress v1.18.7 h1:aUyZsS4kH3QTKurYhAOwAHxllVPnOthb3vPfnF1Ehjw=
+github.com/klauspost/compress v1.18.7/go.mod h1:cwPg85FWrGar70rWktvGQj8/hthj3wpl0PGDogxkrSQ=
 github.com/klauspost/cpuid/v2 v2.3.0 h1:S4CRMLnYUhGeDFDqkGriYKdfoFlDnMtqTiI/sFzhA9Y=
 github.com/klauspost/cpuid/v2 v2.3.0/go.mod h1:hqwkgyIinND0mEev00jJYCxPNVRVXFQeu1XKlok6oO0=
 github.com/minio/asm2plan9s v0.0.0-20200509001527-cdd76441f9d8 h1:AMFGa4R4MiIpspGNG7Z948v4n35fFGB3RR3G/ry4FWs=

server/query_handler_test.go

@@ -20,6 +20,65 @@ type queryResponseExpectation struct {
 	values      []string
 }
 
+func TestHandleQueryRelationFilenode(t *testing.T) {
+	queryHandler := initQueryHandlerForTest(t)
+	for _, query := range []string{
+		`SELECT pg_relation_filenode('"audit"."changes"'::regclass)`,
+		`SELECT pg_catalog.pg_relation_filenode('"audit"."changes"'::pg_catalog.regclass)`,
+		`SELECT pg_relation_filenode(123::oid)`,
+		`SELECT pg_relation_filenode(NULL::regclass)`,
+	} {
+		t.Run(query, func(t *testing.T) {
+			messages, err := queryHandler.HandleSimpleQuery(query)
+			if err != nil {
+				t.Fatal(err)
+			}
+			assertMessageTypes(t, messages, "*pgproto3.RowDescription", "*pgproto3.DataRow", "*pgproto3.CommandComplete")
+			description := messages[0].(*pgproto3.RowDescription)
+			if string(description.Fields[0].Name) != "pg_relation_filenode" || description.Fields[0].DataTypeOID != pgtype.Int8OID {
+				t.Fatalf("unexpected filenode description: %+v", description.Fields[0])
+			}
+			if messages[1].(*pgproto3.DataRow).Values[0] != nil {
+				t.Fatal("filenode must be SQL NULL")
+			}
+		})
+	}
+}
+
+func TestHandlePreparedQueryRelationFilenode(t *testing.T) {
+	queryHandler := initQueryHandlerForTest(t)
+	for _, query := range []string{
+		`SELECT pg_relation_filenode($1::regclass)`,
+		`SELECT pg_catalog.pg_relation_filenode($1::pg_catalog.regclass)`,
+	} {
+		t.Run(query, func(t *testing.T) {
+			_, statement, err := queryHandler.HandleParseQuery(&pgproto3.Parse{Query: query})
+			if err != nil {
+				t.Fatal(err)
+			}
+			defer statement.Statement.Close()
+			_, statement, err = queryHandler.HandleBindQuery(&pgproto3.Bind{
+				Parameters: [][]byte{[]byte(`"audit"."changes"`)},
+			}, statement)
+			if err != nil {
+				t.Fatal(err)
+			}
+			_, statement, err = queryHandler.HandleDescribeQuery(&pgproto3.Describe{ObjectType: 'P'}, statement)
+			if err != nil {
+				t.Fatal(err)
+			}
+			messages, err := queryHandler.HandleExecuteQuery(&pgproto3.Execute{}, statement)
+			if err != nil {
+				t.Fatal(err)
+			}
+			assertMessageTypes(t, messages, "*pgproto3.DataRow", "*pgproto3.CommandComplete")
+			if messages[0].(*pgproto3.DataRow).Values[0] != nil {
+				t.Fatal("filenode must be SQL NULL")
+			}
+		})
+	}
+}
+
 func TestHandleQueryExpandArray(t *testing.T) {
 	queryHandler := initQueryHandlerForTest(t)
 	tests := []struct {

server/query_remapper_function.go

@@ -25,6 +25,7 @@ func CreatePgCatalogMacroQueries(config *Config) []string {
 		"CREATE MACRO pg_get_viewdef(view_oid) AS pg_catalog.pg_get_viewdef(view_oid), (view_oid, pretty_bool) AS pg_catalog.pg_get_viewdef(view_oid)",
 		"CREATE MACRO pg_indexes_size(regclass) AS 0",
 		"CREATE MACRO pg_is_in_recovery() AS false",
+		"CREATE MACRO pg_relation_filenode(relation) AS NULL::BIGINT",
 		"CREATE MACRO pg_table_size(regclass) AS 0",
 		"CREATE MACRO pg_tablespace_location(tablespace_oid) AS ''",
 		"CREATE MACRO pg_total_relation_size(regclass) AS 0",
@@ -140,6 +141,16 @@ func (remapper *QueryRemapperFunction) RemapFunctionCall(functionCall *pgQuery.F
 
 	// pg_catalog.func() -> main.func()
 	case PG_SCHEMA_PG_CATALOG, "":
+		if schemaFunction.Function == "pg_relation_filenode" {
+			parserTypeCast := NewParserTypeCast(remapper.config)
+			for _, argument := range functionCall.Args {
+				typeCast := parserTypeCast.TypeCast(argument)
+				parserTypeCast.RemovePgCatalog(typeCast)
+				if parserTypeCast.TypeName(typeCast) == "regclass" {
+					parserTypeCast.SetTypeName(typeCast, "text")
+				}
+			}
+		}
 		if PG_CATALOG_MACRO_FUNCTION_NAMES.Contains(schemaFunction.Function) || BUILTIN_DUCKDB_PG_FUNCTION_NAMES.Contains(schemaFunction.Function) {
 			remapper.parserFunction.RemapSchemaToMain(functionCall)
 			return schemaFunction