1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495 |
- package irc
-
- import (
- "sync"
- )
-
- type StatsValues struct {
- Unknown int // unregistered clients
- Total int // registered clients, including invisible
- Max int // high-water mark of registered clients
- Invisible int
- Operators int
- }
-
- // Stats tracks statistics for a running server
- type Stats struct {
- StatsValues
-
- mutex sync.Mutex
- }
-
- // Adds an unregistered client
- func (s *Stats) Add() {
- s.mutex.Lock()
- s.Unknown += 1
- s.mutex.Unlock()
- }
-
- // Activates a registered client, e.g., for the initial attach to a persistent client
- func (s *Stats) AddRegistered(invisible, operator bool) {
- s.mutex.Lock()
- if invisible {
- s.Invisible += 1
- }
- if operator {
- s.Operators += 1
- }
- s.Total += 1
- s.setMax()
- s.mutex.Unlock()
- }
-
- // Transition a client from unregistered to registered
- func (s *Stats) Register() {
- s.mutex.Lock()
- s.Unknown -= 1
- s.Total += 1
- s.setMax()
- s.mutex.Unlock()
- }
-
- func (s *Stats) setMax() {
- if s.Max < s.Total {
- s.Max = s.Total
- }
- }
-
- // Modify the Invisible count
- func (s *Stats) ChangeInvisible(increment int) {
- s.mutex.Lock()
- s.Invisible += increment
- s.mutex.Unlock()
- }
-
- // Modify the Operator count
- func (s *Stats) ChangeOperators(increment int) {
- s.mutex.Lock()
- s.Operators += increment
- s.mutex.Unlock()
- }
-
- // Remove a user from the server
- func (s *Stats) Remove(registered, invisible, operator bool) {
- s.mutex.Lock()
- if registered {
- s.Total -= 1
- } else {
- s.Unknown -= 1
- }
- if invisible {
- s.Invisible -= 1
- }
- if operator {
- s.Operators -= 1
- }
- s.mutex.Unlock()
- }
-
- // GetStats retrives total, invisible and oper count
- func (s *Stats) GetValues() (result StatsValues) {
- s.mutex.Lock()
- result = s.StatsValues
- s.mutex.Unlock()
- return
- }
|