website/pkg/worker/barertc.go
2023-06-10 13:25:06 -07:00

101 lines
2.3 KiB
Go

package worker
import (
"encoding/json"
"io/ioutil"
"net/http"
"sync"
"time"
"code.nonshy.com/nonshy/website/pkg/config"
"code.nonshy.com/nonshy/website/pkg/log"
)
// ChatStatistics is the json result of the BareRTC /api/statistics endpoint.
type ChatStatistics struct {
UserCount int
Usernames []string
Cameras struct {
Blue int
Red int
}
}
// GetChatStatistics returns the latest (cached) chat statistics.
func GetChatStatistics() ChatStatistics {
chatStatisticsMu.RLock()
defer chatStatisticsMu.RUnlock()
if cachedChatStatistics != nil {
return *cachedChatStatistics
}
return ChatStatistics{
Usernames: []string{},
}
}
var (
cachedChatStatistics *ChatStatistics
chatStatisticsMu sync.RWMutex
)
// WatchBareRTC is a worker goroutine that caches the current online chatters in the chat room.
func WatchBareRTC() {
if config.Current.BareRTC.JWTSecret == "" || config.Current.BareRTC.URL == "" {
log.Error("Worker (WatchBareRTC): chat room is not configured, will not watch chat room status")
return
}
// Check it immediately.
DoCheckBareRTC()
// And on an interval forever.
for {
time.Sleep(config.ChatStatusRefreshInterval)
DoCheckBareRTC()
}
}
// DoCheckBareRTC invokes the attempt to refresh data from the chat server about who's online.
func DoCheckBareRTC() {
log.Info("Refresh BareRTC")
req, err := http.NewRequest(http.MethodGet, config.Current.BareRTC.URL+"/api/statistics", nil)
// Lock the cached statistics.
chatStatisticsMu.Lock()
defer chatStatisticsMu.Unlock()
if err != nil {
log.Error("WatchBareRTC: couldn't make request: %s", err)
cachedChatStatistics = nil
return
}
client := http.Client{
Timeout: 10 * time.Second,
}
res, err := client.Do(req)
if err != nil {
log.Error("WatchBareRTC: request error: %s", err)
cachedChatStatistics = nil
return
} else if res.StatusCode != http.StatusOK {
log.Error("WatchBareRTC: didn't get expected 200 OK from statistics endpoint, instead got: %s", res.Status)
cachedChatStatistics = nil
return
}
if res.StatusCode == http.StatusOK {
var cs ChatStatistics
body, _ := ioutil.ReadAll(res.Body)
res.Body.Close()
if err = json.Unmarshal(body, &cs); err != nil {
log.Error("WatchBareRTC: json decode error: %s", err)
return
}
cachedChatStatistics = &cs
}
}