LBRYCommunity/tracker
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

storage: add prometheus to memory implementation

Browse source
This commit is contained in:
Leo Balduf 2017-01-25 18:59:10 +01:00
parent 8b1ab73894
commit 18d7e5d51b
1 changed files with 38 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
storage/memory/peer_store.go
View file

@ -9,11 +9,38 @@ import (
"time" "time"
log "github.com/Sirupsen/logrus" log "github.com/Sirupsen/logrus"
"github.com/prometheus/client_golang/prometheus"
"github.com/chihaya/chihaya/bittorrent" "github.com/chihaya/chihaya/bittorrent"
"github.com/chihaya/chihaya/storage" "github.com/chihaya/chihaya/storage"
) )
func init() {
prometheus.MustRegister(promGCDurationMilliseconds)
prometheus.MustRegister(promInfohashesCount)
}
var promGCDurationMilliseconds = prometheus.NewHistogram(prometheus.HistogramOpts{
Name: "chihaya_storage_gc_duration_milliseconds",
Help: "The time it takes to perform storage garbage collection",
Buckets: prometheus.ExponentialBuckets(9.375, 2, 10),
})
var promInfohashesCount = prometheus.NewGauge(prometheus.GaugeOpts{
Name: "chihaya_storage_infohashes_count",
Help: "The number of Infohashes tracked",
})
// recordGCDuration records the duration of a GC sweep.
func recordGCDuration(duration time.Duration) {
promGCDurationMilliseconds.Observe(float64(duration.Nanoseconds()) / float64(time.Millisecond))
}
// recordInfohashesDelta records a change in the number of Infohashes tracked.
func recordInfohashesDelta(delta float64) {
promInfohashesCount.Add(delta)
}
// ErrInvalidGCInterval is returned for a GarbageCollectionInterval that is // ErrInvalidGCInterval is returned for a GarbageCollectionInterval that is
// less than or equal to zero. // less than or equal to zero.
var ErrInvalidGCInterval = errors.New("invalid garbage collection interval") var ErrInvalidGCInterval = errors.New("invalid garbage collection interval")
@ -136,6 +163,7 @@ func (s *peerStore) PutSeeder(ih bittorrent.InfoHash, p bittorrent.Peer) error {
seeders: make(map[serializedPeer]int64), seeders: make(map[serializedPeer]int64),
leechers: make(map[serializedPeer]int64), leechers: make(map[serializedPeer]int64),
} }
recordInfohashesDelta(1)
} }
shard.swarms[ih].seeders[pk] = time.Now().UnixNano() shard.swarms[ih].seeders[pk] = time.Now().UnixNano()
@ -170,6 +198,7 @@ func (s *peerStore) DeleteSeeder(ih bittorrent.InfoHash, p bittorrent.Peer) erro
if len(shard.swarms[ih].seeders)|len(shard.swarms[ih].leechers) == 0 { if len(shard.swarms[ih].seeders)|len(shard.swarms[ih].leechers) == 0 {
delete(shard.swarms, ih) delete(shard.swarms, ih)
recordInfohashesDelta(-1)
} }
shard.Unlock() shard.Unlock()
@ -193,6 +222,7 @@ func (s *peerStore) PutLeecher(ih bittorrent.InfoHash, p bittorrent.Peer) error
seeders: make(map[serializedPeer]int64), seeders: make(map[serializedPeer]int64),
leechers: make(map[serializedPeer]int64), leechers: make(map[serializedPeer]int64),
} }
recordInfohashesDelta(1)
} }
shard.swarms[ih].leechers[pk] = time.Now().UnixNano() shard.swarms[ih].leechers[pk] = time.Now().UnixNano()
@ -227,6 +257,7 @@ func (s *peerStore) DeleteLeecher(ih bittorrent.InfoHash, p bittorrent.Peer) err
if len(shard.swarms[ih].seeders)|len(shard.swarms[ih].leechers) == 0 { if len(shard.swarms[ih].seeders)|len(shard.swarms[ih].leechers) == 0 {
delete(shard.swarms, ih) delete(shard.swarms, ih)
recordInfohashesDelta(-1)
} }
shard.Unlock() shard.Unlock()
@ -250,6 +281,7 @@ func (s *peerStore) GraduateLeecher(ih bittorrent.InfoHash, p bittorrent.Peer) e
seeders: make(map[serializedPeer]int64), seeders: make(map[serializedPeer]int64),
leechers: make(map[serializedPeer]int64), leechers: make(map[serializedPeer]int64),
} }
recordInfohashesDelta(1)
} }
delete(shard.swarms[ih].leechers, pk) delete(shard.swarms[ih].leechers, pk)
@ -356,7 +388,9 @@ func (s *peerStore) collectGarbage(cutoff time.Time) error {
default: default:
} }
var ihDelta float64
cutoffUnix := cutoff.UnixNano() cutoffUnix := cutoff.UnixNano()
start := time.Now()
for _, shard := range s.shards { for _, shard := range s.shards {
shard.RLock() shard.RLock()
var infohashes []bittorrent.InfoHash var infohashes []bittorrent.InfoHash
@ -389,6 +423,7 @@ func (s *peerStore) collectGarbage(cutoff time.Time) error {
if len(shard.swarms[ih].seeders)|len(shard.swarms[ih].leechers) == 0 { if len(shard.swarms[ih].seeders)|len(shard.swarms[ih].leechers) == 0 {
delete(shard.swarms, ih) delete(shard.swarms, ih)
ihDelta--
} }
shard.Unlock() shard.Unlock()
@ -398,6 +433,9 @@ func (s *peerStore) collectGarbage(cutoff time.Time) error {
runtime.Gosched() runtime.Gosched()
} }
recordGCDuration(time.Since(start))
recordInfohashesDelta(ihDelta)
return nil return nil
} }