LogalDeveloper/owncast
0
Fork 0
Code Activity

Fix possible crash for concurrent map writes

Browse Source
This commit is contained in:
Gabe Kangas
2020-07-22 23:09:11 -07:00
parent 515661804a
commit 43df6c432e
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
core/stats.go
View File

@@ -5,6 +5,7 @@ import (
"io/ioutil" "io/ioutil"
"math" "math"
"os" "os"
"sync"
"time" "time"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
@@ -18,6 +19,8 @@ const (
statsFilePath = "stats.json" statsFilePath = "stats.json"
) )
var l = sync.Mutex{}
func setupStats() error { func setupStats() error {
s, err := getSavedStats() s, err := getSavedStats()
if err != nil { if err != nil {
@@ -82,7 +85,9 @@ func SetClientActive(clientID string) {
// fmt.Println("Marking client active:", clientID, s.GetViewerCount()+1, "clients connected.") // fmt.Println("Marking client active:", clientID, s.GetViewerCount()+1, "clients connected.")
// } // }
l.Lock()
_stats.Clients[clientID] = time.Now() _stats.Clients[clientID] = time.Now()
l.Unlock()
_stats.SessionMaxViewerCount = int(math.Max(float64(len(_stats.Clients)), float64(_stats.SessionMaxViewerCount))) _stats.SessionMaxViewerCount = int(math.Max(float64(len(_stats.Clients)), float64(_stats.SessionMaxViewerCount)))
_stats.OverallMaxViewerCount = int(math.Max(float64(_stats.SessionMaxViewerCount), float64(_stats.OverallMaxViewerCount))) _stats.OverallMaxViewerCount = int(math.Max(float64(_stats.SessionMaxViewerCount), float64(_stats.OverallMaxViewerCount)))
} }