明星电力充电桩
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
xuhy
98 分钟以前 fb9e286342938f63af535eb2a363f783a63213db 
 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,6 +28,11 @@


   @Resource


   private TRoleSiteService roleSiteService;


   


   @Resource


   private SiteClient siteClient;


	


	


	


   /**


    * 查询角色站点数据


    * @param roleId


@@ -35,6 +42,14 @@


   @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);


   }