-
Notifications
You must be signed in to change notification settings - Fork 250
/
wrapper.go
44 lines (35 loc) · 993 Bytes
/
wrapper.go
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
package db
import dbm "github.com/cosmos/cosmos-db"
// Wrapper wraps a dbm.DB to implement DB.
type Wrapper struct {
dbm.DB
}
var _ DB = (*Wrapper)(nil)
// NewWrapper returns a new Wrapper.
func NewWrapper(db dbm.DB) *Wrapper {
return &Wrapper{DB: db}
}
// Iterator implements DB.
func (db *Wrapper) Iterator(start, end []byte) (Iterator, error) {
return db.DB.Iterator(start, end)
}
// ReverseIterator implements DB.
func (db *Wrapper) ReverseIterator(start, end []byte) (Iterator, error) {
return db.DB.ReverseIterator(start, end)
}
// NewBatch implements DB.
func (db *Wrapper) NewBatch() Batch {
return db.DB.NewBatch()
}
// NewBatchWithSize implements DB.
func (db *Wrapper) NewBatchWithSize(size int) Batch {
return db.DB.NewBatchWithSize(size)
}
// NewDB returns a new Wrapper.
func NewDB(name, backendType, dir string) (*Wrapper, error) {
db, err := dbm.NewDB(name, dbm.BackendType(backendType), dir)
if err != nil {
return nil, err
}
return NewWrapper(db), nil
}