micromegas 0.11.0

Micromegas is a scalable observability solution.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use datafusion::catalog::SchemaProvider;
use datafusion::{error::DataFusionError, prelude::*, sql::TableReference};
use datafusion_postgres::pg_catalog::{
    PgCatalogSchemaProvider, create_current_schema_udf, create_current_schemas_udf,
    create_has_table_privilege_2param_udf, create_pg_get_userbyid_udf, create_version_udf,
};

/// Sets up the PostgreSQL catalog functions and tables in the DataFusion session context.
pub async fn setup_pg_catalog(ctx: &SessionContext) -> Result<(), Box<DataFusionError>> {
    let pg_catalog_schema = PgCatalogSchemaProvider::new(ctx.state().catalog_list().clone());

    for table_name in pg_catalog_schema.table_names() {
        if let Some(table) = pg_catalog_schema.table(&table_name).await? {
            ctx.register_table(
                TableReference::Bare {
                    table: table_name.into(),
                },
                table,
            )?;
        }
    }

    ctx.register_udf(create_current_schema_udf());
    ctx.register_udf(create_current_schemas_udf());
    ctx.register_udf(create_version_udf());
    ctx.register_udf(create_pg_get_userbyid_udf());
    ctx.register_udf(create_has_table_privilege_2param_udf());

    Ok(())
}