16
data/data.go
16
data/data.go
@@ -2,10 +2,12 @@ package data
|
||||
|
||||
import (
|
||||
"nixcn-cms/data/drivers"
|
||||
"os"
|
||||
|
||||
"log/slog"
|
||||
|
||||
"github.com/meilisearch/meilisearch-go"
|
||||
"github.com/redis/go-redis/v9"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"github.com/spf13/viper"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
@@ -25,19 +27,22 @@ func Init() {
|
||||
}
|
||||
|
||||
if dbType != "postgres" {
|
||||
log.Fatal("[Database] Only support postgras db!")
|
||||
slog.Error("[Database] Only support postgras db!")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// Conect to db
|
||||
db, err := drivers.Postgres(exDSN)
|
||||
if err != nil {
|
||||
log.Fatal("[Database] Error connecting to db!")
|
||||
slog.Error("[Database] Error connecting to db!")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// Auto migrate
|
||||
err = db.AutoMigrate(&User{}, &Event{}, &Attendance{}, &Client{})
|
||||
if err != nil {
|
||||
log.Error("[Database] Error migrating database: ", err)
|
||||
slog.Error("[Database] Error migrating database!", "err", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
Database = db
|
||||
|
||||
@@ -52,7 +57,8 @@ func Init() {
|
||||
}
|
||||
rdb, err := drivers.Redis(rDSN)
|
||||
if err != nil {
|
||||
log.Fatal("[Redis] Error connecting to Redis: ", err)
|
||||
slog.Error("[Redis] Error connecting to Redis!", "err", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
Redis = rdb
|
||||
|
||||
|
||||
Reference in New Issue
Block a user