- update dorm

library.go

@@ -1,5 +1,5 @@
 package go_library
 
 func Version() string {
-	return "1.0.92"
+	return "1.0.93"
 }

utils/dorm/gorm_gen.go

@@ -14,6 +14,7 @@ type GormGenClientTableFun func() (*GormGenClient, string)
 
 type GormGenClientConfig struct {
 	Dns    string     // dns地址
+	Db     *gorm.DB   // db驱动
 	Config gen.Config // gen配置
 }
 

utils/dorm/gorm_gen_mysql.go

@@ -12,8 +12,12 @@ func NewGormGenMysqlClient(config *GormGenClientConfig) (*GormGenClient, error)
 
 	c.generator = gen.NewGenerator(config.Config)
 
-	c.db, _ = gorm.Open(mysql.Open(c.config.Dns), &gorm.Config{})
-	c.generator.UseDB(c.db)
+	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
 }

utils/dorm/gorm_gen_postgresql.go

@@ -12,8 +12,12 @@ func NewGormGenPostgresqlClient(config *GormGenClientConfig) (*GormGenClient, er
 
 	c.generator = gen.NewGenerator(config.Config)
 
-	c.db, _ = gorm.Open(postgres.Open(c.config.Dns), &gorm.Config{})
-	c.generator.UseDB(c.db)
+	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
 }