85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
package contract
|
||
|
||
import (
|
||
"github.com/Superdanda/hade/framework"
|
||
"github.com/go-sql-driver/mysql"
|
||
"gorm.io/gorm"
|
||
"net"
|
||
"strconv"
|
||
"time"
|
||
)
|
||
|
||
const ORMKey = "hade:orm"
|
||
|
||
// ORMService 表示传入的参数
|
||
type ORMService interface {
|
||
GetDB(option ...DBOption) (*gorm.DB, error)
|
||
}
|
||
|
||
// DBOption 代表初始化的时候的选项
|
||
type DBOption func(container framework.Container, config *DBConfig) error
|
||
|
||
// DBConfig 代表数据库连接的所有配置
|
||
type DBConfig struct {
|
||
// 以下配置关于dsn
|
||
WriteTimeout string `yaml:"write_timeout"` // 写超时时间
|
||
Loc string `yaml:"loc"` // 时区
|
||
Port int `yaml:"port"` // 端口
|
||
ReadTimeout string `yaml:"read_timeout"` // 读超时时间
|
||
Charset string `yaml:"charset"` // 字符集
|
||
ParseTime bool `yaml:"parse_time"` // 是否解析时间
|
||
Protocol string `yaml:"protocol"` // 传输协议
|
||
Dsn string `yaml:"dsn"` // 直接传递dsn，如果传递了，其他关于dsn的配置均无效
|
||
Database string `yaml:"database"` // 数据库
|
||
Collation string `yaml:"collation"` // 字符序
|
||
Timeout string `yaml:"timeout"` // 连接超时时间
|
||
Username string `yaml:"username"` // 用户名
|
||
Password string `yaml:"password"` // 密码
|
||
Driver string `yaml:"driver"` // 驱动
|
||
Host string `yaml:"host"` // 数据库地址
|
||
|
||
// 以下配置关于连接池
|
||
ConnMaxIdle int `yaml:"conn_max_idle"` // 最大空闲连接数
|
||
ConnMaxOpen int `yaml:"conn_max_open"` // 最大连接数
|
||
ConnMaxLifetime string `yaml:"conn_max_lifetime"` // 连接最大生命周期
|
||
ConnMaxIdletime string `yaml:"conn_max_idletime"` // 空闲最大生命周期
|
||
|
||
// 以下配置关于gorm
|
||
*gorm.Config // 集成gorm的配置
|
||
}
|
||
|
||
// FormatDsn 生成dsn
|
||
func (conf *DBConfig) FormatDsn() (string, error) {
|
||
port := strconv.Itoa(conf.Port)
|
||
timeout, err := time.ParseDuration(conf.Timeout)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
readTimeout, err := time.ParseDuration(conf.ReadTimeout)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
writeTimeout, err := time.ParseDuration(conf.WriteTimeout)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
location, err := time.LoadLocation(conf.Loc)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
driverConf := &mysql.Config{
|
||
User: conf.Username,
|
||
Passwd: conf.Password,
|
||
Net: conf.Protocol,
|
||
Addr: net.JoinHostPort(conf.Host, port),
|
||
DBName: conf.Database,
|
||
Collation: conf.Collation,
|
||
Loc: location,
|
||
Timeout: timeout,
|
||
ReadTimeout: readTimeout,
|
||
WriteTimeout: writeTimeout,
|
||
ParseTime: conf.ParseTime,
|
||
}
|
||
return driverConf.FormatDSN(), nil
|
||
}
|