Skip to content
This repository has been archived by the owner on Sep 6, 2022. It is now read-only.

Commit

Permalink
remove OpenedStream and ClosedStream from Notifiee interface (#250)
Browse files Browse the repository at this point in the history
* remove TODO for PeerConnected and PeerDisconnected from Notifiee

This is now done via the event bus.

* remove OpenedStream and ClosedStream from Notifiee
  • Loading branch information
marten-seemann committed May 24, 2022
1 parent 02cbdcc commit 13e0150
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 61 deletions.
25 changes: 0 additions & 25 deletions network/notifee.go
Expand Up @@ -11,12 +11,6 @@ type Notifiee interface {
ListenClose(Network, ma.Multiaddr) // called when network stops listening on an addr
Connected(Network, Conn) // called when a connection opened
Disconnected(Network, Conn) // called when a connection closed
OpenedStream(Network, Stream) // called when a stream opened
ClosedStream(Network, Stream) // called when a stream closed

// TODO
// PeerConnected(Network, peer.ID) // called when a peer connected
// PeerDisconnected(Network, peer.ID) // called when a peer disconnected
}

// NotifyBundle implements Notifiee by calling any of the functions set on it,
Expand All @@ -28,9 +22,6 @@ type NotifyBundle struct {

ConnectedF func(Network, Conn)
DisconnectedF func(Network, Conn)

OpenedStreamF func(Network, Stream)
ClosedStreamF func(Network, Stream)
}

var _ Notifiee = (*NotifyBundle)(nil)
Expand Down Expand Up @@ -63,20 +54,6 @@ func (nb *NotifyBundle) Disconnected(n Network, c Conn) {
}
}

// OpenedStream calls OpenedStreamF if it is not null.
func (nb *NotifyBundle) OpenedStream(n Network, s Stream) {
if nb.OpenedStreamF != nil {
nb.OpenedStreamF(n, s)
}
}

// ClosedStream calls ClosedStreamF if it is not null.
func (nb *NotifyBundle) ClosedStream(n Network, s Stream) {
if nb.ClosedStreamF != nil {
nb.ClosedStreamF(n, s)
}
}

// Global noop notifiee. Do not change.
var GlobalNoopNotifiee = &NoopNotifiee{}

Expand All @@ -88,5 +65,3 @@ func (nn *NoopNotifiee) Connected(n Network, c Conn) {}
func (nn *NoopNotifiee) Disconnected(n Network, c Conn) {}
func (nn *NoopNotifiee) Listen(n Network, addr ma.Multiaddr) {}
func (nn *NoopNotifiee) ListenClose(n Network, addr ma.Multiaddr) {}
func (nn *NoopNotifiee) OpenedStream(Network, Stream) {}
func (nn *NoopNotifiee) ClosedStream(Network, Stream) {}
36 changes: 0 additions & 36 deletions network/notifee_test.go
Expand Up @@ -85,39 +85,3 @@ func TestDisconnected(T *testing.T) {
T.Fatal("Disconnected should have been called")
}
}

func TestOpenedStream(T *testing.T) {
var notifee NotifyBundle
notifee.OpenedStream(nil, nil)

called := false
notifee.OpenedStreamF = func(Network, Stream) {
called = true
}
if called {
T.Fatal("called should be false")
}

notifee.OpenedStream(nil, nil)
if !called {
T.Fatal("OpenedStream should have been called")
}
}

func TestClosedStream(T *testing.T) {
var notifee NotifyBundle
notifee.ClosedStream(nil, nil)

called := false
notifee.ClosedStreamF = func(Network, Stream) {
called = true
}
if called {
T.Fatal("called should be false")
}

notifee.ClosedStream(nil, nil)
if !called {
T.Fatal("ClosedStream should have been called")
}
}

0 comments on commit 13e0150

Please sign in to comment.