138 lines
4.1 KiB
Go
138 lines
4.1 KiB
Go
package lib
|
|
|
|
import (
|
|
"os"
|
|
"syscall"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// Хэндлер сигналов закрывает все бд и сваливает из приложения
|
|
func SigHandler() {
|
|
var err error
|
|
|
|
for {
|
|
var s = <-SigChan
|
|
switch s {
|
|
case syscall.SIGINT:
|
|
log.Infoln("Got SIGINT, quitting")
|
|
case syscall.SIGTERM:
|
|
log.Infoln("Got SIGTERM, quitting")
|
|
case syscall.SIGQUIT:
|
|
log.Infoln("Got SIGQUIT, quitting")
|
|
|
|
// Заходим на новую итерацию, если у нас "неинтересный" сигнал
|
|
default:
|
|
continue
|
|
}
|
|
|
|
// Чтобы не срать в логи ошибками от редиски, проставим shutdown state приложения в true
|
|
Shutdown = true
|
|
|
|
// Отпишемся от всех каналов и закроем коннект к редиске
|
|
if err = Subscriber.Unsubscribe(Ctx); err != nil {
|
|
log.Errorf("Unable to unsubscribe from redis channels cleanly: %s", err)
|
|
}
|
|
|
|
if err = Subscriber.Close(); err != nil {
|
|
log.Errorf("Unable to close redis connection cleanly: %s", err)
|
|
}
|
|
|
|
if err := PebbleDB.Fortune.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Fortune.Name, err)
|
|
}
|
|
|
|
// Закроем статичные базы с фразами, тут их многовато :(
|
|
if err = PebbleDB.Fortune.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Fortune.Name, err)
|
|
}
|
|
|
|
if err := PebbleDB.Proverb.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Proverb.Name, err)
|
|
}
|
|
|
|
if err = PebbleDB.Proverb.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Proverb.Name, err)
|
|
}
|
|
|
|
if err := PebbleDB.Count.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Count.Name, err)
|
|
}
|
|
|
|
if err = PebbleDB.Count.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Count.Name, err)
|
|
}
|
|
|
|
if err := PebbleDB.Monday.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Monday.Name, err)
|
|
}
|
|
|
|
if err = PebbleDB.Monday.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Monday.Name, err)
|
|
}
|
|
|
|
if err := PebbleDB.Tuesday.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Tuesday.Name, err)
|
|
}
|
|
|
|
if err = PebbleDB.Tuesday.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Tuesday.Name, err)
|
|
}
|
|
|
|
if err := PebbleDB.Wednesday.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Wednesday.Name, err)
|
|
}
|
|
|
|
if err = PebbleDB.Wednesday.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Wednesday.Name, err)
|
|
}
|
|
|
|
if err := PebbleDB.Thursday.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Thursday.Name, err)
|
|
}
|
|
|
|
if err = PebbleDB.Thursday.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Thursday.Name, err)
|
|
}
|
|
|
|
if err := PebbleDB.Friday.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Friday.Name, err)
|
|
}
|
|
|
|
if err = PebbleDB.Friday.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Friday.Name, err)
|
|
}
|
|
|
|
if err := PebbleDB.Saturday.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Saturday.Name, err)
|
|
}
|
|
|
|
if err = PebbleDB.Saturday.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Saturday.Name, err)
|
|
}
|
|
|
|
if err := PebbleDB.Sunday.DB.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s db: %s", PebbleDB.Sunday.Name, err)
|
|
}
|
|
|
|
if err = PebbleDB.Sunday.DB.Close(); err != nil {
|
|
log.Errorf("Unable to close %s db: %s", PebbleDB.Sunday.Name, err)
|
|
}
|
|
|
|
// Закроем базы с кармой
|
|
for dbName, karma := range KarmaDB {
|
|
if err := karma.Flush(); err != nil {
|
|
log.Errorf("Unable to flush %s karma db: %s", dbName, err)
|
|
}
|
|
|
|
if err := karma.Close(); err != nil {
|
|
log.Errorf("Unable to close %s karma db: %s", dbName, err)
|
|
}
|
|
}
|
|
|
|
os.Exit(0)
|
|
}
|
|
}
|
|
|
|
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */
|