From bfdb3faf4f27df01718f58ac8c4ec0bcc092e7b6 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期六, 30 十一月 2024 18:50:47 +0800
Subject: [PATCH] 生成数据明细导出修改

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java |   23 +++++++++++++++++------
 1 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java
index 4c410e1..d44f85b 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java
@@ -15,6 +15,7 @@
 import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
+import java.util.Objects;
 
 /**
  * <p>
@@ -48,9 +49,10 @@
     public AjaxResult<List<TGoodsType>> list() {
         Long shopId = tokenService.getLoginUser().getObjectId();
         Integer roleType = tokenService.getLoginUser().getRoleType();
+        TShop shop = shopService.getById(shopId);
         LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>();
         if(roleType != 1){
-            wrapper.eq(TGoodsType::getShopId, shopId);
+            wrapper.eq(TGoodsType::getMealType, shop.getMealType());
         }
         return AjaxResult.success(goodsTypeService.list(wrapper));
     }
@@ -60,13 +62,19 @@
      */
     @ApiOperation( value = "查询商品分类列表-后台使用")
     @GetMapping(value = "/getList")
-    public AjaxResult<List<TGoodsType>> getList(@RequestParam("mealType") Integer mealType) {
+    public AjaxResult<List<TGoodsType>> getList(@RequestParam(value = "mealType",required = false) Integer mealType) {
         // 获取餐饮分类店铺id
-        TShop one = shopService.getOne(Wrappers.lambdaQuery(TShop.class)
-                .eq(TShop::getMealType, mealType)
-                .last("LIMIT 1"));
+//        TShop one = shopService.getOne(Wrappers.lambdaQuery(TShop.class)
+//                .eq(TShop::getMealType, mealType)
+//                .last("LIMIT 1"));
+//        if(Objects.isNull(one)){
+//            String str = mealType == 1 ? "中餐" : "火锅";
+//            return AjaxResult.error("无"+str+"类型店铺");
+//        }
         LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>();
-        wrapper.eq(TGoodsType::getShopId, one.getId());
+        if(Objects.nonNull(mealType)){
+            wrapper.eq(TGoodsType::getMealType, mealType);
+        }
         return AjaxResult.success(goodsTypeService.list(wrapper));
     }
 
@@ -76,6 +84,9 @@
     @ApiOperation( value = "添加商品分类")
     @PostMapping(value = "/add")
     public AjaxResult<Boolean> add(@RequestBody TGoodsType dto) {
+        Long shopId = tokenService.getLoginUser().getObjectId();
+        TShop shop = shopService.getById(shopId);
+        dto.setMealType(shop.getMealType());
         dto.setShopId(tokenService.getLoginUser().getObjectId());
         return AjaxResult.success(goodsTypeService.save(dto));
     }

--
Gitblit v1.7.1