feat(dirver): add handler type for connection info (#58)

databendlabs · Apr 14, 2023 · 1bdf232 · 1bdf232
1 parent 22695e7
commit 1bdf232
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/driver/src/conn.rs b/driver/src/conn.rs
@@ -25,6 +25,7 @@ use crate::rows::{Row, RowIterator, RowProgressIterator};
 use crate::schema::Schema;
 
 pub struct ConnectionInfo {
+    pub handler: String,
     pub host: String,
     pub port: u16,
     pub user: String,

diff --git a/driver/src/flight_sql.rs b/driver/src/flight_sql.rs
@@ -44,6 +44,7 @@ pub struct FlightSQLConnection {
 impl Connection for FlightSQLConnection {
     fn info(&self) -> ConnectionInfo {
         ConnectionInfo {
+            handler: "FlightSQL".to_string(),
             host: self.args.host.clone(),
             port: self.args.port,
             user: self.args.user.clone(),

diff --git a/driver/src/rest_api.rs b/driver/src/rest_api.rs
@@ -38,6 +38,7 @@ pub struct RestAPIConnection {
 impl Connection for RestAPIConnection {
     fn info(&self) -> ConnectionInfo {
         ConnectionInfo {
+            handler: "RestAPI".to_string(),
             host: self.client.host.clone(),
             port: self.client.port,
             user: self.client.user.clone(),