forked from mirror/ledisdb
86 lines
1.2 KiB
Go
86 lines
1.2 KiB
Go
package ledis
|
|
|
|
import (
|
|
"github.com/siddontang/go-leveldb/leveldb"
|
|
"github.com/siddontang/ledisdb/replication"
|
|
"sync"
|
|
)
|
|
|
|
type tx struct {
|
|
m sync.Mutex
|
|
|
|
l *Ledis
|
|
wb *leveldb.WriteBatch
|
|
|
|
binlog *replication.BinLog
|
|
batch [][]byte
|
|
}
|
|
|
|
func newTx(l *Ledis) *tx {
|
|
t := new(tx)
|
|
|
|
t.l = l
|
|
t.wb = l.ldb.NewWriteBatch()
|
|
|
|
t.batch = make([][]byte, 0, 4)
|
|
t.binlog = l.binlog
|
|
return t
|
|
}
|
|
|
|
func (t *tx) Close() {
|
|
t.wb.Close()
|
|
}
|
|
|
|
func (t *tx) Put(key []byte, value []byte) {
|
|
t.wb.Put(key, value)
|
|
|
|
if t.binlog != nil {
|
|
buf := encodeBinLogPut(key, value)
|
|
t.batch = append(t.batch, buf)
|
|
}
|
|
}
|
|
|
|
func (t *tx) Delete(key []byte) {
|
|
t.wb.Delete(key)
|
|
|
|
if t.binlog != nil {
|
|
buf := encodeBinLogDelete(key)
|
|
t.batch = append(t.batch, buf)
|
|
}
|
|
}
|
|
|
|
func (t *tx) Lock() {
|
|
t.m.Lock()
|
|
}
|
|
|
|
func (t *tx) Unlock() {
|
|
t.batch = t.batch[0:0]
|
|
t.wb.Rollback()
|
|
t.m.Unlock()
|
|
}
|
|
|
|
func (t *tx) Commit() error {
|
|
var err error
|
|
if t.binlog != nil {
|
|
t.l.Lock()
|
|
err = t.wb.Commit()
|
|
if err != nil {
|
|
t.l.Unlock()
|
|
return err
|
|
}
|
|
|
|
err = t.binlog.Log(t.batch...)
|
|
|
|
t.l.Unlock()
|
|
} else {
|
|
t.l.Lock()
|
|
err = t.wb.Commit()
|
|
t.l.Unlock()
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (t *tx) Rollback() {
|
|
t.wb.Rollback()
|
|
}
|