mirror of https://github.com/ledisdb/ledisdb.git
137 lines
2.0 KiB
Go
137 lines
2.0 KiB
Go
package ledis
|
|
|
|
import (
|
|
"github.com/siddontang/go/log"
|
|
"github.com/siddontang/ledisdb/rpl"
|
|
"github.com/siddontang/ledisdb/store"
|
|
"sync"
|
|
)
|
|
|
|
type batch struct {
|
|
l *Ledis
|
|
|
|
store.WriteBatch
|
|
|
|
sync.Locker
|
|
|
|
tx *Tx
|
|
|
|
eb *eventBatch
|
|
}
|
|
|
|
func (b *batch) Commit() error {
|
|
if b.l.IsReadOnly() {
|
|
return ErrWriteInROnly
|
|
}
|
|
|
|
if b.tx == nil {
|
|
return b.l.handleCommit(b.eb, b.WriteBatch)
|
|
} else {
|
|
if b.l.r != nil {
|
|
b.tx.eb.Write(b.eb.Bytes())
|
|
}
|
|
return b.WriteBatch.Commit()
|
|
}
|
|
}
|
|
|
|
func (b *batch) Lock() {
|
|
b.Locker.Lock()
|
|
}
|
|
|
|
func (b *batch) Unlock() {
|
|
b.eb.Reset()
|
|
|
|
b.WriteBatch.Rollback()
|
|
b.Locker.Unlock()
|
|
}
|
|
|
|
func (b *batch) Put(key []byte, value []byte) {
|
|
if b.l.r != nil {
|
|
b.eb.Put(key, value)
|
|
}
|
|
|
|
b.WriteBatch.Put(key, value)
|
|
}
|
|
|
|
func (b *batch) Delete(key []byte) {
|
|
if b.l.r != nil {
|
|
b.Delete(key)
|
|
}
|
|
|
|
b.WriteBatch.Delete(key)
|
|
}
|
|
|
|
type dbBatchLocker struct {
|
|
l *sync.Mutex
|
|
wrLock *sync.RWMutex
|
|
}
|
|
|
|
func (l *dbBatchLocker) Lock() {
|
|
l.wrLock.RLock()
|
|
l.l.Lock()
|
|
}
|
|
|
|
func (l *dbBatchLocker) Unlock() {
|
|
l.l.Unlock()
|
|
l.wrLock.RUnlock()
|
|
}
|
|
|
|
type txBatchLocker struct {
|
|
}
|
|
|
|
func (l *txBatchLocker) Lock() {}
|
|
func (l *txBatchLocker) Unlock() {}
|
|
|
|
type multiBatchLocker struct {
|
|
}
|
|
|
|
func (l *multiBatchLocker) Lock() {}
|
|
func (l *multiBatchLocker) Unlock() {}
|
|
|
|
func (l *Ledis) newBatch(wb store.WriteBatch, locker sync.Locker, tx *Tx) *batch {
|
|
b := new(batch)
|
|
b.l = l
|
|
b.WriteBatch = wb
|
|
|
|
b.Locker = locker
|
|
|
|
b.tx = tx
|
|
b.eb = new(eventBatch)
|
|
|
|
return b
|
|
}
|
|
|
|
type commiter interface {
|
|
Commit() error
|
|
}
|
|
|
|
func (l *Ledis) handleCommit(eb *eventBatch, c commiter) error {
|
|
l.commitLock.Lock()
|
|
defer l.commitLock.Unlock()
|
|
|
|
var err error
|
|
if l.r != nil {
|
|
var rl *rpl.Log
|
|
if rl, err = l.r.Log(eb.Bytes()); err != nil {
|
|
log.Fatal("write wal error %s", err.Error())
|
|
return err
|
|
}
|
|
|
|
l.propagate(rl)
|
|
|
|
if err = c.Commit(); err != nil {
|
|
log.Fatal("commit error %s", err.Error())
|
|
return err
|
|
}
|
|
|
|
if err = l.r.UpdateCommitID(rl.ID); err != nil {
|
|
log.Fatal("update commit id error %s", err.Error())
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
} else {
|
|
return c.Commit()
|
|
}
|
|
}
|