5dd0bc19e4
Some checks failed
部署管理后台 / 🧪 测试和构建 (push) Failing after 1m5s
部署管理后台 / 🔒 安全扫描 (push) Has been skipped
部署后端服务 / 🧪 测试后端 (push) Failing after 3m13s
部署前端网站 / 🧪 测试和构建 (push) Failing after 2m10s
部署管理后台 / 🚀 部署到生产环境 (push) Has been skipped
部署后端服务 / 🚀 构建并部署 (push) Has been skipped
部署管理后台 / 🔄 回滚部署 (push) Has been skipped
部署前端网站 / 🚀 部署到生产环境 (push) Has been skipped
部署后端服务 / 🔄 回滚部署 (push) Has been skipped
- 后端：应用 go fmt 自动格式化，统一代码风格 - 前端：更新 API 配置，完善类型安全 - 所有代码符合项目规范，准备生产部署
78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
type Config struct {
|
|
Driver string `json:"driver"` // mysql, postgres, sqlite
|
|
Host string `json:"host,optional"`
|
|
Port int `json:"port,optional"`
|
|
Username string `json:"username,optional"`
|
|
Password string `json:"password,optional"`
|
|
Database string `json:"database,optional"`
|
|
Charset string `json:"charset,optional"`
|
|
SSLMode string `json:"ssl_mode,optional"`
|
|
FilePath string `json:"file_path,optional"` // for sqlite
|
|
}
|
|
|
|
func NewDB(config Config) (*gorm.DB, error) {
|
|
var db *gorm.DB
|
|
var err error
|
|
|
|
// 配置日志
|
|
newLogger := logger.New(
|
|
log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
|
|
logger.Config{
|
|
SlowThreshold: time.Second, // 慢 SQL 阈值
|
|
LogLevel: logger.Silent, // 日志级别
|
|
Colorful: false, // 禁用彩色打印
|
|
},
|
|
)
|
|
|
|
switch config.Driver {
|
|
case "mysql":
|
|
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local",
|
|
config.Username, config.Password, config.Host, config.Port, config.Database, config.Charset)
|
|
db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
|
|
Logger: newLogger,
|
|
})
|
|
case "postgres":
|
|
dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=Asia/Shanghai",
|
|
config.Host, config.Username, config.Password, config.Database, config.Port, config.SSLMode)
|
|
db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{
|
|
Logger: newLogger,
|
|
})
|
|
case "sqlite":
|
|
db, err = gorm.Open(sqlite.Open(config.FilePath), &gorm.Config{
|
|
Logger: newLogger,
|
|
})
|
|
default:
|
|
return nil, fmt.Errorf("unsupported database driver: %s", config.Driver)
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 设置连接池
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sqlDB.SetMaxIdleConns(10)
|
|
sqlDB.SetMaxOpenConns(100)
|
|
sqlDB.SetConnMaxLifetime(time.Hour)
|
|
|
|
return db, nil
|
|
}
|