Relay reservations

notify.go

@@ -51,4 +51,19 @@ func (n *RelayNotifiee) Connected(s inet.Network, c inet.Conn) {
 	}(c.RemotePeer())
 }
 
-func (n *RelayNotifiee) Disconnected(s inet.Network, c inet.Conn) {}
+func (n *RelayNotifiee) Disconnected(s inet.Network, c inet.Conn) {
+	r := n.Relay()
+	p := c.RemotePeer()
+
+	if r.host.Network().Connectedness(p) == inet.Connected {
+		// we have multiple connections
+		return
+	}
+
+	r.mx.Lock()
+	defer r.mx.Unlock()
+	if _, ok := r.rsvp[p]; ok {
+		delete(r.rsvp, p)
+		r.host.ConnManager().UntagPeer(p, "relay")
+	}
+}