添加AllModels

generator/gorm_gen_test.go

@@ -7,6 +7,10 @@ import (
 )
 
 func TestXMain(t *testing.T) {
-
+	testGenDir = "../" + testGenDir
+	GenAllModels(testGenDir, TableNameComment{
+		Name:    "FsFont",
+		Comment: "测试",
+	})
 	// Now you can use the generated GORM model to interact with the database
 }

generator/main.go

@@ -65,6 +65,106 @@ func GetColsFromTable(tname string, db *sql.DB) (result []Column, tableName, tab
 	return ParserDDL(ddl)
 }
 
+var gmodelVarStr = `
+package gmodel
+
+import "gorm.io/gorm"
+
+type AllModelsGen struct {
+}
+
+func NewAllModels(gdb *gorm.DB) *AllModelsGen {
+	models := &AllModelsGen{
+	}
+	return models
+}
+`
+
+var gmodelVarStrFormat = `
+package gmodel
+
+import "gorm.io/gorm"
+
+type AllModelsGen struct {
+	 %s
+}
+
+func NewAllModels(gdb *gorm.DB) *AllModelsGen {
+	models := &AllModelsGen{
+		%s
+	}
+	return models
+}
+`
+
+type TableNameComment struct {
+	Name    string
+	GoName  string
+	Comment string
+}
+
+func GenAllModels(filedir string, tmcs ...TableNameComment) {
+	fileName := filedir + "/var_gen.go"
+	if _, err := os.Stat(fileName); err == nil {
+		data, err := os.ReadFile(fileName)
+		if err != nil {
+			panic(err)
+		}
+		filestr := string(data)
+		filelines := strings.Split(filestr, "\n")
+		re := regexp.MustCompile(` +([^:]+)[^/]+ // ([^ ]+) (.?)$`)
+		for _, line := range filelines {
+			result := re.FindStringSubmatch(line)
+			if len(result) > 0 {
+				// key := result[0]
+				log.Println(result)
+				tmcs = append(tmcs, TableNameComment{
+					Name:    result[1],
+					GoName:  result[0],
+					Comment: result[2],
+				})
+
+			}
+		}
+
+		structStr := ""
+		newModelsStr := ""
+		for _, tmc := range tmcs {
+			fsline := fmt.Sprintf("%s %sModel // %s %s\n", tmc.GoName, tmc.GoName, tmc.Name, tmc.Comment)
+			structStr += fsline
+			nmline := fmt.Sprintf("%s: New%sModel(gdb),\n", tmc.GoName, tmc.GoName)
+			newModelsStr += nmline
+		}
+
+		content := fmt.Sprintf(gmodelVarStrFormat, structStr, newModelsStr)
+		f, err := os.OpenFile(fileName, os.O_TRUNC|os.O_CREATE|os.O_WRONLY, 0644)
+		if err != nil {
+			panic(err)
+		}
+		_, err = f.WriteString(content)
+		if err != nil {
+			panic(err)
+		}
+
+	} else if os.IsExist(err) {
+		f, err := os.Create(fileName)
+		if err != nil {
+			panic(err)
+		}
+		_, err = f.WriteString(gmodelVarStr)
+		if err != nil {
+			panic(err)
+		}
+	} else {
+		panic(err)
+	}
+
+	err := exec.Command("gofmt", "-w", fileName).Run()
+	if err != nil {
+		panic(err)
+	}
+}
+
 func main() {
 	var mysqluri string
 	var name string // 需要序列化的单独文件名
@@ -85,13 +185,20 @@ func main() {
 		panic(err)
 	}
 	defer db.Close()
-
+	var tmcs []TableNameComment
 	if name == "-" {
 		tablenames := GetAllTableNames(mysqluri)
 		for _, testName := range tablenames {
 			cols, tname, tcomment := GetColsFromTable(testName, db)
 			GenFromPath(testGenDir, cols, tname, tcomment)
+
+			tmcs = append(tmcs, TableNameComment{
+				Name:    tname,
+				GoName:  toPascalCase(tname),
+				Comment: tcomment,
+			})
 		}
+
 	} else {
 		if name != "" {
 			testName = name
@@ -100,8 +207,16 @@ func main() {
 		// log.Println(testName)
 		cols, tname, tcomment := GetColsFromTable(testName, db)
 		GenFromPath(testGenDir, cols, tname, tcomment)
+
+		tmcs = append(tmcs, TableNameComment{
+			Name:    tname,
+			GoName:  toPascalCase(tname),
+			Comment: tcomment,
+		})
 	}
 
+	GenAllModels(testGenDir, tmcs...)
+
 	// tablenames := GetAllTableNames(mysqluri)
 	// log.Println(tablenames)
 
@@ -162,10 +277,6 @@ func GenFromPath(mdir string, cols []Column, tableName string, tableComment stri
 			typeName = typeName[1:]
 		}
 
-		if col.AutoIncrement {
-
-		}
-
 		tagstr := "`gorm:"
 
 		gormTag := ""
@@ -178,9 +289,6 @@ func GenFromPath(mdir string, cols []Column, tableName string, tableComment stri
 		if col.AutoIncrement {
 			gormTag += "auto_increment;"
 		}
-		// if col.DefaultValue == "" {
-		// 	log.Panic(col, "需要默认值")
-		// }
 
 		tagstr += fmt.Sprintf("\"%s\"", gormTag)
 

generator/main_test.go

@@ -9,6 +9,5 @@ func TestMain(t *testing.T) {
 
 	testGenDir = "../" + testGenDir
 	// os.Args = []string{"cmd", "-name=fs_guest"}
-
 	main()
 }