From 2754ebf74005740b3dd7b388318564dbe2d6fc6d Mon Sep 17 00:00:00 2001
From: Belousov Maksim <mail@singulared.space>
Date: Wed, 30 Jul 2025 16:06:27 +0400
Subject: [PATCH] fix(handlers): pass unhandled SET queries into datafusion

---
 datafusion-postgres/src/handlers.rs | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/datafusion-postgres/src/handlers.rs b/datafusion-postgres/src/handlers.rs
index ab03bcf..24d5beb 100644
--- a/datafusion-postgres/src/handlers.rs
+++ b/datafusion-postgres/src/handlers.rs
@@ -205,8 +205,15 @@ impl DfSessionService {
                     )))
                 }
             } else {
-                // noop: skip any unsupported set statements
-                Ok(Some(Response::Execution(Tag::new("SET"))))
+                // pass SET query to datafusion
+                let df = self
+                    .session_context
+                    .sql(query_lower)
+                    .await
+                    .map_err(|err| PgWireError::ApiError(Box::new(err)))?;
+
+                let resp = df::encode_dataframe(df, &Format::UnifiedText).await?;
+                Ok(Some(Response::Query(resp)))
             }
         } else {
             Ok(None)