wings/internal/cron/activity_cron.go

87 lines
2.4 KiB
Go
Raw Normal View History

package cron
import (
"context"
"net"
2022-10-05 23:24:11 +00:00
"emperror.dev/errors"
2022-10-06 15:58:42 +00:00
2022-07-24 15:43:48 +00:00
"github.com/pterodactyl/wings/internal/database"
"github.com/pterodactyl/wings/internal/models"
"github.com/pterodactyl/wings/server"
"github.com/pterodactyl/wings/system"
)
2022-07-10 20:51:11 +00:00
type activityCron struct {
mu *system.AtomicBool
manager *server.Manager
max int
2022-07-10 20:51:11 +00:00
}
// Run executes the cronjob and ensures we fetch and send all the stored activity to the
2022-07-10 20:51:11 +00:00
// Panel instance. Once activity is sent it is deleted from the local database instance. Any
// SFTP specific events are not handled in this cron, they're handled separately to account
2022-07-10 20:51:11 +00:00
// for de-duplication and event merging.
func (ac *activityCron) Run(ctx context.Context) 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.
2022-07-10 20:51:11 +00:00
if !ac.mu.SwapIf(true) {
return errors.WithStack(ErrCronRunning)
}
2022-07-10 20:51:11 +00:00
defer ac.mu.Store(false)
2022-07-24 15:43:48 +00:00
var activity []models.Activity
tx := database.Instance().WithContext(ctx).
Where("event NOT LIKE ?", "server:sftp.%").
Limit(ac.max).
2022-07-24 15:43:48 +00:00
Find(&activity)
if tx.Error != nil {
return errors.WithStack(tx.Error)
2022-07-10 20:51:11 +00:00
}
2022-07-24 15:43:48 +00:00
if len(activity) == 0 {
2022-07-10 20:51:11 +00:00
return nil
}
2022-07-24 15:43:48 +00:00
// ids to delete from the database.
ids := make([]int, 0, len(activity))
// activities to send to the panel.
activities := make([]models.Activity, 0, len(activity))
for _, v := range activity {
// Delete any activity that has an invalid IP address. This is a fix for
// a bug that truncated the last octet of an IPv6 address in the database.
2023-01-30 16:09:36 +00:00
if ip := net.ParseIP(v.IP); ip == nil {
ids = append(ids, v.ID)
continue
}
activities = append(activities, v)
}
if len(ids) > 0 {
tx = database.Instance().WithContext(ctx).Where("id IN ?", ids).Delete(&models.Activity{})
if tx.Error != nil {
return errors.WithStack(tx.Error)
}
}
if len(activities) == 0 {
return nil
}
if err := ac.manager.Client().SendActivityLogs(ctx, activities); err != nil {
return errors.WrapIf(err, "cron: failed to send activity events to Panel")
}
// Add all the successful activities to the list of IDs to delete.
ids = make([]int, len(activities))
for i, v := range activities {
ids[i] = v.ID
2022-07-24 15:43:48 +00:00
}
// Delete all the activities that were sent to the Panel (or that were invalid).
2022-07-24 15:43:48 +00:00
tx = database.Instance().WithContext(ctx).Where("id IN ?", ids).Delete(&models.Activity{})
if tx.Error != nil {
return errors.WithStack(tx.Error)
2022-07-10 20:51:11 +00:00
}
2022-07-24 15:43:48 +00:00
return nil
}