Ava Affine
e7d229c217
0. tests for event replay Tests are now included for the event replay features. This includes many fixes on top of the last commit as well as some tweaks to config module values. 1. activity module An activity module is created to track the useractive events and provide a counter for them. It also encapsulates logic to discard old useractive events. 2. web module A web module is created. This module serves a static webpage showing runtime information. Currently it only shows a snapshot of the user activity data. It is my intention that it eventually also shows an audit log, known users and channels, uptime, and more. Future work will also be needed in order to use HTML templating so that it doesn't look so... basic. Live updates to the information may also be desired. Signed-off-by: Ava Affine <ava@sunnypup.io>
106 lines
2.3 KiB
Go
106 lines
2.3 KiB
Go
package activity
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
"time"
|
|
|
|
"gitlab.com/whom/bingobot/internal/config"
|
|
"gitlab.com/whom/bingobot/internal/state"
|
|
"gitlab.com/whom/bingobot/internal/logging"
|
|
)
|
|
|
|
/* Activity module
|
|
* This module sits and processes incoming
|
|
*/
|
|
|
|
const (
|
|
ActivityModuleStartFail = "failed to start activity module"
|
|
)
|
|
|
|
var currentUserActivity map[string][]state.UserActiveEvent
|
|
var userActivityLock sync.RWMutex
|
|
|
|
func Start() error {
|
|
ch, err := state.UserActive.Subscribe()
|
|
if err != nil {
|
|
return errors.Join(
|
|
errors.New(ActivityModuleStartFail),
|
|
err,
|
|
)
|
|
}
|
|
|
|
// process incoming events loop
|
|
go func() {
|
|
for {
|
|
ev := <- ch
|
|
emap := ev.Data()
|
|
user := emap[state.UserEventUserKey]
|
|
etime := ev.Time()
|
|
delta := time.Since(etime).Hours() / float64(24)
|
|
|
|
if delta <= float64(config.Get().UserEventLifespanDays) {
|
|
new := []state.UserActiveEvent{ev.(state.UserActiveEvent)}
|
|
userActivityLock.Lock()
|
|
current, found := currentUserActivity[user]
|
|
if found {
|
|
new = append(new, current...)
|
|
}
|
|
userActivityLock.Unlock()
|
|
} else {
|
|
logging.Warn("recieved expired useractive event")
|
|
}
|
|
}
|
|
}()
|
|
|
|
// process expired events loop
|
|
go func() {
|
|
for {
|
|
delta := time.Hour * 24
|
|
delta *= time.Duration(config.Get().UserEventLifespanDays)
|
|
tcur := time.Now().Add(delta)
|
|
|
|
// get next soonest expiration
|
|
userActivityLock.RLock()
|
|
for _, evs := range currentUserActivity {
|
|
for _, ev := range evs {
|
|
hrs := time.Duration(24 * config.Get().UserEventLifespanDays)
|
|
t := ev.Time().Add(time.Hour * hrs)
|
|
if t.Before(tcur) {
|
|
tcur = t
|
|
}
|
|
}
|
|
}
|
|
userActivityLock.RUnlock()
|
|
time.Sleep(tcur.Sub(time.Now()))
|
|
|
|
userActivityLock.Lock()
|
|
for k, v := range currentUserActivity {
|
|
new := []state.UserActiveEvent{}
|
|
for _, ev := range v {
|
|
if !ev.Disposable() {
|
|
new = append(new, ev)
|
|
}
|
|
}
|
|
currentUserActivity[k] = new
|
|
}
|
|
userActivityLock.Unlock()
|
|
}
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
func GetActivitySnapshot() map[string][]state.UserActiveEvent {
|
|
userActivityLock.RLock()
|
|
defer userActivityLock.RUnlock()
|
|
|
|
snapshot := make(map[string][]state.UserActiveEvent)
|
|
for k, v := range currentUserActivity {
|
|
newSl := make([]state.UserActiveEvent, len(v))
|
|
copy(newSl, v)
|
|
snapshot[k] = newSl
|
|
}
|
|
|
|
return snapshot
|
|
}
|