Skip to content

feat: add empty pg_range table to pg_catalog #106

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.

Sign up for GitHub

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

Merged
merged 1 commit into from
Jul 8, 2025
Merged

feat: add empty pg_range table to pg_catalog #106

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 22 additions & 0 deletions datafusion-postgres/src/pg_catalog.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ const PG_CATALOG_TABLE_PG_NAMESPACE: &str = "pg_namespace";
const PG_CATALOG_TABLE_PG_PROC: &str = "pg_proc";
const PG_CATALOG_TABLE_PG_DATABASE: &str = "pg_database";
const PG_CATALOG_TABLE_PG_AM: &str = "pg_am";
const PG_CATALOG_TABLE_PG_RANGE: &str = "pg_range";

/// Determine PostgreSQL table type (relkind) from DataFusion TableProvider
fn get_table_type(table: &Arc<dyn TableProvider>) -> &'static str {
Expand Down Expand Up @@ -64,6 +65,7 @@ pub const PG_CATALOG_TABLES: &[&str] = &[
PG_CATALOG_TABLE_PG_PROC,
PG_CATALOG_TABLE_PG_DATABASE,
PG_CATALOG_TABLE_PG_AM,
PG_CATALOG_TABLE_PG_RANGE,
];

// Data structure to hold pg_type table data
Expand Down Expand Up @@ -249,6 +251,7 @@ impl SchemaProvider for PgCatalogSchemaProvider {
)))
}
PG_CATALOG_TABLE_PG_PROC => Ok(Some(self.create_pg_proc_table())),
PG_CATALOG_TABLE_PG_RANGE => Ok(Some(self.create_pg_range_table())),
_ => Ok(None),
}
}
Expand Down Expand Up @@ -715,6 +718,25 @@ impl PgCatalogSchemaProvider {
Arc::new(provider)
}

/// Create a mock empty table for pg_range
fn create_pg_range_table(&self) -> Arc<dyn TableProvider> {
// Define the schema for pg_range
// This matches PostgreSQL's pg_range table columns
let schema = Arc::new(Schema::new(vec![
Field::new("rngtypid", DataType::Int32, false), // OID of the range type
Field::new("rngsubtype", DataType::Int32, false), // OID of the element type (subtype) of this range type
Field::new("rngmultitypid", DataType::Int32, false), // OID of the multirange type for this range type
Field::new("rngcollation", DataType::Int32, false), // OID of the collation used for range comparisons, or zero if none
Field::new("rngsubopc", DataType::Int32, false), // OID of the subtype's operator class used for range comparisons
Field::new("rngcanonical", DataType::Int32, false), // OID of the function to convert a range value into canonical form, or zero if none
Field::new("rngsubdiff", DataType::Int32, false), // OID of the function to return the difference between two element values as double precision, or zero if none
]));

// Create memory table with schema
let provider = MemTable::try_new(schema, vec![]).unwrap();
Arc::new(provider)
}

/// Create a populated pg_proc table with standard PostgreSQL functions
fn create_pg_proc_table(&self) -> Arc<dyn TableProvider> {
// Define complete schema for pg_proc (matching PostgreSQL)
Expand Down
Loading