package com.ruoyi.account.controller; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.ruoyi.account.api.model.AppUserShop; import com.ruoyi.account.mapper.AppUserMapper; import com.ruoyi.account.mapper.AppUserShopMapper; import com.ruoyi.account.service.AppUserService; import com.ruoyi.account.service.AppUserShopService; import com.ruoyi.common.core.domain.R; import com.ruoyi.common.security.service.TokenService; import com.ruoyi.system.api.model.LoginUser; import org.springframework.util.CollectionUtils; import org.springframework.web.bind.annotation.*; import javax.annotation.Resource; import java.util.List; /** * @author zhibing.pu * @Date 2024/12/2 9:21 */ @RestController @RequestMapping("/appUserShop") public class AppUserShopController { @Resource private AppUserShopService appUserShopService; @Resource private AppUserService appUserService; @Resource private AppUserShopMapper appUserShopMapper; @Resource private TokenService tokenService; @GetMapping("/getUserShopListByUserId") public R> getUserShopListByUserId(@RequestParam("userId") Long userId){ return R.ok(appUserShopMapper.selectList(new LambdaQueryWrapper().eq(AppUserShop::getAppUserId, userId))); } @PostMapping("/addAppUserShop") public R add(@RequestBody AppUserShop appUserShop) { long count = appUserShopService.count(new LambdaQueryWrapper() .eq(AppUserShop::getAppUserId, appUserShop.getAppUserId()).eq(AppUserShop::getShopId, appUserShop.getShopId())); if(0 == count){ appUserShopService.save(appUserShop); } return R.ok(); } @GetMapping("/shop/{userId}") public R> getAppUserShop(@PathVariable("userId") Long userId) { return R.ok(appUserShopService.list(new LambdaQueryWrapper() .eq(AppUserShop::getAppUserId,userId))); } /** * 删除门店用户关系数据 * @param appUserShop * @return */ @PostMapping("/delAppUserShop") public R delAppUserShop(@RequestBody AppUserShop appUserShop){ LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>(); if(null != appUserShop.getAppUserId()){ wrapper.eq(AppUserShop::getAppUserId, appUserShop.getAppUserId()); } if(null != appUserShop.getShopId()){ wrapper.eq(AppUserShop::getShopId, appUserShop.getShopId()); } if(null != appUserShop.getRoleType()){ wrapper.eq(AppUserShop::getRoleType, appUserShop.getRoleType()); } appUserShopService.remove(wrapper); return R.ok(); } /** * 保存关系数据 * @param appUserShop */ @PostMapping("/saveAppUserShop") public void saveAppUserShop(@RequestBody AppUserShop appUserShop){ appUserShopService.save(appUserShop); } /** * 校验登录用户是否为门店店长 */ @GetMapping("/checkUserIsShopManager") public R checkUserIsShopManager(){ LoginUser loginUserApplet = tokenService.getLoginUserApplet(); List list = appUserShopService.list(new LambdaQueryWrapper() .eq(AppUserShop::getAppUserId, loginUserApplet.getUserid()) .eq(AppUserShop::getRoleType, 1)); return R.ok(!CollectionUtils.isEmpty(list)); } @PostMapping("/insert") R insert(@RequestBody AppUserShop appUserShop){ appUserShopService.save(appUserShop); return R.ok(); } @DeleteMapping("/delete") R delete(@RequestParam("id") Integer id,@RequestParam("userId") Long userId){ LambdaQueryWrapper wrapper = new LambdaQueryWrapper(); wrapper.eq(AppUserShop::getShopId, id); wrapper.eq(AppUserShop::getAppUserId, userId); appUserShopService.remove(wrapper); return R.ok(); } }