update mod

vendor/github.com/siddontang/ledisdb/ledis/batch.go

@@ -0,0 +1,139 @@
+package ledis
+
+import (
+	"sync"
+
+	"github.com/siddontang/go/log"
+	"github.com/siddontang/ledisdb/rpl"
+	"github.com/siddontang/ledisdb/store"
+)
+
+type batch struct {
+	l *Ledis
+
+	*store.WriteBatch
+
+	sync.Locker
+
+	//	tx *Tx
+}
+
+func (b *batch) Commit() error {
+	if b.l.cfg.GetReadonly() {
+		return ErrWriteInROnly
+	}
+
+	return b.l.handleCommit(b.WriteBatch, b.WriteBatch)
+
+	// if b.tx == nil {
+	// 	return b.l.handleCommit(b.WriteBatch, b.WriteBatch)
+	// } else {
+	// 	if b.l.r != nil {
+	// 		if err := b.tx.data.Append(b.WriteBatch.BatchData()); err != nil {
+	// 			return err
+	// 		}
+	// 	}
+	// 	return b.WriteBatch.Commit()
+	// }
+}
+
+func (b *batch) Lock() {
+	b.Locker.Lock()
+}
+
+func (b *batch) Unlock() {
+	b.WriteBatch.Rollback()
+	b.Locker.Unlock()
+}
+
+func (b *batch) Put(key []byte, value []byte) {
+	b.WriteBatch.Put(key, value)
+}
+
+func (b *batch) Delete(key []byte) {
+	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) *batch {
+	b := new(batch)
+	b.l = l
+	b.WriteBatch = wb
+
+	b.Locker = locker
+
+	return b
+}
+
+type commiter interface {
+	Commit() error
+}
+
+type commitDataGetter interface {
+	Data() []byte
+}
+
+func (l *Ledis) handleCommit(g commitDataGetter, c commiter) error {
+	l.commitLock.Lock()
+
+	var err error
+	if l.r != nil {
+		var rl *rpl.Log
+		if rl, err = l.r.Log(g.Data()); err != nil {
+			l.commitLock.Unlock()
+
+			log.Fatalf("write wal error %s", err.Error())
+			return err
+		}
+
+		l.propagate(rl)
+
+		if err = c.Commit(); err != nil {
+			l.commitLock.Unlock()
+
+			log.Fatalf("commit error %s", err.Error())
+			l.noticeReplication()
+			return err
+		}
+
+		if err = l.r.UpdateCommitID(rl.ID); err != nil {
+			l.commitLock.Unlock()
+
+			log.Fatalf("update commit id error %s", err.Error())
+			l.noticeReplication()
+			return err
+		}
+	} else {
+		err = c.Commit()
+	}
+
+	l.commitLock.Unlock()
+
+	return err
+}