hongze_cygx/services/industry_map.go

package services

import (
	"hongze/hongze_cygx/models"
	"math/rand"
	"reflect"
	"sort"
	"time"
)

func MakeTree(allNode []*models.CygxIndustryMapItems, node *models.CygxIndustryMapItems) {
	childs, _ := haveChild(allNode, node) //判断节点是否有子节点并返回
	if len(childs) > 0 {
		node.Children = append(node.Children, childs[0:]...) //添加子节点
		for _, v := range childs {                           //查询子节点的子节点，并添加到子节点
			_, has := haveChild(allNode, v)
			if has {
				MakeTree(allNode, v) //递归添加节点
			}
		}
	}
}

func haveChild(allNode []*models.CygxIndustryMapItems, node *models.CygxIndustryMapItems) (childs []*models.CygxIndustryMapItems, yes bool) {
	for _, v := range allNode {
		if v.ParentId == node.IndustryMapId {
			childs = append(childs, v)
		}
	}
	if len(childs) > 0 {
		yes = true
	}
	return
}

func GetIndustryMap() {
	list, err := models.GetCygxIndustryMapByParentId(0)
	if err != nil {
		return
	}
	rootNode := list[0]
	allNodes, err := models.GetCygxIndustryMapAll()
	if err != nil {
		return
	}
	MakeTree(allNodes, rootNode)
}

func GetIndustryTree() (rootNode *models.CygxIndustryMapItems, err error) {
	list, err := models.GetCygxIndustryMapByParentId(0)
	if err != nil {
		return
	}
	rootNode = list[0]
	allNodes, err := models.GetCygxIndustryMapAll()
	if err != nil {
		return
	}
	MakeTree(allNodes, rootNode)
	return
}

//获取对应的层级map
func getMap(tree *models.CygxIndustryMapItems, item *models.CygxIndustryMapItems, hasIdMap map[int]string, otherChildMapSlice map[int][]*models.CygxIndustryMapItems) {
	//深度
	depth := 1

	//获取下级
	childTree := getChildTree(tree, item)
	depth = childTreeToSlice(childTree, hasIdMap, depth, otherChildMapSlice)

	//获取上级
	var tmpParentSlice []*models.CygxIndustryMapItems
	tmpParentSlice, _ = parentTreeToSlice(tree, item, 0, tmpParentSlice, hasIdMap)
	//移除第一个最上级的行业（根节点）
	tmpParentSlice = append(tmpParentSlice[:0], tmpParentSlice[1:len(tmpParentSlice)]...)
	//切片反转
	tmpParentSlice = reverse(tmpParentSlice)
	for _, v := range tmpParentSlice {
		depth++
		if _, ok := hasIdMap[v.IndustryMapId]; ok == false {
			otherChildMapSlice[depth] = append(otherChildMapSlice[depth], v)
		}
	}

	//获取同级
	//fmt.Println("===============")
	depth = siblingTreeToSlice(tree, item, hasIdMap, depth, otherChildMapSlice)

	//上级的同级~
	/*for _,parentSlice := range tmpParentSlice{
		depth = siblingTreeToSlice(tree,parentSlice,hasIdMap,depth,otherChildMapSlice)
	}*/
	return
}

//获取行业图谱切片
func GetIndustryMapNameSlice(industryName string) (nameSlice []string, err error) {
	nameSlice = append(nameSlice, industryName)
	tree, err := GetIndustryTree()
	if err != nil {
		return
	}
	//已经存在的行业id的map集合
	hasIdMap := make(map[int]string)
	itemList, err := models.GetFirstCygxIndustryListByName(industryName)
	if err != nil {
		return
	}
	//找不到对应的数据
	if len(itemList) <= 0 {
		return
	}

	industryMapList := make(map[int][]*models.CygxIndustryMapItems)
	for _, item := range itemList {
		industryMapList[item.Level] = append(industryMapList[item.Level], item)
	}

	//将查出来的根节点数据的key取出来，放在切片中，并对该切片做正序排列
	var sortIndustryList []int
	for k, _ := range industryMapList {
		sortIndustryList = append(sortIndustryList, k)
	}
	sort.Ints(sortIndustryList)

	//最底层节点的数据集合
	list := industryMapList[sortIndustryList[0]]

	otherChildMapSlice := map[int][]*models.CygxIndustryMapItems{}
	for _, item := range list {
		hasIdMap[item.IndustryMapId] = ""
		//将自己的节点给加进去
		otherChildMapSlice[0] = append(otherChildMapSlice[0], item)
		getMap(tree, item, hasIdMap, otherChildMapSlice)
	}
	var tmpSlice []*models.CygxIndustryMapItems

	//将其他规律数据的key取出来，放在切片中，并对该切片做正序排列
	var sortList []int
	for k, _ := range otherChildMapSlice {
		sortList = append(sortList, k)
	}
	sort.Ints(sortList)

	//遍历该切片，根据下标key获取对应的数据，并插入到主数据中
	for _, v := range sortList {
		tmpChildSlice := otherChildMapSlice[v]
		randSlice(tmpChildSlice)
		tmpSlice = append(tmpSlice, tmpChildSlice...)
		//fmt.Println(k,"=====")
		//for _,tmpV := range otherChildMapSlice[v]{
		//	fmt.Println(tmpV.IndustryMapName)
		//}
	}
	//名字切片
	for _, v := range tmpSlice {
		//fmt.Println("k===",k,"=======v=======",v)
		nameSlice = append(nameSlice, v.IndustryMapName)
	}

	//fmt.Println(nameSlice)
	//fmt.Println(strings.Join(nameSlice,","))
	//utils.FileLog.Info("allNodes:%s",strings.Join(nameSlice,","))
	return
}

//切片反转
func reverse(s []*models.CygxIndustryMapItems) []*models.CygxIndustryMapItems {
	for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
		s[i], s[j] = s[j], s[i]
	}
	return s
}

//获取当前节点的树
func getChildTree(rootNode *models.CygxIndustryMapItems, nowNode *models.CygxIndustryMapItems) (returnNode *models.CygxIndustryMapItems) {
	if rootNode.IndustryMapId == nowNode.IndustryMapId {
		returnNode = rootNode
		return
	}
	if rootNode.Children != nil {
		for _, v := range rootNode.Children {
			if v.IndustryMapId == nowNode.IndustryMapId {
				returnNode = v
			} else {
				returnNode = getChildTree(v, nowNode)
			}
			if returnNode != nil {
				return
			}
		}
	}
	return
}

//获取子树V2，与上一个子树规律不一样，因为上级
func childTreeToSlice(rootNode *models.CygxIndustryMapItems, hasIdMap map[int]string, depth int, otherChildMapSlice map[int][]*models.CygxIndustryMapItems) (maxDepth int) {
	if rootNode.Children != nil {
		depth++
		maxDepth = depth
		for _, v := range rootNode.Children {
			//判断是否已经录入，如果已经录入，那么不处理
			if _, ok := hasIdMap[v.IndustryMapId]; ok == false {
				otherChildMapSlice[depth] = append(otherChildMapSlice[depth], v)
			}
		}

		for _, v := range rootNode.Children {
			//判断是否已经录入，如果已经录入，那么不处理
			if _, ok := hasIdMap[v.IndustryMapId]; ok == false {
				hasIdMap[v.IndustryMapId] = ""
				returnDepth := childTreeToSlice(v, hasIdMap, depth, otherChildMapSlice)
				if returnDepth > maxDepth {
					maxDepth = returnDepth
				}
			}
		}
	}
	return maxDepth
}

//获取兄弟级树
func siblingTreeToSlice(rootNode *models.CygxIndustryMapItems, nowNode *models.CygxIndustryMapItems, hasIdMap map[int]string, depth int, otherChildMapSlice map[int][]*models.CygxIndustryMapItems) (maxDepth int) {
	if rootNode.Children != nil {
		depth++
		maxDepth = depth
		for _, v := range rootNode.Children {
			//如果父级id一致的情况下，那么代表这是兄弟节点或者是自己了
			if v.ParentId == nowNode.ParentId {
				//判断是否已经录入，如果已经录入，那么不处理
				if _, ok := hasIdMap[v.IndustryMapId]; ok == false {
					otherChildMapSlice[depth] = append(otherChildMapSlice[depth], v)
					returnDepth := childTreeToSlice(v, hasIdMap, depth, otherChildMapSlice)
					if returnDepth > maxDepth {
						maxDepth = returnDepth
					}
				}
			} else {
				returnDepth := siblingTreeToSlice(v, nowNode, hasIdMap, depth, otherChildMapSlice)
				if returnDepth > maxDepth {
					maxDepth = returnDepth
				}
			}
		}

		for _, v := range rootNode.Children {
			if v.ParentId == nowNode.ParentId {
				if _, ok := hasIdMap[v.IndustryMapId]; ok == false {
					hasIdMap[v.IndustryMapId] = ""
				}

			}
		}
	}
	return
}

//获取上级树
func parentTreeToSlice(rootNode *models.CygxIndustryMapItems, nowNode *models.CygxIndustryMapItems, depth int, tmpReturnSlice []*models.CygxIndustryMapItems, hasIdMap map[int]string) (returnSlice []*models.CygxIndustryMapItems, ok bool) {
	if depth != 0 {
		returnSlice = tmpReturnSlice
	}
	depth++
	returnSlice = append(returnSlice, rootNode)
	//if _,ok := hasIdMap[rootNode.IndustryMapId];ok==false{
	//	hasIdMap[rootNode.IndustryMapId] = ""
	//}
	if rootNode.Children != nil {
		for _, v := range rootNode.Children {
			if v.ParentId == nowNode.ParentId {
				ok = true
				return
			}
			returnSlice, ok = parentTreeToSlice(v, nowNode, depth, returnSlice, hasIdMap)
			//如果返回的匹配完成，那么不进入下一次循环，直接返回
			if ok {
				return
			}

			//一次循环结束后，如果没有匹配上，那么移除临时切片中的数据
			returnSlice = append(returnSlice[:0], returnSlice[0:depth]...)
		}
	}

	return
}

//切片乱序
func randSlice(slice []*models.CygxIndustryMapItems) {
	rv := reflect.ValueOf(slice)
	if rv.Type().Kind() != reflect.Slice {
		return
	}

	length := rv.Len()
	if length < 2 {
		return
	}

	swap := reflect.Swapper(slice)
	rand.Seed(time.Now().Unix())
	for i := length - 1; i >= 0; i-- {
		j := rand.Intn(length)
		swap(i, j)
	}
	return
}

//region数据修复
func FixData() {
	tree, err := GetIndustryTree()
	if err != nil {
	}
	models.FixLevelData(tree.IndustryMapId, 1)
	fixData(tree, 1)
	return
}

func fixData(item *models.CygxIndustryMapItems, depth int) {
	depth++
	if item.Children == nil {
		return
	}
	for _, v := range item.Children {
		models.FixLevelData(v.IndustryMapId, depth)
		fixData(v, depth)
	}
}

//endregion