Browse Source

更新 'orm.go'

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

26
orm.go

@ -2,9 +2,8 @@ package main
import ( import (
"flag" "flag"
"zero-files/api/local/internal/config"
OrmStructs "zero-files/api/local/orm/structs" OrmStructs "RelaseMgt/orm/ormstructs"
"github.com/zeromicro/go-zero/core/conf" "github.com/zeromicro/go-zero/core/conf"
"gorm.io/driver/mysql" "gorm.io/driver/mysql"
@ -12,25 +11,25 @@ import (
"gorm.io/gorm" "gorm.io/gorm"
) )
type Config struct {
MysqlConnStr string
}
func main() { func main() {
var configFile = flag.String("f", "../etc/local-api.yaml", "the config file") // 读取配置文件, 生成数据库表, 生成对应的orm代码,注意修改配置文件路径
var configFile = flag.String("f", "../api/ReleaseMgt/etc/release-mgt-api.yaml", "the config file")
flag.Parse() flag.Parse()
var c config.Config var c Config
conf.MustLoad(*configFile, &c) conf.MustLoad(*configFile, &c, conf.UseEnv())
db, err := gorm.Open(mysql.New(mysql.Config{ db, err := gorm.Open(mysql.New(mysql.Config{
DSN: c.MySQLConn, // DSN data source name DSN: c.MysqlConnStr, // data source name, refer
}), &gorm.Config{}) }), &gorm.Config{})
if err != nil { if err != nil {
panic(err) panic(err)
} }
// 自动迁移 // 自动迁移
err = db.AutoMigrate(&OrmStructs.File{}) err = db.AutoMigrate(&OrmStructs.Project{})
if err != nil {
panic(err)
}
// 自动迁移
err = db.AutoMigrate(&OrmStructs.Path{})
if err != nil { if err != nil {
panic(err) panic(err)
} }
@ -50,3 +49,4 @@ func main() {
// 执行 // 执行
g.Execute() g.Execute()
} }

Loading…
Cancel
Save