Add PeerConnClosed callback

[btrtrc.git] / peerconn.go

diff --git a/peerconn.go b/peerconn.go
index cc8bdeb8580319332205d3f1ef0d7b369c6c56dc..e13e9fa3735578b389bb2acb75aab9a2954dface 100644 (file)
--- a/peerconn.go
+++ b/peerconn.go
@@ -355,6 +355,9 @@ func (cn *PeerConn) _close() {
        if cn.conn != nil {
                cn.conn.Close()
        }
+       if cb := cn.t.cl.config.Callbacks.PeerConnClosed; cb != nil {
+               cb(cn)
+       }
 }
 
 func (cn *peer) peerHasPiece(piece pieceIndex) bool {