fix: handle exit signal properly

This commit is contained in:
Yiyang Kang 2022-11-22 20:33:50 +08:00
parent b73e378de7
commit 18cb9715e4
1 changed files with 12 additions and 5 deletions

13
main.go
View File

@ -6,6 +6,8 @@ import (
"syscall"
"go.uber.org/zap"
"git.gensokyo.cafe/kkyy/tgbot_misaka_5882f7/utils"
)
var logger *zap.SugaredLogger
@ -26,16 +28,21 @@ func runBot() {
logger.Fatalw("Failed to initialize bot", "err", err)
}
go bot.Start()
logger.Info("Bot started")
botFinCh := utils.WaitFor(bot.Start)
logger.Infow("Bot started", "username", bot.Me.Username)
// listen for shutdown signal
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
<-sigCh
select {
case sig := <-sigCh:
logger.Infow("Received signal", "signal", sig)
logger.Info("Bot shutdown...")
bot.Stop()
case <-botFinCh:
logger.Warn("Bot terminated")
}
}
func main() {