mirror of
https://github.com/isar/rusqlite.git
synced 2024-11-23 00:39:20 +08:00
5e2c103a0c
* Add some missing wrappers: sqlite3_value_subtype sqlite3_result_subtype sqlite3_changes64 sqlite3_db_readonly sqlite3_txn_state sqlite3_stmt_isexplain sqlite3_vtab_config sqlite3_index_info.idxFlags sqlite3_index_info.colUsed sqlite3_index_info.idxStr sqlite3_vtab_collation * Mark series VTab as innocuous and csv as direct only
76 lines
2.6 KiB
Rust
76 lines
2.6 KiB
Rust
//! Code related to `sqlite3_context` common to `functions` and `vtab` modules.
|
|
|
|
use std::os::raw::{c_int, c_void};
|
|
#[cfg(feature = "array")]
|
|
use std::rc::Rc;
|
|
|
|
use crate::ffi;
|
|
use crate::ffi::sqlite3_context;
|
|
|
|
use crate::str_for_sqlite;
|
|
use crate::types::{ToSqlOutput, ValueRef};
|
|
#[cfg(feature = "array")]
|
|
use crate::vtab::array::{free_array, ARRAY_TYPE};
|
|
|
|
// This function is inline despite it's size because what's in the ToSqlOutput
|
|
// is often known to the compiler, and thus const prop/DCE can substantially
|
|
// simplify the function.
|
|
#[inline]
|
|
pub(super) unsafe fn set_result(ctx: *mut sqlite3_context, result: &ToSqlOutput<'_>) {
|
|
let value = match *result {
|
|
ToSqlOutput::Borrowed(v) => v,
|
|
ToSqlOutput::Owned(ref v) => ValueRef::from(v),
|
|
|
|
#[cfg(feature = "blob")]
|
|
ToSqlOutput::ZeroBlob(len) => {
|
|
// TODO sqlite3_result_zeroblob64 // 3.8.11
|
|
return ffi::sqlite3_result_zeroblob(ctx, len);
|
|
}
|
|
#[cfg(feature = "array")]
|
|
ToSqlOutput::Array(ref a) => {
|
|
return ffi::sqlite3_result_pointer(
|
|
ctx,
|
|
Rc::into_raw(a.clone()) as *mut c_void,
|
|
ARRAY_TYPE,
|
|
Some(free_array),
|
|
);
|
|
}
|
|
};
|
|
|
|
match value {
|
|
ValueRef::Null => ffi::sqlite3_result_null(ctx),
|
|
ValueRef::Integer(i) => ffi::sqlite3_result_int64(ctx, i),
|
|
ValueRef::Real(r) => ffi::sqlite3_result_double(ctx, r),
|
|
ValueRef::Text(s) => {
|
|
let length = s.len();
|
|
if length > c_int::max_value() as usize {
|
|
ffi::sqlite3_result_error_toobig(ctx);
|
|
} else {
|
|
let (c_str, len, destructor) = match str_for_sqlite(s) {
|
|
Ok(c_str) => c_str,
|
|
// TODO sqlite3_result_error
|
|
Err(_) => return ffi::sqlite3_result_error_code(ctx, ffi::SQLITE_MISUSE),
|
|
};
|
|
// TODO sqlite3_result_text64 // 3.8.7
|
|
ffi::sqlite3_result_text(ctx, c_str, len, destructor);
|
|
}
|
|
}
|
|
ValueRef::Blob(b) => {
|
|
let length = b.len();
|
|
if length > c_int::max_value() as usize {
|
|
ffi::sqlite3_result_error_toobig(ctx);
|
|
} else if length == 0 {
|
|
ffi::sqlite3_result_zeroblob(ctx, 0);
|
|
} else {
|
|
// TODO sqlite3_result_blob64 // 3.8.7
|
|
ffi::sqlite3_result_blob(
|
|
ctx,
|
|
b.as_ptr().cast::<c_void>(),
|
|
length as c_int,
|
|
ffi::SQLITE_TRANSIENT(),
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|