tracker/stats/stats.go

252 lines
5.7 KiB
Go
Raw Normal View History

// Copyright 2014 The Chihaya Authors. All rights reserved.
// Use of this source code is governed by the BSD 2-Clause license,
// which can be found in the LICENSE file.
// Package stats implements a means of tracking processing statistics for a
// BitTorrent tracker.
package stats
import "time"
const (
Announce = iota
Scrape
2014-07-21 17:42:05 +02:00
CompletedIPv4
NewLeechIPv4
DeletedLeechIPv4
ReapedLeechIPv4
NewSeedIPv4
DeletedSeedIPv4
ReapedSeedIPv4
CompletedIPv6
NewLeechIPv6
DeletedLeechIPv6
ReapedLeechIPv6
NewSeedIPv6
DeletedSeedIPv6
ReapedSeedIPv6
NewTorrent
DeletedTorrent
2014-07-21 17:42:05 +02:00
ReapedTorrent
AcceptedConnection
ClosedConnection
HandledRequest
ErroredRequest
2014-07-22 20:57:36 +02:00
ResponseTime
)
2014-07-22 01:08:08 +02:00
// DefaultStats is a default instance of stats tracking that uses an unbuffered
// channel for broadcasting events.
var DefaultStats *Stats
func init() {
DefaultStats = New(0)
}
2014-07-21 17:42:05 +02:00
type PeerStats struct {
// Stats for all peers.
2014-07-23 05:29:30 +02:00
Completed uint64 `json:"completed"` // Number of transitions from leech to seed.
Joined uint64 `json:"joined"` // Total peers that announced.
Left uint64 `json:"left"` // Total peers that paused or stopped.
Reaped uint64 `json:"reaped"` // Total peers cleaned up after inactivity.
Current uint64 `json:"current"` // Current total peer count.
2014-07-21 17:42:05 +02:00
// Stats for seeds only (subset of total).
2014-07-23 05:29:30 +02:00
SeedsJoined uint64 `json:"seeds_joined"` // Seeds that announced (does not included leechers that completed).
SeedsLeft uint64 `json:"seeds_left"` // Seeds that paused or stopped.
SeedsReaped uint64 `json:"seeds_reaped"` // Seeds cleaned up after inactivity.
SeedsCurrent uint64 `json:"seeds_current"` // Current seed count.
2014-07-21 17:42:05 +02:00
}
2014-07-22 20:57:36 +02:00
type PercentileTimes struct {
P50 *Percentile
P90 *Percentile
P95 *Percentile
}
type Stats struct {
2014-07-23 05:29:30 +02:00
Start time.Time `json:"start_time"` // Time at which Chihaya was booted.
2014-07-21 17:42:05 +02:00
2014-07-23 05:29:30 +02:00
Announces uint64 `json:"announces"` // Total number of announces.
Scrapes uint64 `json:"scrapes"` // Total number of scrapes.
2014-07-21 17:42:05 +02:00
2014-07-22 07:12:41 +02:00
IPv4Peers PeerStats `json:"ipv4_peers"`
IPv6Peers PeerStats `json:"ipv6_peers"`
2014-07-21 17:42:05 +02:00
2014-07-22 07:12:41 +02:00
TorrentsAdded uint64 `json:"torrents_added"`
TorrentsRemoved uint64 `json:"torrents_removed"`
TorrentsReaped uint64 `json:"torrents_reaped"`
2014-07-21 17:42:05 +02:00
OpenConnections uint64 `json:"open_connections"`
2014-07-22 07:12:41 +02:00
ConnectionsAccepted uint64 `json:"connections_accepted"`
BytesTransmitted uint64 `json:"bytes_transmitted"`
2014-07-21 17:42:05 +02:00
2014-07-22 07:12:41 +02:00
RequestsHandled uint64 `json:"requests_handled"`
RequestsErrored uint64 `json:"requests_errored"`
2014-07-22 20:57:36 +02:00
ResponseTime PercentileTimes `json:"response_time"`
events chan int
responseTimeEvents chan time.Duration
}
func New(chanSize int) *Stats {
s := &Stats{
Start: time.Now(),
events: make(chan int, chanSize),
2014-07-22 20:57:36 +02:00
responseTimeEvents: make(chan time.Duration, chanSize),
ResponseTime: PercentileTimes{
P50: NewPercentile(0.5),
P90: NewPercentile(0.9),
P95: NewPercentile(0.95),
2014-07-22 20:57:36 +02:00
},
}
go s.handleEvents()
2014-07-22 20:57:36 +02:00
go s.handleTimings()
return s
}
func (s *Stats) Close() {
close(s.events)
}
func (s *Stats) Uptime() time.Duration {
return time.Since(s.Start)
}
func (s *Stats) RecordEvent(event int) {
s.events <- event
}
2014-07-22 18:42:56 +02:00
func (s *Stats) RecordTiming(event int, duration time.Duration) {
2014-07-22 20:57:36 +02:00
switch event {
case ResponseTime:
s.responseTimeEvents <- duration
default:
panic("stats: RecordTiming called with an unknown event")
}
2014-07-22 18:42:56 +02:00
}
func (s *Stats) handleEvents() {
for event := range s.events {
switch event {
case Announce:
s.Announces++
case Scrape:
s.Scrapes++
2014-07-21 17:42:05 +02:00
case CompletedIPv4:
s.IPv4Peers.Completed++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.SeedsCurrent++
2014-07-21 17:42:05 +02:00
case NewLeechIPv4:
s.IPv4Peers.Joined++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.Current++
2014-07-21 17:42:05 +02:00
case DeletedLeechIPv4:
s.IPv4Peers.Left++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.Current--
2014-07-21 17:42:05 +02:00
case ReapedLeechIPv4:
s.IPv4Peers.Reaped++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.Current--
2014-07-21 17:42:05 +02:00
case NewSeedIPv4:
s.IPv4Peers.SeedsJoined++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.SeedsCurrent++
s.IPv4Peers.Joined++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.Current++
2014-07-21 17:42:05 +02:00
case DeletedSeedIPv4:
s.IPv4Peers.SeedsLeft++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.SeedsCurrent--
s.IPv4Peers.Left++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.Current--
2014-07-21 17:42:05 +02:00
case ReapedSeedIPv4:
s.IPv4Peers.SeedsReaped++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.SeedsCurrent--
s.IPv4Peers.Reaped++
2014-07-23 05:29:30 +02:00
s.IPv4Peers.Current--
2014-07-21 17:42:05 +02:00
case CompletedIPv6:
s.IPv6Peers.Completed++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.SeedsCurrent++
2014-07-21 17:42:05 +02:00
case NewLeechIPv6:
s.IPv6Peers.Joined++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.Current++
2014-07-21 17:42:05 +02:00
case DeletedLeechIPv6:
s.IPv6Peers.Left++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.Current--
2014-07-21 17:42:05 +02:00
case ReapedLeechIPv6:
s.IPv6Peers.Reaped++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.Current--
2014-07-21 17:42:05 +02:00
case NewSeedIPv6:
s.IPv6Peers.SeedsJoined++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.SeedsCurrent++
s.IPv6Peers.Joined++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.Current++
2014-07-21 17:42:05 +02:00
case DeletedSeedIPv6:
s.IPv6Peers.SeedsLeft++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.SeedsCurrent--
s.IPv6Peers.Left++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.Current--
2014-07-21 17:42:05 +02:00
case ReapedSeedIPv6:
s.IPv6Peers.SeedsReaped++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.SeedsCurrent--
s.IPv6Peers.Reaped++
2014-07-23 05:29:30 +02:00
s.IPv6Peers.Current--
2014-07-21 17:42:05 +02:00
case NewTorrent:
s.TorrentsAdded++
case DeletedTorrent:
s.TorrentsRemoved++
case ReapedTorrent:
s.TorrentsReaped++
2014-07-21 17:42:05 +02:00
case AcceptedConnection:
s.ConnectionsAccepted++
s.OpenConnections++
2014-07-21 17:42:05 +02:00
case ClosedConnection:
s.OpenConnections--
2014-07-21 17:42:05 +02:00
case HandledRequest:
s.RequestsHandled++
2014-07-21 17:42:05 +02:00
case ErroredRequest:
s.RequestsErrored++
default:
panic("stats: RecordEvent called with an unknown event")
}
}
}
2014-07-22 01:08:08 +02:00
2014-07-22 20:57:36 +02:00
func (s *Stats) handleTimings() {
for {
select {
case duration := <-s.responseTimeEvents:
f := float64(duration) / float64(time.Millisecond)
s.ResponseTime.P50.AddSample(f)
s.ResponseTime.P90.AddSample(f)
s.ResponseTime.P95.AddSample(f)
}
}
}
2014-07-22 18:42:56 +02:00
// RecordEvent broadcasts an event to the default stats queue.
2014-07-22 01:08:08 +02:00
func RecordEvent(event int) {
DefaultStats.RecordEvent(event)
}
2014-07-22 18:42:56 +02:00
// RecordTiming broadcasts a timing event to the default stats queue.
func RecordTiming(event int, duration time.Duration) {
DefaultStats.RecordTiming(event, duration)
}