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
533 B
26 lines
533 B
package dorm
|
|
|
|
import (
|
|
"context"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gen"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// NewGormGenPostgresClient 创建GormGenClient实例 postgres
|
|
func NewGormGenPostgresClient(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(postgres.Open(c.config.Dns), &gorm.Config{})
|
|
c.generator.UseDB(c.db)
|
|
} else {
|
|
c.generator.UseDB(c.config.Db)
|
|
}
|
|
|
|
return c, nil
|
|
}
|