refactor(deps):ORM库切换为Xorm,同时增加保证数据库连接的配置。

This commit is contained in:
徐涛
2022-08-10 15:15:22 +08:00
parent 2cb2cf3cc7
commit 7a55b78ebe
14 changed files with 685 additions and 76 deletions

View File

@@ -2,38 +2,36 @@ package global
import (
"fmt"
"time"
"electricity_bill_calc/config"
"gorm.io/driver/postgres"
"gorm.io/gorm"
_ "github.com/lib/pq"
"xorm.io/xorm"
)
var (
DBConn *gorm.DB
DBConn *xorm.Engine
)
func SetupDatabaseConnection() error {
var err error
DBConn, err = gorm.Open(postgres.New(postgres.Config{
DSN: fmt.Sprintf(
"host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai",
config.DatabaseSettings.Host,
config.DatabaseSettings.User,
config.DatabaseSettings.Pass,
config.DatabaseSettings.DB,
config.DatabaseSettings.Port),
}), &gorm.Config{})
DBConn, err = xorm.NewEngine("postgres", fmt.Sprintf(
"host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai connect_timeout=0",
config.DatabaseSettings.Host,
config.DatabaseSettings.User,
config.DatabaseSettings.Pass,
config.DatabaseSettings.DB,
config.DatabaseSettings.Port,
))
if err != nil {
return err
}
db, err := DBConn.DB()
if err != nil {
return err
}
db.SetMaxIdleConns(config.DatabaseSettings.MaxIdleConns)
db.SetMaxOpenConns(config.DatabaseSettings.MaxOpenConns)
DBConn.Ping()
DBConn.SetMaxIdleConns(config.DatabaseSettings.MaxIdleConns)
DBConn.SetMaxOpenConns(config.DatabaseSettings.MaxOpenConns)
DBConn.SetConnMaxLifetime(60 * time.Second)
return nil
}