-
Notifications
You must be signed in to change notification settings - Fork 85
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Loading status checks…
Merge pull request #147 from blackbeam/json-dom
jsonb: Introduce `JsonDom` type and parsing/conversion
Showing
8 changed files
with
611 additions
and
11 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,208 @@ | ||
| use std::{ | ||
| cmp::min, | ||
| fmt::{self, Write}, | ||
| }; | ||
|
|
||
| use super::misc::{my_packed_time_get_frac_part, my_packed_time_get_int_part}; | ||
|
|
||
| /// Server-side mysql time representation. | ||
| #[derive(Debug, Clone, PartialEq)] | ||
| #[repr(C)] | ||
| pub struct MysqlTime { | ||
| pub year: u32, | ||
| pub month: u32, | ||
| pub day: u32, | ||
| pub hour: u32, | ||
| pub minute: u32, | ||
| pub second: u32, | ||
| /// microseconds | ||
| pub second_part: u32, | ||
| pub neg: bool, | ||
| pub time_type: MysqlTimestampType, | ||
| pub time_zone_displacement: i32, | ||
| } | ||
|
|
||
| impl MysqlTime { | ||
| /// Convert time packed numeric representation to [`MysqlTime`]. | ||
| pub fn from_int64_time_packed(mut packed_value: i64) -> Self { | ||
| let neg = packed_value < 0; | ||
| if neg { | ||
| packed_value = -packed_value | ||
| } | ||
|
|
||
| let hms: i64 = my_packed_time_get_int_part(packed_value); | ||
|
|
||
| let hour = (hms >> 12) as u32 % (1 << 10); /* 10 bits starting at 12th */ | ||
| let minute = (hms >> 6) as u32 % (1 << 6); /* 6 bits starting at 6th */ | ||
| let second = hms as u32 % (1 << 6); /* 6 bits starting at 0th */ | ||
| let second_part = my_packed_time_get_frac_part(packed_value); | ||
|
|
||
| Self { | ||
| year: 0, | ||
| month: 0, | ||
| day: 0, | ||
| hour, | ||
| minute, | ||
| second, | ||
| second_part: second_part as u32, | ||
| neg, | ||
| time_type: MysqlTimestampType::MYSQL_TIMESTAMP_TIME, | ||
| time_zone_displacement: 0, | ||
| } | ||
| } | ||
|
|
||
| /// Convert packed numeric date representation to [`MysqlTime`]. | ||
| pub fn from_int64_date_packed(packed_value: i64) -> Self { | ||
| let mut this = Self::from_int64_datetime_packed(packed_value); | ||
| this.time_type = MysqlTimestampType::MYSQL_TIMESTAMP_DATE; | ||
| this | ||
| } | ||
|
|
||
| /// Convert packed numeric datetime representation to [`MysqlTime`]. | ||
| pub fn from_int64_datetime_packed(mut packed_value: i64) -> Self { | ||
| let neg = packed_value < 0; | ||
| if neg { | ||
| packed_value = -packed_value | ||
| } | ||
|
|
||
| let second_part = my_packed_time_get_frac_part(packed_value); | ||
| let ymdhms: i64 = my_packed_time_get_int_part(packed_value); | ||
|
|
||
| let ymd: i64 = ymdhms >> 17; | ||
| let ym: i64 = ymd >> 5; | ||
| let hms: i64 = ymdhms % (1 << 17); | ||
|
|
||
| let day = ymd % (1 << 5); | ||
| let month = ym % 13; | ||
| let year = (ym / 13) as _; | ||
|
|
||
| let second = hms % (1 << 6); | ||
| let minute = (hms >> 6) % (1 << 6); | ||
| let hour = (hms >> 12) as _; | ||
|
|
||
| Self { | ||
| year, | ||
| month: month as _, | ||
| day: day as _, | ||
| hour, | ||
| minute: minute as _, | ||
| second: second as _, | ||
| second_part: second_part as _, | ||
| neg, | ||
| time_type: MysqlTimestampType::MYSQL_TIMESTAMP_DATETIME, | ||
| time_zone_displacement: 0, | ||
| } | ||
| } | ||
| } | ||
|
|
||
| impl fmt::Display for MysqlTime { | ||
| fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { | ||
| match self.time_type { | ||
| MysqlTimestampType::MYSQL_TIMESTAMP_DATETIME | ||
| | MysqlTimestampType::MYSQL_TIMESTAMP_DATETIME_TZ => format_datetime(self, f), | ||
| MysqlTimestampType::MYSQL_TIMESTAMP_DATE => format_date(self, f), | ||
| MysqlTimestampType::MYSQL_TIMESTAMP_TIME => format_time(self, f), | ||
| MysqlTimestampType::MYSQL_TIMESTAMP_NONE | ||
| | MysqlTimestampType::MYSQL_TIMESTAMP_ERROR => Ok(()), | ||
| } | ||
| } | ||
| } | ||
|
|
||
| fn trim_two_digits(value: u32) -> u32 { | ||
| if value >= 100 { | ||
| 0 | ||
| } else { | ||
| value | ||
| } | ||
| } | ||
|
|
||
| /// Formats a time value as `HH:MM:SS[.fraction]`. | ||
| fn format_time(time: &MysqlTime, f: &mut fmt::Formatter<'_>) -> fmt::Result { | ||
| if time.neg { | ||
| f.write_char('-')?; | ||
| } | ||
|
|
||
| write!( | ||
| f, | ||
| "{:02}:{:02}:{:02}", | ||
| time.hour, | ||
| trim_two_digits(time.minute), | ||
| trim_two_digits(time.second), | ||
| )?; | ||
| format_useconds(time.second_part, f)?; | ||
| Ok(()) | ||
| } | ||
|
|
||
| /// Formats a datetime value with an optional fractional part (if formatter precision is given). | ||
| fn format_datetime(time: &MysqlTime, f: &mut fmt::Formatter<'_>) -> fmt::Result { | ||
| format_date_and_time(time, f)?; | ||
| format_useconds(time.second_part, f)?; | ||
| if time.time_type == MysqlTimestampType::MYSQL_TIMESTAMP_DATETIME_TZ { | ||
| format_tz(time.time_zone_displacement, f)?; | ||
| } | ||
| Ok(()) | ||
| } | ||
|
|
||
| /// Formats date and time part as 'YYYY-MM-DD hh:mm:ss' | ||
| fn format_date_and_time(time: &MysqlTime, f: &mut fmt::Formatter<'_>) -> fmt::Result { | ||
| write!( | ||
| f, | ||
| "{:02}{:02}-{:02}-{:02} {:02}:{:02}:{:02}", | ||
| trim_two_digits(time.year / 100), | ||
| trim_two_digits(time.year % 100), | ||
| trim_two_digits(time.month), | ||
| trim_two_digits(time.day), | ||
| trim_two_digits(time.hour), | ||
| trim_two_digits(time.minute), | ||
| trim_two_digits(time.second), | ||
| ) | ||
| } | ||
|
|
||
| /// Formats a date value as 'YYYY-MM-DD'. | ||
| fn format_date(time: &MysqlTime, f: &mut fmt::Formatter<'_>) -> fmt::Result { | ||
| write!( | ||
| f, | ||
| "{:02}{:02}-{:02}-{:02}", | ||
| trim_two_digits(time.year / 100), | ||
| trim_two_digits(time.year % 100), | ||
| trim_two_digits(time.month), | ||
| trim_two_digits(time.day), | ||
| ) | ||
| } | ||
|
|
||
| /// Only formats useconds if formatter precision is given (will be truncated to 6) | ||
| fn format_useconds(mut useconds: u32, f: &mut fmt::Formatter<'_>) -> fmt::Result { | ||
| let Some(dec) = f.precision().map(|x| min(x, 6)) else { | ||
| return Ok(()); | ||
| }; | ||
|
|
||
| if dec == 0 { | ||
| return Ok(()); | ||
| } | ||
|
|
||
| useconds %= 1_000_000; | ||
|
|
||
| for _ in 0..(6 - dec) { | ||
| useconds /= 10; | ||
| } | ||
|
|
||
| write!(f, ".{:0width$}", useconds, width = dec) | ||
| } | ||
|
|
||
| fn format_tz(tzd: i32, f: &mut fmt::Formatter<'_>) -> fmt::Result { | ||
| write!(f, "+{:02}:{:02}", tzd / 3600, tzd.abs() / 60 % 60) | ||
| } | ||
|
|
||
| #[derive(Debug, Clone, PartialEq)] | ||
| #[repr(C)] | ||
| #[allow(non_camel_case_types)] | ||
| pub enum MysqlTimestampType { | ||
| /// Textual representation of this value is an empty string | ||
| MYSQL_TIMESTAMP_NONE = -2, | ||
| /// Textual representation of this value is an empty string | ||
| MYSQL_TIMESTAMP_ERROR = -1, | ||
| MYSQL_TIMESTAMP_DATE = 0, | ||
| MYSQL_TIMESTAMP_DATETIME = 1, | ||
| MYSQL_TIMESTAMP_TIME = 2, | ||
| MYSQL_TIMESTAMP_DATETIME_TZ = 3, | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Binary file not shown.