supabase-wrappers 0.1.26

Postgres Foreign Data Wrapper development framework in Rust.
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

use std::collections::HashMap;
use std::ffi::CStr;

use crate::prelude::*;
use pgrx::pg_sys::panic::ErrorReport;
use pgrx::{pg_sys::Oid, prelude::*};

#[derive(Debug, Clone, Default)]
pub struct ForeignServer {
    pub server_oid: Oid,
    pub server_name: String,
    pub server_type: Option<String>,
    pub server_version: Option<String>,
    pub options: HashMap<String, String>,
}

// create a fdw instance from its id
pub(super) unsafe fn create_fdw_instance_from_server_id<
    E: Into<ErrorReport>,
    W: ForeignDataWrapper<E>,
>(
    fserver_id: pg_sys::Oid,
) -> W {
    let to_string = |raw: *mut std::ffi::c_char| -> Option<String> {
        if raw.is_null() {
            return None;
        }
        let c_str = unsafe { CStr::from_ptr(raw) };
        let value = c_str
            .to_str()
            .map_err(|_| OptionsError::OptionValueIsInvalidUtf8 {
                option_name: String::from_utf8_lossy(c_str.to_bytes()).to_string(),
            })
            .report_unwrap()
            .to_string();
        Some(value)
    };
    unsafe {
        let fserver = pg_sys::GetForeignServer(fserver_id);
        let server = ForeignServer {
            server_oid: fserver_id,
            server_name: to_string((*fserver).servername).unwrap(),
            server_type: to_string((*fserver).servertype),
            server_version: to_string((*fserver).serverversion),
            options: options_to_hashmap((*fserver).options).report_unwrap(),
        };
        let wrapper = W::new(server);
        wrapper.report_unwrap()
    }
}

// create a fdw instance from a foreign table id
pub(super) unsafe fn create_fdw_instance_from_table_id<
    E: Into<ErrorReport>,
    W: ForeignDataWrapper<E>,
>(
    ftable_id: pg_sys::Oid,
) -> W {
    unsafe {
        let ftable = pg_sys::GetForeignTable(ftable_id);
        create_fdw_instance_from_server_id((*ftable).serverid)
    }
}