1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

use crate::ast::Value;
#[cfg(not(feature = "bigdecimal"))]
use crate::error::*;
#[cfg(feature = "bigdecimal")]
use bigdecimal::BigDecimal;
use std::{borrow::Cow, convert::TryFrom};
#[cfg(feature = "bigdecimal")]
use tiberius::ToSql;
use tiberius::{ColumnData, FromSql, IntoSql};

impl<'a> IntoSql<'a> for &'a Value<'a> {
    fn into_sql(self) -> ColumnData<'a> {
        match self {
            Value::Int32(val) => val.into_sql(),
            Value::Int64(val) => val.into_sql(),
            Value::Float(val) => val.into_sql(),
            Value::Double(val) => val.into_sql(),
            Value::Text(val) => val.as_deref().into_sql(),
            Value::Bytes(val) => val.as_deref().into_sql(),
            Value::Enum(val) => val.as_deref().into_sql(),
            Value::Boolean(val) => val.into_sql(),
            Value::Char(val) => val.as_ref().map(|val| format!("{val}")).into_sql(),
            Value::Xml(val) => val.as_deref().into_sql(),
            Value::Array(_) => panic!("Arrays are not supported on SQL Server."),
            #[cfg(feature = "bigdecimal")]
            Value::Numeric(val) => (*val).to_sql(),
            #[cfg(feature = "json")]
            Value::Json(val) => val.as_ref().map(|val| serde_json::to_string(&val).unwrap()).into_sql(),
            #[cfg(feature = "uuid")]
            Value::Uuid(val) => val.into_sql(),
            #[cfg(feature = "chrono")]
            Value::DateTime(val) => val.into_sql(),
            #[cfg(feature = "chrono")]
            Value::Date(val) => val.into_sql(),
            #[cfg(feature = "chrono")]
            Value::Time(val) => val.into_sql(),
        }
    }
}

impl TryFrom<ColumnData<'static>> for Value<'static> {
    type Error = crate::error::Error;

    fn try_from(cd: ColumnData<'static>) -> crate::Result<Self> {
        let res = match cd {
            ColumnData::U8(num) => Value::Int32(num.map(i32::from)),
            ColumnData::I16(num) => Value::Int32(num.map(i32::from)),
            ColumnData::I32(num) => Value::Int32(num.map(i32::from)),
            ColumnData::I64(num) => Value::Int64(num.map(i64::from)),
            ColumnData::F32(num) => Value::Float(num),
            ColumnData::F64(num) => Value::Double(num),
            ColumnData::Bit(b) => Value::Boolean(b),
            ColumnData::String(s) => Value::Text(s),
            ColumnData::Guid(uuid) => Value::Uuid(uuid),
            ColumnData::Binary(bytes) => Value::Bytes(bytes),
            #[cfg(feature = "bigdecimal")]
            numeric @ ColumnData::Numeric(_) => Value::Numeric(BigDecimal::from_sql(&numeric)?),
            #[cfg(not(feature = "bigdecimal"))]
            _numeric @ ColumnData::Numeric(_) => {
                let kind = ErrorKind::conversion("Please enable `bigdecimal` feature to read numeric values");
                return Err(Error::builder(kind).build());
            }
            #[cfg(feature = "chrono")]
            dt @ ColumnData::DateTime(_) => {
                use tiberius::time::chrono::{DateTime, NaiveDateTime, Utc};

                let dt = NaiveDateTime::from_sql(&dt)?.map(|dt| DateTime::<Utc>::from_utc(dt, Utc));
                Value::DateTime(dt)
            }
            #[cfg(feature = "chrono")]
            dt @ ColumnData::SmallDateTime(_) => {
                use tiberius::time::chrono::{DateTime, NaiveDateTime, Utc};

                let dt = NaiveDateTime::from_sql(&dt)?.map(|dt| DateTime::<Utc>::from_utc(dt, Utc));
                Value::DateTime(dt)
            }
            #[cfg(feature = "chrono")]
            dt @ ColumnData::Time(_) => {
                use tiberius::time::chrono::NaiveTime;

                Value::Time(NaiveTime::from_sql(&dt)?)
            }
            #[cfg(feature = "chrono")]
            dt @ ColumnData::Date(_) => {
                use tiberius::time::chrono::NaiveDate;
                Value::Date(NaiveDate::from_sql(&dt)?)
            }
            #[cfg(feature = "chrono")]
            dt @ ColumnData::DateTime2(_) => {
                use tiberius::time::chrono::{DateTime, NaiveDateTime, Utc};

                let dt = NaiveDateTime::from_sql(&dt)?.map(|dt| DateTime::<Utc>::from_utc(dt, Utc));

                Value::DateTime(dt)
            }
            #[cfg(feature = "chrono")]
            dt @ ColumnData::DateTimeOffset(_) => {
                use tiberius::time::chrono::{DateTime, Utc};

                Value::DateTime(DateTime::<Utc>::from_sql(&dt)?)
            }
            ColumnData::Xml(cow) => Value::Xml(cow.map(|xml_data| Cow::Owned(xml_data.into_owned().into_string()))),
        };

        Ok(res)
    }
}