auto create database

This commit is contained in:
yanghao05
2023-01-30 18:18:25 +08:00
parent 27bb527373
commit 7c1aa71b8e
3 changed files with 110 additions and 29 deletions

View File

@@ -14,11 +14,32 @@ type Database struct {
// MySQL database config
type MySQL struct {
Host string
Port uint
Database string
Username string
Password string
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
@@ -45,6 +66,12 @@ type PostgreSQL struct {
TimeZone string
}
func (m *PostgreSQL) EmptyDsn() string {
dsnTpl := "host=%s user=%s password=%s port=%s 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"