Implement GetAllTxs function in mempool

This commit is contained in:
ChronosX88 2021-05-31 22:24:07 +03:00
parent 00b041fbb0
commit c0e69ff1d0
Signed by: ChronosXYZ
GPG Key ID: 085A69A82C8C511A
4 changed files with 28 additions and 38 deletions

View File

@ -3,7 +3,6 @@ package pool
import ( import (
"encoding/hex" "encoding/hex"
"sort" "sort"
"sync"
"time" "time"
types2 "github.com/Secured-Finance/dione/blockchain/types" types2 "github.com/Secured-Finance/dione/blockchain/types"
@ -19,58 +18,26 @@ const (
) )
type Mempool struct { type Mempool struct {
m sync.RWMutex cache cache.Cache
cache cache.Cache
txDescriptors []string // list of txs in cache
} }
func NewMempool(c cache.Cache) (*Mempool, error) { func NewMempool() (*Mempool, error) {
mp := &Mempool{ mp := &Mempool{
cache: c, cache: cache.NewInMemoryCache(), // here we need to use separate cache
} }
var txDesc []string
err := c.Get("tx_list", &txDesc)
if err != nil || err != cache.ErrNilValue {
return nil, err
}
mp.txDescriptors = txDesc
return mp, nil return mp, nil
} }
func (mp *Mempool) StoreTx(tx *types2.Transaction) error { func (mp *Mempool) StoreTx(tx *types2.Transaction) error {
mp.m.Lock()
defer mp.m.Unlock()
hashStr := hex.EncodeToString(tx.Hash) hashStr := hex.EncodeToString(tx.Hash)
err := mp.cache.StoreWithTTL(DefaultTxPrefix+hashStr, tx, DefaultTxTTL) err := mp.cache.StoreWithTTL(DefaultTxPrefix+hashStr, tx, DefaultTxTTL)
mp.txDescriptors = append(mp.txDescriptors, hashStr)
mp.cache.Store("tx_list", mp.txDescriptors) // update tx list in cache
return err return err
} }
func (mp *Mempool) GetTxsForNewBlock() []*types2.Transaction { func (mp *Mempool) GetTxsForNewBlock() []*types2.Transaction {
mp.m.Lock()
defer mp.m.Unlock()
var txForBlock []*types2.Transaction var txForBlock []*types2.Transaction
var allTxs []*types2.Transaction allTxs := mp.GetAllTxs()
for i, v := range mp.txDescriptors {
var tx types2.Transaction
err := mp.cache.Get(DefaultTxPrefix+v, &tx)
if err != nil {
if err == cache.ErrNilValue {
// descriptor is broken
// delete it and update list
mp.txDescriptors = removeItemFromStringSlice(mp.txDescriptors, i)
mp.cache.Store("tx_list", mp.txDescriptors) // update tx list in cache
}
continue
}
allTxs = append(allTxs, &tx)
}
sort.Slice(allTxs, func(i, j int) bool { sort.Slice(allTxs, func(i, j int) bool {
return allTxs[i].Timestamp.Before(allTxs[j].Timestamp) return allTxs[i].Timestamp.Before(allTxs[j].Timestamp)
}) })
@ -87,6 +54,16 @@ func (mp *Mempool) GetTxsForNewBlock() []*types2.Transaction {
return txForBlock 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 { func removeItemFromStringSlice(s []string, i int) []string {
s[len(s)-1], s[i] = s[i], s[len(s)-1] s[len(s)-1], s[i] = s[i], s[len(s)-1]
return s[:len(s)-1] return s[:len(s)-1]

1
cache/cache.go vendored
View File

@ -12,4 +12,5 @@ type Cache interface {
StoreWithTTL(key string, value interface{}, ttl time.Duration) error StoreWithTTL(key string, value interface{}, ttl time.Duration) error
Get(key string, value interface{}) error Get(key string, value interface{}) error
Delete(key string) Delete(key string)
Items() map[string]interface{}
} }

View File

@ -15,7 +15,7 @@ type InMemoryCache struct {
cache *cache.Cache cache *cache.Cache
} }
func NewInMemoryCache() *InMemoryCache { func NewInMemoryCache() Cache {
return &InMemoryCache{ return &InMemoryCache{
cache: cache.New(DefaultCacheExpiration, DefaultGCInterval), cache: cache.New(DefaultCacheExpiration, DefaultGCInterval),
} }
@ -45,3 +45,11 @@ func (imc *InMemoryCache) Get(key string, value interface{}) error {
func (imc *InMemoryCache) Delete(key string) { func (imc *InMemoryCache) Delete(key string) {
imc.cache.Delete(key) imc.cache.Delete(key)
} }
func (imc *InMemoryCache) Items() map[string]interface{} {
m := make(map[string]interface{})
for k, v := range imc.cache.Items() {
m[k] = v.Object
}
return m
}

View File

@ -81,3 +81,7 @@ func (rc *RedisCache) Get(key string, value interface{}) error {
func (rc *RedisCache) Delete(key string) { func (rc *RedisCache) Delete(key string) {
rc.Client.Del(rc.ctx, key) rc.Client.Del(rc.ctx, key)
} }
func (rc *RedisCache) Items() map[string]interface{} {
return nil // TODO
}