Add support for pg_collation

BemiHQ · Feb 11, 2025 · 1610378 · 1610378
1 parent 345bcb1
commit 1610378
Show file tree

Hide file tree

Showing 4 changed files with 47 additions and 0 deletions.
diff --git a/src/parser_table.go b/src/parser_table.go
@@ -70,6 +70,12 @@ func (parser *ParserTable) MakePgExtensionNode(alias string) *pgQuery.Node {
 	return parser.utils.MakeSubselectWithRowsNode(PG_TABLE_PG_EXTENSION, tableDef, [][]string{tableDef.Values}, alias)
 }
 
+// pg_catalog.pg_collation -> VALUES(values...) t(columns...)
+func (parser *ParserTable) MakePgCollationNode(alias string) *pgQuery.Node {
+	tableDef := PG_COLLATION_DEFINITION
+	return parser.utils.MakeSubselectWithRowsNode(PG_TABLE_PG_COLLATION, tableDef, [][]string{tableDef.Values}, alias)
+}
+
 // pg_catalog.pg_database -> VALUES(values...) t(columns...)
 func (parser *ParserTable) MakePgDatabaseNode(database string, alias string) *pgQuery.Node {
 	tableDef := PG_DATABASE_DEFINITION

diff --git a/src/pg_constants.go b/src/pg_constants.go
@@ -21,6 +21,7 @@ const (
 	PG_TABLE_PG_ATTRIBUTE          = "pg_attribute"
 	PG_TABLE_PG_AUTH_MEMBERS       = "pg_auth_members"
 	PG_TABLE_PG_CLASS              = "pg_class"
+	PG_TABLE_PG_COLLATION          = "pg_collation"
 	PG_TABLE_PG_DATABASE           = "pg_database"
 	PG_TABLE_PG_EXTENSION          = "pg_extension"
 	PG_TABLE_PG_INHERITS           = "pg_inherits"
@@ -252,6 +253,37 @@ var PG_EXTENSION_DEFINITION = TableDefinition{
 	},
 }
 
+var PG_COLLATION_DEFINITION = TableDefinition{
+	Columns: []ColumnDefinition{
+		{"oid", "oid"},
+		{"collname", "text"},
+		{"collnamespace", "oid"},
+		{"collowner", "oid"},
+		{"collprovider", "text"},
+		{"collisdeterministic", "bool"},
+		{"collencoding", "int4"},
+		{"collcollate", "text"},
+		{"collctype", "text"},
+		{"colliculocale", "text"},
+		{"collicurules", "text"},
+		{"collversion", "text"},
+	},
+	Values: []string{
+		"100",
+		"default",
+		"11",
+		"10",
+		"d",
+		"TRUE",
+		"-1",
+		"NULL",
+		"NULL",
+		"NULL",
+		"NULL",
+		"NULL",
+	},
+}
+
 var PG_STAT_USER_TABLES_DEFINITION = TableDefinition{
 	Columns: []ColumnDefinition{
 		{"relid", "oid"},

diff --git a/src/query_handler_test.go b/src/query_handler_test.go
@@ -173,6 +173,11 @@ func TestHandleQuery(t *testing.T) {
 			"description": {"schemaname", "viewname", "viewowner", "definition"},
 			"types":       {Uint32ToString(pgtype.TextOID), Uint32ToString(pgtype.TextOID), Uint32ToString(pgtype.TextOID), Uint32ToString(pgtype.TextOID)},
 		},
+		"SELECT oid FROM pg_collation": {
+			"description": {"oid"},
+			"types":       {Uint32ToString(pgtype.OIDOID)},
+			"values":      {"100"},
+		},
 		"SELECT schemaname, relname, n_live_tup FROM pg_stat_user_tables": {
 			"description": {"schemaname", "relname", "n_live_tup"},
 			"types":       {Uint32ToString(pgtype.TextOID), Uint32ToString(pgtype.TextOID), Uint32ToString(pgtype.Int8OID)},

diff --git a/src/query_remapper_table.go b/src/query_remapper_table.go
@@ -105,6 +105,10 @@ func (remapper *QueryRemapperTable) RemapTable(node *pgQuery.Node) *pgQuery.Node
 		case PG_TABLE_PG_STAT_USER_TABLES:
 			return parser.MakePgStatUserTablesNode(remapper.icebergSchemaTables, qSchemaTable.Alias)
 
+		// pg_collation -> return hard-coded collation (encoding) info
+		case PG_TABLE_PG_COLLATION:
+			return parser.MakePgCollationNode(qSchemaTable.Alias)
+
 		// pg_catalog.pg_* other system tables -> return as is
 		default:
 			return node