From 0f4459e4201fdc19185b7dd03215871f036acd70 Mon Sep 17 00:00:00 2001 From: liujie <1793218484@qq.com> Date: 星期三, 28 五月 2025 16:45:01 +0800 Subject: [PATCH] Merge branch 'dev' of http://120.76.84.145:10101/gitblit/r/java/mx_charging_pile into dev --- ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java | 45 +++++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 43 insertions(+), 2 deletions(-) diff --git a/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java b/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java index 9be842e..7a1f511 100644 --- a/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java +++ b/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java @@ -2,6 +2,8 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.ruoyi.chargingPile.api.feignClient.SiteClient; +import com.ruoyi.chargingPile.api.model.Site; import com.ruoyi.common.core.domain.R; import com.ruoyi.other.api.domain.TRoleSite; import com.ruoyi.other.service.TRoleSiteService; @@ -26,15 +28,54 @@ @Resource private TRoleSiteService roleSiteService; + @Resource + private SiteClient siteClient; + + + /** + * 查询角色站点数据 + * @param roleId + * @return + */ @ResponseBody - @GetMapping("/getSiteIds") - public R<List<Integer>> getSiteIds(@RequestParam("roleId") Long roleId){ + @PostMapping("/getSiteIds/{roleId}") + public R<List<Integer>> getSiteIds(@PathVariable Long roleId){ List<TRoleSite> list = roleSiteService.list(new LambdaQueryWrapper<TRoleSite>().eq(TRoleSite::getRoleId, roleId)); + if(list.size() == 1){ + Integer siteId = list.get(0).getSiteId(); + if(0 == siteId){ + List<Site> data = siteClient.getSiteAll().getData(); + List<Integer> collect = data.stream().map(Site::getId).collect(Collectors.toList()); + return R.ok(collect); + } + } List<Integer> collect = list.stream().map(TRoleSite::getSiteId).collect(Collectors.toList()); return R.ok(collect); } + /** + * 添加角色站点数据 + * @param roleSites + * @return + */ + @ResponseBody + @PostMapping("/addRoleSite") + public R addRoleSite(@RequestBody List<TRoleSite> roleSites){ + roleSiteService.saveBatch(roleSites); + return R.ok(); + } + /** + * 删除角色站点数据 + * @param roleId + * @return + */ + @ResponseBody + @DeleteMapping("/delRoleSite/{roleId}") + public R delRoleSite(@PathVariable("roleId") Long roleId){ + roleSiteService.remove(new LambdaQueryWrapper<TRoleSite>().eq(TRoleSite::getRoleId, roleId)); + return R.ok(); + } } -- Gitblit v1.7.1