ducklake 0.0.5

Rust SDK for DuckLake.
Documentation 
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

mod factory;
mod nested;
mod primitive;

use chrono::{DateTime, NaiveDate, NaiveDateTime, NaiveTime, Utc};
pub use factory::make_column_encoder;
use rust_decimal::Decimal;
use uuid::Uuid;

use crate::DucklakeResult;
use crate::primitives::{Interval, TimeWithTimezone};

/* ----------------------------------- STEP 1: ARRAY -> RUST ----------------------------------- */

pub trait ArrayExtractor<E: TypeEncoder>: Send + Sync {
    fn extract(&self, args: &mut E::Arguments, row_idx: usize) -> DucklakeResult<()>;
    fn extract_text(&self, row_idx: usize) -> String;
}

/* ---------------------------------- STEP 2: RUST -> DATABASE --------------------------------- */

pub trait EncodableArguments: Default {
    type Encoder: TypeEncoder<Arguments = Self> + 'static;
}

pub trait Bindable<DB: sqlx::Database> =
    sqlx::Encode<'static, DB> + sqlx::Type<DB> + Send + 'static;

pub trait TypeEncoder: Send + Sync + 'static {
    type Arguments: sqlx::Arguments<'static, Database = Self::Database> + Default;
    type Database: sqlx::Database;

    fn encode_bool(value: Option<bool>) -> impl Bindable<Self::Database>;
    fn encode_i8(value: Option<i8>) -> impl Bindable<Self::Database>;
    fn encode_i16(value: Option<i16>) -> impl Bindable<Self::Database>;
    fn encode_i32(value: Option<i32>) -> impl Bindable<Self::Database>;
    fn encode_i64(value: Option<i64>) -> impl Bindable<Self::Database>;
    fn encode_i128(value: Option<i128>) -> impl Bindable<Self::Database>;
    fn encode_u8(value: Option<u8>) -> impl Bindable<Self::Database>;
    fn encode_u16(value: Option<u16>) -> impl Bindable<Self::Database>;
    fn encode_u32(value: Option<u32>) -> impl Bindable<Self::Database>;
    fn encode_u64(value: Option<u64>) -> impl Bindable<Self::Database>;
    fn encode_u128(value: Option<u128>) -> impl Bindable<Self::Database>;
    fn encode_f32(value: Option<f32>) -> impl Bindable<Self::Database>;
    fn encode_f64(value: Option<f64>) -> impl Bindable<Self::Database>;
    fn encode_decimal(value: Option<Decimal>) -> impl Bindable<Self::Database>;
    fn encode_time(value: Option<NaiveTime>) -> impl Bindable<Self::Database>;
    fn encode_time_tz(value: Option<TimeWithTimezone>) -> impl Bindable<Self::Database>;
    fn encode_date(value: Option<NaiveDate>) -> impl Bindable<Self::Database>;
    fn encode_timestamp(value: Option<NaiveDateTime>) -> impl Bindable<Self::Database>;
    fn encode_timestamp_tz(value: Option<DateTime<Utc>>) -> impl Bindable<Self::Database>;
    fn encode_interval(value: Option<Interval>) -> impl Bindable<Self::Database>;
    fn encode_string(value: Option<&str>) -> impl Bindable<Self::Database>;
    fn encode_binary(value: Option<&[u8]>) -> impl Bindable<Self::Database>;
    fn encode_uuid(value: Option<Uuid>) -> impl Bindable<Self::Database>;

    /// Encode a value whose representation in the database is always text
    /// (used for nested types — lists, structs, maps). Mirrors
    /// `RowDecoder::decode_text` on the read side.
    fn encode_text(value: Option<&str>) -> impl Bindable<Self::Database>;
}