Merge pull request #12 from anhnmt/v2

Update telegram alert

Author: anhnmt

cmd/backup.go

@@ -39,16 +39,14 @@ func schedule(schedule string) {
 		wg.Add(1)
 		defer wg.Done()
 
-		now := time.Now().Format(time.RFC3339)
-		log.Info().Msgf("Start backup at: %s", now)
-		go OK("Start backup at: %s", now)
+		now := time.Now()
+		log.Info().Msgf("Start backup at: %s", now.Format(time.RFC3339))
 
 		if err := start(); err != nil {
-			log.Err(err).Msg("Failed to start backup")
-			go Err(err, "Failed to start backup")
+			Err(err, "Failed to start backup")
 		}
 
-		go OK("Backup success")
+		OK("Backup successful: %s", time.Since(now).String())
 	})
 	if err != nil {
 		log.Panic().Err(err).Msg("Failed to add cron job")
@@ -69,8 +67,7 @@ func schedule(schedule string) {
 func start() error {
 	defer func() {
 		if err := removeFile(PgDumpFile); err != nil {
-			log.Err(err).Msg("Failed to remove pg_dump file")
-			go Err(err, "Failed to remove pg_dump file")
+			Err(err, "Failed to remove pg_dump file")
 		}
 	}()
 

cmd/telegram.go

@@ -6,6 +6,7 @@ import (
 	"sync/atomic"
 
 	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
+	"github.com/rs/zerolog/log"
 	"github.com/spf13/viper"
 )
 
@@ -43,10 +44,12 @@ func NewTelegram() (*Telegram, error) {
 }
 
 func OK(text string, a ...any) error {
+	log.Info().Msgf(text, a...)
 	return Default().Msg(nil, text, a...)
 }
 
 func Err(err error, text string, a ...any) error {
+	log.Err(err).Msgf(text, a...)
 	return Default().Msg(err, text, a...)
 }