Track peers supporting extension in a nested expvar.Map

[btrtrc.git] / global.go

diff --git a/global.go b/global.go
index 6542ac393daaf324d9b2427db851b559d707aea1..ff237c8ee1edda4f9d464d08d9db1bb49c7fa670 100644 (file)
--- a/global.go
+++ b/global.go
@@ -24,10 +24,15 @@ func defaultPeerExtensionBytes() PeerExtensionBits {
        return pp.NewPeerExtensionBytes(pp.ExtensionBitDHT, pp.ExtensionBitExtended, pp.ExtensionBitFast)
 }
 
+func init() {
+       torrent.Set("peers supporting extension", &peersSupportingExtension)
+}
+
 // I could move a lot of these counters to their own file, but I suspect they
 // may be attached to a Client someday.
 var (
-       torrent = expvar.NewMap("torrent")
+       torrent                  = expvar.NewMap("torrent")
+       peersSupportingExtension expvar.Map
 
        pieceHashedCorrect    = expvar.NewInt("pieceHashedCorrect")
        pieceHashedNotCorrect = expvar.NewInt("pieceHashedNotCorrect")