From f6ab41b829ac391aab6e5705ecfa57627795e7c9 Mon Sep 17 00:00:00 2001
From: Alexander Sporn <github@alexsporn.de>
Date: Wed, 14 Sep 2022 11:45:13 +0200
Subject: [PATCH] Removed OpenedStream/ClosedStream since they were removed
 from network.Notifee

---
 pkg/p2p/manager.go             |  2 --
 pkg/protocol/gossip/service.go | 40 ----------------------------------
 2 files changed, 42 deletions(-)

diff --git a/pkg/p2p/manager.go b/pkg/p2p/manager.go
index 648a2a4f2..9807c129f 100644
--- a/pkg/p2p/manager.go
+++ b/pkg/p2p/manager.go
@@ -1099,5 +1099,3 @@ func (m *netNotifiee) Disconnected(net network.Network, conn network.Conn) {
 	}
 	m.disconnectedChan <- &disconnectmsg{net: net, conn: conn, reason: errors.New("connection closed by libp2p network event")}
 }
-func (m *netNotifiee) OpenedStream(_ network.Network, _ network.Stream) {}
-func (m *netNotifiee) ClosedStream(_ network.Network, _ network.Stream) {}
diff --git a/pkg/protocol/gossip/service.go b/pkg/protocol/gossip/service.go
index 9573de8fe..856a772cc 100644
--- a/pkg/protocol/gossip/service.go
+++ b/pkg/protocol/gossip/service.go
@@ -9,7 +9,6 @@ import (
 	"github.com/libp2p/go-libp2p/core/network"
 	"github.com/libp2p/go-libp2p/core/peer"
 	"github.com/libp2p/go-libp2p/core/protocol"
-	"github.com/multiformats/go-multiaddr"
 	"github.com/pkg/errors"
 
 	"github.com/iotaledger/hive.go/core/events"
@@ -176,7 +175,6 @@ type Service struct {
 	connectedChan       chan *connectionmsg
 	closeStreamChan     chan *closestreammsg
 	disconnectedChan    chan *connectionmsg
-	streamClosedChan    chan *streamclosedmsg
 	relationUpdatedChan chan *relationupdatedmsg
 	streamReqChan       chan *streamreqmsg
 	forEachChan         chan *foreachmsg
@@ -224,7 +222,6 @@ func NewService(
 		connectedChan:       make(chan *connectionmsg, 10),
 		closeStreamChan:     make(chan *closestreammsg, 10),
 		disconnectedChan:    make(chan *connectionmsg, 10),
-		streamClosedChan:    make(chan *streamclosedmsg, 10),
 		relationUpdatedChan: make(chan *relationupdatedmsg, 10),
 		streamReqChan:       make(chan *streamreqmsg, 10),
 		forEachChan:         make(chan *foreachmsg, 10),
@@ -304,17 +301,11 @@ func (s *Service) Start(ctx context.Context) {
 		s.inboundStreamChan <- stream
 	})
 
-	// manage libp2p network events
-	s.host.Network().Notify((*netNotifiee)(s))
-
 	s.eventLoop(ctx)
 
 	// libp2p stream handler
 	s.host.RemoveStreamHandler(s.protocol)
 
-	// de-register libp2p network events
-	s.host.Network().StopNotify((*netNotifiee)(s))
-
 	s.detachEvents()
 	s.peeringMngWP.Stop()
 }
@@ -338,8 +329,6 @@ drainLoop:
 
 		case <-s.disconnectedChan:
 
-		case <-s.streamClosedChan:
-
 		case <-s.relationUpdatedChan:
 
 		case streamReqMsg := <-s.streamReqChan:
@@ -369,11 +358,6 @@ type streamreqmsg struct {
 	back   chan *Protocol
 }
 
-type streamclosedmsg struct {
-	peerID peer.ID
-	stream network.Stream
-}
-
 type relationupdatedmsg struct {
 	peer        *p2p.Peer
 	oldRelation p2p.PeerRelation
@@ -410,11 +394,6 @@ func (s *Service) eventLoop(ctx context.Context) {
 				s.Events.Error.Trigger(err)
 			}
 
-		case streamClosedMsg := <-s.streamClosedChan:
-			if err := s.deregisterProtocol(streamClosedMsg.peerID); err != nil && !errors.Is(err, ErrProtocolDoesNotExist) {
-				s.Events.Error.Trigger(err)
-			}
-
 		case relationUpdatedMsg := <-s.relationUpdatedChan:
 			s.handleRelationUpdated(ctx, relationUpdatedMsg.peer, relationUpdatedMsg.oldRelation)
 
@@ -729,22 +708,3 @@ func (s *Service) detachEvents() {
 	s.Events.InboundStreamCanceled.Detach(s.onGossipServiceInboundStreamCanceled)
 	s.Events.Error.Detach(s.onGossipServiceError)
 }
-
-// lets Service implement network.Notifiee in order to automatically
-// clean up ongoing reset streams.
-type netNotifiee Service
-
-func (m *netNotifiee) Listen(net network.Network, multiaddr multiaddr.Multiaddr)      {}
-func (m *netNotifiee) ListenClose(net network.Network, multiaddr multiaddr.Multiaddr) {}
-func (m *netNotifiee) Connected(net network.Network, conn network.Conn)               {}
-func (m *netNotifiee) Disconnected(net network.Network, conn network.Conn)            {}
-func (m *netNotifiee) OpenedStream(net network.Network, stream network.Stream)        {}
-func (m *netNotifiee) ClosedStream(net network.Network, stream network.Stream) {
-	if stream.Protocol() != m.protocol {
-		return
-	}
-	if m.stopped.IsSet() {
-		return
-	}
-	m.streamClosedChan <- &streamclosedmsg{peerID: stream.Conn().RemotePeer(), stream: stream}
-}