github.com/aergoio/aergo@v1.3.1/mempool/txlist.go

/**
 *  @file
 *  @copyright defined in aergo/LICENSE.txt
 */

package mempool

import (
	"sort"
	"sync"
	"time"

	"github.com/aergoio/aergo/types"
)

// TxList is internal struct for transactions per account
type TxList struct {
	sync.RWMutex
	base     *types.State
	lastTime time.Time
	account  []byte
	ready    int
	list     []types.Transaction // nonce-ordered tx list
}

// NewTxList creates new TxList with given State
func NewTxList(acc []byte, st *types.State) *TxList {
	return &TxList{
		base:    st,
		account: acc,
	}
}

func (tl *TxList) GetLastModifiedTime() time.Time {
	return tl.lastTime
}
func (tl *TxList) GetAccount() []byte {
	return tl.account
}

// Len returns number of transactios which are ready to be processed
func (tl *TxList) Len() int {
	tl.RLock()
	defer tl.RUnlock()
	return tl.ready
}

// Empty check TxList is empty including orphan
func (tl *TxList) Empty() bool {
	tl.RLock()
	defer tl.RUnlock()
	return len(tl.list) == 0
}

func (tl *TxList) search(tx types.Transaction) (int, bool) {
	key := tx.GetBody().GetNonce()
	ind := sort.Search(len(tl.list), func(i int) bool {
		comp := tl.list[i].GetBody().GetNonce()
		return comp >= key
	})
	if ind < len(tl.list) && tl.compare(tx, ind) {
		return ind, true
	}
	return ind, false
}
func (tl *TxList) compare(tx types.Transaction, index int) bool {
	if tx.GetBody().GetNonce() == tl.list[index].GetBody().GetNonce() {
		return true
	}
	return false
}

func (tl *TxList) continuous(index int) bool {
	l := tl.base.Nonce
	r := tl.list[index].GetBody().GetNonce()
	if tl.ready > 0 {
		l = tl.list[tl.ready-1].GetBody().GetNonce()
	}

	if l+1 == r {
		return true
	}
	return false
}

// Put inserts transaction into TxList
// if transaction is processible, it is appended to list
// if not, transaction is managed as orphan
func (tl *TxList) Put(tx types.Transaction) (int, error) {
	tl.Lock()
	defer tl.Unlock()

	nonce := tx.GetBody().GetNonce()
	if nonce <= tl.base.Nonce {
		return 0, types.ErrTxNonceTooLow
	}

	index, found := tl.search(tx)
	if found == true { // exact match
		return 0, types.ErrSameNonceAlreadyInMempool
	}

	oldCnt := len(tl.list) - tl.ready

	if index < len(tl.list) {
		tl.list = append(tl.list[:index], append([]types.Transaction{tx},
			tl.list[index:]...)...)
	} else {
		tl.list = append(tl.list, tx)
	}

	for ; index < len(tl.list); index++ {
		if !tl.continuous(index) {
			break
		}
		tl.ready++
	}
	newCnt := len(tl.list) - tl.ready

	tl.lastTime = time.Now()
	return oldCnt - newCnt, nil
}

// SetMinNonce sets new minimum nonce for TxList
// evict on some transactions is possible due to minimum nonce
func (tl *TxList) FilterByState(st *types.State) (int, []types.Transaction) {
	tl.Lock()
	defer tl.Unlock()

	var balCheck bool

	if tl.base.Nonce == st.Nonce {
		tl.base = st
		return 0, nil
	}

	if tl.base.GetBalanceBigInt().Cmp(st.GetBalanceBigInt()) > 0 {
		balCheck = true
	}
	tl.base = st

	oldCnt := len(tl.list) - tl.ready
	var left []types.Transaction
	removed := tl.list[:0]
	for i, x := range tl.list {
		err := x.ValidateWithSenderState(st)
		if err == nil || err == types.ErrTxNonceToohigh {
			if err != nil && !balCheck {
				left = append(left, tl.list[i:]...)
				break
			}
			left = append(left, x)
		} else {
			removed = append(removed, x)
		}
	}

	tl.list = left
	tl.ready = 0
	for i := 0; i < len(tl.list); i++ {
		if !tl.continuous(i) {
			break
		}
		tl.ready++
	}
	newCnt := len(tl.list) - tl.ready

	tl.lastTime = time.Now()
	return oldCnt - newCnt, removed
}

// FilterByPrice will evict transactions that needs more amount than balance
/*
func (tl *TxList) FilterByPrice(balance uint64) error {
	tl.Lock()
	defer tl.Unlock()
	return nil
}
*/

// Get returns processible transactions
func (tl *TxList) Get() []types.Transaction {
	tl.RLock()
	defer tl.RUnlock()
	return tl.list[:tl.ready]
}

// GetAll returns all transactions including orphans
func (tl *TxList) GetAll() []types.Transaction {
	tl.Lock()
	defer tl.Unlock()
	return tl.list

}

func (tl *TxList) len() int {
	return len(tl.list)
}

/*

func (tl *TxList) printList() {
	fmt.Printf("\t\t")
	for i := 0; i < len(tl.list); i++ {
		cur := tl.list[i].GetBody().GetNonce()
		fmt.Printf("%d, ", cur)
	}
	fmt.Printf("done ready:%d n:%d, b:%d\n", tl.ready, tl.base.Nonce, tl.base.Balance)

}

func (tl *TxList) checkSanity() bool {
	prev := uint64(0)
	for _, v := range tl.list {
		x := v.GetBody().GetNonce()
		if prev >= x {
			return false
		}
		prev = x
	}
	return true
}
func (tl *TxList) printList() {

	var f, l, before uint64
	if tl.list != nil {
		f = tl.list[0].GetBody().GetNonce()
		l = tl.list[len(tl.list)-1].GetBody().GetNonce()
	}
	fmt.Printf("min: %d ready(nr:%d)[%d~%d]", tl.min, len(tl.list), f, l)

	for i := 0; i < len(tl.list); i++ {
		cur := tl.list[i].GetBody().GetNonce()
		if i != 0 && before+1 != cur {
			fmt.Printf("WARN: List is not sequential")
		}
		before = cur
	}

	fmt.Println()
	fmt.Printf("deps 1st(%d):", len(tl.deps))
	for k, v := range tl.deps {
		f := v[0].GetBody().GetNonce()
		l := v[len(v)-1].GetBody().GetNonce()
		fmt.Printf("%d=>[%d~%d],", k, f, l)
	}
	fmt.Println()

	fmt.Printf("dep parent(%d):", len(tl.parent))
	for k, v := range tl.parent {
		fmt.Printf("(%d, %d)", k, v)
	}
	fmt.Println()
}
*/