176 lines
3.1 KiB
Go
176 lines
3.1 KiB
Go
// 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
|
|
|
|
CompletedIPv4
|
|
NewLeechIPv4
|
|
DeletedLeechIPv4
|
|
ReapedLeechIPv4
|
|
NewSeedIPv4
|
|
DeletedSeedIPv4
|
|
ReapedSeedIPv4
|
|
|
|
CompletedIPv6
|
|
NewLeechIPv6
|
|
DeletedLeechIPv6
|
|
ReapedLeechIPv6
|
|
NewSeedIPv6
|
|
DeletedSeedIPv6
|
|
ReapedSeedIPv6
|
|
|
|
NewTorrent
|
|
DeletedTorrent
|
|
ReapedTorrent
|
|
|
|
AcceptedConnection
|
|
ClosedConnection
|
|
|
|
HandledRequest
|
|
ErroredRequest
|
|
)
|
|
|
|
// DefaultStats is a default instance of stats tracking that uses an unbuffered
|
|
// channel for broadcasting events.
|
|
var DefaultStats *Stats
|
|
|
|
func init() {
|
|
DefaultStats = New(0)
|
|
}
|
|
|
|
type PeerStats struct {
|
|
// Stats for all peers.
|
|
Completed uint64
|
|
Joined uint64
|
|
Left uint64
|
|
Reaped uint64
|
|
|
|
// Stats for seeds only (subset of total).
|
|
SeedsJoined uint64
|
|
SeedsLeft uint64
|
|
SeedsReaped uint64
|
|
}
|
|
|
|
type Stats struct {
|
|
Start time.Time
|
|
|
|
Announces uint64
|
|
Scrapes uint64
|
|
|
|
IPv4Peers PeerStats
|
|
IPv6Peers PeerStats
|
|
|
|
TorrentsAdded uint64
|
|
TorrentsRemoved uint64
|
|
TorrentsReaped uint64
|
|
|
|
ActiveConnections uint64
|
|
ConnectionsAccepted uint64
|
|
BytesTransmitted uint64
|
|
|
|
RequestsHandled uint64
|
|
RequestsErrored uint64
|
|
|
|
events chan int
|
|
}
|
|
|
|
func New(chanSize int) *Stats {
|
|
s := &Stats{
|
|
Start: time.Now(),
|
|
events: make(chan int, chanSize),
|
|
}
|
|
|
|
go s.handleEvents()
|
|
|
|
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
|
|
}
|
|
|
|
func (s *Stats) handleEvents() {
|
|
for event := range s.events {
|
|
switch event {
|
|
case Announce:
|
|
s.Announces++
|
|
case Scrape:
|
|
s.Scrapes++
|
|
|
|
case CompletedIPv4:
|
|
s.IPv4Peers.Completed++
|
|
case NewLeechIPv4:
|
|
s.IPv4Peers.Joined++
|
|
case DeletedLeechIPv4:
|
|
s.IPv4Peers.Left++
|
|
case ReapedLeechIPv4:
|
|
s.IPv4Peers.Reaped++
|
|
case NewSeedIPv4:
|
|
s.IPv4Peers.SeedsJoined++
|
|
case DeletedSeedIPv4:
|
|
s.IPv4Peers.SeedsLeft++
|
|
case ReapedSeedIPv4:
|
|
s.IPv4Peers.SeedsReaped++
|
|
|
|
case CompletedIPv6:
|
|
s.IPv6Peers.Completed++
|
|
case NewLeechIPv6:
|
|
s.IPv6Peers.Joined++
|
|
case DeletedLeechIPv6:
|
|
s.IPv6Peers.Left++
|
|
case ReapedLeechIPv6:
|
|
s.IPv6Peers.Reaped++
|
|
case NewSeedIPv6:
|
|
s.IPv6Peers.SeedsJoined++
|
|
case DeletedSeedIPv6:
|
|
s.IPv6Peers.SeedsLeft++
|
|
case ReapedSeedIPv6:
|
|
s.IPv6Peers.SeedsReaped++
|
|
|
|
case NewTorrent:
|
|
s.TorrentsAdded++
|
|
case DeletedTorrent:
|
|
s.TorrentsRemoved++
|
|
case ReapedTorrent:
|
|
s.TorrentsReaped++
|
|
|
|
case AcceptedConnection:
|
|
s.ConnectionsAccepted++
|
|
s.ActiveConnections++
|
|
|
|
case ClosedConnection:
|
|
s.ActiveConnections--
|
|
|
|
case HandledRequest:
|
|
s.RequestsHandled++
|
|
|
|
case ErroredRequest:
|
|
s.RequestsErrored++
|
|
|
|
default:
|
|
panic("stats: RecordEvent called with an unknown event")
|
|
}
|
|
}
|
|
}
|
|
|
|
// RecordEvent broadcasts an event to the default stats tracking.
|
|
func RecordEvent(event int) {
|
|
DefaultStats.RecordEvent(event)
|
|
}
|