re
/
ledisdb
forked from mirror/ledisdb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bugfix for lmdb

This commit is contained in:
siddontang 2014-09-04 14:55:26 +08:00
parent c77f938b14
commit 42f6a3679f
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
store/mdb/mdb.go
View File

@ -92,14 +92,16 @@ func (s Store) Repair(path string, c *config.Config) error {
func (db MDB) Put(key, value []byte) error { func (db MDB) Put(key, value []byte) error {
itr := db.iterator(false) itr := db.iterator(false)
defer itr.Close()
itr.err = itr.c.Put(key, value, 0) itr.err = itr.c.Put(key, value, 0)
itr.setState() itr.setState()
return itr.Close() return itr.err
} }
func (db MDB) BatchPut(writes []driver.Write) error { func (db MDB) BatchPut(writes []driver.Write) error {
itr := db.iterator(false) itr := db.iterator(false)
defer itr.Close()
for _, w := range writes { for _, w := range writes {
if w.Value == nil { if w.Value == nil {
@ -117,7 +119,7 @@ func (db MDB) BatchPut(writes []driver.Write) error {
} }
itr.setState() itr.setState()
return itr.Close() return itr.err
} }
func (db MDB) Get(key []byte) ([]byte, error) { func (db MDB) Get(key []byte) ([]byte, error) {
@ -208,6 +210,8 @@ func (itr *MDBIterator) setState() {
itr.err = nil itr.err = nil
} }
itr.valid = false itr.valid = false
} else {
itr.valid = true
} }
} }