184 lines
3.6 KiB
Go
184 lines
3.6 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 `json:"completed"`
|
|
Joined uint64 `json:"joined"`
|
|
Left uint64 `json:"left"`
|
|
Reaped uint64 `json:"reaped"`
|
|
|
|
// Stats for seeds only (subset of total).
|
|
SeedsJoined uint64 `json:"seeds_joined"`
|
|
SeedsLeft uint64 `json:"seeds_left"`
|
|
SeedsReaped uint64 `json:"seeds_reaped"`
|
|
}
|
|
|
|
type Stats struct {
|
|
Start time.Time `json:"start_time"`
|
|
|
|
Announces uint64 `json:"announces"`
|
|
Scrapes uint64 `json:"scrapes"`
|
|
|
|
IPv4Peers PeerStats `json:"ipv4_peers"`
|
|
IPv6Peers PeerStats `json:"ipv6_peers"`
|
|
|
|
TorrentsAdded uint64 `json:"torrents_added"`
|
|
TorrentsRemoved uint64 `json:"torrents_removed"`
|
|
TorrentsReaped uint64 `json:"torrents_reaped"`
|
|
|
|
OpenConnections uint64 `json:"open_connections"`
|
|
ConnectionsAccepted uint64 `json:"connections_accepted"`
|
|
BytesTransmitted uint64 `json:"bytes_transmitted"`
|
|
|
|
RequestsHandled uint64 `json:"requests_handled"`
|
|
RequestsErrored uint64 `json:"requests_errored"`
|
|
|
|
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++
|
|
s.IPv4Peers.Joined++
|
|
case DeletedSeedIPv4:
|
|
s.IPv4Peers.SeedsLeft++
|
|
s.IPv4Peers.Left++
|
|
case ReapedSeedIPv4:
|
|
s.IPv4Peers.SeedsReaped++
|
|
s.IPv4Peers.Reaped++
|
|
|
|
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++
|
|
s.IPv6Peers.Joined++
|
|
case DeletedSeedIPv6:
|
|
s.IPv6Peers.SeedsLeft++
|
|
s.IPv6Peers.Left++
|
|
case ReapedSeedIPv6:
|
|
s.IPv6Peers.SeedsReaped++
|
|
s.IPv6Peers.Reaped++
|
|
|
|
case NewTorrent:
|
|
s.TorrentsAdded++
|
|
case DeletedTorrent:
|
|
s.TorrentsRemoved++
|
|
case ReapedTorrent:
|
|
s.TorrentsReaped++
|
|
|
|
case AcceptedConnection:
|
|
s.ConnectionsAccepted++
|
|
s.OpenConnections++
|
|
|
|
case ClosedConnection:
|
|
s.OpenConnections--
|
|
|
|
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)
|
|
}
|