59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package global
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"electricity_bill_calc/config"
|
|
"electricity_bill_calc/logger"
|
|
|
|
// _ "github.com/lib/pq"
|
|
_ "github.com/jackc/pgx/v5/stdlib"
|
|
"xorm.io/xorm"
|
|
"xorm.io/xorm/log"
|
|
)
|
|
|
|
var (
|
|
DBConn *xorm.Engine
|
|
)
|
|
|
|
func SetupDatabaseConnection() error {
|
|
var err error
|
|
// 以下连接方式是采用pgx驱动的时候使用的。
|
|
DBConn, err = xorm.NewEngine("pgx", fmt.Sprintf(
|
|
"postgresql://%s:%s@%s:%d/%s?sslmode=disable&",
|
|
config.DatabaseSettings.User,
|
|
config.DatabaseSettings.Pass,
|
|
config.DatabaseSettings.Host,
|
|
config.DatabaseSettings.Port,
|
|
config.DatabaseSettings.DB,
|
|
))
|
|
// 以下连接方式是采用lib/pq驱动的时候使用的。
|
|
// 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
|
|
}
|
|
xLogger := logger.NewXormZeroLogger(logger.GetLogger())
|
|
DBConn.SetLogger(log.NewLoggerAdapter(xLogger))
|
|
|
|
DBConn.Ping()
|
|
DBConn.SetMaxIdleConns(config.DatabaseSettings.MaxIdleConns)
|
|
DBConn.SetMaxOpenConns(config.DatabaseSettings.MaxOpenConns)
|
|
DBConn.SetConnMaxLifetime(60 * time.Second)
|
|
|
|
if strings.ToLower(config.ServerSettings.RunMode) == "debug" {
|
|
DBConn.ShowSQL(true)
|
|
DBConn.Logger().SetLevel(log.LOG_DEBUG)
|
|
}
|
|
|
|
return nil
|
|
}
|