From dcf9d1324c4ecc23bc0b120ee9af9f0f74cde034 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期一, 08 一月 2024 19:04:50 +0800
Subject: [PATCH] swagger提交

---
 controllers/location.go |    8 +++++++-
 1 files changed, 7 insertions(+), 1 deletions(-)

diff --git a/controllers/location.go b/controllers/location.go
index e1102fa..80adda5 100644
--- a/controllers/location.go
+++ b/controllers/location.go
@@ -68,7 +68,8 @@
 	if params.PageInfo.Check() {
 		search.SetPage(params.Page, params.PageSize)
 	}
-	list, total, err := search.SetKeyword(params.KeyWord).SetType(params.Type).SetJointName(params.JointName).SetOrder("created_at desc").Find()
+	list, total, err := search.SetKeyword(params.KeyWord).SetType(params.Type).SetJointName(params.JointName).
+		SetIsScrapLocation(params.IsScrapLocation).SetOrder("created_at desc").Find()
 	if err != nil {
 		util.ResponseFormat(c, code.RequestParamError, "鏌ユ壘澶辫触")
 		return
@@ -98,6 +99,11 @@
 		if location.ParentId == 0 {
 			tree = append(tree, location)
 		} else {
+			if _, ok := m[location.ParentId]; !ok {
+				tree = append(tree, location)
+				continue
+			}
+			m[location.ParentId].Children = make([]*models.Location, 0)
 			m[location.ParentId].Children = append(m[location.ParentId].Children, location)
 		}
 	}

--
Gitblit v1.8.0