feat: run cleanup task periodically

internal/repository/cleaner.go

@@ -11,7 +11,18 @@ var (
 	cleanupSQL string
 )
 
-func (s *Service) Cleanup(threshold time.Time) error {
+func (s *Service) startCleaner() {
+	ticker := time.NewTicker(s.config.History)
+
+	for threshold := range ticker.C {
+		err := s.cleanup(threshold)
+		if err != nil {
+			log.Error().Err(err).Msg("failed to clean history")
+		}
+	}
+}
+
+func (s *Service) cleanup(threshold time.Time) error {
 	exec, err := s.conn.Exec(cleanupSQL, threshold)
 	if err != nil {
 		return err

internal/repository/initialize.go

@@ -28,6 +28,8 @@ func (s *Service) Initialize(container *injector.Container) error {
 		return err
 	}
 
+	go s.startCleaner()
+
 	return nil
 }
 

internal/repository/model.go

@@ -0,0 +1,11 @@
+package repository
+
+import (
+	"database/sql"
+	"sendgrid-mock/internal/config"
+)
+
+type Service struct {
+	config config.Config
+	conn   *sql.DB
+}

internal/repository/storage.go

@@ -2,11 +2,9 @@ package repository
 
 import (
 	"context"
-	"database/sql"
 	_ "embed"
 	"encoding/json"
 	_ "github.com/mattn/go-sqlite3"
-	"sendgrid-mock/internal/config"
 	"sendgrid-mock/internal/model"
 )
 
@@ -18,11 +16,6 @@ var (
 	insertSQL string
 )
 
-type Service struct {
-	config config.Config
-	conn   *sql.DB
-}
-
 func (s *Service) Save(ctx context.Context, message *model.Message) error {
 	customArgs, err := json.Marshal(message.CustomArgs)
 	if err != nil {

internal/sendgrid/initialize.go

@@ -35,9 +35,6 @@ func (s *Service) Initialize(container *injector.Container) error {
 
 	s.event = &event
 
-	s.cleaner = make(chan bool, 1)
-	go s.startCleaner()
-
 	return nil
 }
 

internal/sendgrid/router.go

@@ -21,10 +21,9 @@ var (
 )
 
 type Service struct {
-	config  *config.Config
-	repo    *repository.Service
-	event   *eventsender.Service
-	cleaner chan bool
+	config *config.Config
+	repo   *repository.Service
+	event  *eventsender.Service
 }
 
 func (s *Service) Routes() []restrouters.Route {

internal/sendgrid/service.go

@@ -66,7 +66,5 @@ func (s *Service) persist(ctx context.Context, body []byte) (string, error) {
 		}
 	}
 
-	s.triggerCleaner()
-
 	return messageID, err
 }