dione/blockchain/pool/mempool.go

package pool

import (
	"encoding/hex"
	"sort"
	"time"

	types2 "github.com/Secured-Finance/dione/blockchain/types"

	"github.com/Secured-Finance/dione/consensus/policy"

	"github.com/Secured-Finance/dione/cache"
)

const (
	DefaultTxTTL    = 10 * time.Minute
	DefaultTxPrefix = "tx_"
)

type Mempool struct {
	cache cache.Cache
}

func NewMempool() (*Mempool, error) {
	mp := &Mempool{
		cache: cache.NewInMemoryCache(), // here we need to use separate cache
	}

	return mp, nil
}

func (mp *Mempool) StoreTx(tx *types2.Transaction) error {
	hashStr := hex.EncodeToString(tx.Hash)
	err := mp.cache.StoreWithTTL(DefaultTxPrefix+hashStr, tx, DefaultTxTTL)
	return err
}

func (mp *Mempool) GetTxsForNewBlock() []*types2.Transaction {
	var txForBlock []*types2.Transaction
	allTxs := mp.GetAllTxs()
	sort.Slice(allTxs, func(i, j int) bool {
		return allTxs[i].Timestamp.Before(allTxs[j].Timestamp)
	})

	for i := 0; i < policy.BlockMaxTransactionCount; i++ {
		if len(allTxs) == 0 {
			break
		}
		tx := allTxs[0]     // get oldest tx
		allTxs = allTxs[1:] // pop tx
		txForBlock = append(txForBlock, tx)
	}

	return txForBlock
}

func (mp *Mempool) GetAllTxs() []*types2.Transaction {
	var allTxs []*types2.Transaction

	for _, v := range mp.cache.Items() {
		tx := v.(types2.Transaction)
		allTxs = append(allTxs, &tx)
	}
	return allTxs
}

func removeItemFromStringSlice(s []string, i int) []string {
	s[len(s)-1], s[i] = s[i], s[len(s)-1]
	return s[:len(s)-1]
}