推荐分类修改

business/jxstore/cms/store_sku.go

@@ -2140,18 +2140,38 @@ func GetTopSkusByStoreIDs(ctx *jxcontext.Context, storeIDs []int) (storeSkuNameE
 }
 
 func GetTopCategorysByStoreIDs(ctx *jxcontext.Context, storeIDs []int) (skuCategory []*model.SkuCategory, err error) {
-	var skuCategory2 []*model.SkuCategory
+	var (
+		skuCategory2   []*model.SkuCategory
+		skuCategoryMap = make(map[int]*model.SkuCategory)
+		limit          = 10
+	)
 	if len(storeIDs) == 0 {
 		return skuCategory, err
 	}
 	db := dao.GetDB()
-	skuCategory, err = dao.GetTopCategorysByStoreIDs(db, storeIDs)
-	if len(skuCategory) < 10 {
-		j := 10 - len(skuCategory)
+	skuCategory, err = dao.GetTopCategorysByStoreIDs(db, storeIDs, limit)
+	//若大于等于10个就不用做下面的操作
+	if len(skuCategory) >= limit {
+		return skuCategory, err
+	}
+	if len(skuCategory) > 0 {
+		for _, v := range skuCategory {
+			skuCategoryMap[v.ID] = v
+		}
+	}
+	//推荐分类，若不满10个，则填满10个
+	if (len(skuCategory) < limit && len(skuCategory) > 0) || len(skuCategory) == 0 {
 		skuCategory2, err = dao.GetCategories(db, -1, 1, nil)
 		if len(skuCategory2) > 0 {
-			for i := 0; i < j; i++ {
-				skuCategory = append(skuCategory, skuCategory2[i])
+			for _, v := range skuCategory2 {
+				if skuCategoryMap[v.ID] == nil {
+					if !strings.Contains(v.Name, "赠品") {
+						skuCategory = append(skuCategory, v)
+					}
+				}
+				if len(skuCategory) >= limit {
+					break
+				}
 			}
 		}
 	}