TudbuT/conduit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

optimize increment / increment batch

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-07-02 06:24:01 +00:00
parent ee64fb149c
commit 2e2cf08bb2
1 changed files with 7 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
src/database/map.rs
View file

@ -143,54 +143,34 @@ impl Map {
Box::new(Iter::new(&self.db, &self.cf, read_options, &mode).take_while(move |(k, _)| k.starts_with(&prefix)))
}
pub fn increment(&self, key: &[u8]) -> Result<Vec<u8>> {
let read_options = &self.read_options;
let old = self
.db
.db
.get_cf_opt(&self.cf(), key, read_options)
.or_else(or_else)?;
pub fn increment(&self, key: &[u8]) -> Result<[u8; 8]> {
let old = self.get(key)?;
let new = utils::increment(old.as_deref());
let write_options = &self.write_options;
self.db
.db
.put_cf_opt(&self.cf(), key, new, write_options)
.or_else(or_else)?;
self.insert(key, &new)?;
if !self.db.corked() {
self.db.flush()?;
}
Ok(new.to_vec())
Ok(new)
}
pub fn increment_batch(&self, iter: &mut dyn Iterator<Item = Val>) -> Result<()> {
let mut batch = WriteBatchWithTransaction::<false>::default();
let read_options = &self.read_options;
for key in iter {
let old = self
.db
.db
.get_cf_opt(&self.cf(), &key, read_options)
.or_else(or_else)?;
let old = self.get(&key)?;
let new = utils::increment(old.as_deref());
batch.put_cf(&self.cf(), key, new);
}
let write_options = &self.write_options;
self.db
.db
.write_opt(batch, write_options)
.or_else(or_else)?;
let res = self.db.db.write_opt(batch, write_options);
if !self.db.corked() {
self.db.flush()?;
}
Ok(())
result(res)
}
pub fn watch_prefix<'a>(&'a self, prefix: &[u8]) -> Pin<Box<dyn Future<Output = ()> + Send + 'a>> {