Skip to content

Commit

Permalink
clean
Browse files Browse the repository at this point in the history
  • Loading branch information
KeXiangWang committed Nov 2, 2023
1 parent b2e145f commit 6f67496
Show file tree
Hide file tree
Showing 4 changed files with 17 additions and 112 deletions.
7 changes: 0 additions & 7 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -45,13 +45,6 @@
target="_blank"
>
<img alt="Docker" src="https://img.shields.io/docker/v/risingwavelabs/risingwave" />
</a>
<a
href="https://buildkite.com/risingwavelabs/main"
target="_blank"
>
<img alt="Build status" src="https://badge.buildkite.com/9394d2bca0f87e2e97aa78b25f765c92d4207c0b65e7f6648f.svg" />
</a>
<a
href="https://twitter.com/risingwavelabs"
target="_blank"
Expand Down
116 changes: 14 additions & 102 deletions src/common/src/types/jsonb.rs
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,10 @@ use std::fmt;
use std::hash::Hash;

use bytes::Buf;
use jsonbb::{Builder, Value, ValueRef};
use jsonbb::{Value, ValueRef};

use crate::estimate_size::EstimateSize;
use crate::types::{
Date, Decimal, Int256Ref, Interval, Scalar, ScalarRef, Serial, Time, Timestamp, ToText, F32,
F64,
};
use crate::types::{Scalar, ScalarRef};

#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct JsonbVal(pub(crate) Value);
Expand Down Expand Up @@ -196,6 +193,18 @@ impl From<Value> for JsonbVal {
}
}

impl From<JsonbRef<'_>> for JsonbVal {
fn from(v: JsonbRef<'_>) -> Self {
Self(v.0.to_owned())
}
}

impl From<f64> for JsonbVal {
fn from(v: f64) -> Self {
Self(v.into())
}
}

impl<'a> From<JsonbRef<'a>> for ValueRef<'a> {
fn from(v: JsonbRef<'a>) -> Self {
v.0
Expand Down Expand Up @@ -416,100 +425,3 @@ impl<F: std::fmt::Write> std::io::Write for FmtToIoUnchecked<F> {
Ok(())
}
}
impl From<F32> for Value {
fn from(v: F32) -> Self {
if v.0 == f32::INFINITY {
"Infinity".into()
} else if v.0 == f32::NEG_INFINITY {
"-Infinity".into()
} else if v.0.is_nan() {
"NaN".into()
} else {
v.0.into()
}
}
}

// NOTE: Infinite or NaN values are not JSON numbers. They are stored as strings in Postgres.
impl From<F64> for Value {
fn from(v: F64) -> Self {
if v.0 == f64::INFINITY {
"Infinity".into()
} else if v.0 == f64::NEG_INFINITY {
"-Infinity".into()
} else if v.0.is_nan() {
"NaN".into()
} else {
v.0.into()
}
}
}

impl From<JsonbRef<'_>> for Value {
fn from(v: JsonbRef<'_>) -> Self {
v.0.to_owned()
}
}

impl From<Decimal> for Value {
fn from(v: Decimal) -> Self {
// Make it numeric if it can be converted to IEEE 754 double, otherwise a string.
// May loss precision here.
let res: Result<F64, _> = v.try_into();
if let Ok(value) = res {
value.into()
} else {
v.to_text().as_str().into()
}
}
}

impl From<Timestamp> for Value {
fn from(v: Timestamp) -> Self {
let mut builder = Builder::default();
builder.display(format!("{}T{}", v.0.date(), v.0.time()));
builder.finish()
}
}

macro_rules! impl_convert_to_value {
($scalar_type:ty) => {
impl From<$scalar_type> for Value {
fn from(v: $scalar_type) -> Self {
v.to_text().as_str().into()
}
}
};
}

impl_convert_to_value!(Serial);
impl_convert_to_value!(Int256Ref<'_>);
impl_convert_to_value!(Date);
impl_convert_to_value!(Time);
impl_convert_to_value!(Interval);

macro_rules! impl_convert_to_jsonb_val {
($scalar_ref:ty) => {
impl<'scalar> From<$scalar_ref> for JsonbVal {
fn from(val: $scalar_ref) -> Self {
Self(val.into())
}
}
};
}

impl_convert_to_jsonb_val!(bool);
impl_convert_to_jsonb_val!(i16);
impl_convert_to_jsonb_val!(i32);
impl_convert_to_jsonb_val!(i64);
impl_convert_to_jsonb_val!(Int256Ref<'_>);
impl_convert_to_jsonb_val!(F32);
impl_convert_to_jsonb_val!(F64);
impl_convert_to_jsonb_val!(Decimal);
impl_convert_to_jsonb_val!(Serial);
impl_convert_to_jsonb_val!(Date);
impl_convert_to_jsonb_val!(Time);
impl_convert_to_jsonb_val!(Timestamp);
impl_convert_to_jsonb_val!(Interval);
impl_convert_to_jsonb_val!(&str);
impl_convert_to_jsonb_val!(JsonbRef<'_>);
2 changes: 1 addition & 1 deletion src/expr/impl/benches/expr.rs
Original file line number Diff line number Diff line change
Expand Up @@ -170,7 +170,7 @@ fn bench_expr(c: &mut Criterion) {
// 25: serial array
SerialArray::from_iter((1..=CHUNK_SIZE).map(|i| Serial::from(i as i64))).into_ref(),
// 26: jsonb array
JsonbArray::from_iter((1..=CHUNK_SIZE).map(|i| JsonbVal::from(i as i64))).into_ref(),
JsonbArray::from_iter((1..=CHUNK_SIZE).map(|i| JsonbVal::from(i as f64))).into_ref(),
// 27: int256 array
Int256Array::from_iter((1..=CHUNK_SIZE).map(|_| Int256::from(1))).into_ref(),
// 28: extract field for interval
Expand Down
4 changes: 2 additions & 2 deletions src/expr/impl/src/scalar/to_jsonb.rs
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,7 @@ impl ToJsonb for F64 {
} else if self.0.is_nan() {
"NaN".add_to(builder)?;
} else {
builder.add_f64(self.0 as f64);
builder.add_f64(self.0);
}
Ok(())
}
Expand Down Expand Up @@ -267,7 +267,7 @@ impl ToJsonb for Interval {

impl ToJsonb for Timestamp {
fn add_to(self, builder: &mut Builder) -> Result<()> {
builder.display(format!("{}T{}", self.0.date(), self.0.time()));
builder.display(format_args!("{}T{}", self.0.date(), self.0.time()));
Ok(())
}
}
Expand Down

0 comments on commit 6f67496

Please sign in to comment.