github.com/inklabsfoundation/inkchain@v0.17.1-0.20181025012015-c3cef8062f19/core/ledger/kvledger/txmgmt/statedb/statedb.go

/*
Copyright IBM Corp. 2016 All Rights Reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

		 http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package statedb

import (
	"sort"

	"math/big"

	"encoding/json"

	"github.com/inklabsfoundation/inkchain/core/ledger/kvledger/txmgmt/version"
	"github.com/inklabsfoundation/inkchain/core/ledger/util"
)

// VersionedDBProvider provides an instance of an versioned DB
type VersionedDBProvider interface {
	// GetDBHandle returns a handle to a VersionedDB
	GetDBHandle(id string) (VersionedDB, error)
	// Close closes all the VersionedDB instances and releases any resources held by VersionedDBProvider
	Close()
}

// VersionedDB lists methods that a db is supposed to implement
type VersionedDB interface {
	// GetState gets the value for given namespace and key. For a chaincode, the namespace corresponds to the chaincodeId
	GetState(namespace string, key string) (*VersionedValue, error)
	// GetStateMultipleKeys gets the values for multiple keys in a single call
	GetStateMultipleKeys(namespace string, keys []string) ([]*VersionedValue, error)
	// GetStateRangeScanIterator returns an iterator that contains all the key-values between given key ranges.
	// startKey is inclusive
	// endKey is exclusive
	// The returned ResultsIterator contains results of type *VersionedKV
	GetStateRangeScanIterator(namespace string, startKey string, endKey string) (ResultsIterator, error)
	// ExecuteQuery executes the given query and returns an iterator that contains results of type *VersionedKV.
	ExecuteQuery(namespace, query string) (ResultsIterator, error)
	// ApplyUpdates applies the batch to the underlying db.
	// height is the height of the highest transaction in the Batch that
	// a state db implementation is expected to ues as a save point
	ApplyUpdates(batch *UpdateBatch, height *version.Height) error
	// GetLatestSavePoint returns the height of the highest transaction upto which
	// the state db is consistent
	GetLatestSavePoint() (*version.Height, error)
	// ValidateKey tests whether the key is supported by the db implementation.
	// For instance, leveldb supports any bytes for the key while the couchdb supports only valid utf-8 string
	ValidateKey(key string) error
	// Open opens the db
	Open() error
	// Close closes the db
	Close()
}

// CompositeKey encloses Namespace and Key components
type CompositeKey struct {
	Namespace string
	Key       string
}

// VersionedValue encloses value and corresponding version
type VersionedValue struct {
	Value   []byte
	Version *version.Height
}

// VersionedKV encloses key and corresponding VersionedValue
type VersionedKV struct {
	CompositeKey
	VersionedValue
}

// ResultsIterator hepls in iterates over query results
type ResultsIterator interface {
	Next() (QueryResult, error)
	Close()
}

// QueryResult - a general interface for supporting different types of query results. Actual types differ for different queries
type QueryResult interface{}

type nsUpdates struct {
	m map[string]*VersionedValue
}

func newNsUpdates() *nsUpdates {
	return &nsUpdates{make(map[string]*VersionedValue)}
}

// UpdateBatch encloses the details of multiple `updates`
type UpdateBatch struct {
	updates map[string]*nsUpdates
}

// NewUpdateBatch constructs an instance of a Batch
func NewUpdateBatch() *UpdateBatch {
	return &UpdateBatch{make(map[string]*nsUpdates)}
}

// Get returns the VersionedValue for the given namespace and key
func (batch *UpdateBatch) Get(ns string, key string) *VersionedValue {
	nsUpdates, ok := batch.updates[ns]
	if !ok {
		return nil
	}
	vv, ok := nsUpdates.m[key]
	if !ok {
		return nil
	}
	return vv
}

// Put adds a VersionedKV
func (batch *UpdateBatch) Put(ns string, key string, value []byte, version *version.Height) {
	if value == nil {
		panic("Nil value not allowed")
	}
	nsUpdates := batch.getOrCreateNsUpdates(ns)
	nsUpdates.m[key] = &VersionedValue{value, version}
}

// Delete deletes a Key and associated value
func (batch *UpdateBatch) Delete(ns string, key string, version *version.Height) {
	nsUpdates := batch.getOrCreateNsUpdates(ns)
	nsUpdates.m[key] = &VersionedValue{nil, version}
}

// Exists checks whether the given key exists in the batch
func (batch *UpdateBatch) Exists(ns string, key string) bool {
	nsUpdates, ok := batch.updates[ns]
	if !ok {
		return false
	}
	_, ok = nsUpdates.m[key]
	return ok
}

// GetUpdatedNamespaces returns the names of the namespaces that are updated
func (batch *UpdateBatch) GetUpdatedNamespaces() []string {
	namespaces := make([]string, len(batch.updates))
	i := 0
	for ns := range batch.updates {
		namespaces[i] = ns
		i++
	}
	return namespaces
}

// GetUpdates returns all the updates for a namespace
func (batch *UpdateBatch) GetUpdates(ns string) map[string]*VersionedValue {
	nsUpdates, ok := batch.updates[ns]
	if !ok {
		return nil
	}
	return nsUpdates.m
}

// GetRangeScanIterator returns an iterator that iterates over keys of a specific namespace in sorted order
// In other word this gives the same functionality over the contents in the `UpdateBatch` as
// `VersionedDB.GetStateRangeScanIterator()` method gives over the contents in the statedb
// This function can be used for querying the contents in the updateBatch before they are committed to the statedb.
// For instance, a validator implementation can used this to verify the validity of a range query of a transaction
// where the UpdateBatch represents the union of the modifications performed by the preceding valid transactions in the same block
// (Assuming Group commit approach where we commit all the updates caused by a block together).
func (batch *UpdateBatch) GetRangeScanIterator(ns string, startKey string, endKey string) ResultsIterator {
	return newNsIterator(ns, startKey, endKey, batch)
}

func (batch *UpdateBatch) getOrCreateNsUpdates(ns string) *nsUpdates {
	nsUpdates := batch.updates[ns]
	if nsUpdates == nil {
		nsUpdates = newNsUpdates()
		batch.updates[ns] = nsUpdates
	}
	return nsUpdates
}

type nsIterator struct {
	ns         string
	nsUpdates  *nsUpdates
	sortedKeys []string
	nextIndex  int
	lastIndex  int
}

func newNsIterator(ns string, startKey string, endKey string, batch *UpdateBatch) *nsIterator {
	nsUpdates, ok := batch.updates[ns]
	if !ok {
		return &nsIterator{}
	}
	sortedKeys := util.GetSortedKeys(nsUpdates.m)
	var nextIndex int
	var lastIndex int
	if startKey == "" {
		nextIndex = 0
	} else {
		nextIndex = sort.SearchStrings(sortedKeys, startKey)
	}
	if endKey == "" {
		lastIndex = len(sortedKeys)
	} else {
		lastIndex = sort.SearchStrings(sortedKeys, endKey)
	}
	return &nsIterator{ns, nsUpdates, sortedKeys, nextIndex, lastIndex}
}

// Next gives next key and versioned value. It returns a nil when exhausted
func (itr *nsIterator) Next() (QueryResult, error) {
	if itr.nextIndex >= itr.lastIndex {
		return nil, nil
	}
	key := itr.sortedKeys[itr.nextIndex]
	vv := itr.nsUpdates.m[key]
	itr.nextIndex++
	return &VersionedKV{CompositeKey{itr.ns, key}, VersionedValue{vv.Value, vv.Version}}, nil
}

// Close implements the method from QueryResult interface
func (itr *nsIterator) Close() {
	// do nothing
}

//******************
type TransferAmount struct {
	balanceType string   `json:"balanceType"`
	amount      *big.Int `json:"amount"`
}

type transferUpdates struct {
	balanceUpdate map[string]*big.Int
	version       *version.Height
	counter       uint64
	ink           *big.Int
	m             map[string]*VersionedValue
}
type TransferBatch struct {
	Updates map[string]*transferUpdates
}

func newTransferUpdates() *transferUpdates {
	return &transferUpdates{make(map[string]*big.Int), nil, 0, nil, make(map[string]*VersionedValue)}
}

func NewTransferBatch() *TransferBatch {
	return &TransferBatch{make(map[string]*transferUpdates)}
}

func (batch *TransferBatch) Put(from string, to string, balanceType string, amount []byte, version *version.Height) {
	if amount == nil {
		return
	}
	fromUpdates := batch.getOrCreateTransferUpdates(from)
	fromBalance, ok := fromUpdates.balanceUpdate[balanceType]
	if !ok {
		fromBalance = big.NewInt(0)
		fromUpdates.balanceUpdate[balanceType] = fromBalance
	}
	amountInt := new(big.Int).SetBytes(amount)
	fromBalance = fromBalance.Sub(fromBalance, amountInt)
	transAmount := &TransferAmount{balanceType: balanceType, amount: amountInt}
	transAmountBytes, _ := json.Marshal(transAmount)
	fromUpdates.m[to] = &VersionedValue{transAmountBytes, version}
	toUpdates := batch.getOrCreateTransferUpdates(to)
	toBalance, ok := toUpdates.balanceUpdate[balanceType]
	if !ok {
		toBalance = big.NewInt(0)
		toUpdates.balanceUpdate[balanceType] = toBalance
	}

	toBalance = toBalance.Add(toBalance, new(big.Int).SetBytes(amount))
	if fromUpdates.version == nil || (fromUpdates.version != nil && fromUpdates.version.Compare(version) < 0) {
		fromUpdates.version = version
	}
	if toUpdates.version == nil || (toUpdates.version != nil && toUpdates.version.Compare(version) < 0) {
		toUpdates.version = version
	}
}

func (batch *TransferBatch) getOrCreateTransferUpdates(from string) *transferUpdates {
	transferUpdates, ok := batch.Updates[from]
	if !ok {
		transferUpdates = newTransferUpdates()
		batch.Updates[from] = transferUpdates
	}
	return transferUpdates
}

func (batch *TransferBatch) UpdateSender(from string, counter uint64, ink *big.Int, version *version.Height) {
	transferUpdates, ok := batch.Updates[from]
	if !ok {
		transferUpdates = newTransferUpdates()
		batch.Updates[from] = transferUpdates
	}
	transferUpdates.version = version
	transferUpdates.counter = counter + 1
	if transferUpdates.ink == nil {
		transferUpdates.ink = ink
	} else {
		transferUpdates.ink = ink.Add(transferUpdates.ink, ink)
	}
}
func (batch *TransferBatch) GetSenderCounter(from string) (uint64, bool) {
	transferUpdates, ok := batch.Updates[from]
	if !ok {
		return 0, ok
	}
	return transferUpdates.counter, ok
}

func (batch *TransferBatch) GetSenderInk(from string) (*big.Int, bool) {
	transferUpdates, ok := batch.Updates[from]
	if !ok {
		return nil, ok
	}
	return transferUpdates.ink, ok
}
func (batch *TransferBatch) ExistsFrom(from string) bool {
	_, ok := batch.Updates[from]
	return ok
}
func (batch *TransferBatch) ExistsTransfer(from string, to string) bool {
	transferUpdates, ok := batch.Updates[from]
	if !ok {
		return false
	}
	_, ok = transferUpdates.m[to]
	return ok
}
func (batch *TransferBatch) GetBalanceUpdate(from string, balanceType string) *big.Int {
	transferUpdates, ok := batch.Updates[from]
	if !ok {
		return nil
	}
	balance, ok := transferUpdates.balanceUpdate[balanceType]
	if !ok {
		return nil
	}
	return balance
}
func (batch *TransferBatch) GetAllBalanceUpdates(from string) map[string]*big.Int {
	transferUpdates, ok := batch.Updates[from]
	if !ok {
		return nil
	}
	return transferUpdates.balanceUpdate
}
func (batch *TransferBatch) GetBalanceVersion(from string) *version.Height {
	transferUpdates, ok := batch.Updates[from]
	if !ok {
		return nil
	}
	return transferUpdates.version
}
func (batch *TransferBatch) Get(from string, to string) *VersionedValue {
	transferUpdates, ok := batch.Updates[from]
	if !ok {
		return nil
	}
	vv, ok := transferUpdates.m[to]
	if !ok {
		return nil
	}
	return vv
}