修改接口

xuhy

2024-09-19 45403c689e2e0d6e69d32e1833faabc116f859e2

 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java

@@ -1,7 +1,10 @@
package com.ruoyi.web.controller.api;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.domain.TGoodsType;
import com.ruoyi.system.service.TGoodsTypeService;
import io.swagger.annotations.Api;
@@ -25,10 +28,12 @@
public class TGoodsTypeController {

    private final TGoodsTypeService goodsTypeService;
    private final TokenService tokenService;

    @Autowired
    public TGoodsTypeController(TGoodsTypeService goodsTypeService) {
    public TGoodsTypeController(TGoodsTypeService goodsTypeService, TokenService tokenService) {
        this.goodsTypeService = goodsTypeService;
        this.tokenService = tokenService;
    }

    /**
@@ -37,7 +42,24 @@
    @ApiOperation( value = "查询商品分类列表")
    @PostMapping(value = "/list")
    public AjaxResult<List<TGoodsType>> list() {
        return AjaxResult.success(goodsTypeService.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(wrapper));
    }

    /**
     * 查询商品分类列表
     */
    @ApiOperation( value = "查询商品分类列表-后台使用")
    @GetMapping(value = "/getList")
    public AjaxResult<List<TGoodsType>> getList(@RequestParam("shopId") Long shopId) {
        LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TGoodsType::getShopId, shopId);
        return AjaxResult.success(goodsTypeService.list(wrapper));
    }

    /**
@@ -46,6 +68,7 @@
    @ApiOperation( value = "添加商品分类")
    @PostMapping(value = "/add")
    public AjaxResult<Boolean> add(@RequestBody TGoodsType dto) {
        dto.setShopId(tokenService.getLoginUser().getObjectId());
        return AjaxResult.success(goodsTypeService.save(dto));
    }