langdb_core 0.3.2

AI gateway Core for LangDB AI Gateway.
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
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

pub mod clickhouse;
pub mod error;
pub mod executor;
pub mod ssh_tunnel;
pub mod user;

use error::QueryError;
use futures::Stream;
use serde::Deserialize;
use serde_json::Value;
use std::fmt::Debug;
use std::pin::Pin;

const MAX_LIMIT: i32 = 10000;

type Result<T> = std::result::Result<T, QueryError>;

#[async_trait::async_trait]
pub trait DatabaseTransport: Send + Sync + DatabaseTransportClone + 'static {
    async fn execute_binary(&self, sql: &str) -> Result<Vec<u8>>;
    async fn execute(&self, sql: &str) -> Result<String>;
    async fn execute_compact_json(&self, sql: &str) -> Result<String>;
    async fn execute_delete(&self, sql: &str) -> Result<String>;
    async fn insert_values(
        &self,
        table_name: &str,
        columns: &[&str],
        body: Vec<Vec<Value>>,
    ) -> Result<String>;
    async fn query_stream(
        &self,
        sql: &str,
    ) -> Result<(
        Vec<String>,
        Pin<Box<dyn Stream<Item = Result<Vec<Value>>> + Send>>,
    )>;
    async fn test_connection(&self) -> Result<()>;
}

#[derive(Deserialize, Debug, Clone)]
pub struct QueryResponse<T>
where
    T: Sized,
{
    pub meta: Vec<QueryMetaItem>,
    pub data: Vec<T>,
    #[serde(default)]
    pub rows: usize,
    pub statistics: Option<QueryStatistics>,
    pub exception: Option<String>,
}

#[derive(Deserialize, Debug, Clone)]
pub struct QueryMetaItem {
    pub name: String,
    pub r#type: String,
}

#[derive(Deserialize, Debug, Clone)]
pub struct QueryStatistics {
    pub elapsed: f64,
    pub rows_read: usize,
    pub bytes_read: usize,
}

pub trait DatabaseTransportClone {
    fn clone_box(&self) -> Box<dyn DatabaseTransport + Send + Sync>;
}

impl<T> DatabaseTransportClone for T
where
    T: 'static + DatabaseTransport + Clone + Send + Sync,
{
    fn clone_box(&self) -> Box<dyn DatabaseTransport + Send + Sync> {
        Box::new(self.clone())
    }
}

impl Clone for Box<dyn DatabaseTransport + Send + Sync> {
    fn clone(&self) -> Box<dyn DatabaseTransport + Send + Sync> {
        self.clone_box()
    }
}

impl Clone for Box<dyn DatabaseTransport> {
    fn clone(&self) -> Self {
        self.clone_box()
    }
}