25 lines
332 B
Go
25 lines
332 B
Go
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()
|
|
}
|