forked from mirror/ledisdb
51 lines
770 B
Go
51 lines
770 B
Go
|
package ledis
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"github.com/siddontang/go-leveldb/leveldb"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DB struct {
|
||
|
|
db *leveldb.DB
|
||
|
|
|
||
|
|
kvTx *tx
|
||
|
|
listTx *tx
|
||
|
|
hashTx *tx
|
||
|
|
zsetTx *tx
|
||
|
|
}
|
||
|
|
|
||
|
|
func OpenDB(configJson json.RawMessage) (*DB, error) {
|
||
|
|
db, err := leveldb.Open(configJson)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return newDB(db)
|
||
|
|
}
|
||
|
|
|
||
|
|
func OpenDBWithConfig(cfg *leveldb.Config) (*DB, error) {
|
||
|
|
db, err := leveldb.OpenWithConfig(cfg)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return newDB(db)
|
||
|
|
}
|
||
|
|
|
||
|
|
func newDB(db *leveldb.DB) (*DB, error) {
|
||
|
|
d := new(DB)
|
||
|
|
|
||
|
|
d.db = db
|
||
|
|
|
||
|
|
d.kvTx = &tx{wb: db.NewWriteBatch()}
|
||
|
|
d.listTx = &tx{wb: db.NewWriteBatch()}
|
||
|
|
d.hashTx = &tx{wb: db.NewWriteBatch()}
|
||
|
|
d.zsetTx = &tx{wb: db.NewWriteBatch()}
|
||
|
|
|
||
|
|
return d, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *DB) Close() {
|
||
|
|
db.db.Close()
|
||
|
|
}
|