update tracking

This commit is contained in:
Fran Jurmanović
2025-05-26 20:55:22 +02:00
parent edf5a2c8c4
commit e52894c663
5 changed files with 186 additions and 58 deletions

View File

@@ -0,0 +1,26 @@
package regexHandler
import (
"acc-server-manager/local/model"
"regexp"
)
type AccServerInstance struct {
Model *model.Server
State *model.ServerState
}
type RegexHandler struct {
regex *regexp.Regexp
}
func (rh *RegexHandler) Contains(line string, callback func(...string)) {
match := rh.regex.FindStringSubmatch(line)
callback(match...)
}
func New(str string) *RegexHandler {
return &RegexHandler{
regex: regexp.MustCompile(str),
}
}

View File

@@ -1,11 +1,80 @@
package tracking
import (
"acc-server-manager/local/model"
"acc-server-manager/local/utl/regexHandler"
"bufio"
"os"
"strconv"
"strings"
"time"
)
type StateChange int
const (
PlayerCount StateChange = iota
Session
)
var StateChanges = map[StateChange]string {
PlayerCount: "player-count",
Session: "session",
}
type AccServerInstance struct {
Model *model.Server
State *model.ServerState
OnStateChange func(...StateChange)
}
func NewAccServerInstance(server *model.Server, onStateChange func(...StateChange)) *AccServerInstance {
return &AccServerInstance{
Model: server,
State: &model.ServerState{PlayerCount: 0},
OnStateChange: onStateChange,
}
}
type StateRegexHandler struct {
*regexHandler.RegexHandler
test string
}
func NewRegexHandler(str string, test string) *StateRegexHandler {
return &StateRegexHandler{
RegexHandler: regexHandler.New(str),
test: test,
}
}
func (rh *StateRegexHandler) Test(line string) bool{
return strings.Contains(line, rh.test)
}
func (rh *StateRegexHandler) Count(line string) int{
var count int = 0
rh.Contains(line, func (strs ...string) {
if len(strs) == 2 {
if ct, err := strconv.Atoi(strs[1]); err == nil {
count = ct
}
}
})
return count
}
func (rh *StateRegexHandler) Change(line string) (string, string){
var old string = ""
var new string = ""
rh.Contains(line, func (strs ...string) {
if len(strs) == 3 {
old = strs[1]
new = strs[2]
}
})
return old, new
}
func TailLogFile(path string, callback func(string)) {
file, _ := os.Open(path)
defer file.Close()
@@ -21,4 +90,85 @@ func TailLogFile(path string, callback func(string)) {
time.Sleep(500 * time.Millisecond) // wait for new data
}
}
}
type LogStateType int
const (
SessionChange LogStateType = iota
LeaderboardUpdate
UDPCount
ClientsOnline
)
var logStateContain = map[LogStateType]string {
SessionChange: "Session changed",
LeaderboardUpdate: "Updated leaderboard for",
UDPCount: "Udp message count",
ClientsOnline: "client(s) online",
}
var sessionChangeRegex = NewRegexHandler(`Session changed: (\w+) -> (\w+)`, logStateContain[SessionChange])
var leaderboardUpdateRegex = NewRegexHandler(`Updated leaderboard for (\d+) clients`, logStateContain[LeaderboardUpdate])
var udpCountRegex = NewRegexHandler(`Udp message count ((\d+) client`, logStateContain[UDPCount])
var clientsOnlineRegex = NewRegexHandler(`(\d+) client(s) online`, logStateContain[ClientsOnline])
var logStateRegex = map[LogStateType]*StateRegexHandler {
SessionChange: sessionChangeRegex,
LeaderboardUpdate: leaderboardUpdateRegex,
UDPCount: udpCountRegex,
ClientsOnline: clientsOnlineRegex,
}
func (instance *AccServerInstance) HandleLogLine(line string) {
for logState, regexHandler := range logStateRegex {
if (regexHandler.Test(line)) {
switch logState {
case LeaderboardUpdate:
case UDPCount:
case ClientsOnline:
count := regexHandler.Count(line)
instance.UpdatePlayerCount(count)
case SessionChange:
_, new := regexHandler.Change(line)
instance.UpdateSessionChange(new)
}
}
}
}
func (instance *AccServerInstance) UpdateState(callback func(state *model.ServerState)) {
state := instance.State
state.Lock()
defer state.Unlock()
callback(state)
}
func (instance *AccServerInstance) UpdatePlayerCount(count int) {
changes := []StateChange{}
instance.UpdateState(func (state *model.ServerState) {
if (count > 0 && state.PlayerCount == 0) {
state.SessionStart = time.Now()
changes = append(changes, Session)
} else if (count == 0) {
state.SessionStart = time.Time{}
changes = append(changes, Session)
}
state.PlayerCount = count
changes = append(changes, PlayerCount)
})
instance.OnStateChange(changes...)
}
func (instance *AccServerInstance) UpdateSessionChange(session string) {
changes := []StateChange{}
instance.UpdateState(func (state *model.ServerState) {
if (state.PlayerCount > 0) {
state.SessionStart = time.Now()
} else {
state.SessionStart = time.Time{}
}
state.Session = session
changes = append(changes, Session)
})
instance.OnStateChange(changes...)
}