Debug Spread Get Peer

daemons/node/indexer/service.go

@@ -3,10 +3,13 @@ package indexer
 import (
 	"context"
 	"oc-discovery/daemons/node/common"
+	"sync"
 
 	oclib "cloud.o-forge.io/core/oc-lib"
+	pp "cloud.o-forge.io/core/oc-lib/models/peer"
 	pubsub "github.com/libp2p/go-libp2p-pubsub"
 	"github.com/libp2p/go-libp2p/core/host"
+	"github.com/libp2p/go-libp2p/core/peer"
 )
 
 // Index Record is the model for the specialized registry of node connected to Indexer
@@ -14,6 +17,8 @@ type IndexerService struct {
 	*common.LongLivedStreamRecordedService[PeerRecord]
 	PS              *pubsub.PubSub
 	isStrictIndexer bool
+	mu              sync.RWMutex
+	DisposedPeers   map[peer.ID]*common.TopicNodeActivityPub
 }
 
 // if a pubsub is given... indexer is also an active oc-node. If not... your a strict indexer
@@ -37,11 +42,22 @@ func NewIndexerService(h host.Host, ps *pubsub.PubSub, maxNode int) *IndexerServ
 		logger.Info().Msg("connect to indexers as strict indexer...")
 		common.ConnectToIndexers(h, 0, 5, ix.Host.ID()) // TODO : make var to change how many indexers are allowed.
 		logger.Info().Msg("subscribe to node activity as strict indexer...")
-		ix.SubscribeToNodeActivity(ix.PS) // now we subscribe to a long run topic named node-activity, to relay message.
+
 		logger.Info().Msg("subscribe to decentralized search flow as strict indexer...")
-		ix.SubscribeToSearch(ix.PS)
+		ix.SubscribeToSearch(ix.PS, nil)
 	}
-	ix.initNodeHandler() // then listen up on every protocol expected
+	f := func(ctx context.Context, evt common.TopicNodeActivityPub, _ string) {
+		ix.mu.Lock()
+		if evt.NodeActivity == pp.OFFLINE {
+			delete(ix.DisposedPeers, evt.Disposer.ID)
+		}
+		if evt.NodeActivity == pp.ONLINE {
+			ix.DisposedPeers[evt.Disposer.ID] = &evt
+		}
+		ix.mu.Unlock()
+	}
+	ix.SubscribeToNodeActivity(ix.PS, &f) // now we subscribe to a long run topic named node-activity, to relay message.
+	ix.initNodeHandler()                  // then listen up on every protocol expected
 	return ix
 }