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) }) }