feat: geodatafusion integration

datafusion-postgres/Cargo.toml

@@ -19,6 +19,8 @@ async-trait = "0.1"
 chrono.workspace = true
 datafusion.workspace = true
 datafusion-pg-catalog = { path = "../datafusion-pg-catalog", version = "0.14.0" }
+#geodatafusion = { version = "0.1.1", optional = true }
+geodatafusion = { git = "https://github.com/sunng87/geodatafusion", branch = "feature/datafusion-52", optional = true }
 futures.workspace = true
 getset = "0.1"
 log = "0.4"
@@ -32,3 +34,7 @@ rustls-pki-types = "1.13"
 
 [dev-dependencies]
 env_logger = "0.11"
+
+[features]
+default = []
+postgis = ["geodatafusion", "arrow-pg/geo"]

datafusion-postgres/src/lib.rs

@@ -86,6 +86,9 @@ pub async fn serve(
     session_context: Arc<SessionContext>,
     opts: &ServerOptions,
 ) -> Result<(), std::io::Error> {
+    #[cfg(feature = "postgis")]
+    geodatafusion::register(&session_context);
+
     // Create the handler factory with authentication
     let factory = Arc::new(HandlerFactory::new(session_context));
 
@@ -99,6 +102,9 @@ pub async fn serve_with_hooks(
     opts: &ServerOptions,
     hooks: Vec<Arc<dyn QueryHook>>,
 ) -> Result<(), std::io::Error> {
+    #[cfg(feature = "postgis")]
+    geodatafusion::register(&session_context);
+
     // Create the handler factory with authentication
     let factory = Arc::new(HandlerFactory::new_with_hooks(session_context, hooks));