From 3bfc8df523f0ad8c46c85f6f7c23bc0f83db46e1 Mon Sep 17 00:00:00 2001
From: gfanton <8671905+gfanton@users.noreply.github.com>
Date: Thu, 16 Feb 2023 11:30:09 +0100
Subject: [PATCH] feat: add `remoteheads` in head exchange

Signed-off-by: gfanton <8671905+gfanton@users.noreply.github.com>
---
 stores/basestore/base_store.go | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/stores/basestore/base_store.go b/stores/basestore/base_store.go
index 8617397..fd30c3b 100644
--- a/stores/basestore/base_store.go
+++ b/stores/basestore/base_store.go
@@ -1184,16 +1184,15 @@ func (b *BaseStore) exchangeHeads(p peer.ID) error {
 		return fmt.Errorf("unable to get local heads from cache: %w", err)
 	}
 
-	// @FIXME(gfanton): looks like activate this break the exchange
-	// rawRemoteHeads, err := store.Cache().Get(b.ctx, datastore.NewKey("_remoteHeads"))
-	// if err != nil && err != datastore.ErrNotFound {
-	// 	return fmt.Errorf("unable to get data from cache: %w", err)
-	// }
+	rawRemoteHeads, err := b.Cache().Get(b.ctx, datastore.NewKey("_remoteHeads"))
+	if err != nil && err != datastore.ErrNotFound {
+		return fmt.Errorf("unable to get data from cache: %w", err)
+	}
 
 	heads := []*entry.Entry{}
 
-	for _, rawHeads := range [][]byte{rawLocalHeads} {
-		if len(rawLocalHeads) > 0 {
+	for _, rawHeads := range [][]byte{rawLocalHeads, rawRemoteHeads} {
+		if len(rawHeads) > 0 {
 			var dHeads []*entry.Entry
 			err = json.Unmarshal(rawHeads, &dHeads)
 			if err != nil {