add Money support to mssql

Author: c5soft

Cargo.lock

@@ -28,6 +28,7 @@ impl FromStr for MssqlConnectOptions {
     ///   - `strict`: Requires encryption and validates the server certificate.
     ///   - `mandatory` or `true` or `yes`: Requires encryption but doesn't validate the server certificate.
     ///   - `optional` or `false` or `no`: Uses encryption if available, falls back to unencrypted.
+    ///   - `not_supported`: No encryption.
     /// - `sslrootcert` or `ssl-root-cert` or `ssl-ca`: Path to the root certificate for validating the server's SSL certificate.
     /// - `trust_server_certificate`: When true, skips validation of the server's SSL certificate. Use with caution as it makes the connection vulnerable to man-in-the-middle attacks.
     /// - `hostname_in_certificate`: The hostname expected in the server's SSL certificate. Use this when the server's hostname doesn't match the certificate.

sqlx-core/src/mssql/options/parse.rs

@@ -23,7 +23,13 @@ impl Type<Mssql> for BigDecimal {
     fn compatible(ty: &MssqlTypeInfo) -> bool {
         matches!(
             ty.0.ty,
-            DataType::Numeric | DataType::NumericN | DataType::Decimal | DataType::DecimalN
+            DataType::Numeric
+                | DataType::NumericN
+                | DataType::Decimal
+                | DataType::DecimalN
+                | DataType::MoneyN
+                | DataType::Money
+                | DataType::SmallMoney
         )
     }
 }
@@ -58,15 +64,32 @@ impl Encode<'_, Mssql> for BigDecimal {
 impl Decode<'_, Mssql> for BigDecimal {
     fn decode(value: MssqlValueRef<'_>) -> Result<Self, BoxDynError> {
         let ty = value.type_info.0.ty;
-        if !matches!(
-            ty,
-            DataType::Decimal | DataType::DecimalN | DataType::Numeric | DataType::NumericN
-        ) {
-            return Err(err_protocol!("expected numeric type, got {:?}", value.type_info.0).into());
+        match ty {
+            DataType::Decimal | DataType::DecimalN | DataType::Numeric | DataType::NumericN => {
+                let precision = value.type_info.0.precision;
+                let scale = value.type_info.0.scale;
+                decode_numeric(value.as_bytes()?, precision, scale)
+            }
+            DataType::MoneyN | DataType::Money | DataType::SmallMoney => {
+                // Money is stored as an 8-byte integer representing the amount in ten-thousandths of a currency unit
+                let bytes = value.as_bytes()?;
+                // println!("bytes: {:?}", bytes);
+                if bytes.len() != 8 && bytes.len() != 4 {
+                    return Err(
+                        err_protocol!("expected 8/4 bytes for Money, got {}", bytes.len()).into(),
+                    );
+                }
+                let amount = if bytes.len() == 8 {
+                    let amount_h = i32::from_le_bytes(bytes[0..4].try_into()?) as i64;
+                    let amount_l = u32::from_le_bytes(bytes[4..8].try_into()?) as i64;
+                    (amount_h << 32) | amount_l
+                } else {
+                    i32::from_le_bytes(bytes.try_into()?) as i64
+                };
+                Ok(BigDecimal::new(BigInt::from(amount), 4))
+            }
+            _ => Err(err_protocol!("expected numeric type, got {:?}", value.type_info.0).into()),
         }
-        let precision = value.type_info.0.precision;
-        let scale = value.type_info.0.scale;
-        decode_numeric(value.as_bytes()?, precision, scale)
     }
 }
 

sqlx-core/src/mssql/types/bigdecimal.rs

@@ -21,7 +21,13 @@ impl Type<Mssql> for Decimal {
     fn compatible(ty: &MssqlTypeInfo) -> bool {
         matches!(
             ty.0.ty,
-            DataType::Numeric | DataType::NumericN | DataType::Decimal | DataType::DecimalN
+            DataType::Numeric
+                | DataType::NumericN
+                | DataType::Decimal
+                | DataType::DecimalN
+                | DataType::MoneyN
+                | DataType::Money
+                | DataType::SmallMoney
         )
     }
 }
@@ -49,15 +55,32 @@ impl Encode<'_, Mssql> for Decimal {
 impl Decode<'_, Mssql> for Decimal {
     fn decode(value: MssqlValueRef<'_>) -> Result<Self, BoxDynError> {
         let ty = value.type_info.0.ty;
-        if !matches!(
-            ty,
-            DataType::Decimal | DataType::DecimalN | DataType::Numeric | DataType::NumericN
-        ) {
-            return Err(err_protocol!("expected numeric type, got {:?}", value.type_info.0).into());
+        match ty {
+            DataType::Decimal | DataType::DecimalN | DataType::Numeric | DataType::NumericN => {
+                let precision = value.type_info.0.precision;
+                let scale = value.type_info.0.scale;
+                decode_numeric(value.as_bytes()?, precision, scale)
+            }
+            DataType::MoneyN | DataType::Money | DataType::SmallMoney => {
+                // Money is stored as an 8-byte integer representing the amount in ten-thousandths of a currency unit
+                let bytes = value.as_bytes()?;
+                // println!("bytes: {:?}", bytes);
+                if bytes.len() != 8 && bytes.len() != 4 {
+                    return Err(
+                        err_protocol!("expected 8/4 bytes for Money, got {}", bytes.len()).into(),
+                    );
+                }
+                let amount: i64 = if bytes.len() == 8 {
+                    let amount_h = i32::from_le_bytes(bytes[0..4].try_into()?) as i64;
+                    let amount_l = u32::from_le_bytes(bytes[4..8].try_into()?) as i64;
+                    (amount_h << 32) | amount_l
+                } else {
+                    i32::from_le_bytes(bytes.try_into()?) as i64
+                };
+                Ok(Decimal::new(amount, 4))
+            }
+            _ => Err(err_protocol!("expected numeric type, got {:?}", value.type_info.0).into()),
         }
-        let precision = value.type_info.0.precision;
-        let scale = value.type_info.0.scale;
-        decode_numeric(value.as_bytes()?, precision, scale)
     }
 }
 

sqlx-core/src/mssql/types/decimal.rs

@@ -25,6 +25,16 @@ impl_database_ext! {
 
         #[cfg(feature = "chrono")]
         sqlx::types::chrono::DateTime<sqlx::types::chrono::FixedOffset>,
+
+        #[cfg(feature = "bigdecimal")]
+        sqlx::types::BigDecimal,
+
+        #[cfg(feature = "decimal")]
+        sqlx::types::Decimal,
+
+        #[cfg(feature = "json")]
+        sqlx::types::JsonValue,
+
     },
     ParamChecking::Weak,
     feature-types: _info => None,