1

src/main/java/com/cdf/controller/api/ShopApiController.java

@@ -34,11 +34,11 @@ public class ShopApiController {
     private IDeptService deptService;
 
     @GetMapping("/getShopCategory")
-    public ResultData getShopCategory(){
+    public ResultData getShopCategory(@RequestParam(required = false) String sceneNum){
         List<ShopCategory> list = shopCategoryService.list();
         List<ShopCategory> listVo = new ArrayList<>();
         for (ShopCategory category : list) {
-            List<Shop> byCategory = shopService.getByCategory(category.getId());
+            List<Shop> byCategory = shopService.getByCategoryAndSceneNum(category.getId(),sceneNum);
             if(byCategory != null && byCategory.size() >0){
                 listVo.add(category);
             }

src/main/java/com/cdf/service/IShopService.java

@@ -21,4 +21,6 @@ public interface IShopService extends IService<Shop> {
     PageInfo pageList(ShopRequest param);
 
     List<Shop>  getByCategory(Integer categoryId);
+
+    List<Shop> getByCategoryAndSceneNum(Integer categoryId, String sceneNum);
 }

src/main/java/com/cdf/service/impl/ShopServiceImpl.java

@@ -89,4 +89,12 @@ public class ShopServiceImpl extends ServiceImpl<IShopMapper, Shop> implements I
         wrapper.eq(Shop::getCategoryId,categoryId);
         return this.list(wrapper);
     }
+
+    @Override
+    public List<Shop> getByCategoryAndSceneNum(Integer categoryId, String sceneNum) {
+        LambdaQueryWrapper<Shop> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(Shop::getCategoryId,categoryId);
+        wrapper.eq(Shop::getSceneUrl,sceneNum);
+        return this.list(wrapper);
+    }
 }