feat: add garbagecollection

Author: n3rdc4ptn

internal/runnable/UsageRunnable.go

@@ -2,6 +2,8 @@ package runnable
 
 import (
 	"context"
+	"errors"
+	"fmt"
 	"time"
 
 	"sigs.k8s.io/controller-runtime/pkg/client"
@@ -47,11 +49,17 @@ func (u *UsageRunnable) Start(ctx context.Context) error {
 	}
 }
 
-func (u *UsageRunnable) loop(ctx context.Context) error {
+func (u *UsageRunnable) loop(ctx context.Context) (errs error) {
+
 	err := u.usageTracker.ScheduledEvent(ctx)
 	if err != nil {
-		return err
+		errs = errors.Join(errs, fmt.Errorf("error in scheduled event: %w", err))
+	}
+
+	err = u.usageTracker.GarbageCollection(ctx)
+	if err != nil {
+		errs = errors.Join(errs, fmt.Errorf("error in garbage collection: %w", err))
 	}
 
-	return nil
+	return
 }

internal/usage/tracking.go

@@ -183,8 +183,41 @@ func (u *UsageTracker) ScheduledEvent(ctx context.Context) error {
 		errs = errors.Join(errs, err)
 	}
 
+	if errs != nil {
+		return fmt.Errorf("error when updating the usage: %w", errs)
+	}
+
+	return nil
+}
+
+func (u *UsageTracker) GarbageCollection(ctx context.Context) error {
+	_ = u.log.WithName("scheduled")
+
+	var mcpUsages v1.MCPUsageList
+	err := u.client.List(ctx, &mcpUsages)
 	if err != nil {
-		return fmt.Errorf("error when updating the usage: %w", err)
+		return fmt.Errorf("error when getting list of mcp usages: %w", err)
+	}
+
+	now := time.Now().UTC().Truncate(time.Hour * 24)
+	oneMonth := time.Hour * 24 * 32
+	latestTimestamp := now.Add(-oneMonth)
+
+	var errs error
+	for _, mcpUsage := range mcpUsages.Items {
+		usagesToKeep := make([]v1.DailyUsage, 0, len(mcpUsage.Spec.Usage))
+		for _, usage := range mcpUsage.Spec.Usage {
+			if !usage.Date.Time.Before(latestTimestamp) {
+				usagesToKeep = append(usagesToKeep, usage)
+			}
+		}
+		mcpUsage.Spec.Usage = usagesToKeep
+		err = u.client.Update(ctx, &mcpUsage)
+		errs = errors.Join(errs, err)
+	}
+
+	if errs != nil {
+		return fmt.Errorf("error when updating the usage: %w", errs)
 	}
 
 	return nil