mirror of
https://github.com/ergochat/ergo.git
synced 2024-12-23 11:12:44 +01:00
0241e0c31d
Previously, we were applying defaults before the user had completed registration. This meant that the number of invisible users was incremented when the user connected, and then the total was incremented when registration was completed. Now both counters are updated at the same time. If a user disconnects prior to registration, +i has not yet been applied so it would not be decremented.
99 lines
1.7 KiB
Go
99 lines
1.7 KiB
Go
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(invisible bool) {
|
|
s.mutex.Lock()
|
|
s.Unknown -= 1
|
|
if invisible {
|
|
s.Invisible += 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
|
|
}
|