fifos/top-seed

diff --git a/cmd/btrtrc/USAGE b/cmd/btrtrc/USAGE
index 40d58782a38d62031bd4ddd9c4bd5dfaa41177f8..8800d9b368061b370eff3bec68c95f90798eb057 100644 (file)
--- a/cmd/btrtrc/USAGE
+++ b/cmd/btrtrc/USAGE
@@ -46,6 +46,7 @@ fifos subdirectory will be created with following FIFO files:
   * amount of downloaded/uploaded traffic during current session
   * remote address with port
   * client's name
+* fifos/top-seed -- list torrents sorted by total transfer data amount
 
 For each torrent, corresponding .torrent file will be created.
 Additional symbolic link with torrent's name will lead to HASH.torrent.

diff --git a/cmd/btrtrc/fifos.go b/cmd/btrtrc/fifos.go
index cb6139c8a7bd7023be0df3189fd01e3ea7fa4bc3..c7466926bdf662a61adab449eaa55316c6071046 100644 (file)
--- a/cmd/btrtrc/fifos.go
+++ b/cmd/btrtrc/fifos.go
@@ -223,6 +223,47 @@ func fifoDHTList(c *torrent.Client) {
        }
 }
 
+type topTorrent struct {
+       infoHash metainfo.Hash
+       name     string
+       tx       int64
+}
+
+func fifoTopSeed(c *torrent.Client) {
+       pth := path.Join(FIFOsDir, "top-seed")
+       recreateFIFO(pth)
+       for {
+               fd, err := os.OpenFile(pth, os.O_WRONLY|os.O_APPEND, os.FileMode(0666))
+               if err != nil {
+                       log.Println("OpenFile:", pth, err)
+                       time.Sleep(time.Second)
+                       continue
+               }
+               var ts []*topTorrent
+               for _, t := range c.Torrents() {
+                       if t.Info() == nil {
+                               continue
+                       }
+                       ts = append(ts, &topTorrent{
+                               infoHash: t.InfoHash(),
+                               name:     t.Name(),
+                               tx:       TxStats[t.InfoHash()],
+                       })
+               }
+               sort.Sort(ByTxTraffic(ts))
+               for _, t := range ts {
+                       fmt.Fprintf(fd,
+                               "%s%s%s %s%40s%s %s\n",
+                               Blue, t.infoHash.HexString(), Reset,
+                               Green, shortenName(t.name), Reset,
+                               humanize.IBytes(uint64(t.tx)),
+                       )
+               }
+               fd.Close()
+               time.Sleep(time.Second)
+       }
+}
+
 type stringAddr string
 
 func (stringAddr) Network() string   { return "" }
@@ -358,6 +399,7 @@ func fifosCleanup() {
        os.Remove(path.Join(FIFOsDir, "dht"))
        os.Remove(path.Join(FIFOsDir, "add"))
        os.Remove(path.Join(FIFOsDir, "del"))
+       os.Remove(path.Join(FIFOsDir, "top-seed"))
        os.RemoveAll(path.Join(FIFOsDir, PeersDir))
        os.RemoveAll(path.Join(FIFOsDir, FilesDir))
 }

diff --git a/cmd/btrtrc/main.go b/cmd/btrtrc/main.go
index 880348be67ec56af8ffd1c05e187c6da9280656f..63e6c4fff24a18c9c4e0d2cba14ad5019eae61d6 100644 (file)
--- a/cmd/btrtrc/main.go
+++ b/cmd/btrtrc/main.go
@@ -81,6 +81,7 @@ func main() {
        Jobs.Add(1)
        go overallStatus(client)
        go fifoList(client)
+       go fifoTopSeed(client)
        go fifoDHTList(client)
        go fifoAdd(client)
        go fifoDel(client)

diff --git a/cmd/btrtrc/sort.go b/cmd/btrtrc/sort.go
index cfb6478c68eda6be163cc026075db62998cdd5af..5b1ad858ff88de7c1e6a006bf0de92281ad1a2ce 100644 (file)
--- a/cmd/btrtrc/sort.go
+++ b/cmd/btrtrc/sort.go
@@ -20,6 +20,20 @@ func (a ByInfoHash) Less(i, j int) bool {
        return a[i].InfoHash().HexString() < a[j].InfoHash().HexString()
 }
 
+type ByTxTraffic []*topTorrent
+
+func (a ByTxTraffic) Len() int {
+       return len(a)
+}
+
+func (a ByTxTraffic) Swap(i, j int) {
+       a[i], a[j] = a[j], a[i]
+}
+
+func (a ByTxTraffic) Less(i, j int) bool {
+       return a[i].tx < a[j].tx
+}
+
 type ByPeerID []*torrent.PeerConn
 
 func (a ByPeerID) Len() int {