NextGraph
/
lmdb-rs
mirror of https://github.com/nextgraph-org/lmdb-rs.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Generic key/data arguments

Browse Source
without.crypto
Dan Burkert 9 years ago
parent c101fa5c1f
commit f749811730
3 changed files with 153 additions and 129 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 88
      src/cursor.rs
  2. 6
      src/lib.rs
  3. 178
      src/transaction.rs

88
src/cursor.rs
Unescape View File

@ -13,11 +13,12 @@ pub trait Cursor<'txn> {
/// Returns a raw pointer to the underlying LMDB cursor. /// Returns a raw pointer to the underlying LMDB cursor.
/// ///
/// The caller **must** ensure that the pointer is not used after the lifetime of the cursor. /// The caller **must** ensure that the pointer is not used after the
/// lifetime of the cursor.
fn cursor(&self) -> *mut ffi::MDB_cursor; fn cursor(&self) -> *mut ffi::MDB_cursor;
/// Retrieves a key/data pair from the cursor. Depending on the cursor op, the current key may /// Retrieves a key/data pair from the cursor. Depending on the cursor op,
/// be returned. /// the current key may be returned.
fn get(&self, fn get(&self,
key: Option<&[u8]>, key: Option<&[u8]>,
data: Option<&[u8]>, data: Option<&[u8]>,
@ -27,30 +28,29 @@ pub trait Cursor<'txn> {
let mut key_val = slice_to_val(key); let mut key_val = slice_to_val(key);
let mut data_val = slice_to_val(data); let mut data_val = slice_to_val(data);
let key_ptr = key_val.mv_data; let key_ptr = key_val.mv_data;
try!(lmdb_result(ffi::mdb_cursor_get(self.cursor(), try!(lmdb_result(ffi::mdb_cursor_get(self.cursor(), &mut key_val, &mut data_val, op)));
&mut key_val,
&mut data_val,
op)));
let key_out = if key_ptr != key_val.mv_data { Some(val_to_slice(key_val)) } else { None }; let key_out = if key_ptr != key_val.mv_data { Some(val_to_slice(key_val)) } else { None };
let data_out = val_to_slice(data_val); let data_out = val_to_slice(data_val);
Ok((key_out, data_out)) Ok((key_out, data_out))
} }
} }
/// Iterate over database items. The iterator will begin with item next after the cursor, and /// Iterate over database items. The iterator will begin with item next
/// continue until the end of the database. For new cursors, the iterator will begin with the /// after the cursor, and continue until the end of the database. For new
/// first item in the database. /// cursors, the iterator will begin with the first item in the database.
/// ///
/// For databases with duplicate data items (`DatabaseFlags::DUP_SORT`), the duplicate data /// For databases with duplicate data items (`DatabaseFlags::DUP_SORT`), the
/// items of each key will be returned before moving on to the next key. /// duplicate data items of each key will be returned before moving on to
/// the next key.
fn iter(&mut self) -> Iter<'txn> { fn iter(&mut self) -> Iter<'txn> {
Iter::new(self.cursor(), ffi::MDB_NEXT, ffi::MDB_NEXT) Iter::new(self.cursor(), ffi::MDB_NEXT, ffi::MDB_NEXT)
} }
/// Iterate over database items starting from the beginning of the database. /// Iterate over database items starting from the beginning of the database.
/// ///
/// For databases with duplicate data items (`DatabaseFlags::DUP_SORT`), the duplicate data /// For databases with duplicate data items (`DatabaseFlags::DUP_SORT`), the
/// items of each key will be returned before moving on to the next key. /// duplicate data items of each key will be returned before moving on to
/// the next key.
fn iter_start(&mut self) -> Iter<'txn> { fn iter_start(&mut self) -> Iter<'txn> {
self.get(None, None, ffi::MDB_FIRST).unwrap(); self.get(None, None, ffi::MDB_FIRST).unwrap();
Iter::new(self.cursor(), ffi::MDB_GET_CURRENT, ffi::MDB_NEXT) Iter::new(self.cursor(), ffi::MDB_GET_CURRENT, ffi::MDB_NEXT)
@ -58,37 +58,40 @@ pub trait Cursor<'txn> {
/// Iterate over database items starting from the given key. /// Iterate over database items starting from the given key.
/// ///
/// For databases with duplicate data items (`DatabaseFlags::DUP_SORT`), the duplicate data /// For databases with duplicate data items (`DatabaseFlags::DUP_SORT`), the
/// items of each key will be returned before moving on to the next key. /// duplicate data items of each key will be returned before moving on to
fn iter_from(&mut self, key: &[u8]) -> Iter<'txn> { /// the next key.
self.get(Some(key), None, ffi::MDB_SET_RANGE).unwrap(); fn iter_from<K>(&mut self, key: K) -> Iter<'txn> where K: AsRef<[u8]> {
self.get(Some(key.as_ref()), None, ffi::MDB_SET_RANGE).unwrap();
Iter::new(self.cursor(), ffi::MDB_GET_CURRENT, ffi::MDB_NEXT) Iter::new(self.cursor(), ffi::MDB_GET_CURRENT, ffi::MDB_NEXT)
} }
/// Iterate over duplicate database items. The iterator will begin with the item next after the /// Iterate over duplicate database items. The iterator will begin with the
/// cursor, and continue until the end of the database. Each item will be returned as an /// item next after the cursor, and continue until the end of the database.
/// iterator of its duplicates. /// Each item will be returned as an iterator of its duplicates.
fn iter_dup(&mut self) -> IterDup<'txn> { fn iter_dup(&mut self) -> IterDup<'txn> {
IterDup::new(self.cursor(), ffi::MDB_NEXT) IterDup::new(self.cursor(), ffi::MDB_NEXT)
} }
/// Iterate over duplicate database items starting from the beginning of the database. Each item /// Iterate over duplicate database items starting from the beginning of the
/// will be returned as an iterator of its duplicates. /// database. Each item will be returned as an iterator of its duplicates.
fn iter_dup_start(&mut self) -> IterDup<'txn> { fn iter_dup_start(&mut self) -> IterDup<'txn> {
self.get(None, None, ffi::MDB_FIRST).unwrap(); self.get(None, None, ffi::MDB_FIRST).unwrap();
IterDup::new(self.cursor(), ffi::MDB_GET_CURRENT) IterDup::new(self.cursor(), ffi::MDB_GET_CURRENT)
} }
/// Iterate over duplicate items in the database starting from the given key. Each item will be /// Iterate over duplicate items in the database starting from the given
/// returned as an iterator of its duplicates. /// key. Each item will be returned as an iterator of its duplicates.
fn iter_dup_from(&mut self, key: &[u8]) -> IterDup<'txn> { fn iter_dup_from<K>(&mut self, key: &K) -> IterDup<'txn> where K: AsRef<[u8]> {
self.get(Some(key), None, ffi::MDB_SET_RANGE).unwrap(); self.get(Some(key.as_ref()), None, ffi::MDB_SET_RANGE).unwrap();
IterDup::new(self.cursor(), ffi::MDB_GET_CURRENT) IterDup::new(self.cursor(), ffi::MDB_GET_CURRENT)
} }
/// Iterate over the duplicates of the item in the database with the given key. /// Iterate over the duplicates of the item in the database with the given
fn iter_dup_of(&mut self, key: &[u8]) -> Result<Iter<'txn>> { /// key.
try!(self.get(Some(key), None, ffi::MDB_SET)); fn iter_dup_of<K>(&mut self, key: &K) -> Result<Iter<'txn>> where K:
AsRef<[u8]> {
try!(self.get(Some(key.as_ref()), None, ffi::MDB_SET));
Ok(Iter::new(self.cursor(), ffi::MDB_GET_CURRENT, ffi::MDB_NEXT_DUP)) Ok(Iter::new(self.cursor(), ffi::MDB_GET_CURRENT, ffi::MDB_NEXT_DUP))
} }
} }
@ -113,8 +116,8 @@ impl <'txn> Drop for RoCursor<'txn> {
impl <'txn> RoCursor<'txn> { impl <'txn> RoCursor<'txn> {
/// Creates a new read-only cursor in the given database and transaction. Prefer using /// Creates a new read-only cursor in the given database and transaction.
/// `Transaction::open_cursor`. /// Prefer using `Transaction::open_cursor`.
#[doc(hidden)] #[doc(hidden)]
pub fn new<T>(txn: &'txn T, db: Database) -> Result<RoCursor<'txn>> where T: Transaction { pub fn new<T>(txn: &'txn T, db: Database) -> Result<RoCursor<'txn>> where T: Transaction {
let mut cursor: *mut ffi::MDB_cursor = ptr::null_mut(); let mut cursor: *mut ffi::MDB_cursor = ptr::null_mut();
@ -146,8 +149,8 @@ impl <'txn> Drop for RwCursor<'txn> {
impl <'txn> RwCursor<'txn> { impl <'txn> RwCursor<'txn> {
/// Creates a new read-only cursor in the given database and transaction. Prefer using /// Creates a new read-only cursor in the given database and transaction.
/// `RwTransaction::open_rw_cursor`. /// Prefer using `RwTransaction::open_rw_cursor`.
#[doc(hidden)] #[doc(hidden)]
pub fn new<T>(txn: &'txn T, db: Database) -> Result<RwCursor<'txn>> where T: Transaction { pub fn new<T>(txn: &'txn T, db: Database) -> Result<RwCursor<'txn>> where T: Transaction {
let mut cursor: *mut ffi::MDB_cursor = ptr::null_mut(); let mut cursor: *mut ffi::MDB_cursor = ptr::null_mut();
@ -155,9 +158,12 @@ impl <'txn> RwCursor<'txn> {
Ok(RwCursor { cursor: cursor, _marker: PhantomData }) Ok(RwCursor { cursor: cursor, _marker: PhantomData })
} }
/// Puts a key/data pair into the database. The cursor will be positioned at the new data item, /// Puts a key/data pair into the database. The cursor will be positioned at
/// or on failure usually near it. /// the new data item, or on failure usually near it.
pub fn put(&mut self, key: &[u8], data: &[u8], flags: WriteFlags) -> Result<()> { pub fn put<K, D>(&mut self, key: &K, data: &D, flags: WriteFlags) -> Result<()>
where K: AsRef<[u8]>, D: AsRef<[u8]> {
let key = key.as_ref();
let data = data.as_ref();
let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t, let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t,
mv_data: key.as_ptr() as *mut c_void }; mv_data: key.as_ptr() as *mut c_void };
let mut data_val: ffi::MDB_val = ffi::MDB_val { mv_size: data.len() as size_t, let mut data_val: ffi::MDB_val = ffi::MDB_val { mv_size: data.len() as size_t,
@ -174,8 +180,8 @@ impl <'txn> RwCursor<'txn> {
/// ///
/// ### Flags /// ### Flags
/// ///
/// `WriteFlags::NO_DUP_DATA` may be used to delete all data items for the current key, if the /// `WriteFlags::NO_DUP_DATA` may be used to delete all data items for the
/// database was opened with `DatabaseFlags::DUP_SORT`. /// current key, if the database was opened with `DatabaseFlags::DUP_SORT`.
pub fn del(&mut self, flags: WriteFlags) -> Result<()> { pub fn del(&mut self, flags: WriteFlags) -> Result<()> {
unsafe { lmdb_result(ffi::mdb_cursor_del(self.cursor(), flags.bits())) } unsafe { lmdb_result(ffi::mdb_cursor_del(self.cursor(), flags.bits())) }
} }
@ -396,7 +402,7 @@ mod test {
{ {
let mut txn = env.begin_rw_txn().unwrap(); let mut txn = env.begin_rw_txn().unwrap();
for &(key, data) in items.iter() { for &(ref key, ref data) in &items {
txn.put(db, key, data, WriteFlags::empty()).unwrap(); txn.put(db, key, data, WriteFlags::empty()).unwrap();
} }
txn.commit().unwrap(); txn.commit().unwrap();
@ -434,7 +440,7 @@ mod test {
{ {
let mut txn = env.begin_rw_txn().unwrap(); let mut txn = env.begin_rw_txn().unwrap();
for &(key, data) in items.iter() { for &(ref key, ref data) in &items {
txn.put(db, key, data, WriteFlags::empty()).unwrap(); txn.put(db, key, data, WriteFlags::empty()).unwrap();
} }
txn.commit().unwrap(); txn.commit().unwrap();

6
src/lib.rs
Unescape View File

@ -78,11 +78,7 @@ mod test_utils {
let db = env.open_db(None).unwrap(); let db = env.open_db(None).unwrap();
let mut txn = env.begin_rw_txn().unwrap(); let mut txn = env.begin_rw_txn().unwrap();
for i in 0..num_rows { for i in 0..num_rows {
txn.put(db, txn.put(db, &get_key(i), &get_data(i), WriteFlags::empty()).unwrap();
get_key(i).as_bytes(),
get_data(i).as_bytes(),
WriteFlags::empty())
.unwrap();
} }
txn.commit().unwrap(); txn.commit().unwrap();
} }

178
src/transaction.rs
Unescape View File

@ -17,8 +17,8 @@ pub trait Transaction : Sized {
/// Returns a raw pointer to the underlying LMDB transaction. /// Returns a raw pointer to the underlying LMDB transaction.
/// ///
/// The caller **must** ensure that the pointer is not used after the lifetime of the /// The caller **must** ensure that the pointer is not used after the
/// transaction. /// lifetime of the transaction.
fn txn(&self) -> *mut ffi::MDB_txn; fn txn(&self) -> *mut ffi::MDB_txn;
/// Commits the transaction. /// Commits the transaction.
@ -41,31 +41,39 @@ pub trait Transaction : Sized {
/// Opens a database in the transaction. /// Opens a database in the transaction.
/// ///
/// If `name` is `None`, then the default database will be opened, otherwise a named database /// If `name` is `None`, then the default database will be opened, otherwise
/// will be opened. The database handle will be private to the transaction until the transaction /// a named database will be opened. The database handle will be private to
/// is successfully committed. If the transaction is aborted the returned database handle /// the transaction until the transaction is successfully committed. If the
/// should no longer be used. /// transaction is aborted the returned database handle should no longer be
/// used.
/// ///
/// Prefer using `Environment::open_db`. /// Prefer using `Environment::open_db`.
/// ///
/// ## Safety /// ## Safety
/// ///
/// This function (as well as `Environment::open_db`, `Environment::create_db`, and /// This function (as well as `Environment::open_db`,
/// `Database::create`) **must not** be called from multiple concurrent transactions in the same /// `Environment::create_db`, and `Database::create`) **must not** be called
/// environment. A transaction which uses this function must finish (either commit or abort) /// from multiple concurrent transactions in the same environment. A
/// before any other transaction may use this function. /// transaction which uses this function must finish (either commit or
/// abort) before any other transaction may use this function.
unsafe fn open_db(&self, name: Option<&str>) -> Result<Database> { unsafe fn open_db(&self, name: Option<&str>) -> Result<Database> {
Database::new(self.txn(), name, 0) Database::new(self.txn(), name, 0)
} }
/// Gets an item from a database. /// Gets an item from a database.
/// ///
/// This function retrieves the data associated with the given key in the database. If the /// This function retrieves the data associated with the given key in the
/// database supports duplicate keys (`DatabaseFlags::DUP_SORT`) then the first data item for /// database. If the database supports duplicate keys
/// the key will be returned. Retrieval of other items requires the use of /// (`DatabaseFlags::DUP_SORT`) then the first data item for the key will be
/// `Transaction::cursor_get`. If the item is not in the database, then `Error::NotFound` /// returned. Retrieval of other items requires the use of
/// will be returned. /// `Transaction::cursor_get`. If the item is not in the database, then
fn get<'txn>(&'txn self, database: Database, key: &[u8]) -> Result<&'txn [u8]> { /// `Error::NotFound` will be returned.
fn get<'txn, K>(&'txn self,
database: Database,
key: &K)
-> Result<&'txn [u8]>
where K: AsRef<[u8]> {
let key = key.as_ref();
let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t, let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t,
mv_data: key.as_ptr() as *mut c_void }; mv_data: key.as_ptr() as *mut c_void };
let mut data_val: ffi::MDB_val = ffi::MDB_val { mv_size: 0, let mut data_val: ffi::MDB_val = ffi::MDB_val { mv_size: 0,
@ -110,8 +118,8 @@ impl <'env> Drop for RoTransaction<'env> {
impl <'env> RoTransaction<'env> { impl <'env> RoTransaction<'env> {
/// Creates a new read-only transaction in the given environment. Prefer using /// Creates a new read-only transaction in the given environment. Prefer
/// `Environment::begin_ro_txn`. /// using `Environment::begin_ro_txn`.
#[doc(hidden)] #[doc(hidden)]
pub fn new(env: &'env Environment) -> Result<RoTransaction<'env>> { pub fn new(env: &'env Environment) -> Result<RoTransaction<'env>> {
let mut txn: *mut ffi::MDB_txn = ptr::null_mut(); let mut txn: *mut ffi::MDB_txn = ptr::null_mut();
@ -126,14 +134,16 @@ impl <'env> RoTransaction<'env> {
/// Resets the read-only transaction. /// Resets the read-only transaction.
/// ///
/// Abort the transaction like `Transaction::abort`, but keep the transaction handle. /// Abort the transaction like `Transaction::abort`, but keep the
/// `InactiveTransaction::renew` may reuse the handle. This saves allocation overhead if the /// transaction handle. `InactiveTransaction::renew` may reuse the handle.
/// process will start a new read-only transaction soon, and also locking overhead if /// This saves allocation overhead if the process will start a new read-only
/// `EnvironmentFlags::NO_TLS` is in use. The reader table lock is released, but the table slot /// transaction soon, and also locking overhead if
/// stays tied to its thread or transaction. Reader locks generally don't interfere with /// `EnvironmentFlags::NO_TLS` is in use. The reader table lock is released,
/// writers, but they keep old versions of database pages allocated. Thus they prevent the old /// but the table slot stays tied to its thread or transaction. Reader locks
/// pages from being reused when writers commit new data, and so under heavy load the database /// generally don't interfere with writers, but they keep old versions of
/// size may grow much more rapidly than otherwise. /// database pages allocated. Thus they prevent the old pages from being
/// reused when writers commit new data, and so under heavy load the
/// database size may grow much more rapidly than otherwise.
pub fn reset(self) -> InactiveTransaction<'env> { pub fn reset(self) -> InactiveTransaction<'env> {
let txn = self.txn; let txn = self.txn;
unsafe { unsafe {
@ -164,10 +174,11 @@ impl <'env> Drop for InactiveTransaction<'env> {
impl <'env> InactiveTransaction<'env> { impl <'env> InactiveTransaction<'env> {
/// Renews the inactive transaction, returning an active read-only transaction. /// Renews the inactive transaction, returning an active read-only
/// transaction.
/// ///
/// This acquires a new reader lock for a transaction handle that had been released by /// This acquires a new reader lock for a transaction handle that had been
/// `RoTransaction::reset`. /// released by `RoTransaction::reset`.
pub fn renew(self) -> Result<RoTransaction<'env>> { pub fn renew(self) -> Result<RoTransaction<'env>> {
let txn = self.txn; let txn = self.txn;
unsafe { unsafe {
@ -192,8 +203,8 @@ impl <'env> Drop for RwTransaction<'env> {
impl <'env> RwTransaction<'env> { impl <'env> RwTransaction<'env> {
/// Creates a new read-write transaction in the given environment. Prefer using /// Creates a new read-write transaction in the given environment. Prefer
/// `Environment::begin_ro_txn`. /// using `Environment::begin_ro_txn`.
#[doc(hidden)] #[doc(hidden)]
pub fn new(env: &'env Environment) -> Result<RwTransaction<'env>> { pub fn new(env: &'env Environment) -> Result<RwTransaction<'env>> {
let mut txn: *mut ffi::MDB_txn = ptr::null_mut(); let mut txn: *mut ffi::MDB_txn = ptr::null_mut();
@ -208,24 +219,25 @@ impl <'env> RwTransaction<'env> {
/// Opens a database in the provided transaction, creating it if necessary. /// Opens a database in the provided transaction, creating it if necessary.
/// ///
/// If `name` is `None`, then the default database will be opened, otherwise a named database /// If `name` is `None`, then the default database will be opened, otherwise
/// will be opened. The database handle will be private to the transaction until the transaction /// a named database will be opened. The database handle will be private to
/// is successfully committed. If the transaction is aborted the returned database handle /// the transaction until the transaction is successfully committed. If the
/// should no longer be used. /// transaction is aborted the returned database handle should no longer be
/// used.
/// ///
/// Prefer using `Environment::create_db`. /// Prefer using `Environment::create_db`.
/// ///
/// ## Safety /// ## Safety
/// ///
/// * This function (as well as `Environment::open_db`, `Environment::create_db`, and /// * This function (as well as `Environment::open_db`,
/// `Database::open`) **must not** be called from multiple concurrent transactions in the same /// `Environment::create_db`, and `Database::open`) **must not** be called
/// environment. A transaction which uses this function must finish (either commit or abort) /// from multiple concurrent transactions in the same environment. A
/// before any other transaction may use this function. /// transaction which uses this function must finish (either commit or
/// abort) before any other transaction may use this function.
pub unsafe fn create_db(&self, name: Option<&str>, flags: DatabaseFlags) -> Result<Database> { pub unsafe fn create_db(&self, name: Option<&str>, flags: DatabaseFlags) -> Result<Database> {
Database::new(self.txn(), name, flags.bits() | ffi::MDB_CREATE) Database::new(self.txn(), name, flags.bits() | ffi::MDB_CREATE)
} }
/// Opens a new read-write cursor on the given database and transaction. /// Opens a new read-write cursor on the given database and transaction.
pub fn open_rw_cursor<'txn>(&'txn mut self, db: Database) -> Result<RwCursor<'txn>> { pub fn open_rw_cursor<'txn>(&'txn mut self, db: Database) -> Result<RwCursor<'txn>> {
RwCursor::new(self, db) RwCursor::new(self, db)
@ -233,15 +245,19 @@ impl <'env> RwTransaction<'env> {
/// Stores an item into a database. /// Stores an item into a database.
/// ///
/// This function stores key/data pairs in the database. The default behavior is to enter the /// This function stores key/data pairs in the database. The default
/// new key/data pair, replacing any previously existing key if duplicates are disallowed, or /// behavior is to enter the new key/data pair, replacing any previously
/// adding a duplicate data item if duplicates are allowed (`DatabaseFlags::DUP_SORT`). /// existing key if duplicates are disallowed, or adding a duplicate data
pub fn put(&mut self, /// item if duplicates are allowed (`DatabaseFlags::DUP_SORT`).
pub fn put<K, D>(&mut self,
database: Database, database: Database,
key: &[u8], key: &K,
data: &[u8], data: &D,
flags: WriteFlags) flags: WriteFlags)
-> Result<()> { -> Result<()>
where K: AsRef<[u8]>, D: AsRef<[u8]> {
let key = key.as_ref();
let data = data.as_ref();
let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t, let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t,
mv_data: key.as_ptr() as *mut c_void }; mv_data: key.as_ptr() as *mut c_void };
let mut data_val: ffi::MDB_val = ffi::MDB_val { mv_size: data.len() as size_t, let mut data_val: ffi::MDB_val = ffi::MDB_val { mv_size: data.len() as size_t,
@ -255,14 +271,17 @@ impl <'env> RwTransaction<'env> {
} }
} }
/// Returns a `BufWriter` which can be used to write a value into the item at the given key /// Returns a buffer which can be used to write a value into the item at the
/// and with the given length. The buffer must be completely filled by the caller. /// given key and with the given length. The buffer must be completely
pub fn reserve<'txn>(&'txn mut self, /// filled by the caller.
pub fn reserve<'txn, K>(&'txn mut self,
database: Database, database: Database,
key: &[u8], key: &K,
len: size_t, len: size_t,
flags: WriteFlags) flags: WriteFlags)
-> Result<&'txn mut [u8]> { -> Result<&'txn mut [u8]>
where K: AsRef<[u8]> {
let key = key.as_ref();
let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t, let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t,
mv_data: key.as_ptr() as *mut c_void }; mv_data: key.as_ptr() as *mut c_void };
let mut data_val: ffi::MDB_val = ffi::MDB_val { mv_size: len, let mut data_val: ffi::MDB_val = ffi::MDB_val { mv_size: len,
@ -280,17 +299,21 @@ impl <'env> RwTransaction<'env> {
/// Deletes an item from a database. /// Deletes an item from a database.
/// ///
/// This function removes key/data pairs from the database. If the database does not support /// This function removes key/data pairs from the database. If the database
/// sorted duplicate data items (`DatabaseFlags::DUP_SORT`) the data parameter is ignored. /// does not support sorted duplicate data items (`DatabaseFlags::DUP_SORT`)
/// If the database supports sorted duplicates and the data parameter is `None`, all of the /// the data parameter is ignored. If the database supports sorted
/// duplicate data items for the key will be deleted. Otherwise, if the data parameter is /// duplicates and the data parameter is `None`, all of the duplicate data
/// `Some` only the matching data item will be deleted. This function will return /// items for the key will be deleted. Otherwise, if the data parameter is
/// `Error::NotFound` if the specified key/data pair is not in the database. /// `Some` only the matching data item will be deleted. This function will
pub fn del(&mut self, /// return `Error::NotFound` if the specified key/data pair is not in the
/// database.
pub fn del<K>(&mut self,
database: Database, database: Database,
key: &[u8], key: &K,
data: Option<&[u8]>) data: Option<&[u8]>)
-> Result<()> { -> Result<()>
where K: AsRef<[u8]> {
let key = key.as_ref();
let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t, let mut key_val: ffi::MDB_val = ffi::MDB_val { mv_size: key.len() as size_t,
mv_data: key.as_ptr() as *mut c_void }; mv_data: key.as_ptr() as *mut c_void };
let data_val: Option<ffi::MDB_val> = let data_val: Option<ffi::MDB_val> =
@ -314,8 +337,8 @@ impl <'env> RwTransaction<'env> {
/// ///
/// ## Safety /// ## Safety
/// ///
/// This method is unsafe in the same ways as `Environment::close_db`, and should be used /// This method is unsafe in the same ways as `Environment::close_db`, and
/// accordingly. /// should be used accordingly.
pub unsafe fn drop_db(&mut self, db: Database) -> Result<()> { pub unsafe fn drop_db(&mut self, db: Database) -> Result<()> {
lmdb_result(ffi::mdb_drop(self.txn, db.dbi(), 1)) lmdb_result(ffi::mdb_drop(self.txn, db.dbi(), 1))
} }
@ -489,7 +512,7 @@ mod test {
let n = 10usize; // Number of concurrent readers let n = 10usize; // Number of concurrent readers
let barrier = Arc::new(Barrier::new(n + 1)); let barrier = Arc::new(Barrier::new(n + 1));
let mut futures: Vec<JoinHandle<bool>> = Vec::with_capacity(n); let mut threads: Vec<JoinHandle<bool>> = Vec::with_capacity(n);
let key = b"key"; let key = b"key";
let val = b"val"; let val = b"val";
@ -498,7 +521,7 @@ mod test {
let reader_env = env.clone(); let reader_env = env.clone();
let reader_barrier = barrier.clone(); let reader_barrier = barrier.clone();
futures.push(thread::spawn(move|| { threads.push(thread::spawn(move|| {
let db = reader_env.open_db(None).unwrap(); let db = reader_env.open_db(None).unwrap();
{ {
let txn = reader_env.begin_ro_txn().unwrap(); let txn = reader_env.begin_ro_txn().unwrap();
@ -521,7 +544,7 @@ mod test {
txn.commit().unwrap(); txn.commit().unwrap();
barrier.wait(); barrier.wait();
assert!(futures.into_iter().all(|b| b.join().unwrap())) assert!(threads.into_iter().all(|b| b.join().unwrap()))
} }
#[test] #[test]
@ -530,7 +553,7 @@ mod test {
let env = Arc::new(Environment::new().open(dir.path()).unwrap()); let env = Arc::new(Environment::new().open(dir.path()).unwrap());
let n = 10usize; // Number of concurrent writers let n = 10usize; // Number of concurrent writers
let mut futures: Vec<JoinHandle<bool>> = Vec::with_capacity(n); let mut threads: Vec<JoinHandle<bool>> = Vec::with_capacity(n);
let key = "key"; let key = "key";
let val = "val"; let val = "val";
@ -538,26 +561,25 @@ mod test {
for i in 0..n { for i in 0..n {
let writer_env = env.clone(); let writer_env = env.clone();
futures.push(thread::spawn(move|| { threads.push(thread::spawn(move|| {
let db = writer_env.open_db(None).unwrap(); let db = writer_env.open_db(None).unwrap();
let mut txn = writer_env.begin_rw_txn().unwrap(); let mut txn = writer_env.begin_rw_txn().unwrap();
txn.put(db, txn.put(db,
format!("{}{}", key, i).as_bytes(), &format!("{}{}", key, i),
format!("{}{}", val, i).as_bytes(), &format!("{}{}", val, i),
WriteFlags::empty()) WriteFlags::empty())
.unwrap(); .unwrap();
txn.commit().is_ok() txn.commit().is_ok()
})); }));
} }
assert!(futures.into_iter().all(|b| b.join().unwrap())); assert!(threads.into_iter().all(|b| b.join().unwrap()));
let db = env.open_db(None).unwrap(); let db = env.open_db(None).unwrap();
let txn = env.begin_ro_txn().unwrap(); let txn = env.begin_ro_txn().unwrap();
for i in 0..n { for i in 0..n {
assert_eq!( assert_eq!(format!("{}{}", val, i).as_bytes(),
format!("{}{}", val, i).as_bytes(), txn.get(db, &format!("{}{}", key, i)).unwrap());
txn.get(db, format!("{}{}", key, i).as_bytes()).unwrap());
} }
} }
@ -573,8 +595,8 @@ mod test {
b.iter(|| { b.iter(|| {
let mut i = 0usize; let mut i = 0usize;
for key in keys.iter() { for key in &keys {
i = i + txn.get(db, key.as_bytes()).unwrap().len(); i = i + txn.get(db, key).unwrap().len();
} }
black_box(i); black_box(i);
}); });
@ -598,7 +620,7 @@ mod test {
b.iter(|| unsafe { b.iter(|| unsafe {
let mut i: size_t = 0; let mut i: size_t = 0;
for key in keys.iter() { for key in &keys {
key_val.mv_size = key.len() as size_t; key_val.mv_size = key.len() as size_t;
key_val.mv_data = key.as_bytes().as_ptr() as *mut _; key_val.mv_data = key.as_bytes().as_ptr() as *mut _;
@ -622,7 +644,7 @@ mod test {
b.iter(|| { b.iter(|| {
let mut txn = env.begin_rw_txn().unwrap(); let mut txn = env.begin_rw_txn().unwrap();
for &(ref key, ref data) in items.iter() { for &(ref key, ref data) in items.iter() {
txn.put(db, key.as_bytes(), data.as_bytes(), WriteFlags::empty()).unwrap(); txn.put(db, key, data, WriteFlags::empty()).unwrap();
} }
txn.abort(); txn.abort();
}); });

Write Preview
Loading…
Cancel
Save