77 lines
2.4 KiB
Go
77 lines
2.4 KiB
Go
package indexer
|
|
|
|
import (
|
|
"context"
|
|
"oc-discovery/daemons/node/common"
|
|
"sync"
|
|
|
|
oclib "cloud.o-forge.io/core/oc-lib"
|
|
dht "github.com/libp2p/go-libp2p-kad-dht"
|
|
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
|
record "github.com/libp2p/go-libp2p-record"
|
|
"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
|
|
type IndexerService struct {
|
|
*common.LongLivedStreamRecordedService[PeerRecord]
|
|
PS *pubsub.PubSub
|
|
DHT *dht.IpfsDHT
|
|
isStrictIndexer bool
|
|
mu sync.RWMutex
|
|
DisposedPeers map[peer.ID]*common.TopicNodeActivityPub
|
|
|
|
SeenQueries map[string]bool
|
|
SeenMU sync.Mutex
|
|
}
|
|
|
|
// if a pubsub is given... indexer is also an active oc-node. If not... your a strict indexer
|
|
func NewIndexerService(h host.Host, ps *pubsub.PubSub, maxNode int) *IndexerService {
|
|
logger := oclib.GetLogger()
|
|
logger.Info().Msg("open indexer mode...")
|
|
var err error
|
|
ix := &IndexerService{
|
|
LongLivedStreamRecordedService: common.NewStreamRecordedService[PeerRecord](h, maxNode, false),
|
|
isStrictIndexer: ps == nil,
|
|
}
|
|
if ps == nil { // generate your fresh gossip for the flow of killed node... EVERYBODY should know !
|
|
ps, err = pubsub.NewGossipSub(context.Background(), ix.Host)
|
|
if err != nil {
|
|
panic(err) // can't run your indexer without a propalgation pubsub, of state of node.
|
|
}
|
|
}
|
|
ix.PS = ps
|
|
// later TODO : all indexer laucnh a private replica of them self. DEV OPS
|
|
if ix.isStrictIndexer {
|
|
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...")
|
|
|
|
logger.Info().Msg("subscribe to decentralized search flow as strict indexer...")
|
|
ix.SubscribeToSearch(ix.PS, nil)
|
|
}
|
|
if ix.DHT, err = dht.New(
|
|
context.Background(),
|
|
ix.Host,
|
|
dht.Mode(dht.ModeServer),
|
|
dht.Validator(record.NamespacedValidator{
|
|
"node": PeerRecordValidator{},
|
|
}),
|
|
); err != nil {
|
|
return nil
|
|
}
|
|
ix.initNodeHandler() // then listen up on every protocol expected
|
|
return ix
|
|
}
|
|
|
|
func (ix *IndexerService) Close() {
|
|
ix.DHT.Close()
|
|
ix.PS.UnregisterTopicValidator(common.TopicPubSubSearch)
|
|
for _, s := range ix.StreamRecords {
|
|
for _, ss := range s {
|
|
ss.HeartbeatStream.Stream.Close()
|
|
}
|
|
}
|
|
}
|