65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
package cron
|
|
|
|
import (
|
|
"context"
|
|
"emperror.dev/errors"
|
|
"github.com/apex/log"
|
|
"github.com/goccy/go-json"
|
|
"github.com/pterodactyl/wings/internal/database"
|
|
"github.com/pterodactyl/wings/server"
|
|
"github.com/pterodactyl/wings/system"
|
|
"github.com/xujiajun/nutsdb"
|
|
)
|
|
|
|
var processing system.AtomicBool
|
|
|
|
func processActivityLogs(m *server.Manager) error {
|
|
// Don't execute this cron if there is currently one running. Once this task is completed
|
|
// go ahead and mark it as no longer running.
|
|
if !processing.SwapIf(true) {
|
|
return nil
|
|
}
|
|
defer processing.Store(false)
|
|
|
|
var list [][]byte
|
|
err := database.DB().View(func(tx *nutsdb.Tx) error {
|
|
// Grab the oldest 100 activity events that have been logged and send them back to the
|
|
// Panel for processing. Once completed, delete those events from the database and then
|
|
// release the lock on this process.
|
|
l, err := tx.LRange(database.ServerActivityBucket, []byte("events"), 0, 1)
|
|
if err != nil {
|
|
// This error is returned when the bucket doesn't exist, which is likely on the
|
|
// first invocations of Wings since we haven't yet logged any data. There is nothing
|
|
// that needs to be done if this error occurs.
|
|
if errors.Is(err, nutsdb.ErrBucket) {
|
|
return nil
|
|
}
|
|
return errors.WithStackIf(err)
|
|
}
|
|
list = l
|
|
return nil
|
|
})
|
|
|
|
if err != nil || len(list) == 0 {
|
|
return errors.WithStackIf(err)
|
|
}
|
|
|
|
var processed []json.RawMessage
|
|
for _, l := range list {
|
|
var v json.RawMessage
|
|
if err := json.Unmarshal(l, &v); err != nil {
|
|
log.WithField("error", errors.WithStack(err)).Warn("failed to parse activity event json, skipping entry")
|
|
continue
|
|
}
|
|
processed = append(processed, v)
|
|
}
|
|
|
|
if err := m.Client().SendActivityLogs(context.Background(), processed); err != nil {
|
|
return errors.WrapIf(err, "cron: failed to send activity events to Panel")
|
|
}
|
|
|
|
return database.DB().Update(func(tx *nutsdb.Tx) error {
|
|
return tx.LTrim(database.ServerActivityBucket, []byte("events"), 2, -1)
|
|
})
|
|
}
|