94 lines
2.1 KiB
Go
Executable File
94 lines
2.1 KiB
Go
Executable File
package config
|
||
|
||
import (
|
||
"fmt"
|
||
)
|
||
|
||
// Database database config
|
||
type Database struct {
|
||
Driver string
|
||
MySQL MySQL
|
||
Redis Redis
|
||
PostgreSQL PostgreSQL
|
||
}
|
||
|
||
// MySQL database config
|
||
type MySQL struct {
|
||
Host string
|
||
Port uint
|
||
Database string
|
||
Username string
|
||
Password string
|
||
Prefix string // 表前缀
|
||
Singular bool //是否开启全局禁用复数，true表示开启
|
||
MaxIdleConns int // 空闲中的最大连接数
|
||
MaxOpenConns int // 打开到数据库的最大连接数
|
||
Engine string //数据库引擎，默认InnoDB
|
||
}
|
||
|
||
func (m *MySQL) CreateDatabaseSql() string {
|
||
return fmt.Sprintf("CREATE DATABASE IF NOT EXISTS `%s` DEFAULT CHARACTER SET utf8mb4 DEFAULT COLLATE utf8mb4_general_ci;", m.Database)
|
||
}
|
||
func (m *MySQL) EmptyDsn() string {
|
||
dsnTpl := "%s@tcp(%s:%d)/"
|
||
|
||
authString := func() string {
|
||
if len(m.Password) > 0 {
|
||
return m.Username + ":" + m.Password
|
||
}
|
||
return m.Username
|
||
}
|
||
|
||
return fmt.Sprintf(dsnTpl, authString(), m.Host, m.Port)
|
||
}
|
||
|
||
// DSN connection dsn
|
||
func (m *MySQL) DSN() string {
|
||
dsnTpl := "%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local"
|
||
|
||
authString := func() string {
|
||
if len(m.Password) > 0 {
|
||
return m.Username + ":" + m.Password
|
||
}
|
||
return m.Username
|
||
}
|
||
|
||
return fmt.Sprintf(dsnTpl, authString(), m.Host, m.Port, m.Database)
|
||
}
|
||
|
||
type PostgreSQL struct {
|
||
User string
|
||
Password string
|
||
Database string
|
||
Host string
|
||
Port uint
|
||
SslMode string
|
||
TimeZone string
|
||
}
|
||
|
||
func (m *PostgreSQL) EmptyDsn() string {
|
||
dsnTpl := "host=%s user=%s password=%s port=%d dbname=postgres sslmode=disable TimeZone=Asia/Shanghai"
|
||
|
||
return fmt.Sprintf(dsnTpl, m.Host, m.User, m.Password, m.Port)
|
||
}
|
||
|
||
// DSN connection dsn
|
||
func (m *PostgreSQL) DSN() string {
|
||
dsnTpl := "host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=%s"
|
||
return fmt.Sprintf(dsnTpl, m.Host, m.User, m.Password, m.Database, m.Port, m.SslMode, m.TimeZone)
|
||
}
|
||
|
||
type Redis struct {
|
||
Host string
|
||
Port uint
|
||
Database uint
|
||
Username string
|
||
Password string
|
||
}
|
||
|
||
// DSN connection dsn
|
||
func (m *Redis) DSN() string {
|
||
dsnTpl := "%s:%d"
|
||
return fmt.Sprintf(dsnTpl, m.Host, m.Port)
|
||
}
|