golang遞歸實(shí)現(xiàn)類別樹(CategoryTree)

需求描述

點(diǎn)擊一級(jí)或二級(jí)或三級(jí)類別都會(huì)進(jìn)行查詢踱蠢，選中的類別變?yōu)樽仙?/p>

效果圖

數(shù)據(jù)庫(kù)設(shè)計(jì)

model

// ArtAssetCategory 資產(chǎn)分類表 mapped from table <art_asset_category>
type ArtAssetCategory struct {
    ID        int32  `gorm:"column:id;primaryKey" json:"id"`
    ParentID  int32  `gorm:"column:parent_id;not null" json:"parent_id"`         // 上級(jí)分類
    Name      string `gorm:"column:name;not null" json:"name"`                   // 分類名稱
    CreatedAt int64  `gorm:"column:created_at;autoCreateTime" json:"created_at"` // 創(chuàng)建日期
    Remark    string `gorm:"column:remark;not null" json:"remark"`               // 備注
    Order     int32  `gorm:"column:order;not null" json:"order"`                 // 位置
}

表

實(shí)現(xiàn)思想

  1. 將根類別（ParentID為0）的類別裝入 list

  2. 否則裝入categoryMap（key：id暮屡，value：類別）

  3. 遍歷 list揉燃，根據(jù)根類別 ID 從 categoryMap 中查出當(dāng)前類別下的子類別

    1. 若一個(gè)類別的 ParentID == categroyId（list.id）帽馋，則該類別為 categroyId(list.id) 下的子類別
    2. 將全部子類別裝入 list
      1. 遍歷 list窥妇，遞歸查詢

  6. 根據(jù)order(位置)從小到大 排序

  7. 返回 包含子類別（Child）的類別樹（CategoryTree）

golang代碼實(shí)現(xiàn)

handler

// CategoryTree 所有分類 樹形下拉接口
func (ah *ArtAssetHandler) CategoryTree(c *gin.Context) {
    ctx := c.Request.Context()
    resp, err := ah.artAssetLogic.AssetCategoryTree(ctx)
    if err != nil {
        ah.Fail(c, err)
        return
    }
    ah.Success(c, resp)
}

logic

func (aal *ArtAssetLogic) AssetCategoryTree(ctx context.Context) (*bean.AssetCategoryTreeResp, error) {

    alllist, err := aal.artAssetService.GetAllCategoryList(ctx)
    if err != nil {
        return nil, err
    }
    // 將根類別（ParentID為0）的類別裝入list吉执； 
    // 否則裝入categoryMap（key：id涕蚤，value：類別）
    categoryMap := make(map[int32]*model.ArtAssetCategory)
    list := make([]*model.ArtAssetCategory, 0) // 根級(jí)別 有可能是多個(gè) 所以是數(shù)組
    for _, v := range alllist {
        if v.ParentID == 0 { // 第一級(jí)別的列表 parentid  ==0 默認(rèn)0
            list = append(list, v)
        } else {
            categoryMap[v.ID] = v
        }
    }
    // 包含子類別（Child）的類別樹（CategoryTree）
    childList := make([]*bean.AssetCategoryTreeChild, 0, len(list))
    for _, v := range list {
        child := &bean.AssetCategoryTreeChild{
            ID:       v.ID,
            Name:     v.Name,
            ParentId: v.ParentID,
            Order:    v.Order,
        }
        // 根據(jù) ID 查出當(dāng)前類別下的子類別
        child.Child = aal.aggreCategoryTree(categoryMap, v.ID)
        childList = append(childList, child)
    }

    // 根據(jù)order(位置)從小到大 排序
    sort.Slice(childList, func(i, j int) bool {
        return childList[i].Order < childList[j].Order
    })
    return &bean.AssetCategoryTreeResp{
        Tree: childList,
    }, nil
}

func (aal *ArtAssetLogic) aggreCategoryTree(cmap map[int32]*model.ArtAssetCategory, categroyId int32) []*bean.AssetCategoryTreeChild {
    if categroyId <= 0 || len(cmap) <= 0 {
        return nil
    }
    // 查詢 categroyId 下的子類別
    // 若一個(gè)類別的 ParentID == categroyId，則該類別為 categroyId 下的子類別
    // 將全部子類別裝入list
    list := make([]*model.ArtAssetCategory, 0)
    for _, v := range cmap {
        if v.ParentID == categroyId { // 第一級(jí)別的列表
            list = append(list, v)
        }
    }
    if len(list) <= 0 {
        return nil
    }
    // 包含子類別（Child）的類別樹（CategoryTree）
    childList := make([]*bean.AssetCategoryTreeChild, 0, len(list))
    for _, v := range list {
        child := &bean.AssetCategoryTreeChild{
            ID:       v.ID,
            Name:     v.Name,
            ParentId: v.ParentID,
            Order:    v.Order,
        }
        // 根據(jù) ID 查出當(dāng)前類別下的子類別（遞歸）
        sublist := aal.aggreCategoryTree(cmap, v.ID)
        if sublist != nil && len(sublist) > 0 {
            child.Child = sublist
        }
        childList = append(childList, child)
    }
    // 根據(jù)order(位置)從小到大 排序
    sort.Slice(childList, func(i, j int) bool {
        return childList[i].Order < childList[j].Order
    })
    return childList
}

service

// 獲取資產(chǎn)分類 所有乘寒，內(nèi)存做樹形結(jié)構(gòu)組裝 只適合數(shù)量少的情況望众，數(shù)量大的一定讓前端組裝
func (aal *ArtAssetService) GetAllCategoryList(ctx context.Context) ([]*model.ArtAssetCategory, error) {
    resList := make([]*model.ArtAssetCategory, 0)
    db := gmysql.DB(ctx, config.GlobConfig.Mysql.DBName) //連接到數(shù)據(jù)庫(kù)
    categorys := query.Use(db).ArtAssetCategory
    err := categorys.WithContext(ctx).Order(categorys.Order).Scan(&resList)
    if err != nil {
        return nil, err
    }
    return resList, nil

}

bean

type AssetCategoryTreeResp struct {
    Tree []*AssetCategoryTreeChild `json:"list"`
}

type AssetCategoryTreeChild struct {
    ID       int32                     `json:"id"`
    Name     string                    `json:"name"`
    ParentId int32                     `json:"parent_id"`
    Child    []*AssetCategoryTreeChild `json:"child"`
    Order    int32                     `json:"order"`
}

請(qǐng)求結(jié)果

{
    "code": 0,
    "message": "",
    "data": {
        "list": [
            {
                "id": 1,
                "name": "場(chǎng)景",
                "parent_id": 0,
                "child": [
                    {
                        "id": 9,
                        "name": "自然場(chǎng)景",
                        "parent_id": 1,
                        "child": null,
                        "order": 1
                    },
                    {
                        "id": 10,
                        "name": "建筑",
                        "parent_id": 1,
                        "child": null,
                        "order": 2
                    },
                    {
                        "id": 11,
                        "name": "地形",
                        "parent_id": 1,
                        "child": null,
                        "order": 3
                    },
                    {
                        "id": 12,
                        "name": "科技風(fēng)",
                        "parent_id": 1,
                        "child": null,
                        "order": 4
                    },
                    {
                        "id": 13,
                        "name": "現(xiàn)代風(fēng)",
                        "parent_id": 1,
                        "child": null,
                        "order": 5
                    },
                    {
                        "id": 14,
                        "name": "鄉(xiāng)村風(fēng)",
                        "parent_id": 1,
                        "child": null,
                        "order": 6
                    },
                    {
                        "id": 15,
                        "name": "場(chǎng)景部件",
                        "parent_id": 1,
                        "child": null,
                        "order": 7
                    },
                    {
                        "id": 16,
                        "name": "其他（場(chǎng)景）",
                        "parent_id": 1,
                        "child": null,
                        "order": 8
                    }
                ],
                "order": 0
            },
            {
                "id": 2,
                "name": "人物",
                "parent_id": 0,
                "child": [
                    {
                        "id": 17,
                        "name": "成年人-男",
                        "parent_id": 2,
                        "child": null,
                        "order": 1
                    },
                    {
                        "id": 18,
                        "name": "成年人-女",
                        "parent_id": 2,
                        "child": null,
                        "order": 2
                    },
                    {
                        "id": 19,
                        "name": "孩子-男",
                        "parent_id": 2,
                        "child": null,
                        "order": 3
                    },
                    {
                        "id": 20,
                        "name": "孩子-女",
                        "parent_id": 2,
                        "child": null,
                        "order": 4
                    },
                    {
                        "id": 21,
                        "name": "老人-男",
                        "parent_id": 2,
                        "child": null,
                        "order": 5
                    },
                    {
                        "id": 22,
                        "name": "老人-女",
                        "parent_id": 2,
                        "child": null,
                        "order": 6
                    },
                    {
                        "id": 23,
                        "name": "其他（人物）",
                        "parent_id": 2,
                        "child": null,
                        "order": 7
                    }
                ],
                "order": 1
            },
            {
                "id": 3,
                "name": "動(dòng)物",
                "parent_id": 0,
                "child": [
                    {
                        "id": 24,
                        "name": "哺乳",
                        "parent_id": 3,
                        "child": null,
                        "order": 1
                    },
                    {
                        "id": 25,
                        "name": "飛禽",
                        "parent_id": 3,
                        "child": null,
                        "order": 2
                    },
                    {
                        "id": 26,
                        "name": "爬行",
                        "parent_id": 3,
                        "child": null,
                        "order": 3
                    },
                    {
                        "id": 27,
                        "name": "恐龍",
                        "parent_id": 3,
                        "child": null,
                        "order": 4
                    },
                    {
                        "id": 28,
                        "name": "昆蟲",
                        "parent_id": 3,
                        "child": null,
                        "order": 5
                    },
                    {
                        "id": 29,
                        "name": "魚類",
                        "parent_id": 3,
                        "child": null,
                        "order": 6
                    },
                    {
                        "id": 30,
                        "name": "兩棲",
                        "parent_id": 3,
                        "child": null,
                        "order": 7
                    },
                    {
                        "id": 31,
                        "name": "機(jī)器動(dòng)物",
                        "parent_id": 3,
                        "child": null,
                        "order": 8
                    },
                    {
                        "id": 32,
                        "name": "其他（動(dòng)物）",
                        "parent_id": 3,
                        "child": null,
                        "order": 9
                    }
                ],
                "order": 2
            },
            {
                "id": 4,
                "name": "植物",
                "parent_id": 0,
                "child": [
                    {
                        "id": 33,
                        "name": "石頭",
                        "parent_id": 4,
                        "child": null,
                        "order": 1
                    },
                    {
                        "id": 34,
                        "name": "花卉",
                        "parent_id": 4,
                        "child": null,
                        "order": 2
                    },
                    {
                        "id": 35,
                        "name": "樹木",
                        "parent_id": 4,
                        "child": null,
                        "order": 3
                    },
                    {
                        "id": 36,
                        "name": "草類",
                        "parent_id": 4,
                        "child": null,
                        "order": 4
                    },
                    {
                        "id": 37,
                        "name": "水果蔬菜",
                        "parent_id": 4,
                        "child": null,
                        "order": 5
                    },
                    {
                        "id": 38,
                        "name": "其他（植物）",
                        "parent_id": 4,
                        "child": null,
                        "order": 6
                    }
                ],
                "order": 3
            },
            {
                "id": 5,
                "name": "道具",
                "parent_id": 0,
                "child": [
                    {
                        "id": 39,
                        "name": "武器",
                        "parent_id": 5,
                        "child": null,
                        "order": 1
                    },
                    {
                        "id": 40,
                        "name": "家具/生活用具",
                        "parent_id": 5,
                        "child": [
                            {
                                "id": 49,
                                "name": "沙發(fā)",
                                "parent_id": 40,
                                "child": null,
                                "order": 1
                            },
                            {
                                "id": 50,
                                "name": "桌椅",
                                "parent_id": 40,
                                "child": null,
                                "order": 2
                            },
                            {
                                "id": 51,
                                "name": "床",
                                "parent_id": 40,
                                "child": null,
                                "order": 3
                            },
                            {
                                "id": 52,
                                "name": "柜子",
                                "parent_id": 40,
                                "child": null,
                                "order": 4
                            },
                            {
                                "id": 53,
                                "name": "門窗",
                                "parent_id": 40,
                                "child": null,
                                "order": 5
                            },
                            {
                                "id": 54,
                                "name": "燈具",
                                "parent_id": 40,
                                "child": null,
                                "order": 6
                            }
                        ],
                        "order": 2
                    },
                    {
                        "id": 41,
                        "name": "食品/飲料/藥品",
                        "parent_id": 5,
                        "child": [
                            {
                                "id": 55,
                                "name": "水果",
                                "parent_id": 41,
                                "child": null,
                                "order": 1
                            },
                            {
                                "id": 56,
                                "name": "蔬菜",
                                "parent_id": 41,
                                "child": null,
                                "order": 2
                            },
                            {
                                "id": 57,
                                "name": "零食",
                                "parent_id": 41,
                                "child": null,
                                "order": 3
                            },
                            {
                                "id": 58,
                                "name": "飲料",
                                "parent_id": 41,
                                "child": null,
                                "order": 4
                            }
                        ],
                        "order": 3
                    },
                    {
                        "id": 42,
                        "name": "其他（道具）",
                        "parent_id": 5,
                        "child": null,
                        "order": 4
                    }
                ],
                "order": 4
            },
            {
                "id": 6,
                "name": "載具",
                "parent_id": 0,
                "child": [
                    {
                        "id": 43,
                        "name": "車輛",
                        "parent_id": 6,
                        "child": null,
                        "order": 1
                    },
                    {
                        "id": 44,
                        "name": "船艇",
                        "parent_id": 6,
                        "child": null,
                        "order": 2
                    },
                    {
                        "id": 45,
                        "name": "飛機(jī)/航空器",
                        "parent_id": 6,
                        "child": null,
                        "order": 3
                    },
                    {
                        "id": 46,
                        "name": "其他（載具）",
                        "parent_id": 6,
                        "child": null,
                        "order": 4
                    }
                ],
                "order": 5
            },
            {
                "id": 7,
                "name": "怪物",
                "parent_id": 0,
                "child": [
                    {
                        "id": 47,
                        "name": "其他（怪物）",
                        "parent_id": 7,
                        "child": null,
                        "order": 0
                    }
                ],
                "order": 6
            },
            {
                "id": 8,
                "name": "其他",
                "parent_id": 0,
                "child": [
                    {
                        "id": 48,
                        "name": "其他（其他）",
                        "parent_id": 8,
                        "child": null,
                        "order": 0
                    }
                ],
                "order": 7
            }
        ]
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 人面猴
    序言：七十年代末，一起剝皮案震驚了整個(gè)濱河市伞辛，隨后出現(xiàn)的幾起案子烂翰，更是在濱河造成了極大的恐慌，老刑警劉巖蚤氏，帶你破解...
    沈念sama閱讀 216,692評(píng)論 6贊 501
  • 死咒
    序言：濱河連續(xù)發(fā)生了三起死亡事件甘耿，死亡現(xiàn)場(chǎng)離奇詭異，居然都是意外死亡竿滨，警方通過查閱死者的電腦和手機(jī)佳恬，發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,482評(píng)論 3贊 392
  • 救了他兩次的神仙讓他今天三更去死
    文/潘曉璐 我一進(jìn)店門，熙熙樓的掌柜王于貴愁眉苦臉地迎上來姐呐，“玉大人殿怜，你說我怎么就攤上這事∈锷埃” “怎么了头谜？”我有些...
    開封第一講書人閱讀 162,995評(píng)論 0贊 353
  • 道士緝兇錄：失蹤的賣姜人
    文/不壞的土叔 我叫張陵，是天一觀的道長(zhǎng)鸠澈。 經(jīng)常有香客問我柱告，道長(zhǎng)截驮，這世上最難降的妖魔是什么？ 我笑而不...
    開封第一講書人閱讀 58,223評(píng)論 1贊 292
  • ?港島之戀（遺憾婚禮）
    正文 為了忘掉前任际度，我火速辦了婚禮葵袭，結(jié)果婚禮上，老公的妹妹穿的比我還像新娘乖菱。我一直安慰自己坡锡，他們只是感情好，可當(dāng)我...
    茶點(diǎn)故事閱讀 67,245評(píng)論 6贊 388
  • 惡毒庶女頂嫁案：這布局不是一般人想出來的
    文/花漫 我一把揭開白布窒所。 她就那樣靜靜地躺著鹉勒，像睡著了一般。 火紅的嫁衣襯著肌膚如雪吵取。 梳的紋絲不亂的頭發(fā)上禽额，一...
    開封第一講書人閱讀 51,208評(píng)論 1贊 299
  • 城市分裂傳說
    那天，我揣著相機(jī)與錄音皮官，去河邊找鬼脯倒。 笑死，一個(gè)胖子當(dāng)著我的面吹牛捺氢，可吹牛的內(nèi)容都是我干的藻丢。 我是一名探鬼主播，決...
    沈念sama閱讀 40,091評(píng)論 3贊 418
  • 雙鴛鴦連環(huán)套：你想象不到人心有多黑
    文/蒼蘭香墨 我猛地睜開眼讯沈，長(zhǎng)吁一口氣：“原來是場(chǎng)噩夢(mèng)啊……” “哼郁岩！你這毒婦竟也來了？” 一聲冷哼從身側(cè)響起缺狠，我...
    開封第一講書人閱讀 38,929評(píng)論 0贊 274
  • 萬榮殺人案實(shí)錄
    序言：老撾萬榮一對(duì)情侶失蹤，失蹤者是張志新（化名）和其女友劉穎萍摊，沒想到半個(gè)月后挤茄，有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體，經(jīng)...
    沈念sama閱讀 45,346評(píng)論 1贊 311
  • ?護(hù)林員之死
    正文 獨(dú)居荒郊野嶺守林人離奇死亡冰木，尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,570評(píng)論 2贊 333
  • ?白月光啟示錄
    正文 我和宋清朗相戀三年穷劈，在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片踊沸。...
    茶點(diǎn)故事閱讀 39,739評(píng)論 1贊 348
  • 活死人
    序言：一個(gè)原本活蹦亂跳的男人離奇死亡歇终，死狀恐怖，靈堂內(nèi)的尸體忽然破棺而出逼龟，到底是詐尸還是另有隱情评凝，我是刑警寧澤，帶...
    沈念sama閱讀 35,437評(píng)論 5贊 344
  • ?日本核電站爆炸內(nèi)幕
    正文 年R本政府宣布腺律，位于F島的核電站奕短，受9級(jí)特大地震影響宜肉，放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜翎碑，卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,037評(píng)論 3贊 326
  • 男人毒藥：我在死后第九天來索命
    文/蒙蒙 一谬返、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧日杈，春花似錦遣铝、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,677評(píng)論 0贊 22
  • 一樁弒父案翰蠢，背后竟有這般陰謀
    文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至啰劲，卻和暖如春梁沧，著一層夾襖步出監(jiān)牢的瞬間，已是汗流浹背蝇裤。 一陣腳步聲響...
    開封第一講書人閱讀 32,833評(píng)論 1贊 269
  • 情欲美人皮
    我被黑心中介騙來泰國(guó)打工廷支， 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留，地道東北人栓辜。 一個(gè)月前我還...
    沈念sama閱讀 47,760評(píng)論 2贊 369
  • 代替公主和親
    正文 我出身青樓恋拍，卻偏偏與公主長(zhǎng)得像，于是被迫代替她去往敵國(guó)和親藕甩。 傳聞我的和親對(duì)象是個(gè)殘疾皇子施敢，可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,647評(píng)論 2贊 354

推薦閱讀更多精彩內(nèi)容

  • 1.abstract class 和interface 有什么區(qū)別? 抽象類可以有構(gòu)造方法 接口不行 抽象類可以有...
    sunnysans閱讀 852評(píng)論 0贊 1
  • 操作系統(tǒng) 操作系統(tǒng)的特點(diǎn)？ – 共享：資源可被多個(gè)并發(fā)執(zhí)行的進(jìn)程使用– 并發(fā)：可以在同一時(shí)間間隔處理多個(gè)進(jìn)程狭莱，需要...
    不在窩里閱讀 718評(píng)論 0贊 1
  • 1. Java基礎(chǔ)部分 基礎(chǔ)部分的順序：基本語(yǔ)法僵娃，類相關(guān)的語(yǔ)法，內(nèi)部類的語(yǔ)法腋妙，繼承相關(guān)的語(yǔ)法默怨，異常的語(yǔ)法，線程的語(yǔ)...
    子非魚_t_閱讀 31,624評(píng)論 18贊 399
  • 一. Java基礎(chǔ)部分.................................................
    wy_sure閱讀 3,810評(píng)論 0贊 11
  • 程序設(shè)計(jì)中常使用樹型結(jié)構(gòu)來表征某些數(shù)據(jù)的關(guān)聯(lián)關(guān)系骤素，如上下級(jí)匙睹、欄目結(jié)構(gòu)、商品分類济竹、菜單痕檬、回復(fù)等。 分類的層級(jí)關(guān)系可以...
    JunChow520閱讀 4,098評(píng)論 4贊 3