From 4a9bc4b7c84985047c7ebe0b991e8c8364bb56a2 Mon Sep 17 00:00:00 2001 From: liujiandao <274878379@qq.com> Date: 星期五, 03 十一月 2023 10:58:25 +0800 Subject: [PATCH] 重订货规则修改 --- controllers/location.go | 28 ++++++++++++++++++++++++++++ 1 files changed, 28 insertions(+), 0 deletions(-) diff --git a/controllers/location.go b/controllers/location.go index 5bfec45..2ef2a95 100644 --- a/controllers/location.go +++ b/controllers/location.go @@ -76,6 +76,34 @@ util.ResponseFormatList(c, code.Success, list, int(total)) } +// GetLocationTreeList +// @Tags 浣嶇疆 +// @Summary 鑾峰彇浣嶇疆鍒楄〃鏍� +// @Produce application/json +// @Success 200 {object} util.ResponseList{data=[]models.Location} "鎴愬姛" +// @Router /api-wms/v1/location/getLocationTreeList [get] +func (slf LocationController) GetLocationTreeList(c *gin.Context) { + all, err := models.NewLocationSearch().SetType(3).FindAll() + if err != nil { + util.ResponseFormat(c, code.RequestParamError, "鏌ユ壘澶辫触") + return + } + var tree []*models.Location + m := make(map[int]*models.Location) + for _, location := range all { + m[location.Id] = location + } + for _, location := range all { + if location.ParentId == 0 { + tree = append(tree, location) + } else { + m[location.ParentId].Children = append(m[location.ParentId].Children, location) + } + } + + util.ResponseFormat(c, code.Success, tree) +} + // GetLocationDetails // @Tags 浣嶇疆 // @Summary 鑾峰彇浣嶇疆璇︽儏 -- Gitblit v1.8.0