ruint2/support/
sqlx.rs

1//! Support for the [`sqlx`](https://crates.io/crates/sqlx) crate.
2//!
3//! Currently only encodes to/from a big-endian byte array.
4#![cfg(feature = "sqlx")]
5#![cfg_attr(has_doc_cfg, doc(cfg(feature = "sqlx")))]
6
7use crate::Uint;
8use sqlx_core::{
9    database::{Database, HasArguments, HasValueRef},
10    decode::Decode,
11    encode::{Encode, IsNull},
12    error::BoxDynError,
13    types::Type,
14};
15use thiserror::Error;
16
17#[derive(Error, Debug)]
18enum DecodeError {
19    #[error("Value too large for target type")]
20    Overflow,
21}
22
23impl<const BITS: usize, const LIMBS: usize, DB: Database> Type<DB> for Uint<BITS, LIMBS>
24where
25    Vec<u8>: Type<DB>,
26{
27    fn type_info() -> DB::TypeInfo {
28        <Vec<u8> as Type<DB>>::type_info()
29    }
30
31    fn compatible(ty: &DB::TypeInfo) -> bool {
32        <Vec<u8> as Type<DB>>::compatible(ty)
33    }
34}
35
36impl<'a, const BITS: usize, const LIMBS: usize, DB: Database> Encode<'a, DB> for Uint<BITS, LIMBS>
37where
38    Vec<u8>: Encode<'a, DB>,
39{
40    fn encode_by_ref(&self, buf: &mut <DB as HasArguments<'a>>::ArgumentBuffer) -> IsNull {
41        self.to_be_bytes_vec().encode_by_ref(buf)
42    }
43}
44
45impl<'a, const BITS: usize, const LIMBS: usize, DB: Database> Decode<'a, DB> for Uint<BITS, LIMBS>
46where
47    Vec<u8>: Decode<'a, DB>,
48{
49    fn decode(value: <DB as HasValueRef<'a>>::ValueRef) -> Result<Self, BoxDynError> {
50        let bytes = Vec::<u8>::decode(value)?;
51        Self::try_from_be_slice(bytes.as_slice()).ok_or_else(|| DecodeError::Overflow.into())
52    }
53}
ruint2/support/sqlx.rs

ruint2/support/
sqlx.rs
