feat: support kick

This commit is contained in:
Vaala Cat
2023-06-19 19:21:26 +08:00
parent 730aea3e3b
commit c278b3f516
16 changed files with 447 additions and 90 deletions

24
utils/database/db.go Normal file
View File

@@ -0,0 +1,24 @@
package database
import (
"github.com/joho/godotenv"
"github.com/sirupsen/logrus"
"gorm.io/gorm"
)
func init() {
godotenv.Load()
initSqlite()
}
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()
}