mirror of https://github.com/ledisdb/ledisdb.git
55 lines
625 B
Go
55 lines
625 B
Go
package ledis
|
|
|
|
import (
|
|
"github.com/siddontang/go-leveldb/leveldb"
|
|
"sync"
|
|
)
|
|
|
|
type tx struct {
|
|
m sync.Mutex
|
|
|
|
app *App
|
|
|
|
wb *leveldb.WriteBatch
|
|
}
|
|
|
|
func (app *App) newTx() *tx {
|
|
t := new(tx)
|
|
|
|
t.app = app
|
|
|
|
t.wb = app.db.NewWriteBatch()
|
|
|
|
return t
|
|
}
|
|
|
|
func (t *tx) Close() {
|
|
t.wb.Close()
|
|
}
|
|
|
|
func (t *tx) Put(key []byte, value []byte) {
|
|
t.wb.Put(key, value)
|
|
}
|
|
|
|
func (t *tx) Delete(key []byte) {
|
|
t.wb.Delete(key)
|
|
}
|
|
|
|
func (t *tx) Lock() {
|
|
t.m.Lock()
|
|
}
|
|
|
|
func (t *tx) Unlock() {
|
|
t.wb.Rollback()
|
|
t.m.Unlock()
|
|
}
|
|
|
|
func (t *tx) Commit() error {
|
|
err := t.wb.Commit()
|
|
return err
|
|
}
|
|
|
|
func (t *tx) Rollback() {
|
|
t.wb.Rollback()
|
|
}
|