You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
518 B
26 lines
518 B
package dorm
|
|
|
|
import (
|
|
"context"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gen"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// NewGormGenMysqlClient 创建GormGenClient实例 mysql
|
|
func NewGormGenMysqlClient(ctx context.Context, config *GormGenClientConfig) (*GormGenClient, error) {
|
|
|
|
c := &GormGenClient{config: config}
|
|
|
|
c.generator = gen.NewGenerator(config.Config)
|
|
|
|
if c.config.Dns != "" {
|
|
c.db, _ = gorm.Open(mysql.Open(c.config.Dns), &gorm.Config{})
|
|
c.generator.UseDB(c.db)
|
|
} else {
|
|
c.generator.UseDB(c.config.Db)
|
|
}
|
|
|
|
return c, nil
|
|
}
|