From 2102859f868dfe7dcfa18d470a7c6806d1d16516 Mon Sep 17 00:00:00 2001
From: dark <nndaa@qq.com>
Date: Tue, 24 Dec 2024 01:10:00 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'orm.go'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 orm.go | 104 ++++++++++++++++++++++++++++-----------------------------
 1 file changed, 52 insertions(+), 52 deletions(-)

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