增加menu自动化添加功能

code/makecode.go

@@ -13,6 +13,7 @@ import (
 )
 
 type MakeCode struct {
+	IndexMenus   Map
 	TableConfig  Map
 	TableColumns map[string]map[string]Map
 	Config       Map
@@ -42,6 +43,39 @@ func (that *MakeCode) Db2JSON(name string, path string, db db.HoTimeDB) {
 	} else {
 		that.Error.SetError(errors.New("配置文件不存在，或者配置出错，使用缺省默认配置"))
 	}
+	that.IndexMenus = Map{}
+	menusConfig := that.Config.GetSlice("menus")
+	//将配置写入到内存中仅作为判断用
+	if menusConfig != nil {
+		for kmenu, _ := range menusConfig {
+			menu := menusConfig.GetMap(kmenu)
+			if menu != nil {
+				mname := menu.GetString("table")
+				if mname == "" { //如果为空则不是表格
+					mname = menu.GetString("name")
+				}
+				that.IndexMenus[mname] = menu
+				childMenus := menu.GetSlice("menus")
+				if childMenus != nil {
+					for ckmenu, _ := range childMenus {
+						cmenu := childMenus.GetMap(ckmenu)
+						if cmenu != nil {
+							cname := cmenu.GetString("table")
+							if cmenu.GetString("table") == "" {
+								continue
+							}
+							that.IndexMenus[mname+"/"+cname] = cmenu
+						}
+
+					}
+
+				}
+
+			}
+
+		}
+	}
+
 	that.TableConfig = that.Config.GetMap("tables")
 	if that.TableConfig == nil {
 		that.TableConfig = Map{}
@@ -76,7 +110,7 @@ func (that *MakeCode) Db2JSON(name string, path string, db db.HoTimeDB) {
 	for _, v := range nowTables {
 		if that.TableConfig.GetMap(v.GetString("name")) == nil {
 			if v.GetString("label") == "" {
-				v["label"] = "备注"
+				v["label"] = v.GetString("name")
 			}
 			that.TableConfig[v.GetString("name")] = Map{
 				"label":   v.GetString("label"),
@@ -108,6 +142,7 @@ func (that *MakeCode) Db2JSON(name string, path string, db db.HoTimeDB) {
 		if that.TableColumns[v.GetString("name")] == nil {
 			that.TableColumns[v.GetString("name")] = make(map[string]Map)
 		}
+
 		tableInfo := make([]Map, 0)
 		if db.Type == "mysql" {
 			tableInfo = db.Select("INFORMATION_SCHEMA.COLUMNS", "COLUMN_NAME AS name,COLUMN_TYPE AS type,COLUMN_COMMENT AS label", Map{"AND": Map{"TABLE_SCHEMA": db.DBName, "TABLE_NAME": v.GetString("name")}})
@@ -119,7 +154,7 @@ func (that *MakeCode) Db2JSON(name string, path string, db db.HoTimeDB) {
 		idSlice = append(idSlice, tableInfo)
 		for _, info := range tableInfo {
 			if info.GetString("label") == "" {
-				info["label"] = "备注"
+				info["label"] = info.GetString("name")
 			}
 			coloum := that.TableColumns[v.GetString("name")][info.GetString("name")]
 
@@ -220,16 +255,68 @@ func (that *MakeCode) Db2JSON(name string, path string, db db.HoTimeDB) {
 		}
 		return
 	}
-	//数据生成完后开始关联id
+	//数据生成完后进一步解析
 	for fk, fv := range that.TableColumns {
+
+		//判断是否将表写入menu中
+		isMenusGet := false //判断是否被目录收录
+		for indexKey, _ := range that.IndexMenus {
+			indexCode := strings.Index(indexKey, fk)
+			//如果相等或者表名在目录中已经设置（主要前一位是/并且他是最后一个字符串）
+			if indexKey == fk || (indexCode != -1 && indexKey[indexCode-1] == '/' && indexKey[indexCode:] == fk) {
+				isMenusGet = true
+				break
+			}
+		}
+
+		//目录没有收录
+		if !isMenusGet {
+
+			tablePrefixCode := strings.Index(fk, "_")
+			isNewPrefix := false //假定表名没有前缀
+			prefixName := ""
+			//并且代表有前缀，根据数据表分库设定使用
+			if tablePrefixCode != -1 {
+				prefixName = fk[:tablePrefixCode]
+			}
+			if tablePrefixCode != -1 {
+				for ck, _ := range that.TableColumns {
+					//判断不止一个前缀相同
+					if strings.Index(ck, prefixName) == 0 && ck != prefixName && ck != fk {
+						isNewPrefix = true
+						break
+					}
+				}
+			}
+			menuIns := Map{"label": that.TableConfig.GetMap(fk).GetString("label"), "table": fk}
+			//多耗费一点内存
+			mMenu := Map{"menus": Slice{menuIns}, "label": that.TableConfig.GetMap(fk).GetString("label"), "name": prefixName, "icon": "el-icon-setting"}
+			//表名有前缀
+			if !isNewPrefix {
+				//是否已有对应前缀，已经有对应的menu只需要push进去即可
+				prefixName = DefaultMenuParentName
+				mMenu = Map{"menus": Slice{menuIns}, "label": "系统管理", "name": prefixName, "icon": "el-icon-setting"}
+			}
+			//没有新前缀
+			if that.IndexMenus[prefixName] != nil {
+				that.IndexMenus.GetMap(prefixName)["menus"] = append(that.IndexMenus.GetMap(prefixName).GetSlice("menus"), menuIns)
+				that.IndexMenus[prefixName+"/"+fk] = menuIns
+			} else {
+
+				that.Config["menus"] = append(that.Config.GetSlice("menus"), mMenu) //注入配置
+				//写入Index
+				that.IndexMenus[prefixName] = mMenu
+				that.IndexMenus[prefixName+"/"+fk] = menuIns
+			}
+		}
+
 		for k, v := range fv {
+			//虚招后缀是_id结尾的表字段
 			if len(k) <= 3 || strings.LastIndex(k, "_id") != len(k)-3 {
 				continue
 			}
-
 			//普通表匹配
 			oldTableName := k[:len(k)-3]
-
 			//上级ID匹配
 			if oldTableName == "parent" {
 				oldTableName = fk
@@ -258,9 +345,9 @@ func (that *MakeCode) Db2JSON(name string, path string, db db.HoTimeDB) {
 						tableName = oldTableName
 					}
 				}
-
 			}
 
+			//如果数据匹配则写入到配置中
 			if that.TableConfig[tableName] != nil {
 				v["link"] = tableName
 				//一般查询name字段，如果没有name字段则默认第二个地段