1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
use crate::db::cf_handle;
use crate::{
internal_error, Result, RocksDB, RocksDBSnapshot, RocksDBTransaction,
RocksDBTransactionSnapshot,
};
use ckb_db_schema::Col;
use rocksdb::{ops::IterateCF, ReadOptions};
pub use rocksdb::{DBIterator as DBIter, Direction, IteratorMode};
pub trait DBIterator {
fn iter(&self, col: Col, mode: IteratorMode) -> Result<DBIter> {
let opts = ReadOptions::default();
self.iter_opt(col, mode, &opts)
}
fn iter_opt(&self, col: Col, mode: IteratorMode, readopts: &ReadOptions) -> Result<DBIter>;
}
impl DBIterator for RocksDB {
fn iter_opt(&self, col: Col, mode: IteratorMode, readopts: &ReadOptions) -> Result<DBIter> {
let cf = cf_handle(&self.inner, col)?;
self.inner
.iterator_cf_opt(cf, mode, readopts)
.map_err(internal_error)
}
}
impl DBIterator for RocksDBTransaction {
fn iter_opt(&self, col: Col, mode: IteratorMode, readopts: &ReadOptions) -> Result<DBIter> {
let cf = cf_handle(&self.db, col)?;
self.inner
.iterator_cf_opt(cf, mode, readopts)
.map_err(internal_error)
}
}
impl<'a> DBIterator for RocksDBTransactionSnapshot<'a> {
fn iter_opt(&self, col: Col, mode: IteratorMode, readopts: &ReadOptions) -> Result<DBIter> {
let cf = cf_handle(&self.db, col)?;
self.inner
.iterator_cf_opt(cf, mode, readopts)
.map_err(internal_error)
}
}
impl DBIterator for RocksDBSnapshot {
fn iter_opt(&self, col: Col, mode: IteratorMode, readopts: &ReadOptions) -> Result<DBIter> {
let cf = cf_handle(&self.db, col)?;
self.iterator_cf_opt(cf, mode, readopts)
.map_err(internal_error)
}
}