1 changed files with 52 additions and 52 deletions
@ -1,52 +1,52 @@ |
|||||
package main |
package main |
||||
|
|
||||
import ( |
import ( |
||||
"flag" |
"flag" |
||||
"zero-files/api/local/internal/config" |
|
||||
|
OrmStructs "RelaseMgt/orm/ormstructs" |
||||
OrmStructs "zero-files/api/local/orm/structs" |
|
||||
|
"github.com/zeromicro/go-zero/core/conf" |
||||
"github.com/zeromicro/go-zero/core/conf" |
"gorm.io/driver/mysql" |
||||
"gorm.io/driver/mysql" |
"gorm.io/gen" |
||||
"gorm.io/gen" |
"gorm.io/gorm" |
||||
"gorm.io/gorm" |
) |
||||
) |
|
||||
|
type Config struct { |
||||
func main() { |
MysqlConnStr string |
||||
var configFile = flag.String("f", "../etc/local-api.yaml", "the config file") |
} |
||||
flag.Parse() |
|
||||
var c config.Config |
func main() { |
||||
conf.MustLoad(*configFile, &c) |
// 读取配置文件, 生成数据库表, 生成对应的orm代码,注意修改配置文件路径
|
||||
db, err := gorm.Open(mysql.New(mysql.Config{ |
var configFile = flag.String("f", "../api/ReleaseMgt/etc/release-mgt-api.yaml", "the config file") |
||||
DSN: c.MySQLConn, // DSN data source name
|
flag.Parse() |
||||
}), &gorm.Config{}) |
var c Config |
||||
if err != nil { |
conf.MustLoad(*configFile, &c, conf.UseEnv()) |
||||
panic(err) |
|
||||
} |
db, err := gorm.Open(mysql.New(mysql.Config{ |
||||
// 自动迁移
|
DSN: c.MysqlConnStr, // data source name, refer
|
||||
err = db.AutoMigrate(&OrmStructs.File{}) |
}), &gorm.Config{}) |
||||
if err != nil { |
if err != nil { |
||||
panic(err) |
panic(err) |
||||
} |
} |
||||
|
// 自动迁移
|
||||
// 自动迁移
|
err = db.AutoMigrate(&OrmStructs.Project{}) |
||||
err = db.AutoMigrate(&OrmStructs.Path{}) |
if err != nil { |
||||
if err != nil { |
panic(err) |
||||
panic(err) |
} |
||||
} |
|
||||
|
// 配置生成器
|
||||
// 配置生成器
|
g := gen.NewGenerator(gen.Config{ |
||||
g := gen.NewGenerator(gen.Config{ |
OutPath: "./dal", |
||||
OutPath: "./dal", |
Mode: gen.WithoutContext | gen.WithDefaultQuery | gen.WithQueryInterface, // generate mode
|
||||
Mode: gen.WithoutContext | gen.WithDefaultQuery | gen.WithQueryInterface, // generate mode
|
}) |
||||
}) |
// 使用数据库
|
||||
// 使用数据库
|
g.UseDB(db) |
||||
g.UseDB(db) |
|
||||
|
g.ApplyBasic( |
||||
g.ApplyBasic( |
// 生成所有表
|
||||
// 生成所有表
|
g.GenerateAllTable()..., |
||||
g.GenerateAllTable()..., |
) |
||||
) |
// 执行
|
||||
// 执行
|
g.Execute() |
||||
g.Execute() |
} |
||||
} |
|
||||
|
|||||
Loading…
Reference in new issue