worse-conns.go

   1 package torrent
   2
   3 import (
   4         "container/heap"
   5         "fmt"
   6         "unsafe"
   7
   8         "github.com/anacrolix/multiless"
   9 )
  10
  11 func worseConn(l, r *Peer) bool {
  12         less, ok := multiless.New().Bool(
  13                 l.useful(), r.useful()).CmpInt64(
  14                 l.lastHelpful().Sub(r.lastHelpful()).Nanoseconds()).CmpInt64(
  15                 l.completedHandshake.Sub(r.completedHandshake).Nanoseconds()).LazySameLess(
  16                 func() (same, less bool) {
  17                         lpp, err := l.peerPriority()
  18                         if err != nil {
  19                                 same = true
  20                                 return
  21                         }
  22                         rpp, err := r.peerPriority()
  23                         if err != nil {
  24                                 same = true
  25                                 return
  26                         }
  27                         return lpp == rpp, lpp < rpp
  28                 }).Uintptr(
  29                 uintptr(unsafe.Pointer(l)), uintptr(unsafe.Pointer(r)),
  30         ).LessOk()
  31         if !ok {
  32                 panic(fmt.Sprintf("cannot differentiate %#v and %#v", l, r))
  33         }
  34         return less
  35 }
  36
  37 type worseConnSlice struct {
  38         conns []*PeerConn
  39 }
  40
  41 var _ heap.Interface = &worseConnSlice{}
  42
  43 func (me worseConnSlice) Len() int {
  44         return len(me.conns)
  45 }
  46
  47 func (me worseConnSlice) Less(i, j int) bool {
  48         return worseConn(&me.conns[i].Peer, &me.conns[j].Peer)
  49 }
  50
  51 func (me *worseConnSlice) Pop() interface{} {
  52         i := len(me.conns) - 1
  53         ret := me.conns[i]
  54         me.conns = me.conns[:i]
  55         return ret
  56 }
  57
  58 func (me *worseConnSlice) Push(x interface{}) {
  59         me.conns = append(me.conns, x.(*PeerConn))
  60 }
  61
  62 func (me worseConnSlice) Swap(i, j int) {
  63         me.conns[i], me.conns[j] = me.conns[j], me.conns[i]
  64 }