to options mode

This commit is contained in:
yanghao05
2023-04-27 18:30:39 +08:00
parent 8ae3d19aa9
commit 574736a878
20 changed files with 180 additions and 50 deletions

View File

@@ -4,6 +4,8 @@ import (
"fmt"
)
const DefaultPrefix = "MySQL"
// MySQL database config
type Config struct {
Host string

View File

@@ -4,15 +4,20 @@ import (
"database/sql"
"github.com/rogeecn/atom/container"
"github.com/rogeecn/atom/providers"
"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 {
func Provide(o *providers.Options) error {
var conf Config
if err := o.UnmarshalConfig(&conf); err != nil {
return err
}
return container.Container.Provide(func() (*gorm.DB, error) {
if err := createMySQLDatabase(conf.EmptyDsn(), "mysql", conf.CreateDatabaseSql()); err != nil {
return nil, err
@@ -53,7 +58,7 @@ func Provide(conf *Config, opts ...dig.ProvideOption) error {
sqlDB.SetMaxOpenConns(conf.MaxOpenConns)
return db, err
}, opts...)
}, o.DiOptions()...)
}
// createDatabase 创建数据库