feat: vacuum database with cron

Author: ItsNotGoodName

cmd/smtpbridge/main.go

@@ -144,6 +144,16 @@ func run(flags *flag.FlagSet) lieut.Executor {
 				}
 			}
 
+			{
+				job := cron.NewDatabaseVacuum(app)
+				trigger := quartz.NewSimpleTrigger(24 * time.Hour)
+				if err := scheduler.ScheduleJob(ctx, job, trigger); err != nil {
+					return err
+				}
+
+				job.Execute(ctx)
+			}
+
 			return nil
 		}))
 

cron/cron.go

@@ -95,3 +95,30 @@ func (r Healthcheck) Execute(ctx context.Context) {
 func (r Healthcheck) Key() int {
 	return quartz.HashCode(r.Description())
 }
+
+// DatabaseVacuum
+type DatabaseVacuum struct {
+	app core.App
+}
+
+func NewDatabaseVacuum(app core.App) DatabaseVacuum {
+	return DatabaseVacuum{
+		app: app,
+	}
+}
+
+func (DatabaseVacuum) Description() string {
+	return "database-vacuum"
+}
+
+func (r DatabaseVacuum) Execute(ctx context.Context) {
+	err := r.app.DatabaseVacuum(ctx)
+	if err != nil {
+		log.Err(err).Msg("Failed to vacuum database")
+		return
+	}
+}
+
+func (r DatabaseVacuum) Key() int {
+	return quartz.HashCode(r.Description())
+}

internal/app/app.go

@@ -38,6 +38,11 @@ type App struct {
 	webTestFileStore WebTestFileStore
 }
 
+// DatabaseVacuum implements core.App.
+func (a App) DatabaseVacuum(ctx context.Context) error {
+	return repo.Vacuum(ctx, a.db)
+}
+
 func (a App) RuleEndpointsGet(ctx context.Context, id int64) (models.RuleEndpoints, error) {
 	return repo.RuleEndpointsGet(ctx, a.db, id)
 }

internal/core/app.go

@@ -51,4 +51,5 @@ type App interface {
 	TraceDrop(ctx context.Context) error
 	TraceList(ctx context.Context, page pagination.Page, req models.DTOTraceListRequest) (models.DTOTraceListResult, error)
 	Tracer(source string) trace.Tracer
+	DatabaseVacuum(ctx context.Context) error
 }