Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: print more logs

Browse Source
This commit is contained in:
JustSong 2023-06-22 01:12:28 +08:00
parent 567916bd80
commit b86de464b5
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
common/redis.go
View File

@ -17,9 +17,10 @@ func InitRedisClient() (err error) {
SysLog("REDIS_CONN_STRING not set, Redis is not enabled") SysLog("REDIS_CONN_STRING not set, Redis is not enabled")
return nil return nil
} }
SysLog("Redis is enabled")
opt, err := redis.ParseURL(os.Getenv("REDIS_CONN_STRING")) opt, err := redis.ParseURL(os.Getenv("REDIS_CONN_STRING"))
if err != nil { if err != nil {
panic(err) FatalLog(err)
} }
RDB = redis.NewClient(opt) RDB = redis.NewClient(opt)

5
model/main.go
View File

@ -42,17 +42,19 @@ func InitDB() (err error) {
var db *gorm.DB var db *gorm.DB
if os.Getenv("SQL_DSN") != "" { if os.Getenv("SQL_DSN") != "" {
// Use MySQL // Use MySQL
common.SysLog("using MySQL as database")
db, err = gorm.Open(mysql.Open(os.Getenv("SQL_DSN")), &gorm.Config{ db, err = gorm.Open(mysql.Open(os.Getenv("SQL_DSN")), &gorm.Config{
PrepareStmt: true, // precompile SQL PrepareStmt: true, // precompile SQL
}) })
} else { } else {
// Use SQLite // Use SQLite
common.SysLog("SQL_DSN not set, using SQLite as database")
common.UsingSQLite = true common.UsingSQLite = true
db, err = gorm.Open(sqlite.Open(common.SQLitePath), &gorm.Config{ db, err = gorm.Open(sqlite.Open(common.SQLitePath), &gorm.Config{
PrepareStmt: true, // precompile SQL PrepareStmt: true, // precompile SQL
}) })
common.SysLog("SQL_DSN not set, using SQLite as database")
} }
common.SysLog("Database connected")
if err == nil { if err == nil {
DB = db DB = db
if !common.IsMasterNode { if !common.IsMasterNode {
@ -86,6 +88,7 @@ func InitDB() (err error) {
if err != nil { if err != nil {
return err return err
} }
common.SysLog("Database migrated")
err = createRootAccountIfNeed() err = createRootAccountIfNeed()
return err return err
} else { } else {