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

@@ -1,5 +1,7 @@
package sqlite
const DefaultPrefix = "SQLite"
type Config struct {
File string
}

View File

@@ -2,14 +2,19 @@ package sqlite
import (
"github.com/rogeecn/atom/container"
"go.uber.org/dig"
"github.com/rogeecn/atom/providers"
// "gorm.io/driver/sqlite"
"github.com/glebarez/sqlite"
"gorm.io/gorm"
)
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) {
db, err := gorm.Open(sqlite.Open(conf.File), &gorm.Config{})
if err != nil {
@@ -17,5 +22,5 @@ func Provide(conf *Config, opts ...dig.ProvideOption) error {
}
return db, err
}, opts...)
}, o.DiOptions()...)
}