@@ -7,7 +7,7 @@ use datafusion::arrow::datatypes::DataType;
77use datafusion:: logical_expr:: LogicalPlan ;
88use datafusion:: prelude:: * ;
99use pgwire:: api:: auth:: noop:: NoopStartupHandler ;
10- use pgwire:: api:: copy :: NoopCopyHandler ;
10+ use pgwire:: api:: auth :: StartupHandler ;
1111use pgwire:: api:: portal:: { Format , Portal } ;
1212use pgwire:: api:: query:: { ExtendedQueryHandler , SimpleQueryHandler } ;
1313use pgwire:: api:: results:: {
@@ -16,7 +16,7 @@ use pgwire::api::results::{
1616} ;
1717use pgwire:: api:: stmt:: QueryParser ;
1818use pgwire:: api:: stmt:: StoredStatement ;
19- use pgwire:: api:: { ClientInfo , NoopErrorHandler , PgWireServerHandlers , Type } ;
19+ use pgwire:: api:: { ClientInfo , PgWireServerHandlers , Type } ;
2020use pgwire:: error:: { PgWireError , PgWireResult } ;
2121use tokio:: sync:: Mutex ;
2222
@@ -50,31 +50,17 @@ impl HandlerFactory {
5050}
5151
5252impl PgWireServerHandlers for HandlerFactory {
53- type StartupHandler = SimpleStartupHandler ;
54- type SimpleQueryHandler = DfSessionService ;
55- type ExtendedQueryHandler = DfSessionService ;
56- type CopyHandler = NoopCopyHandler ;
57- type ErrorHandler = NoopErrorHandler ;
58-
59- fn simple_query_handler ( & self ) -> Arc < Self :: SimpleQueryHandler > {
53+ fn simple_query_handler ( & self ) -> Arc < impl SimpleQueryHandler > {
6054 self . session_service . clone ( )
6155 }
6256
63- fn extended_query_handler ( & self ) -> Arc < Self :: ExtendedQueryHandler > {
57+ fn extended_query_handler ( & self ) -> Arc < impl ExtendedQueryHandler > {
6458 self . session_service . clone ( )
6559 }
6660
67- fn startup_handler ( & self ) -> Arc < Self :: StartupHandler > {
61+ fn startup_handler ( & self ) -> Arc < impl StartupHandler > {
6862 Arc :: new ( SimpleStartupHandler )
6963 }
70-
71- fn copy_handler ( & self ) -> Arc < Self :: CopyHandler > {
72- Arc :: new ( NoopCopyHandler )
73- }
74-
75- fn error_handler ( & self ) -> Arc < Self :: ErrorHandler > {
76- Arc :: new ( NoopErrorHandler )
77- }
7864}
7965
8066pub struct DfSessionService {
0 commit comments