ci: make output of ttc server with arrow consist with http handler.

Author: youngsofun

sql/src/raw_rows.rs

@@ -16,14 +16,25 @@ use crate::error::Error;
 use crate::error::Result;
 use crate::rows::Row;
 use crate::rows::ServerStats;
+use crate::value::FormatOptions;
 use crate::value::Value;
 use chrono_tz::Tz;
 use databend_client::schema::SchemaRef;
+use lexical_core::WriteFloatOptionsBuilder;
 use std::pin::Pin;
 use std::task::Context;
 use std::task::Poll;
 use tokio_stream::{Stream, StreamExt};
 
+pub static HTTP_HANDLER_OPTIONS: FormatOptions = FormatOptions {
+    true_string: b"1",
+    false_string: b"0",
+    float_options: WriteFloatOptionsBuilder::new()
+        .nan_string(Some(b"NaN"))
+        .nan_string(Some(b"Infinity"))
+        .build_unchecked(),
+};
+
 #[derive(Clone, Debug)]
 pub enum RawRowWithStats {
     Row(RawRow),
@@ -76,7 +87,7 @@ impl From<Row> for RawRow {
     fn from(row: Row) -> Self {
         let mut raw_row: Vec<Option<String>> = Vec::with_capacity(row.values().len());
         for val in row.values() {
-            raw_row.push(Some(val.to_string()));
+            raw_row.push(Some(val.to_string_with_options(&HTTP_HANDLER_OPTIONS)));
         }
         RawRow::new(row, raw_row)
     }

sql/src/value/convert.rs

@@ -20,40 +20,7 @@ use chrono_tz::Tz;
 
 use crate::error::{ConvertError, Error, Result};
 
-use super::{
-    display_decimal_128, display_decimal_256, NumberValue, Value, DAYS_FROM_CE, TIMESTAMP_FORMAT,
-};
-
-impl TryFrom<Value> for String {
-    type Error = Error;
-    fn try_from(val: Value) -> Result<Self> {
-        match val {
-            Value::String(s) => Ok(s),
-            Value::Bitmap(s) => Ok(s),
-            Value::Number(NumberValue::Decimal128(v, s)) => Ok(display_decimal_128(v, s.scale)),
-            Value::Number(NumberValue::Decimal256(v, s)) => Ok(display_decimal_256(v, s.scale)),
-            Value::Geometry(s) => Ok(s),
-            Value::Geography(s) => Ok(s),
-            Value::Interval(s) => Ok(s),
-            Value::Variant(s) => Ok(s),
-            Value::Date(d) => {
-                let date =
-                    NaiveDate::from_num_days_from_ce_opt(d + DAYS_FROM_CE).ok_or_else(|| {
-                        ConvertError::new("date", format!("invalid date value: {}", d))
-                    })?;
-                Ok(date.format("%Y-%m-%d").to_string())
-            }
-            Value::Timestamp(ts, tz) => {
-                let dt = DateTime::from_timestamp_micros(ts).ok_or_else(|| {
-                    ConvertError::new("timestamp", format!("invalid timestamp: {}", ts))
-                })?;
-                let dt = dt.with_timezone(&tz);
-                Ok(dt.format(TIMESTAMP_FORMAT).to_string())
-            }
-            _ => Err(ConvertError::new("string", format!("{val:?}")).into()),
-        }
-    }
-}
+use super::{NumberValue, Value, DAYS_FROM_CE};
 
 impl TryFrom<Value> for bool {
     type Error = Error;