feat: pgwire 0.33 (#197)

* feat: update pgwire to 0.33

* test: update duckdb example

Author: sunng87

Cargo.lock

@@ -19,7 +19,7 @@ bytes = "1.10.1"
 chrono = { version = "0.4", features = ["std"] }
 datafusion = { version = "50", default-features = false }
 futures = "0.3"
-pgwire = { version = "0.32", default-features = false }
+pgwire = { version = "0.33", default-features = false }
 postgres-types = "0.2"
 rust_decimal = { version = "1.38", features = ["db-postgres"] }
 tokio = { version = "1", default-features = false }

Cargo.toml

@@ -26,6 +26,7 @@ pub struct DuckDBBackend {
     query_parser: Arc<NoopQueryParser>,
 }
 
+#[derive(Debug)]
 struct DummyAuthSource;
 
 #[async_trait]
@@ -44,7 +45,7 @@ impl AuthSource for DummyAuthSource {
 
 #[async_trait]
 impl SimpleQueryHandler for DuckDBBackend {
-    async fn do_query<'a, C>(&self, _client: &mut C, query: &str) -> PgWireResult<Vec<Response<'a>>>
+    async fn do_query<C>(&self, _client: &mut C, query: &str) -> PgWireResult<Vec<Response>>
     where
         C: ClientInfo + Unpin + Send + Sync,
     {
@@ -152,12 +153,12 @@ impl ExtendedQueryHandler for DuckDBBackend {
         self.query_parser.clone()
     }
 
-    async fn do_query<'a, C>(
+    async fn do_query<C>(
         &self,
         _client: &mut C,
         portal: &Portal<Self::Statement>,
         _max_rows: usize,
-    ) -> PgWireResult<Response<'a>>
+    ) -> PgWireResult<Response>
     where
         C: ClientInfo + Unpin + Send + Sync,
     {

arrow-pg/examples/duckdb.rs

@@ -18,10 +18,7 @@ use rust_decimal::Decimal;
 
 use super::{arrow_schema_to_pg_fields, encode_recordbatch, into_pg_type};
 
-pub async fn encode_dataframe<'a>(
-    df: DataFrame,
-    format: &Format,
-) -> PgWireResult<QueryResponse<'a>> {
+pub async fn encode_dataframe(df: DataFrame, format: &Format) -> PgWireResult<QueryResponse> {
     let fields = Arc::new(arrow_schema_to_pg_fields(df.schema().as_arrow(), format)?);
 
     let recordbatch_stream = df

arrow-pg/src/datatypes/df.rs

@@ -22,7 +22,7 @@ datafusion-pg-catalog = { path = "../datafusion-pg-catalog", version = "0.11.0"
 futures.workspace = true
 getset = "0.1"
 log = "0.4"
-pgwire = { workspace = true, features = ["server-api-ring", "scram"] }
+pgwire = { workspace = true, features = ["server-api-ring"] }
 postgres-types.workspace = true
 rust_decimal.workspace = true
 tokio = { version = "1.47", features = ["sync", "net"] }

datafusion-postgres/Cargo.toml

@@ -460,7 +460,7 @@ impl PgCatalogContextProvider for AuthManager {
 
 /// AuthSource implementation for integration with pgwire authentication
 /// Provides proper password-based authentication instead of custom startup handler
-#[derive(Clone)]
+#[derive(Clone, Debug)]
 pub struct DfAuthSource {
     pub auth_manager: Arc<AuthManager>,
 }
@@ -550,6 +550,7 @@ impl AuthSource for DfAuthSource {
 // ```
 
 /// Simple AuthSource implementation that accepts any user with empty password
+#[derive(Debug)]
 pub struct SimpleAuthSource {
     auth_manager: Arc<AuthManager>,
 }

datafusion-postgres/src/auth.rs

@@ -209,7 +209,7 @@ impl DfSessionService {
         ResourceType::All
     }
 
-    fn mock_show_response<'a>(name: &str, value: &str) -> PgWireResult<QueryResponse<'a>> {
+    fn mock_show_response(name: &str, value: &str) -> PgWireResult<QueryResponse> {
         let fields = vec![FieldInfo::new(
             name.to_string(),
             None,
@@ -228,11 +228,11 @@ impl DfSessionService {
         Ok(QueryResponse::new(Arc::new(fields), Box::pin(row_stream)))
     }
 
-    async fn try_respond_set_statements<'a, C>(
+    async fn try_respond_set_statements<C>(
         &self,
         client: &mut C,
         query_lower: &str,
-    ) -> PgWireResult<Option<Response<'a>>>
+    ) -> PgWireResult<Option<Response>>
     where
         C: ClientInfo,
     {
@@ -310,11 +310,11 @@ impl DfSessionService {
         }
     }
 
-    async fn try_respond_transaction_statements<'a, C>(
+    async fn try_respond_transaction_statements<C>(
         &self,
         client: &C,
         query_lower: &str,
-    ) -> PgWireResult<Option<Response<'a>>>
+    ) -> PgWireResult<Option<Response>>
     where
         C: ClientInfo,
     {
@@ -361,11 +361,11 @@ impl DfSessionService {
         }
     }
 
-    async fn try_respond_show_statements<'a, C>(
+    async fn try_respond_show_statements<C>(
         &self,
         client: &C,
         query_lower: &str,
-    ) -> PgWireResult<Option<Response<'a>>>
+    ) -> PgWireResult<Option<Response>>
     where
         C: ClientInfo,
     {
@@ -423,7 +423,7 @@ impl DfSessionService {
 
 #[async_trait]
 impl SimpleQueryHandler for DfSessionService {
-    async fn do_query<'a, C>(&self, client: &mut C, query: &str) -> PgWireResult<Vec<Response<'a>>>
+    async fn do_query<C>(&self, client: &mut C, query: &str) -> PgWireResult<Vec<Response>>
     where
         C: ClientInfo + Unpin + Send + Sync,
     {
@@ -599,12 +599,12 @@ impl ExtendedQueryHandler for DfSessionService {
         Ok(DescribePortalResponse::new(fields))
     }
 
-    async fn do_query<'a, C>(
+    async fn do_query<C>(
         &self,
         client: &mut C,
         portal: &Portal<Self::Statement>,
         _max_rows: usize,
-    ) -> PgWireResult<Response<'a>>
+    ) -> PgWireResult<Response>
     where
         C: ClientInfo + Unpin + Send + Sync,
     {