This commit is contained in:
yanghao05
2023-01-29 19:00:18 +08:00
parent 8ade8728e4
commit e07cf08d34
8 changed files with 243 additions and 22 deletions

View File

@@ -0,0 +1,30 @@
package mysql
import (
"atom/container"
"atom/providers/config"
"atom/providers/logger"
"database/sql"
"gorm.io/driver/mysql"
"gorm.io/gorm"
)
func init() {
if err := container.Container.Provide(NewDatabase); err != nil {
logger.Fatal(err)
}
}
func NewDatabase(config *config.Config) (*gorm.DB, error) {
sqlDB, err := sql.Open("mysql", config.Database.MySQL.DSN())
if err != nil {
return nil, err
}
gormDB, err := gorm.Open(mysql.New(mysql.Config{
Conn: sqlDB,
}), &gorm.Config{})
return gormDB, err
}