Browse Source

更新 'orm.go'

master
dark 1 year ago
parent
commit
2102859f86
  1. 104
      orm.go

104
orm.go

@ -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…
Cancel
Save