search

daemons/node/stream/service.go

@@ -51,22 +51,24 @@ var protocolsPartners = map[protocol.ID]*common.ProtocolInfo{
 }
 
 type StreamService struct {
-	Key          pp.ID
-	Host         host.Host
-	Node         common.DiscoveryPeer
-	Streams      common.ProtocolStream
-	maxNodesConn int
-	Mu           sync.RWMutex
+	Key             pp.ID
+	Host            host.Host
+	Node            common.DiscoveryPeer
+	Streams         common.ProtocolStream
+	maxNodesConn    int
+	Mu              sync.RWMutex
+	ResourceSearches *common.SearchTracker
 }
 
 func InitStream(ctx context.Context, h host.Host, key pp.ID, maxNode int, node common.DiscoveryPeer) (*StreamService, error) {
 	logger := oclib.GetLogger()
 	service := &StreamService{
-		Key:          key,
-		Node:         node,
-		Host:         h,
-		Streams:      common.ProtocolStream{},
-		maxNodesConn: maxNode,
+		Key:              key,
+		Node:             node,
+		Host:             h,
+		Streams:          common.ProtocolStream{},
+		maxNodesConn:     maxNode,
+		ResourceSearches: common.NewSearchTracker(),
 	}
 	for proto := range protocols {
 		service.Host.SetStreamHandler(proto, service.HandleResponse)