Merge branch 'main' into feat/add-transaction-method

Author: sundy-li

bindings/nodejs/Cargo.toml

@@ -15,7 +15,7 @@ doc = false
 [dependencies]
 databend-driver = { workspace = true, features = ["rustls", "flight-sql"] }
 
-chrono = { version = "0.4", default-features = false }
+chrono = { version = "0.4.35", default-features = false }
 napi = { version = "2.14", default-features = false, features = [
     "napi6",
     "async",

cli/Cargo.toml

@@ -15,7 +15,7 @@ databend-driver = { workspace = true, features = ["rustls", "flight-sql"] }
 
 anyhow = "1.0"
 async-trait = "0.1"
-chrono = { version = "0.4.31", default-features = false, features = ["clock"] }
+chrono = { version = "0.4.35", default-features = false, features = ["clock"] }
 clap = { version = "4.4", features = ["derive", "env"] }
 comfy-table = "7.1"
 csv = "1.3"

core/Cargo.toml

@@ -34,4 +34,4 @@ url = { version = "2.5", default-features = false }
 uuid = { version = "1.6", features = ["v4"] }
 
 [dev-dependencies]
-chrono = { version = "0.4", default-features = false, features = ["clock"] }
+chrono = { version = "0.4.35", default-features = false, features = ["clock"] }

core/src/client.rs

@@ -323,10 +323,10 @@ impl APIClient {
         }
 
         let resp: QueryResponse = resp.json().await?;
+        self.handle_session(&resp.session).await;
         if let Some(err) = resp.error {
             return Err(Error::InvalidResponse(err));
         }
-        self.handle_session(&resp.session).await;
         self.handle_warnings(&resp);
         Ok(resp)
     }

driver/Cargo.toml

@@ -31,7 +31,7 @@ databend-driver-macros = { workspace = true }
 databend-sql = { workspace = true }
 
 async-trait = "0.1"
-chrono = { version = "0.4.31", default-features = false, features = ["clock"] }
+chrono = { version = "0.4.35", default-features = false, features = ["clock"] }
 csv = "1.3"
 dyn-clone = "1.0"
 glob = "0.3"

sql/Cargo.toml

@@ -16,7 +16,7 @@ flight-sql = ["dep:arrow-array", "dep:arrow-schema", "dep:tonic"]
 [dependencies]
 databend-client = { workspace = true }
 
-chrono = { version = "0.4", default-features = false }
+chrono = { version = "0.4.35", default-features = false }
 geozero = { version = "0.12.0", features = ["default", "with-wkb"] }
 glob = "0.3"
 itertools = "0.12"

sql/src/value.rs

@@ -13,7 +13,7 @@
 // limitations under the License.
 
 use arrow::datatypes::{i256, ArrowNativeTypeOp};
-use chrono::{Datelike, NaiveDate, NaiveDateTime};
+use chrono::{DateTime, Datelike, NaiveDate, NaiveDateTime};
 
 use crate::{
     error::{ConvertError, Error, Result},
@@ -190,6 +190,7 @@ impl TryFrom<(&DataType, &str)> for Value {
 
             DataType::Timestamp => Ok(Self::Timestamp(
                 chrono::NaiveDateTime::parse_from_str(v, "%Y-%m-%d %H:%M:%S%.6f")?
+                    .and_utc()
                     .timestamp_micros(),
             )),
             DataType::Date => Ok(Self::Date(
@@ -527,9 +528,9 @@ impl TryFrom<Value> for NaiveDateTime {
             Value::Timestamp(i) => {
                 let secs = i / 1_000_000;
                 let nanos = ((i % 1_000_000) * 1000) as u32;
-                let t = NaiveDateTime::from_timestamp_opt(secs, nanos);
+                let t = DateTime::from_timestamp(secs, nanos);
                 match t {
-                    Some(t) => Ok(t),
+                    Some(t) => Ok(t.naive_utc()),
                     None => Err(ConvertError::new("NaiveDateTime", "".to_string()).into()),
                 }
             }
@@ -636,7 +637,8 @@ fn encode_value(f: &mut std::fmt::Formatter<'_>, val: &Value, raw: bool) -> std:
         Value::Timestamp(i) => {
             let secs = i / 1_000_000;
             let nanos = ((i % 1_000_000) * 1000) as u32;
-            let t = NaiveDateTime::from_timestamp_opt(secs, nanos).unwrap_or_default();
+            let t = DateTime::from_timestamp(secs, nanos).unwrap_or_default();
+            let t = t.naive_utc();
             if raw {
                 write!(f, "{}", t)
             } else {