rogee/atom
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

restructure

Browse Source
This commit is contained in:
yanghao05
2023-04-20 12:11:34 +08:00
parent 6757e00d73
commit 5b8eca5d87
120 changed files with 546 additions and 7303 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
providers/database/database.go
View File

@@ -1,35 +0,0 @@
package database
import (
"atom/container"
"atom/providers/config"
"errors"
"log"
"gorm.io/gorm"
)
const (
DriverMySQL = "mysql"
DriverSQLite = "sqlite"
DriverPostgres = "postgres"
DriverSQLServer = "sqlserver"
)
func init() {
if err := container.Container.Provide(NewDatabase); err != nil {
log.Fatal(err)
}
}
func NewDatabase(config *config.Config) (*gorm.DB, error) {
switch config.Database.Driver {
case DriverMySQL:
return NewMySQL(config.Database.MySQL)
case DriverSQLite:
return NewSQLite(config.Database.SQLite)
case DriverPostgres:
return NewPostgres(config.Database.PostgreSQL)
}
return nil, errors.New("failed to connect to db")
}

73
providers/database/mysql.go
View File

@@ -1,73 +0,0 @@
package database
import (
"atom/providers/config"
"atom/providers/log"
"database/sql"
"gorm.io/driver/mysql"
"gorm.io/gorm"
"gorm.io/gorm/schema"
)
func NewMySQL(conf *config.MySQL) (*gorm.DB, error) {
if err := createMySQLDatabase(conf.EmptyDsn(), "mysql", conf.CreateDatabaseSql()); err != nil {
return nil, err
}
mysqlConfig := mysql.Config{
DSN: conf.DSN(), // DSN data source name
DefaultStringSize: 191, // string 类型字段的默认长度
SkipInitializeWithVersion: false, // 根据版本自动配置
}
gormConfig := gorm.Config{
NamingStrategy: schema.NamingStrategy{
TablePrefix: conf.Prefix,
SingularTable: conf.Singular,
},
DisableForeignKeyConstraintWhenMigrating: true,
}
// TODO: config logger
// _default := logger.New(NewWriter(log.New(os.Stdout, "\r\n", log.LstdFlags)), logger.Config{
// SlowThreshold: 200 * time.Millisecond,
// LogLevel: logger.Warn,
// Colorful: true,
// })
// conf.Logger = _default.LogMode(logger.Warn)
db, err := gorm.Open(mysql.New(mysqlConfig), &gormConfig)
if err != nil {
return nil, err
}
// config instance
db.InstanceSet("gorm:table_options", "ENGINE="+conf.Engine)
sqlDB, _ := db.DB()
sqlDB.SetMaxIdleConns(conf.MaxIdleConns)
sqlDB.SetMaxOpenConns(conf.MaxOpenConns)
return db, err
}
// createDatabase 创建数据库
func createMySQLDatabase(dsn, driver, createSql string) error {
db, err := sql.Open(driver, dsn)
if err != nil {
return err
}
defer func(db *sql.DB) {
err = db.Close()
if err != nil {
log.Error(err)
}
}(db)
err = db.Ping()
if err != nil {
return err
}
_, err = db.Exec(createSql)
return err
}

50
providers/database/mysql/config.go Executable file
View File

@@ -0,0 +1,50 @@
package mysql
import (
"fmt"
)
// MySQL database config
type Config 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 *Config) CreateDatabaseSql() string {
return fmt.Sprintf("CREATE DATABASE IF NOT EXISTS `%s` DEFAULT CHARACTER SET utf8mb4 DEFAULT COLLATE utf8mb4_general_ci;", m.Database)
}
func (m *Config) 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 *Config) 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)
}

77
providers/database/mysql/mysql.go Normal file
View File

@@ -0,0 +1,77 @@
package mysql
import (
"database/sql"
"github.com/rogeecn/atom/container"
"github.com/rogeecn/atom/providers/log"
"go.uber.org/dig"
"gorm.io/driver/mysql"
"gorm.io/gorm"
"gorm.io/gorm/schema"
)
func Provide(conf *Config, opts ...dig.ProvideOption) error {
return container.Container.Provide(func() (*gorm.DB, error) {
if err := createMySQLDatabase(conf.EmptyDsn(), "mysql", conf.CreateDatabaseSql()); err != nil {
return nil, err
}
mysqlConfig := mysql.Config{
DSN: conf.DSN(), // DSN data source name
DefaultStringSize: 191, // string 类型字段的默认长度
SkipInitializeWithVersion: false, // 根据版本自动配置
}
gormConfig := gorm.Config{
NamingStrategy: schema.NamingStrategy{
TablePrefix: conf.Prefix,
SingularTable: conf.Singular,
},
DisableForeignKeyConstraintWhenMigrating: true,
}
// TODO: config logger
// _default := logger.New(NewWriter(log.New(os.Stdout, "\r\n", log.LstdFlags)), logger.Config{
// SlowThreshold: 200 * time.Millisecond,
// LogLevel: logger.Warn,
// Colorful: true,
// })
// conf.Logger = _default.LogMode(logger.Warn)
db, err := gorm.Open(mysql.New(mysqlConfig), &gormConfig)
if err != nil {
return nil, err
}
// config instance
db.InstanceSet("gorm:table_options", "ENGINE="+conf.Engine)
sqlDB, _ := db.DB()
sqlDB.SetMaxIdleConns(conf.MaxIdleConns)
sqlDB.SetMaxOpenConns(conf.MaxOpenConns)
return db, err
}, opts...)
}
// createDatabase 创建数据库
func createMySQLDatabase(dsn, driver, createSql string) error {
db, err := sql.Open(driver, dsn)
if err != nil {
return err
}
defer func(db *sql.DB) {
err = db.Close()
if err != nil {
log.Error(err)
}
}(db)
err = db.Ping()
if err != nil {
return err
}
_, err = db.Exec(createSql)
return err
}

36
providers/database/postgres.go
View File

@@ -1,36 +0,0 @@
package database
import (
"atom/providers/config"
"log"
"gorm.io/driver/postgres"
"gorm.io/gorm"
"gorm.io/gorm/schema"
)
func NewPostgres(conf *config.PostgreSQL) (*gorm.DB, error) {
dbConfig := postgres.Config{
DSN: conf.DSN(), // DSN data source name
}
log.Println("PostgreSQL DSN: ", dbConfig.DSN)
gormConfig := gorm.Config{
NamingStrategy: schema.NamingStrategy{
TablePrefix: conf.Prefix,
SingularTable: conf.Singular,
},
DisableForeignKeyConstraintWhenMigrating: true,
}
db, err := gorm.Open(postgres.New(dbConfig), &gormConfig)
if err != nil {
return nil, err
}
sqlDB, _ := db.DB()
sqlDB.SetMaxIdleConns(conf.MaxIdleConns)
sqlDB.SetMaxOpenConns(conf.MaxOpenConns)
return db, err
}

31
providers/database/postgres/config.go Executable file
View File

@@ -0,0 +1,31 @@
package postgres
import (
"fmt"
)
type Config struct {
Username string
Password string
Database string
Host string
Port uint
SslMode string
TimeZone string
Prefix string // 表前缀
Singular bool // 是否开启全局禁用复数，true表示开启
MaxIdleConns int // 空闲中的最大连接数
MaxOpenConns int // 打开到数据库的最大连接数
}
func (m *Config) 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.Username, m.Password, m.Port)
}
// DSN connection dsn
func (m *Config) DSN() string {
dsnTpl := "host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=%s"
return fmt.Sprintf(dsnTpl, m.Host, m.Username, m.Password, m.Database, m.Port, m.SslMode, m.TimeZone)
}

40
providers/database/postgres/postgres.go Normal file
View File

@@ -0,0 +1,40 @@
package postgres
import (
"log"
"github.com/rogeecn/atom/container"
"go.uber.org/dig"
"gorm.io/driver/postgres"
"gorm.io/gorm"
"gorm.io/gorm/schema"
)
func Provide(conf *Config, opts ...dig.ProvideOption) error {
return container.Container.Provide(func() (*gorm.DB, error) {
dbConfig := postgres.Config{
DSN: conf.DSN(), // DSN data source name
}
log.Println("PostgreSQL DSN: ", dbConfig.DSN)
gormConfig := gorm.Config{
NamingStrategy: schema.NamingStrategy{
TablePrefix: conf.Prefix,
SingularTable: conf.Singular,
},
DisableForeignKeyConstraintWhenMigrating: true,
}
db, err := gorm.Open(postgres.New(dbConfig), &gormConfig)
if err != nil {
return nil, err
}
sqlDB, _ := db.DB()
sqlDB.SetMaxIdleConns(conf.MaxIdleConns)
sqlDB.SetMaxOpenConns(conf.MaxOpenConns)
return db, err
}, opts...)
}

1
providers/database/redis/todo.go Normal file
View File

@@ -0,0 +1 @@
package redis

18
providers/database/sqlite.go
View File

@@ -1,18 +0,0 @@
package database
import (
"atom/providers/config"
// "gorm.io/driver/sqlite"
"github.com/glebarez/sqlite"
"gorm.io/gorm"
)
func NewSQLite(conf *config.SQLite) (*gorm.DB, error) {
db, err := gorm.Open(sqlite.Open(conf.File), &gorm.Config{})
if err != nil {
return nil, err
}
return db, err
}

13
providers/database/sqlite/config.go Executable file
View File

@@ -0,0 +1,13 @@
package sqlite
type Config struct {
File string
}
func (m *Config) CreateDatabaseSql() string {
return ""
}
func (m *Config) EmptyDsn() string {
return m.File
}

21
providers/database/sqlite/sqlite.go Normal file
View File

@@ -0,0 +1,21 @@
package sqlite
import (
"github.com/rogeecn/atom/container"
"go.uber.org/dig"
// "gorm.io/driver/sqlite"
"github.com/glebarez/sqlite"
"gorm.io/gorm"
)
func Provide(conf *Config, opts ...dig.ProvideOption) error {
return container.Container.Provide(func() (*gorm.DB, error) {
db, err := gorm.Open(sqlite.Open(conf.File), &gorm.Config{})
if err != nil {
return nil, err
}
return db, err
}, opts...)
}