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
use crate::as_string;
use crate::duckly::{
duckdb_function_get_bind_data, duckdb_function_get_extra_info, duckdb_function_get_init_data,
duckdb_function_get_local_init_data, duckdb_function_info, duckdb_function_set_error,
};
use std::os::raw::c_char;
#[derive(Debug)]
pub struct FunctionInfo(duckdb_function_info);
impl FunctionInfo {
/// Report that an error has occurred while executing the function.
///
/// # Arguments
/// * `error`: The error message
pub fn set_error(&self, error: &str) {
unsafe {
duckdb_function_set_error(self.0, as_string!(error));
}
}
/// Gets the bind data set by [`BindInfo#set_bind_data`] during the bind.
///
/// Note that the bind data should be considered as read-only.
/// For tracking state, use the init data instead.
///
/// # Arguments
/// * `returns`: The bind data object
pub fn get_bind_data<T>(&self) -> *mut T {
unsafe { duckdb_function_get_bind_data(self.0).cast() }
}
/// Gets the init data set by [`InitInfo#set_init_data`] during the init.
///
/// # Arguments
/// * `returns`: The init data object
pub fn get_init_data<T>(&self) -> *mut T {
unsafe { duckdb_function_get_init_data(self.0).cast() }
}
/// Retrieves the extra info of the function as set in [`TableFunction#set_extra_info`]
///
/// # Arguments
/// * `returns`: The extra info
pub fn get_extra_info<T>(&self) -> *mut T {
unsafe { duckdb_function_get_extra_info(self.0).cast() }
}
/// Gets the thread-local init data set by [`InitInfo#set_init_data`] during the local_init.
///
/// # Arguments
/// * `returns`: The init data object
pub fn get_local_init_data<T>(&self) -> *mut T {
unsafe { duckdb_function_get_local_init_data(self.0).cast() }
}
}
impl From<duckdb_function_info> for FunctionInfo {
fn from(ptr: duckdb_function_info) -> Self {
Self(ptr)
}
}