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 (
"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"
"gorm.io/driver/mysql"
@ -12,25 +11,25 @@ import (
"gorm.io/gorm"
)
type Config struct {
MysqlConnStr string
}
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()
var c config.Config
conf.MustLoad(*configFile, &c)
var c Config
conf.MustLoad(*configFile, &c, conf.UseEnv())
db, err := gorm.Open(mysql.New(mysql.Config{
DSN: c.MySQLConn, // DSN data source name
DSN: c.MysqlConnStr, // data source name, refer
}), &gorm.Config{})
if err != nil {
panic(err)
}
// 自动迁移
err = db.AutoMigrate(&OrmStructs.File{})
if err != nil {
panic(err)
}
// 自动迁移
err = db.AutoMigrate(&OrmStructs.Path{})
err = db.AutoMigrate(&OrmStructs.Project{})
if err != nil {
panic(err)
}
@ -50,3 +49,4 @@ func main() {
// 执行
g.Execute()
}

Loading…
Cancel
Save