mirror of
https://github.com/fluencelabs/sqlite-wasm-connector
synced 2025-03-15 06:20:50 +00:00
Move Database into a separate file
This commit is contained in:
parent
9f95b9f645
commit
1dfeee881b
80
src/database.rs
Normal file
80
src/database.rs
Normal file
@ -0,0 +1,80 @@
|
||||
use libc::{c_char, c_int, c_void};
|
||||
use raw;
|
||||
use std::marker::PhantomData;
|
||||
use std::path::Path;
|
||||
|
||||
use Result;
|
||||
|
||||
/// A database.
|
||||
pub struct Database<'d> {
|
||||
db: *mut raw::sqlite3,
|
||||
_phantom: PhantomData<&'d raw::sqlite3>,
|
||||
}
|
||||
|
||||
/// A callback executed for each row of the result of an SQL query.
|
||||
pub type ExecuteCallback<'c> = FnMut(Vec<(String, String)>) -> bool + 'c;
|
||||
|
||||
impl<'d> Database<'d> {
|
||||
/// Open a database.
|
||||
pub fn open(path: &Path) -> Result<Database<'d>> {
|
||||
let mut db = 0 as *mut _;
|
||||
unsafe {
|
||||
success!(raw::sqlite3_open(path_to_c_str!(path), &mut db));
|
||||
}
|
||||
Ok(Database { db: db, _phantom: PhantomData })
|
||||
}
|
||||
|
||||
/// Execute an SQL statement.
|
||||
pub fn execute<'c>(&mut self, sql: &str,
|
||||
callback: Option<&mut ExecuteCallback<'c>>) -> Result<()> {
|
||||
|
||||
unsafe {
|
||||
match callback {
|
||||
Some(callback) => {
|
||||
let mut callback = Box::new(callback);
|
||||
success!(raw::sqlite3_exec(self.db, str_to_c_str!(sql), Some(execute_callback),
|
||||
&mut callback as *mut _ as *mut _, 0 as *mut _));
|
||||
},
|
||||
None => {
|
||||
success!(raw::sqlite3_exec(self.db, str_to_c_str!(sql), None,
|
||||
0 as *mut _, 0 as *mut _));
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
impl<'d> Drop for Database<'d> {
|
||||
#[inline]
|
||||
fn drop(&mut self) {
|
||||
unsafe { ::raw::sqlite3_close(self.db) };
|
||||
}
|
||||
}
|
||||
|
||||
extern fn execute_callback(callback: *mut c_void, count: c_int, values: *mut *mut c_char,
|
||||
columns: *mut *mut c_char) -> c_int {
|
||||
|
||||
macro_rules! c_str_to_string(
|
||||
($string:expr) => (
|
||||
match ::std::str::from_utf8(::std::ffi::CStr::from_ptr($string).to_bytes()) {
|
||||
Ok(string) => String::from(string),
|
||||
Err(_) => return 1,
|
||||
}
|
||||
);
|
||||
);
|
||||
|
||||
unsafe {
|
||||
let mut pairs = Vec::with_capacity(count as usize);
|
||||
|
||||
for i in 0..(count as isize) {
|
||||
let column = c_str_to_string!(*columns.offset(i) as *const _);
|
||||
let value = c_str_to_string!(*values.offset(i) as *const _);
|
||||
pairs.push((column, value));
|
||||
}
|
||||
|
||||
let ref mut callback = *(callback as *mut Box<&mut ExecuteCallback>);
|
||||
if callback(pairs) { 0 } else { 1 }
|
||||
}
|
||||
}
|
148
src/lib.rs
148
src/lib.rs
@ -3,8 +3,6 @@
|
||||
extern crate libc;
|
||||
extern crate sqlite3_sys as raw;
|
||||
|
||||
use libc::{c_char, c_int, c_void};
|
||||
use std::marker::PhantomData;
|
||||
use std::path::Path;
|
||||
|
||||
/// A result.
|
||||
@ -17,6 +15,42 @@ pub struct Error {
|
||||
pub message: Option<String>,
|
||||
}
|
||||
|
||||
/// An error code.
|
||||
#[derive(Clone, Copy, Debug)]
|
||||
pub enum ErrorCode {
|
||||
Abort = raw::SQLITE_ABORT as isize,
|
||||
Authorization = raw::SQLITE_AUTH as isize,
|
||||
Busy = raw::SQLITE_BUSY as isize,
|
||||
CantOpen = raw::SQLITE_CANTOPEN as isize,
|
||||
Constraint = raw::SQLITE_CONSTRAINT as isize,
|
||||
Corruption = raw::SQLITE_CORRUPT as isize,
|
||||
Done = raw::SQLITE_DONE as isize,
|
||||
Empty = raw::SQLITE_EMPTY as isize,
|
||||
Error = raw::SQLITE_ERROR as isize,
|
||||
Format = raw::SQLITE_FORMAT as isize,
|
||||
Full = raw::SQLITE_FULL as isize,
|
||||
Internal = raw::SQLITE_INTERNAL as isize,
|
||||
Interruption = raw::SQLITE_INTERRUPT as isize,
|
||||
IOError = raw::SQLITE_IOERR as isize,
|
||||
Locked = raw::SQLITE_LOCKED as isize,
|
||||
Mismatch = raw::SQLITE_MISMATCH as isize,
|
||||
Misuse = raw::SQLITE_MISUSE as isize,
|
||||
NoLargeFileSupport = raw::SQLITE_NOLFS as isize,
|
||||
NoMemory = raw::SQLITE_NOMEM as isize,
|
||||
NotDatabase = raw::SQLITE_NOTADB as isize,
|
||||
NotFound = raw::SQLITE_NOTFOUND as isize,
|
||||
Notice = raw::SQLITE_NOTICE as isize,
|
||||
OK = raw::SQLITE_OK as isize,
|
||||
Permission = raw::SQLITE_PERM as isize,
|
||||
Protocol = raw::SQLITE_PROTOCOL as isize,
|
||||
Range = raw::SQLITE_RANGE as isize,
|
||||
ReadOnly = raw::SQLITE_READONLY as isize,
|
||||
Row = raw::SQLITE_ROW as isize,
|
||||
Schema = raw::SQLITE_SCHEMA as isize,
|
||||
TooBig = raw::SQLITE_TOOBIG as isize,
|
||||
Warning = raw::SQLITE_WARNING as isize,
|
||||
}
|
||||
|
||||
macro_rules! raise(
|
||||
($message:expr) => (
|
||||
return Err(::Error { code: ::ErrorCode::Error, message: Some($message.to_string()) })
|
||||
@ -56,118 +90,12 @@ macro_rules! str_to_c_str(
|
||||
);
|
||||
);
|
||||
|
||||
/// An error code.
|
||||
#[derive(Clone, Copy, Debug)]
|
||||
pub enum ErrorCode {
|
||||
Abort = raw::SQLITE_ABORT as isize,
|
||||
Authorization = raw::SQLITE_AUTH as isize,
|
||||
Busy = raw::SQLITE_BUSY as isize,
|
||||
CantOpen = raw::SQLITE_CANTOPEN as isize,
|
||||
Constraint = raw::SQLITE_CONSTRAINT as isize,
|
||||
Corruption = raw::SQLITE_CORRUPT as isize,
|
||||
Done = raw::SQLITE_DONE as isize,
|
||||
Empty = raw::SQLITE_EMPTY as isize,
|
||||
Error = raw::SQLITE_ERROR as isize,
|
||||
Format = raw::SQLITE_FORMAT as isize,
|
||||
Full = raw::SQLITE_FULL as isize,
|
||||
Internal = raw::SQLITE_INTERNAL as isize,
|
||||
Interruption = raw::SQLITE_INTERRUPT as isize,
|
||||
IOError = raw::SQLITE_IOERR as isize,
|
||||
Locked = raw::SQLITE_LOCKED as isize,
|
||||
Mismatch = raw::SQLITE_MISMATCH as isize,
|
||||
Misuse = raw::SQLITE_MISUSE as isize,
|
||||
NoLargeFileSupport = raw::SQLITE_NOLFS as isize,
|
||||
NoMemory = raw::SQLITE_NOMEM as isize,
|
||||
NotDatabase = raw::SQLITE_NOTADB as isize,
|
||||
NotFound = raw::SQLITE_NOTFOUND as isize,
|
||||
Notice = raw::SQLITE_NOTICE as isize,
|
||||
OK = raw::SQLITE_OK as isize,
|
||||
Permission = raw::SQLITE_PERM as isize,
|
||||
Protocol = raw::SQLITE_PROTOCOL as isize,
|
||||
Range = raw::SQLITE_RANGE as isize,
|
||||
ReadOnly = raw::SQLITE_READONLY as isize,
|
||||
Row = raw::SQLITE_ROW as isize,
|
||||
Schema = raw::SQLITE_SCHEMA as isize,
|
||||
TooBig = raw::SQLITE_TOOBIG as isize,
|
||||
Warning = raw::SQLITE_WARNING as isize,
|
||||
}
|
||||
mod database;
|
||||
|
||||
/// A database.
|
||||
pub struct Database<'d> {
|
||||
db: *mut raw::sqlite3,
|
||||
_phantom: PhantomData<&'d raw::sqlite3>,
|
||||
}
|
||||
|
||||
/// A callback executed for each row of the result of an SQL query.
|
||||
pub type ExecuteCallback<'c> = FnMut(Vec<(String, String)>) -> bool + 'c;
|
||||
|
||||
impl<'d> Database<'d> {
|
||||
/// Open a database.
|
||||
pub fn open(path: &Path) -> Result<Database<'d>> {
|
||||
let mut db = 0 as *mut _;
|
||||
unsafe {
|
||||
success!(raw::sqlite3_open(path_to_c_str!(path), &mut db));
|
||||
}
|
||||
Ok(Database { db: db, _phantom: PhantomData })
|
||||
}
|
||||
|
||||
/// Execute an SQL statement.
|
||||
pub fn execute<'c>(&mut self, sql: &str,
|
||||
callback: Option<&mut ExecuteCallback<'c>>) -> Result<()> {
|
||||
|
||||
unsafe {
|
||||
match callback {
|
||||
Some(callback) => {
|
||||
let mut callback = Box::new(callback);
|
||||
success!(raw::sqlite3_exec(self.db, str_to_c_str!(sql), Some(execute_callback),
|
||||
&mut callback as *mut _ as *mut _, 0 as *mut _));
|
||||
},
|
||||
None => {
|
||||
success!(raw::sqlite3_exec(self.db, str_to_c_str!(sql), None,
|
||||
0 as *mut _, 0 as *mut _));
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
impl<'d> Drop for Database<'d> {
|
||||
#[inline]
|
||||
fn drop(&mut self) {
|
||||
unsafe { ::raw::sqlite3_close(self.db) };
|
||||
}
|
||||
}
|
||||
pub use database::{Database, ExecuteCallback};
|
||||
|
||||
/// Open a database.
|
||||
#[inline]
|
||||
pub fn open(path: &Path) -> Result<Database> {
|
||||
Database::open(path)
|
||||
}
|
||||
|
||||
extern fn execute_callback(callback: *mut c_void, count: c_int, values: *mut *mut c_char,
|
||||
columns: *mut *mut c_char) -> c_int {
|
||||
|
||||
macro_rules! c_str_to_string(
|
||||
($string:expr) => (
|
||||
match ::std::str::from_utf8(::std::ffi::CStr::from_ptr($string).to_bytes()) {
|
||||
Ok(string) => String::from(string),
|
||||
Err(_) => return 1,
|
||||
}
|
||||
);
|
||||
);
|
||||
|
||||
unsafe {
|
||||
let mut pairs = Vec::with_capacity(count as usize);
|
||||
|
||||
for i in 0..(count as isize) {
|
||||
let column = c_str_to_string!(*columns.offset(i) as *const _);
|
||||
let value = c_str_to_string!(*values.offset(i) as *const _);
|
||||
pairs.push((column, value));
|
||||
}
|
||||
|
||||
let ref mut callback = *(callback as *mut Box<&mut ExecuteCallback>);
|
||||
if callback(pairs) { 0 } else { 1 }
|
||||
}
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user