yanweidong
dbf68c38c1
docs(readme): 重构 README 文档结构并补充模块说明 - 重新组织 README 内容，明确划分私有仓库设置、功能模块等章节 - 补充 crypto、cache、database、middleware、queue 和 utils 模块的功能简介 - 统一代码示例格式，增加 bash 和 go 语言标识 feat(crypto): 优化 PKCS7 填充与去填充函数实现 - 新增 PKCS7Padding 和 PKCS7UnPadding 函数的详细注释 - 添加对输入参数的有效性校验，提升健壮性 - 修复可能引发越界 panic 的潜在问题 feat(database): 完善数据库连接及初始化逻辑 - 为 NewDatabase、NewMysql 和 NewPostgres 函数添加完整注释 - 修复 MaxOpenConns 配置未正确赋值的问题 - 在获取 *sql.DB 实例时增加错误处理逻辑 - 支持通过 Init 变量在连接建立后执行自定义初始化函数 feat(go.mod): 初始化项目依赖管理并引入核心组件 - 添加项目所需的主要依赖包，包括 gin、gorm、redis、nats 等 - 引入常用的工具库如 uuid、ulid、gopsutil 等 - 自动拉取并锁定所有间接依赖版本 ```
72 lines
1.9 KiB
Go
72 lines
1.9 KiB
Go
package sql
|
|
|
|
import (
|
|
"git.apinb.com/bsm-sdk/core/types"
|
|
"git.apinb.com/bsm-sdk/core/vars"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
func SetOptions(options *types.SqlOptions) *types.SqlOptions {
|
|
if options == nil {
|
|
options = &types.SqlOptions{
|
|
MaxIdleConns: vars.SqlOptionMaxIdleConns,
|
|
MaxOpenConns: vars.SqlOptionMaxOpenConns,
|
|
ConnMaxLifetime: vars.SqlOptionConnMaxLifetime,
|
|
LogStdout: false,
|
|
Debug: false,
|
|
}
|
|
}
|
|
|
|
return options
|
|
}
|
|
|
|
// new grom db.
|
|
func NewPostgreSql(dsn string, options *types.SqlOptions) (*gorm.DB, error) {
|
|
var err error
|
|
|
|
//set connection default val.
|
|
if options == nil {
|
|
options = &types.SqlOptions{
|
|
MaxIdleConns: vars.SqlOptionMaxIdleConns,
|
|
MaxOpenConns: vars.SqlOptionMaxOpenConns,
|
|
ConnMaxLifetime: vars.SqlOptionConnMaxLifetime,
|
|
LogStdout: false,
|
|
Debug: true,
|
|
}
|
|
}
|
|
|
|
gormDb, err := gorm.Open(postgres.New(postgres.Config{
|
|
DSN: dsn,
|
|
// PreferSimpleProtocol: true, disables implicit prepared statement usage
|
|
|
|
}), &gorm.Config{
|
|
//Logger:newLogger,
|
|
DisableForeignKeyConstraintWhenMigrating: true,
|
|
NamingStrategy: schema.NamingStrategy{
|
|
SingularTable: true, // 使用单数表名，启用该选项，此时，`User` 的表名应该是 `t_user`
|
|
}})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if options.Debug {
|
|
gormDb = gormDb.Debug()
|
|
}
|
|
|
|
// 获取通用数据库对象 sql.DB ，然后使用其提供的功能
|
|
sqlDB, err := gormDb.DB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
|
|
sqlDB.SetMaxIdleConns(options.MaxIdleConns)
|
|
// SetMaxOpenConns 设置打开数据库连接的最大数量。
|
|
sqlDB.SetMaxOpenConns(options.MaxOpenConns)
|
|
// SetConnMaxLifetime 设置了连接可复用的最大时间。
|
|
sqlDB.SetConnMaxLifetime(options.ConnMaxLifetime)
|
|
|
|
return gormDb, nil
|
|
} |