ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-15 07:20:26 +00:00
Code Issues 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
Download Patch File Download Diff File Expand all files Collapse all files

5
model/main.go
View File

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