You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
2.3 KiB
119 lines
2.3 KiB
package remote
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
"github.com/harmony-one/harmony/internal/tikv/common"
|
|
)
|
|
|
|
type RemoteBatch struct {
|
|
db *RemoteDatabase
|
|
lock sync.Mutex
|
|
|
|
size int
|
|
batchWriteKey [][]byte
|
|
batchWriteValue [][]byte
|
|
batchDeleteKey [][]byte
|
|
}
|
|
|
|
func newRemoteBatch(db *RemoteDatabase) *RemoteBatch {
|
|
return &RemoteBatch{db: db}
|
|
}
|
|
|
|
// Put inserts the given value into the key-value data store.
|
|
func (b *RemoteBatch) Put(key []byte, value []byte) error {
|
|
if len(key) == 0 {
|
|
return common.ErrEmptyKey
|
|
}
|
|
|
|
if len(value) == 0 {
|
|
value = EmptyValueStub
|
|
}
|
|
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
b.batchWriteKey = append(b.batchWriteKey, key)
|
|
b.batchWriteValue = append(b.batchWriteValue, value)
|
|
b.size += len(key) + len(value)
|
|
return nil
|
|
}
|
|
|
|
// Delete removes the key from the key-value data store.
|
|
func (b *RemoteBatch) Delete(key []byte) error {
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
b.batchDeleteKey = append(b.batchDeleteKey, key)
|
|
b.size += len(key)
|
|
return nil
|
|
}
|
|
|
|
// ValueSize retrieves the amount of data queued up for writing.
|
|
func (b *RemoteBatch) ValueSize() int {
|
|
return b.size
|
|
}
|
|
|
|
// Write flushes any accumulated data to disk.
|
|
func (b *RemoteBatch) Write() error {
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
if len(b.batchWriteKey) > 0 {
|
|
err := b.db.client.BatchPut(context.Background(), b.batchWriteKey, b.batchWriteValue)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if len(b.batchDeleteKey) > 0 {
|
|
err := b.db.client.BatchDelete(context.Background(), b.batchDeleteKey)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Reset resets the batch for reuse.
|
|
func (b *RemoteBatch) Reset() {
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
b.batchWriteKey = b.batchWriteKey[:0]
|
|
b.batchWriteValue = b.batchWriteValue[:0]
|
|
b.batchDeleteKey = b.batchDeleteKey[:0]
|
|
b.size = 0
|
|
}
|
|
|
|
// Replay replays the batch contents.
|
|
func (b *RemoteBatch) Replay(w ethdb.KeyValueWriter) error {
|
|
for i, key := range b.batchWriteKey {
|
|
if bytes.Compare(b.batchWriteValue[i], EmptyValueStub) == 0 {
|
|
err := w.Put(key, []byte{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
err := w.Put(key, b.batchWriteValue[i])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(b.batchDeleteKey) > 0 {
|
|
for _, key := range b.batchDeleteKey {
|
|
err := w.Delete(key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|