40 lines
765 B
Go
40 lines
765 B
Go
package global
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"electricity_bill_calc/config"
|
|
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var (
|
|
DBConn *gorm.DB
|
|
)
|
|
|
|
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{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
db, err := DBConn.DB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
db.SetMaxIdleConns(config.DatabaseSettings.MaxIdleConns)
|
|
db.SetMaxOpenConns(config.DatabaseSettings.MaxOpenConns)
|
|
|
|
return nil
|
|
}
|