Try to fix memory leak

cmd/mempool/cache.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"sync"
 	"time"
 )
@@ -9,35 +10,27 @@ import (
 type Cache struct {
 	mux    sync.RWMutex
 	lookup map[string]int64
+	ticker *time.Ticker
 	ttl    time.Duration
+
+	wg sync.WaitGroup
 }
 
 // NewCache -
 func NewCache(ttl time.Duration) *Cache {
 	return &Cache{
 		lookup: make(map[string]int64),
 		ttl:    ttl,
+		ticker: time.NewTicker(time.Minute),
 	}
 }
 
 // Has -
 func (c *Cache) Has(key string) bool {
 	c.mux.RLock()
-	expires, ok := c.lookup[key]
+	_, ok := c.lookup[key]
 	c.mux.RUnlock()
-
-	if !ok {
-		return false
-	}
-
-	if time.Now().UnixNano() > expires {
-		c.mux.Lock()
-		delete(c.lookup, key)
-		c.mux.Unlock()
-		return false
-	}
-
-	return true
+	return ok
 }
 
 // Set -
@@ -47,3 +40,32 @@ func (c *Cache) Set(key string) {
 	c.lookup[key] = expires
 	c.mux.Unlock()
 }
+
+// Start -
+func (c *Cache) Start(ctx context.Context) {
+	c.wg.Add(1)
+	go c.checkExpiration(ctx)
+}
+
+func (c *Cache) checkExpiration(ctx context.Context) {
+	defer c.wg.Done()
+
+	for {
+		select {
+		case <-ctx.Done():
+			c.ticker.Stop()
+			return
+		case <-c.ticker.C:
+			c.mux.RLock()
+			for key, expiration := range c.lookup {
+				if time.Now().UnixNano() <= expiration {
+					continue
+				}
+				c.mux.Lock()
+				delete(c.lookup, key)
+				c.mux.Unlock()
+			}
+			c.mux.RUnlock()
+		}
+	}
+}

cmd/mempool/endorsement/check.go

@@ -77,12 +77,9 @@ func getCoordinates(data []byte) (*big.Int, *big.Int, error) {
 
 	// The equation is y^2 = x^3 - 3x + b
 	// x^3, mod P
-	xCubed := new(big.Int).Exp(x, three, c.P)
-
+	xCubed := new(big.Int).Exp(x, three, nil)
 	// 3x, mod P
 	threeX := new(big.Int).Mul(x, three)
-	threeX.Mod(threeX, c.P)
-
 	// x^3 - 3x
 	ySquared := new(big.Int).Sub(xCubed, threeX)
 

cmd/mempool/indexer.go

@@ -110,6 +110,7 @@ func NewIndexer(ctx context.Context, network string, indexerCfg config.Indexer,
 		endorsements:     make(chan *models.Endorsement, 1024*32),
 		rights:           ccache.New(ccache.Configure().MaxSize(60)),
 	}
+	indexer.cache.Start(ctx)
 
 	indexer.state = &database.State{
 		IndexType: models.IndexTypeMempool,
@@ -263,6 +264,9 @@ func (indexer *Indexer) listen(ctx context.Context) {
 					indexer.error().Msgf("invalid applied operation %v", applied)
 					continue
 				}
+				if !indexer.branches.Contains(applied.Branch) {
+					continue
+				}
 				if indexer.isHashProcessed(applied.Hash) {
 					continue
 				}
@@ -276,6 +280,10 @@ func (indexer *Indexer) listen(ctx context.Context) {
 					indexer.error().Msgf("invalid %s operation %v", msg.Status, failed)
 					continue
 				}
+
+				if !indexer.branches.Contains(failed.Branch) {
+					continue
+				}
 				if indexer.isHashProcessed(failed.Hash) {
 					continue
 				}