Project restructure (#18)

* First pass at restructuring the project; untested but it does compile * Restructure builds and runs 🎉 * Add the dist folder to the gitignore * Update core/playlist/monitor.go * golint and reorganize the monitor.go file Co-authored-by: Gabe Kangas <gabek@real-ity.com>
2020-06-22 20:11:56 -05:00
parent b0768de6c0
commit 487bd12444
42 changed files with 1309 additions and 1000 deletions
--- a/core/stats.go
+++ b/core/stats.go
@@ -0,0 +1,135 @@
+package core
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"math"
+	"os"
+	"time"
+
+	log "github.com/sirupsen/logrus"
+
+	"github.com/gabek/owncast/models"
+	"github.com/gabek/owncast/utils"
+)
+
+const (
+	statsFilePath = "stats.json"
+)
+
+func setupStats() error {
+	s, err := getSavedStats()
+	if err != nil {
+		return err
+	}
+
+	_stats = &s
+
+	statsSaveTimer := time.NewTicker(1 * time.Minute)
+	go func() {
+		for {
+			select {
+			case <-statsSaveTimer.C:
+				if err := saveStatsToFile(); err != nil {
+					panic(err)
+				}
+			}
+		}
+	}()
+
+	staleViewerPurgeTimer := time.NewTicker(3 * time.Second)
+	go func() {
+		for {
+			select {
+			case <-staleViewerPurgeTimer.C:
+				purgeStaleViewers()
+			}
+		}
+	}()
+
+	return nil
+}
+
+func purgeStaleViewers() {
+	for clientID, lastConnectedtime := range _stats.Clients {
+		timeSinceLastActive := time.Since(lastConnectedtime).Minutes()
+		if timeSinceLastActive > 2 {
+			RemoveClient(clientID)
+		}
+	}
+}
+
+//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.
+	timeSinceLastConnected := time.Since(_stats.LastConnectTime).Seconds()
+	if timeSinceLastConnected < 10 {
+		return false
+	}
+
+	return _stats.StreamConnected
+}
+
+//SetClientActive sets a client as active and connected
+func SetClientActive(clientID string) {
+	// if _, ok := s.clients[clientID]; !ok {
+	// 	fmt.Println("Marking client active:", clientID, s.GetViewerCount()+1, "clients connected.")
+	// }
+
+	_stats.Clients[clientID] = time.Now()
+	_stats.SessionMaxViewerCount = int(math.Max(float64(len(_stats.Clients)), float64(_stats.SessionMaxViewerCount)))
+	_stats.OverallMaxViewerCount = int(math.Max(float64(_stats.SessionMaxViewerCount), float64(_stats.OverallMaxViewerCount)))
+}
+
+//RemoveClient removes a client from the active clients record
+func RemoveClient(clientID string) {
+	log.Println("Removing the client:", clientID)
+
+	delete(_stats.Clients, clientID)
+}
+
+func saveStatsToFile() error {
+	jsonData, err := json.Marshal(_stats)
+	if err != nil {
+		return err
+	}
+
+	f, err := os.Create(statsFilePath)
+	if err != nil {
+		return err
+	}
+
+	defer f.Close()
+
+	if _, err := f.Write(jsonData); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func getSavedStats() (models.Stats, error) {
+	result := models.Stats{
+		Clients: make(map[string]time.Time),
+	}
+
+	if !utils.DoesFileExists(statsFilePath) {
+		return result, nil
+	}
+
+	jsonFile, err := ioutil.ReadFile(statsFilePath)
+	if err != nil {
+		return result, nil
+	}
+
+	if err := json.Unmarshal(jsonFile, &result); err != nil {
+		return result, nil
+	}
+
+	return result, nil
+}