迁移配置表

internal/apps/openflare/tasks/database_cleanup.go

@@ -11,6 +11,7 @@ import (
 	"time"
 
 	"github.com/Rain-kl/Wavelet/internal/model"
+	"github.com/Rain-kl/Wavelet/internal/repository"
 )
 
 const (
@@ -91,14 +92,20 @@ func CleanupDatabaseObservability(ctx context.Context, input DatabaseCleanupInpu
 
 // RunDatabaseAutoCleanupOnce runs retention-based cleanup for all observability targets.
 func RunDatabaseAutoCleanupOnce(ctx context.Context, now time.Time) (*DatabaseAutoCleanupSummary, error) {
-	if !model.DatabaseAutoCleanupEnabled {
+	enabled, err := repository.GetBoolByKey(ctx, model.ConfigKeyDatabaseAutoCleanupEnabled)
+	if err != nil {
+		return nil, fmt.Errorf("failed to read database_auto_cleanup_enabled: %w", err)
+	}
+	if !enabled {
 		return nil, nil
 	}
-	if model.DatabaseAutoCleanupRetentionDays < 1 {
-		return nil, fmt.Errorf("database auto cleanup retention_days must be at least 1")
+
+	retentionDays, err := repository.GetIntByKey(ctx, model.ConfigKeyDatabaseAutoCleanupRetentionDays)
+	if err != nil || retentionDays <= 0 {
+		// Use default value 30 if config read fails or value is invalid
+		retentionDays = 30
 	}
 
-	retentionDays := model.DatabaseAutoCleanupRetentionDays
 	results := make([]DatabaseCleanupResult, 0, len(databaseCleanupTargets))
 	for _, target := range []string{
 		DatabaseCleanupTargetAccessLogs,

internal/apps/openflare/tasks/database_cleanup_test.go

@@ -10,6 +10,7 @@ import (
 
 	"github.com/Rain-kl/Wavelet/internal/db"
 	"github.com/Rain-kl/Wavelet/internal/model"
+	"github.com/Rain-kl/Wavelet/internal/repository"
 	"github.com/glebarez/sqlite"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
@@ -23,6 +24,7 @@ func setupDatabaseCleanupTestDB(t *testing.T) context.Context {
 		DisableForeignKeyConstraintWhenMigrating: true,
 	})
 	require.NoError(t, err)
+	require.NoError(t, sqliteDB.AutoMigrate(&model.SystemConfig{}))
 	db.SetDB(sqliteDB)
 	resetAccessLogStore := model.SetAccessLogStoreForTest(model.NewMemoryAccessLogStore())
 	resetObservabilityStore := model.SetObservabilityStoreForTest(model.NewMemoryObservabilityStore())
@@ -123,14 +125,8 @@ func TestRunDatabaseAutoCleanupOnceDeletesAllObservabilityTargets(t *testing.T)
 		RequestCount:    15,
 	}))
 
-	previousEnabled := model.DatabaseAutoCleanupEnabled
-	previousRetentionDays := model.DatabaseAutoCleanupRetentionDays
-	model.DatabaseAutoCleanupEnabled = true
-	model.DatabaseAutoCleanupRetentionDays = 1
-	t.Cleanup(func() {
-		model.DatabaseAutoCleanupEnabled = previousEnabled
-		model.DatabaseAutoCleanupRetentionDays = previousRetentionDays
-	})
+	require.NoError(t, repository.SaveOrUpdateSystemConfig(ctx, model.ConfigKeyDatabaseAutoCleanupEnabled, "true"))
+	require.NoError(t, repository.SaveOrUpdateSystemConfig(ctx, model.ConfigKeyDatabaseAutoCleanupRetentionDays, "1"))
 
 	summary, err := RunDatabaseAutoCleanupOnce(ctx, now)
 	require.NoError(t, err)