gбновление зависимостей, рефакторинг кода, чтобы он выглядел как гошный проект

2025-04-06 20:22:16 +03:00
parent 00257b8e6c
commit cfb98fc0b3
389 changed files with 18649 additions and 9105 deletions
--- a/internal/lib/sighandler.go
+++ b/internal/lib/sighandler.go
@@ -0,0 +1,137 @@
+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: */