jybxseo/internal/service/areaStruct.go

package service

import (
	"context"
	"fmt"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gctx"
	"github.com/gogf/gf/v2/util/gconv"
	"jybxseo/internal/consts"
)

var (
	JyBxSeoAreaRoot *AreaRoot = &AreaRoot{}
)

type (
	AreaRoot struct {
		areaTree    map[string]*AreaNode //层级关系
		areaCodeMap map[string]*AreaNode //所有节点平铺 key:code
		areaNameMap map[string]*AreaNode //所有节点平铺 key:name
		rootList    []*AreaNode          //根节点
	}
	AreaNode struct {
		Name  string      `json:"name" doc:"名称"`
		Code  string      `json:"code" doc:"代码"`
		PNode *AreaNode   `json:"pNode" doc:"父节点代码"`
		Type  int         `json:"type" doc:"1省份 2城市 3直辖市"`
		Child []*AreaNode `json:"child" doc:"子栏目"`
	}
)

func init() {
	JyBxSeoAreaRoot.LoadAreasFrom(gctx.New())
}

// LoadAreasFrom 加载地区表示
func (aRoot *AreaRoot) LoadAreasFrom(ctx context.Context) {
	areaTree, areaCodeMap, areaNameMap := map[string]*AreaNode{}, map[string]*AreaNode{}, map[string]*AreaNode{}
	var rootArr []*AreaNode
	for _, m := range g.Cfg("global").MustGet(context.Background(), "allAreaTree").Maps() {
		root := &AreaNode{
			Name: gconv.String(m["name"]),
			Code: gconv.String(m["code"]),
		}
		if gconv.Bool(m["special"]) {
			root.Type = consts.CitySpecialType
		} else {
			root.Type = consts.AreaType
		}
		for _, n := range gconv.Maps(m["child"]) {
			node := &AreaNode{
				Name:  gconv.String(n["name"]),
				Code:  gconv.String(n["code"]),
				Type:  consts.CityType,
				PNode: root,
			}
			root.Child = append(root.Child, node)
			areaCodeMap[node.Code] = node
			areaNameMap[node.Name] = node
		}
		areaTree[root.Code] = root
		areaCodeMap[root.Code] = root
		areaCodeMap[root.Name] = root
		rootArr = append(rootArr, root)
	}
	aRoot.areaTree, aRoot.areaCodeMap = areaTree, areaCodeMap
	aRoot.areaNameMap = areaNameMap
	aRoot.rootList = rootArr
}

// GetData 获取地区数据
func (aRoot *AreaRoot) GetData(ctx context.Context, maxTotal int, query *SeoBiddingQuery) []map[string]interface{} {
	var sql string
	var values []interface{}
	if query.district != "" {
		sql += " AND b.district=? "
		values = append(values, query.district)
	} else if query.city != "" {
		sql += " AND b.city=? "
		values = append(values, query.city)
	} else if query.area != "" {
		sql += " AND b.area=? "
		values = append(values, query.area)
	}

	if query.topType != "" {
		if val, _ := consts.TopTypeMap[query.topType]; val != "" {
			sql += " AND b.toptype=? "
			values = append(values, val)
		} else {
			sql += " AND b.toptype=? "
			values = append(values, query.topType)
		}
	} else if query.subType != "" && query.keys == "" {
		sql += " AND b.subtype=? "
		values = append(values, query.subType)
	} else if query.topType == "" && query.subType == "" {
		sql += " AND (b.toptype !='拟建' AND b.toptype !='采购意向') "
	}
	values = append(values, maxTotal+50)
	queryRes, err := g.DB().Query(ctx, fmt.Sprintf(`SELECT b.bid_id
			FROM jyseo.new_areaClass b
			WHERE 1=1 %s
			ORDER BY b.publish_time DESC
			LIMIT 0,?`, sql), values...)
	if err != nil || queryRes.IsEmpty() {
		return nil
	}
	return FillingBiddingBaseFields(ctx, queryRes.List())
}

func (an *AreaNode) GetChildNode(code string) *AreaNode {
	if len(an.Child) == 0 {
		return nil
	}
	for _, node := range an.Child {
		if node.Code == code {
			return node
		}
	}
	return nil
}

// GetNodeByCode 根据code
func (aRoot *AreaRoot) GetNodeByCode(code string) *AreaNode {
	return aRoot.areaCodeMap[code]
}

// GetNodeByName 根据省份名字获取node
func (aRoot *AreaRoot) GetNodeByName(name string) *AreaNode {
	return aRoot.areaNameMap[name]
}

// GetAllRootNodes 获取全部根地区节点
func (aRoot *AreaRoot) GetAllRootNodes() []*AreaNode {
	return aRoot.rootList
}