chore(query): introduce TzLUT util

src/query/expression/Cargo.toml

@@ -23,6 +23,7 @@ chrono = { workspace = true }
 chrono-tz = { workspace = true }
 comfy-table = "6"
 common-jsonb = { path = "../../common/jsonb" }
+dashmap = "5.4"
 educe = "0.4"
 enum-as-inner = "0.5"
 enum_dispatch = "0.3.8"
@@ -34,6 +35,7 @@ lexical-core = "0.8.5"
 match-template = "0.0.1"
 micromarshal = "0.2.1"
 num-traits = "0.2.15"
+once_cell = "1.15.0"
 ordered-float = { workspace = true, features = ["serde", "rand"] }
 parking_lot = "0.12.1"
 primitive-types = "0.12.0"

src/query/expression/src/function.rs

@@ -16,7 +16,6 @@ use std::collections::HashMap;
 use std::ops::BitAnd;
 use std::sync::Arc;
 
-use chrono_tz::Tz;
 use common_arrow::arrow::bitmap::Bitmap;
 use common_arrow::arrow::bitmap::MutableBitmap;
 use common_exception::ErrorCode;
@@ -26,6 +25,7 @@ use itertools::Itertools;
 use serde::Deserialize;
 use serde::Serialize;
 
+use crate::date_helper::TzLUT;
 use crate::property::Domain;
 use crate::property::FunctionProperty;
 use crate::types::nullable::NullableColumn;
@@ -49,22 +49,16 @@ pub struct FunctionSignature {
 
 pub type AutoCastSignature = Vec<(DataType, DataType)>;
 
-#[derive(Clone, Copy)]
+#[derive(Clone, Copy, Default)]
 pub struct FunctionContext {
-    pub tz: Tz,
-}
-
-impl Default for FunctionContext {
-    fn default() -> Self {
-        Self { tz: Tz::UTC }
-    }
+    pub tz: TzLUT,
 }
 
 #[derive(Clone)]
 pub struct EvalContext<'a> {
     pub generics: &'a GenericMap,
     pub num_rows: usize,
-    pub tz: Tz,
+    pub tz: TzLUT,
 
     /// Validity bitmap of outer nullable column. This is an optimization
     /// to avoid recording errors on the NULL value which has a coresponding

src/query/expression/src/types/variant.rs

@@ -14,11 +14,10 @@
 
 use std::ops::Range;
 
-use chrono_tz::Tz;
-
 use super::date::date_to_string;
 use super::number::NumberScalar;
 use super::timestamp::timestamp_to_string;
+use crate::date_helper::TzLUT;
 use crate::property::Domain;
 use crate::types::string::StringColumn;
 use crate::types::string::StringColumnBuilder;
@@ -170,7 +169,8 @@ impl ArgType for VariantType {
     }
 }
 
-pub fn cast_scalar_to_variant(scalar: ScalarRef, tz: Tz, buf: &mut Vec<u8>) {
+pub fn cast_scalar_to_variant(scalar: ScalarRef, tz: TzLUT, buf: &mut Vec<u8>) {
+    let inner_tz = tz.tz;
     let value = match scalar {
         ScalarRef::Null => common_jsonb::Value::Null,
         ScalarRef::EmptyArray => common_jsonb::Value::Array(vec![]),
@@ -189,8 +189,8 @@ pub fn cast_scalar_to_variant(scalar: ScalarRef, tz: Tz, buf: &mut Vec<u8>) {
         ScalarRef::Decimal(_) => todo!("decimal"),
         ScalarRef::Boolean(b) => common_jsonb::Value::Bool(b),
         ScalarRef::String(s) => common_jsonb::Value::String(String::from_utf8_lossy(s)),
-        ScalarRef::Timestamp(ts) => timestamp_to_string(ts, tz).to_string().into(),
-        ScalarRef::Date(d) => date_to_string(d, tz).to_string().into(),
+        ScalarRef::Timestamp(ts) => timestamp_to_string(ts, inner_tz).to_string().into(),
+        ScalarRef::Date(d) => date_to_string(d, inner_tz).to_string().into(),
         ScalarRef::Array(col) => {
             let items = cast_scalars_to_variants(col.iter(), tz);
             common_jsonb::build_array(items.iter(), buf).expect("failed to build jsonb array");
@@ -218,7 +218,7 @@ pub fn cast_scalar_to_variant(scalar: ScalarRef, tz: Tz, buf: &mut Vec<u8>) {
 
 pub fn cast_scalars_to_variants(
     scalars: impl IntoIterator<Item = ScalarRef>,
-    tz: Tz,
+    tz: TzLUT,
 ) -> StringColumn {
     let iter = scalars.into_iter();
     let mut builder = StringColumnBuilder::with_capacity(iter.size_hint().0, 0);