A pretty basic pastes cleanup procedure.

Fixes #16.

internal/database/database.go

@@ -27,6 +27,7 @@ package database
 import (
 	// stdlib
 	"database/sql"
+	"time"
 
 	// local
 	"go.dev.pztrn.name/fastpastebin/internal/database/dialects/flatfiles"
@@ -44,6 +45,46 @@ type Database struct {
 	db dialectinterface.Interface
 }
 
+// Database cleanup function. Executes once per hour, hardcoded for now and is
+// a subject of change in future.
+func (db *Database) cleanup() {
+	for {
+		c.Logger.Info().Msg("Starting pastes cleanup procedure...")
+
+		pages := db.db.GetPastesPages()
+
+		var pasteIDsToRemove []int
+
+		for i := 0; i < pages; i++ {
+			pastes, err := db.db.GetPagedPastes(i)
+			if err != nil {
+				c.Logger.Error().Err(err).Int("page", i).Msg("Failed to perform database cleanup")
+			}
+
+			for _, paste := range pastes {
+				if paste.IsExpired() {
+					pasteIDsToRemove = append(pasteIDsToRemove, paste.ID)
+				}
+			}
+		}
+
+		for _, pasteID := range pasteIDsToRemove {
+			err := db.DeletePaste(pasteID)
+			if err != nil {
+				c.Logger.Error().Err(err).Int("paste", pasteID).Msg("Failed to delete paste!")
+			}
+		}
+
+		c.Logger.Info().Msg("Pastes cleanup done.")
+
+		time.Sleep(time.Hour)
+	}
+}
+
+func (db *Database) DeletePaste(pasteID int) error {
+	return db.db.DeletePaste(pasteID)
+}
+
 func (db *Database) GetDatabaseConnection() *sql.DB {
 	if db.db != nil {
 		return db.db.GetDatabaseConnection()
@@ -77,6 +118,8 @@ func (db *Database) Initialize() {
 	} else {
 		c.Logger.Fatal().Str("type", c.Config.Database.Type).Msg("Unknown database type")
 	}
+
+	go db.cleanup()
 }
 
 func (db *Database) RegisterDialect(di dialectinterface.Interface) {