1 files changed, 4 insertions, 1 deletions
diff --git a/main.go b/main.go
index 9d80cee..174854d 100644
--- a/main.go
+++ b/main.go
@@ -153,7 +153,7 @@ func main() {
 		os.Exit(1)
 	}
 
-	router.BlogTrigger, err = blogtrigger.NewBlogTriggerScheduler(b2Client, cfg.AvailableLanguages, func(bp []*b2.BlogPage) error {
+	router.BlogTrigger, err = blogtrigger.NewBlogTriggerScheduler(b2Client, cfg.AvailableLanguages, cfg.Mail.Trigger.OnNewPost, func(bp []*b2.BlogPage) error {
 		for _, post := range bp {
 			if err := router.Mailer.NewPost(post); err != nil {
 				return err
@@ -202,6 +202,9 @@ func main() {
 
 	<-sigChan
 	slog.Info("gracefully shutting down...")
+	if err = router.BlogTrigger.Close(); err != nil {
+		slog.Error("fail to shutdown blog trigger scheduler", slog.String("error", err.Error()))
+	}
 	if err = app.Shutdown(); err != nil {
 		slog.Error("fail to shutdown fiber server", slog.String("error", err.Error()))
 	}