Account for the root

[rbdr/mobius] / hotline / stats.go

diff --git a/hotline/stats.go b/hotline/stats.go
index 4b9119d6a21aa60516fc9af8e3c2e323d4d7400b..316a67df9429e675e740b447b6cbf2df458b557d 100644 (file)
--- a/hotline/stats.go
+++ b/hotline/stats.go
@@ -1,34 +1,96 @@
 package hotline
 
 import (
-       "fmt"
+       "sync"
        "time"
 )
 
+// Stat counter keys
+const (
+       StatCurrentlyConnected = iota
+       StatDownloadsInProgress
+       StatUploadsInProgress
+       StatWaitingDownloads
+       StatConnectionPeak
+       StatConnectionCounter
+       StatDownloadCounter
+       StatUploadCounter
+)
+
+type Counter interface {
+       Increment(keys ...int)
+       Decrement(key int)
+       Set(key, val int)
+       Get(key int) int
+       Values() map[string]interface{}
+}
+
 type Stats struct {
-       LoginCount      int       `yaml:"login count"`
-       StartTime       time.Time `yaml:"start time"`
-       DownloadCounter int
-       UploadCounter   int
-}
-
-func (s *Stats) String() string {
-       template := `
-Server Stats:
-  Start Time:          %v
-  Uptime:                      %s
-  Login Count: %v
-`
-       d := time.Since(s.StartTime)
-       d = d.Round(time.Minute)
-       h := d / time.Hour
-       d -= h * time.Hour
-       m := d / time.Minute
-
-       return fmt.Sprintf(
-               template,
-               s.StartTime.Format(time.RFC1123Z),
-               fmt.Sprintf("%02d:%02d", h, m),
-               s.LoginCount,
-       )
+       stats map[int]int
+       since time.Time
+
+       mu sync.RWMutex
+}
+
+func NewStats() *Stats {
+       return &Stats{
+               since: time.Now(),
+               stats: map[int]int{
+                       StatCurrentlyConnected:  0,
+                       StatDownloadsInProgress: 0,
+                       StatUploadsInProgress:   0,
+                       StatWaitingDownloads:    0,
+                       StatConnectionPeak:      0,
+                       StatDownloadCounter:     0,
+                       StatUploadCounter:       0,
+                       StatConnectionCounter:   0,
+               },
+       }
+}
+
+func (s *Stats) Increment(keys ...int) {
+       s.mu.Lock()
+       defer s.mu.Unlock()
+
+       for _, key := range keys {
+               s.stats[key]++
+       }
+}
+
+func (s *Stats) Decrement(key int) {
+       s.mu.Lock()
+       defer s.mu.Unlock()
+
+       s.stats[key]--
+}
+
+func (s *Stats) Set(key, val int) {
+       s.mu.Lock()
+       defer s.mu.Unlock()
+
+       s.stats[key] = val
+}
+
+func (s *Stats) Get(key int) int {
+       s.mu.RLock()
+       defer s.mu.RUnlock()
+
+       return s.stats[key]
+}
+
+func (s *Stats) Values() map[string]interface{} {
+       s.mu.RLock()
+       defer s.mu.RUnlock()
+
+       return map[string]interface{}{
+               "CurrentlyConnected":  s.stats[StatCurrentlyConnected],
+               "DownloadsInProgress": s.stats[StatDownloadsInProgress],
+               "UploadsInProgress":   s.stats[StatUploadsInProgress],
+               "WaitingDownloads":    s.stats[StatWaitingDownloads],
+               "ConnectionPeak":      s.stats[StatConnectionPeak],
+               "ConnectionCounter":   s.stats[StatConnectionCounter],
+               "DownloadCounter":     s.stats[StatDownloadCounter],
+               "UploadCounter":       s.stats[StatUploadCounter],
+               "Since":               s.since,
+       }
 }