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

fix: set connection limits for database

Browse Source
This commit is contained in:
JustSong
2023-08-12 10:05:25 +08:00
parent 2b088a1678
commit 466005de07
3 changed files with 28 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
model/main.go
View File

@@ -6,6 +6,7 @@ import (
"gorm.io/gorm"
"one-api/common"
"os"
"time"
)
var DB *gorm.DB
@@ -57,10 +58,18 @@ func InitDB() (err error) {
common.SysLog("database connected")
if err == nil {
DB = db
sqlDB, err := DB.DB()
if err != nil {
return err
}
sqlDB.SetMaxIdleConns(common.GetOrDefault("SQL_MAX_IDLE_CONNS", 10))
sqlDB.SetMaxOpenConns(common.GetOrDefault("SQL_MAX_OPEN_CONNS", 100))
sqlDB.SetConnMaxLifetime(time.Second * time.Duration(common.GetOrDefault("SQL_MAX_LIFETIME", 60)))
if !common.IsMasterNode {
return nil
}
err := db.AutoMigrate(&Channel{})
err = db.AutoMigrate(&Channel{})
if err != nil {
return err
}