From 11a5e71b3e664cc96e292150dbf6beeba40b4282 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期四, 31 十月 2024 16:55:34 +0800
Subject: [PATCH] 修改

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java |   22 +++++++++++++---------
 ruoyi-system/src/main/resources/mapper/system/TGoodsTypeMapper.xml               |    3 ++-
 ruoyi-system/src/main/java/com/ruoyi/system/domain/TGoodsType.java               |    4 ++++
 3 files changed, 19 insertions(+), 10 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 b5e28f5..e19d6fe 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
@@ -47,13 +47,14 @@
     @ApiOperation( value = "查询商品分类列表")
     @PostMapping(value = "/list")
     public AjaxResult<List<TGoodsType>> list() {
-//        Long shopId = tokenService.getLoginUser().getObjectId();
-//        Integer roleType = tokenService.getLoginUser().getRoleType();
-//        LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>();
-//        if(roleType != 1){
-//            wrapper.eq(TGoodsType::getShopId, shopId);
-//        }
-        return AjaxResult.success(goodsTypeService.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::getMealType, shop.getMealType());
+        }
+        return AjaxResult.success(goodsTypeService.list(wrapper));
     }
 
     /**
@@ -70,8 +71,8 @@
 //            String str = mealType == 1 ? "中餐" : "火锅";
 //            return AjaxResult.error("无"+str+"类型店铺");
 //        }
-//        LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>();
-//        wrapper.eq(TGoodsType::getShopId, one.getId());
+        LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(TGoodsType::getMealType, mealType);
         return AjaxResult.success(goodsTypeService.list());
     }
 
@@ -81,6 +82,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));
     }
diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/TGoodsType.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/TGoodsType.java
index e8c0975..e274c72 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/domain/TGoodsType.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/TGoodsType.java
@@ -37,4 +37,8 @@
     @TableField("typeName")
     private String typeName;
 
+    @ApiModelProperty(value = "餐饮分类 1=中餐 2=火锅")
+    @TableField("mealType")
+    private Integer mealType;
+
 }
diff --git a/ruoyi-system/src/main/resources/mapper/system/TGoodsTypeMapper.xml b/ruoyi-system/src/main/resources/mapper/system/TGoodsTypeMapper.xml
index dcad072..97f2eeb 100644
--- a/ruoyi-system/src/main/resources/mapper/system/TGoodsTypeMapper.xml
+++ b/ruoyi-system/src/main/resources/mapper/system/TGoodsTypeMapper.xml
@@ -12,11 +12,12 @@
         <result column="createBy" property="createBy" />
         <result column="updateBy" property="updateBy" />
         <result column="shopId" property="shopId" />
+        <result column="mealType" property="mealType" />
     </resultMap>
 
     <!-- 通用查询结果列 -->
     <sql id="Base_Column_List">
-        id, typeName, createTime, updateTime, disabled, createBy, updateBy,shopId
+        id, typeName, createTime, updateTime, disabled, createBy, updateBy,shopId,mealType
     </sql>
     <delete id="deleteByShopId">
         delete from t_goods_type where shopId = #{shopId}

--
Gitblit v1.7.1