wip diff

Signed-off-by: Jakub Sztandera <[email protected]>

Author: Kubuxu

chainexchange/pubsub.go

@@ -3,6 +3,7 @@ package chainexchange
 import (
 	"context"
 	"fmt"
+	"runtime"
 	"sync"
 	"time"
 
@@ -266,7 +267,9 @@ func (p *PubSubChainExchange) cacheAsDiscoveredChain(ctx context.Context, cmsg M
 	keysBatch := cmsg.Chain.KeysForPrefixes()
 	for offset := cmsg.Chain.Len() - 1; offset >= 0 && ctx.Err() == nil; offset-- {
 		prefix := cmsg.Chain.Prefix(offset)
-		key := keysBatch[offset]
+		//key := keysBatch[offset]
+		runtime.KeepAlive(keysBatch)
+		key := prefix.Key()
 
 		if portion, found := wanted.Peek(key); !found {
 			// Not a wanted key; add it to discovered chains if they are not there already,
@@ -332,7 +335,9 @@ func (p *PubSubChainExchange) cacheAsWantedChain(ctx context.Context, cmsg Messa
 	keysBatch := cmsg.Chain.KeysForPrefixes()
 	for offset := cmsg.Chain.Len() - 1; offset >= 0 && ctx.Err() == nil; offset-- {
 		prefix := cmsg.Chain.Prefix(offset)
-		key := keysBatch[offset]
+		//key := keysBatch[offset]
+		runtime.KeepAlive(keysBatch)
+		key := prefix.Key()
 
 		if portion, found := wanted.Peek(key); !found || portion.IsPlaceholder() {
 			wanted.Add(key, &chainPortion{

merkle/merkle_test.go

@@ -3,6 +3,7 @@ package merkle
 import (
 	"encoding/hex"
 	"fmt"
+	"runtime"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -131,3 +132,23 @@ func TestHashTreeGolden(t *testing.T) {
 	}
 	assert.Equal(t, expectedHex, batchResHash)
 }
+
+func BenchmarkIndividualPrefix(b *testing.B) {
+	K := 128
+	inputs := generateInputs("golden", K)
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		for i := 1; i < len(inputs); i++ {
+			runtime.KeepAlive(Tree(inputs[:i+1]))
+		}
+	}
+}
+
+func BenchmarkBatchPrefix(b *testing.B) {
+	K := 128
+	inputs := generateInputs("golden", K)
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		runtime.KeepAlive(BatchTree(inputs))
+	}
+}