146 lines
3.9 KiB
Go
146 lines
3.9 KiB
Go
package core
|
|
|
|
import (
|
|
"math"
|
|
"sync"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/owncast/owncast/core/data"
|
|
"github.com/owncast/owncast/models"
|
|
"github.com/owncast/owncast/services/geoip"
|
|
)
|
|
|
|
var (
|
|
l = &sync.RWMutex{}
|
|
_activeViewerPurgeTimeout = time.Second * 15
|
|
_geoIPClient = geoip.NewClient()
|
|
)
|
|
|
|
func setupStats() error {
|
|
s := getSavedStats()
|
|
_stats = &s
|
|
|
|
statsSaveTimer := time.NewTicker(1 * time.Minute)
|
|
go func() {
|
|
for range statsSaveTimer.C {
|
|
saveStats()
|
|
}
|
|
}()
|
|
|
|
viewerCountPruneTimer := time.NewTicker(5 * time.Second)
|
|
go func() {
|
|
for range viewerCountPruneTimer.C {
|
|
pruneViewerCount()
|
|
}
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
// IsStreamConnected checks if the stream is connected or not.
|
|
func IsStreamConnected() bool {
|
|
if !_stats.StreamConnected {
|
|
return false
|
|
}
|
|
|
|
// Kind of a hack. It takes a handful of seconds between a RTMP connection and when HLS data is available.
|
|
// So account for that with an artificial buffer of four segments.
|
|
timeSinceLastConnected := time.Since(_stats.LastConnectTime.Time).Seconds()
|
|
waitTime := math.Max(float64(data.GetStreamLatencyLevel().SecondsPerSegment)*3.0, 7)
|
|
if timeSinceLastConnected < waitTime {
|
|
return false
|
|
}
|
|
|
|
return _stats.StreamConnected
|
|
}
|
|
|
|
// RemoveChatClient removes a client from the active clients record.
|
|
func RemoveChatClient(clientID string) {
|
|
log.Trace("Removing the client:", clientID)
|
|
|
|
l.Lock()
|
|
delete(_stats.ChatClients, clientID)
|
|
l.Unlock()
|
|
}
|
|
|
|
// SetViewerActive sets a client as active and connected.
|
|
func SetViewerActive(viewer *models.Viewer) {
|
|
// Don't update viewer counts if a live stream session is not active.
|
|
if !_stats.StreamConnected {
|
|
return
|
|
}
|
|
|
|
l.Lock()
|
|
defer l.Unlock()
|
|
|
|
// Asynchronously, optionally, fetch GeoIP data.
|
|
go func(viewer *models.Viewer) {
|
|
viewer.Geo = _geoIPClient.GetGeoFromIP(viewer.IPAddress)
|
|
}(viewer)
|
|
|
|
if _, exists := _stats.Viewers[viewer.ClientID]; exists {
|
|
_stats.Viewers[viewer.ClientID].LastSeen = time.Now()
|
|
} else {
|
|
_stats.Viewers[viewer.ClientID] = viewer
|
|
}
|
|
_stats.SessionMaxViewerCount = int(math.Max(float64(len(_stats.Viewers)), float64(_stats.SessionMaxViewerCount)))
|
|
_stats.OverallMaxViewerCount = int(math.Max(float64(_stats.SessionMaxViewerCount), float64(_stats.OverallMaxViewerCount)))
|
|
}
|
|
|
|
// GetActiveViewers will return the active viewers.
|
|
func GetActiveViewers() map[string]*models.Viewer {
|
|
return _stats.Viewers
|
|
}
|
|
|
|
func pruneViewerCount() {
|
|
viewers := make(map[string]*models.Viewer)
|
|
|
|
l.Lock()
|
|
defer l.Unlock()
|
|
|
|
for viewerID, viewer := range _stats.Viewers {
|
|
viewerLastSeenTime := _stats.Viewers[viewerID].LastSeen
|
|
if time.Since(viewerLastSeenTime) < _activeViewerPurgeTimeout {
|
|
viewers[viewerID] = viewer
|
|
}
|
|
}
|
|
|
|
_stats.Viewers = viewers
|
|
}
|
|
|
|
func saveStats() {
|
|
if err := data.SetPeakOverallViewerCount(_stats.OverallMaxViewerCount); err != nil {
|
|
log.Errorln("error saving viewer count", err)
|
|
}
|
|
if err := data.SetPeakSessionViewerCount(_stats.SessionMaxViewerCount); err != nil {
|
|
log.Errorln("error saving viewer count", err)
|
|
}
|
|
if _stats.LastDisconnectTime != nil && _stats.LastDisconnectTime.Valid {
|
|
if err := data.SetLastDisconnectTime(_stats.LastDisconnectTime.Time); err != nil {
|
|
log.Errorln("error saving disconnect time", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func getSavedStats() models.Stats {
|
|
savedLastDisconnectTime, _ := data.GetLastDisconnectTime()
|
|
|
|
result := models.Stats{
|
|
ChatClients: make(map[string]models.Client),
|
|
Viewers: make(map[string]*models.Viewer),
|
|
SessionMaxViewerCount: data.GetPeakSessionViewerCount(),
|
|
OverallMaxViewerCount: data.GetPeakOverallViewerCount(),
|
|
LastDisconnectTime: savedLastDisconnectTime,
|
|
}
|
|
|
|
// If the stats were saved > 5min ago then ignore the
|
|
// peak session count value, since the session is over.
|
|
if result.LastDisconnectTime == nil || !result.LastDisconnectTime.Valid || time.Since(result.LastDisconnectTime.Time).Minutes() > 5 {
|
|
result.SessionMaxViewerCount = 0
|
|
}
|
|
|
|
return result
|
|
}
|