feat: gateway

utils/database/db.go

@@ -2,7 +2,6 @@ package database
 
 import (
 	"github.com/joho/godotenv"
-	"github.com/sirupsen/logrus"
 	"gorm.io/gorm"
 )
 
@@ -15,10 +14,10 @@ func GetDB() *gorm.DB {
 	return GetSqlite()
 }
 
-func CloseDB(db *gorm.DB) {
-	tdb, err := db.DB()
-	if err != nil {
-		logrus.WithError(err).Errorf("Close DB error")
-	}
-	tdb.Close()
-}
+// func CloseDB(db *gorm.DB) {
+// 	tdb, err := db.DB()
+// 	if err != nil {
+// 		logrus.WithError(err).Errorf("Close DB error")
+// 	}
+// 	tdb.Close()
+// }

utils/database/sqlite.go

@@ -9,23 +9,25 @@ import (
 	"gorm.io/gorm"
 )
 
+var (
+	db *gorm.DB
+)
+
 func initSqlite() {
 	var err error
 	godotenv.Load()
 
 	dbPath := conf.GetBotSettings().DBPath
-	db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
+	db, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
 	if err != nil {
 		logrus.Panic(err, "Initializing DB Error")
 	}
-	CloseDB(db)
+	logrus.Info("Initialized DB at ", dbPath)
 }
 
 func GetSqlite() *gorm.DB {
-	dbPath := conf.GetBotSettings().DBPath
-	db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
-	if err != nil {
-		return nil
+	if db == nil {
+		initSqlite()
 	}
 	return db
 }