mirror of
https://github.com/isar/rusqlite.git
synced 2024-11-29 21:52:12 +08:00
Rustfmt
This commit is contained in:
parent
6f0f121d61
commit
888dce0d8f
60
src/blob.rs
60
src/blob.rs
@ -26,15 +26,37 @@ impl Connection {
|
|||||||
///
|
///
|
||||||
/// Will return `Err` if `db`/`table`/`column` cannot be converted to a C-compatible string or if the
|
/// Will return `Err` if `db`/`table`/`column` cannot be converted to a C-compatible string or if the
|
||||||
/// underlying SQLite BLOB open call fails.
|
/// underlying SQLite BLOB open call fails.
|
||||||
pub fn blob_open<'a>(&'a self, db: &str, table: &str, column: &str, row: i64, read_only: bool) -> Result<Blob<'a>> {
|
pub fn blob_open<'a>(&'a self,
|
||||||
|
db: &str,
|
||||||
|
table: &str,
|
||||||
|
column: &str,
|
||||||
|
row: i64,
|
||||||
|
read_only: bool)
|
||||||
|
-> Result<Blob<'a>> {
|
||||||
let mut c = self.db.borrow_mut();
|
let mut c = self.db.borrow_mut();
|
||||||
let mut blob = ptr::null_mut();
|
let mut blob = ptr::null_mut();
|
||||||
let db = try!(super::str_to_cstring(db));
|
let db = try!(super::str_to_cstring(db));
|
||||||
let table = try!(super::str_to_cstring(table));
|
let table = try!(super::str_to_cstring(table));
|
||||||
let column = try!(super::str_to_cstring(column));
|
let column = try!(super::str_to_cstring(column));
|
||||||
let rc = unsafe{ ffi::sqlite3_blob_open(c.db(), db.as_ptr(), table.as_ptr(), column.as_ptr(), row, if read_only { 0 } else { 1 }, &mut blob) };
|
let rc = unsafe {
|
||||||
|
ffi::sqlite3_blob_open(c.db(),
|
||||||
|
db.as_ptr(),
|
||||||
|
table.as_ptr(),
|
||||||
|
column.as_ptr(),
|
||||||
|
row,
|
||||||
|
if read_only {
|
||||||
|
0
|
||||||
|
} else {
|
||||||
|
1
|
||||||
|
},
|
||||||
|
&mut blob)
|
||||||
|
};
|
||||||
c.decode_result(rc).map(|_| {
|
c.decode_result(rc).map(|_| {
|
||||||
Blob{ conn: self, blob: blob, pos: 0 }
|
Blob {
|
||||||
|
conn: self,
|
||||||
|
blob: blob,
|
||||||
|
pos: 0,
|
||||||
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -46,7 +68,7 @@ impl<'conn> Blob<'conn> {
|
|||||||
///
|
///
|
||||||
/// Will return `Err` if the underlying SQLite BLOB reopen call fails.
|
/// Will return `Err` if the underlying SQLite BLOB reopen call fails.
|
||||||
pub fn reopen(&mut self, row: i64) -> Result<()> {
|
pub fn reopen(&mut self, row: i64) -> Result<()> {
|
||||||
let rc = unsafe{ ffi::sqlite3_blob_reopen(self.blob, row) };
|
let rc = unsafe { ffi::sqlite3_blob_reopen(self.blob, row) };
|
||||||
if rc != ffi::SQLITE_OK {
|
if rc != ffi::SQLITE_OK {
|
||||||
return self.conn.decode_result(rc);
|
return self.conn.decode_result(rc);
|
||||||
}
|
}
|
||||||
@ -56,7 +78,7 @@ impl<'conn> Blob<'conn> {
|
|||||||
|
|
||||||
/// Return the size in bytes of the BLOB
|
/// Return the size in bytes of the BLOB
|
||||||
pub fn size(&self) -> i32 {
|
pub fn size(&self) -> i32 {
|
||||||
unsafe{ ffi::sqlite3_blob_bytes(self.blob) }
|
unsafe { ffi::sqlite3_blob_bytes(self.blob) }
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Read data from a BLOB incrementally
|
/// Read data from a BLOB incrementally
|
||||||
@ -68,7 +90,7 @@ impl<'conn> Blob<'conn> {
|
|||||||
if buf.len() > ::std::i32::MAX as usize {
|
if buf.len() > ::std::i32::MAX as usize {
|
||||||
return Err(Error {
|
return Err(Error {
|
||||||
code: ffi::SQLITE_TOOBIG,
|
code: ffi::SQLITE_TOOBIG,
|
||||||
message: "buffer too long".to_string()
|
message: "buffer too long".to_string(),
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
let mut n = buf.len() as i32;
|
let mut n = buf.len() as i32;
|
||||||
@ -79,7 +101,9 @@ impl<'conn> Blob<'conn> {
|
|||||||
if n <= 0 {
|
if n <= 0 {
|
||||||
return Ok(0);
|
return Ok(0);
|
||||||
}
|
}
|
||||||
let rc = unsafe { ffi::sqlite3_blob_read(self.blob, mem::transmute(buf.as_ptr()), n, self.pos) };
|
let rc = unsafe {
|
||||||
|
ffi::sqlite3_blob_read(self.blob, mem::transmute(buf.as_ptr()), n, self.pos)
|
||||||
|
};
|
||||||
self.conn.decode_result(rc).map(|_| {
|
self.conn.decode_result(rc).map(|_| {
|
||||||
self.pos += n;
|
self.pos += n;
|
||||||
n
|
n
|
||||||
@ -98,18 +122,23 @@ impl<'conn> Blob<'conn> {
|
|||||||
if buf.len() > ::std::i32::MAX as usize {
|
if buf.len() > ::std::i32::MAX as usize {
|
||||||
return Err(Error {
|
return Err(Error {
|
||||||
code: ffi::SQLITE_TOOBIG,
|
code: ffi::SQLITE_TOOBIG,
|
||||||
message: "buffer too long".to_string()
|
message: "buffer too long".to_string(),
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
let n = buf.len() as i32;
|
let n = buf.len() as i32;
|
||||||
let size = self.size();
|
let size = self.size();
|
||||||
if self.pos + n > size {
|
if self.pos + n > size {
|
||||||
return Err(Error{code: ffi::SQLITE_MISUSE, message: format!("pos = {} + n = {} > size = {}", self.pos, n, size)});
|
return Err(Error {
|
||||||
|
code: ffi::SQLITE_MISUSE,
|
||||||
|
message: format!("pos = {} + n = {} > size = {}", self.pos, n, size),
|
||||||
|
});
|
||||||
}
|
}
|
||||||
if n <= 0 {
|
if n <= 0 {
|
||||||
return Ok(0);
|
return Ok(0);
|
||||||
}
|
}
|
||||||
let rc = unsafe { ffi::sqlite3_blob_write(self.blob, mem::transmute(buf.as_ptr()), n, self.pos) };
|
let rc = unsafe {
|
||||||
|
ffi::sqlite3_blob_write(self.blob, mem::transmute(buf.as_ptr()), n, self.pos)
|
||||||
|
};
|
||||||
self.conn.decode_result(rc).map(|_| {
|
self.conn.decode_result(rc).map(|_| {
|
||||||
self.pos += n;
|
self.pos += n;
|
||||||
n
|
n
|
||||||
@ -118,10 +147,10 @@ impl<'conn> Blob<'conn> {
|
|||||||
|
|
||||||
/// Seek to an offset, in bytes, in BLOB.
|
/// Seek to an offset, in bytes, in BLOB.
|
||||||
pub fn seek(&mut self, pos: SeekFrom) {
|
pub fn seek(&mut self, pos: SeekFrom) {
|
||||||
self.pos = match pos {
|
self.pos = match pos {
|
||||||
SeekFrom::Start(offset) => offset,
|
SeekFrom::Start(offset) => offset,
|
||||||
SeekFrom::Current(offset) => self.pos + offset,
|
SeekFrom::Current(offset) => self.pos + offset,
|
||||||
SeekFrom::End(offset) => self.size() + offset
|
SeekFrom::End(offset) => self.size() + offset,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -152,7 +181,8 @@ impl<'conn> Drop for Blob<'conn> {
|
|||||||
mod test {
|
mod test {
|
||||||
use Connection;
|
use Connection;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
#[cfg_attr(rustfmt, rustfmt_skip)]
|
||||||
fn test_blob() {
|
fn test_blob() {
|
||||||
let db = Connection::open_in_memory().unwrap();
|
let db = Connection::open_in_memory().unwrap();
|
||||||
let sql = "BEGIN;
|
let sql = "BEGIN;
|
||||||
@ -165,7 +195,7 @@ mod test {
|
|||||||
let mut blob = db.blob_open("main", "test", "content", rowid, false).unwrap();
|
let mut blob = db.blob_open("main", "test", "content", rowid, false).unwrap();
|
||||||
blob.write(b"Clob").unwrap();
|
blob.write(b"Clob").unwrap();
|
||||||
let err = blob.write(b"5678901");
|
let err = blob.write(b"5678901");
|
||||||
//writeln!(io::stderr(), "{:?}", err);
|
// writeln!(io::stderr(), "{:?}", err);
|
||||||
assert!(err.is_err());
|
assert!(err.is_err());
|
||||||
|
|
||||||
assert!(blob.reopen(rowid).is_ok());
|
assert!(blob.reopen(rowid).is_ok());
|
||||||
@ -180,4 +210,4 @@ mod test {
|
|||||||
assert!(blob.reopen(rowid).is_ok());
|
assert!(blob.reopen(rowid).is_ok());
|
||||||
blob.seek(super::SeekFrom::Start(0));
|
blob.seek(super::SeekFrom::Start(0));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user