From c5224303fbe084e9a2d9b2e34fd4f7ec806d3f91 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期四, 31 十月 2024 17:58:58 +0800
Subject: [PATCH] 修改
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java | 30 +++++++++++++++++++-----------
1 files changed, 19 insertions(+), 11 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..6247028 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));
}
/**
@@ -61,7 +62,7 @@
*/
@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)
@@ -70,9 +71,13 @@
// String str = mealType == 1 ? "中餐" : "火锅";
// return AjaxResult.error("无"+str+"类型店铺");
// }
-// LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>();
-// wrapper.eq(TGoodsType::getShopId, one.getId());
- return AjaxResult.success(goodsTypeService.list());
+ LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>();
+ if(Objects.isNull(mealType)){
+ wrapper.eq(TGoodsType::getMealType, 2);
+ }else {
+ wrapper.eq(TGoodsType::getMealType, mealType);
+ }
+ return AjaxResult.success(goodsTypeService.list(wrapper));
}
/**
@@ -81,6 +86,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