diff --git a/Cargo.lock b/Cargo.lock index 4197de1..3addaa1 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -2360,9 +2360,9 @@ dependencies = [ [[package]] name = "md5" -version = "0.7.0" +version = "0.8.0" source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "490cc448043f947bae3cbee9c203358d62dbee0db12107a74be5c30ccfd09771" +checksum = "ae960838283323069879657ca3de837e9f7bbb4c7bf6ea7f1b290d5e9476d2e0" [[package]] name = "memchr" @@ -2623,9 +2623,9 @@ dependencies = [ [[package]] name = "pgwire" -version = "0.30.2" +version = "0.31.0" source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "4ca6c26b25be998208a13ff2f0c55b567363f34675410e6d6f1c513a150583fd" +checksum = "449fecabd6a04033ec9c12e6c0bb7e663e03c3731f59d1e196c1ae9f1b65a9a9" dependencies = [ "async-trait", "base64", diff --git a/Cargo.toml b/Cargo.toml index 66d3a96..1f81280 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -19,7 +19,7 @@ bytes = "1.10.1" chrono = { version = "0.4", features = ["std"] } datafusion = { version = "48", default-features = false } futures = "0.3" -pgwire = { version = "0.30.2", default-features = false } +pgwire = { version = "0.31", default-features = false } postgres-types = "0.2" rust_decimal = { version = "1.37", features = ["db-postgres"] } tokio = { version = "1", default-features = false } diff --git a/datafusion-postgres/src/handlers.rs b/datafusion-postgres/src/handlers.rs index 00d88fe..ac62f13 100644 --- a/datafusion-postgres/src/handlers.rs +++ b/datafusion-postgres/src/handlers.rs @@ -7,7 +7,7 @@ use datafusion::arrow::datatypes::DataType; use datafusion::logical_expr::LogicalPlan; use datafusion::prelude::*; use pgwire::api::auth::noop::NoopStartupHandler; -use pgwire::api::copy::NoopCopyHandler; +use pgwire::api::auth::StartupHandler; use pgwire::api::portal::{Format, Portal}; use pgwire::api::query::{ExtendedQueryHandler, SimpleQueryHandler}; use pgwire::api::results::{ @@ -16,7 +16,7 @@ use pgwire::api::results::{ }; use pgwire::api::stmt::QueryParser; use pgwire::api::stmt::StoredStatement; -use pgwire::api::{ClientInfo, NoopErrorHandler, PgWireServerHandlers, Type}; +use pgwire::api::{ClientInfo, PgWireServerHandlers, Type}; use pgwire::error::{PgWireError, PgWireResult}; use tokio::sync::Mutex; @@ -50,31 +50,17 @@ impl HandlerFactory { } impl PgWireServerHandlers for HandlerFactory { - type StartupHandler = SimpleStartupHandler; - type SimpleQueryHandler = DfSessionService; - type ExtendedQueryHandler = DfSessionService; - type CopyHandler = NoopCopyHandler; - type ErrorHandler = NoopErrorHandler; - - fn simple_query_handler(&self) -> Arc { + fn simple_query_handler(&self) -> Arc { self.session_service.clone() } - fn extended_query_handler(&self) -> Arc { + fn extended_query_handler(&self) -> Arc { self.session_service.clone() } - fn startup_handler(&self) -> Arc { + fn startup_handler(&self) -> Arc { Arc::new(SimpleStartupHandler) } - - fn copy_handler(&self) -> Arc { - Arc::new(NoopCopyHandler) - } - - fn error_handler(&self) -> Arc { - Arc::new(NoopErrorHandler) - } } pub struct DfSessionService {