Pu Zhibing
2024-09-20 a146097a15d719c60af3cdeb4de1f21aa7b5ca8d
ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java
@@ -20,6 +20,8 @@
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.common.core.web.page.PageInfo;
import com.ruoyi.common.log.annotation.Log;
import com.ruoyi.common.log.enums.BusinessType;
import com.ruoyi.common.security.utils.SecurityUtils;
import com.ruoyi.order.api.feignClient.ChargingOrderClient;
import com.ruoyi.order.api.model.TChargingOrder;
@@ -27,6 +29,8 @@
import com.ruoyi.other.api.feignClient.UserSiteClient;
import com.ruoyi.system.api.domain.SysUser;
import com.ruoyi.system.api.feignClient.SysUserClient;
import com.ruoyi.system.api.feignClient.SysUserRoleClient;
import com.ruoyi.system.api.model.SysUserRoleVo;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
@@ -82,6 +86,22 @@
   @Resource
   private ChargingOrderClient chargingOrderClient;
   @Resource
   private SysUserRoleClient userRoleClient;
   /**
    * 根据站点id 查询所有充电桩
    * @param siteId
    * @return
    */
   @GetMapping("/getPileListBySiteId/{siteId}")
   public R<List<TChargingPile>> getPileListBySiteId(@PathVariable("siteId")Integer siteId){
      return R.ok(chargingPileService.lambdaQuery().eq(TChargingPile::getSiteId, siteId).list());
   }
   /**
    * 小程序扫一扫
    * @param number
@@ -287,6 +307,7 @@
   @ResponseBody
   @PostMapping("/addSite")
   @ApiOperation(value = "添加站点", tags = {"管理后台-站点管理"})
   @Log(title = "【站点管理】添加站点", businessType = BusinessType.INSERT)
   public AjaxResult addSite(@RequestBody Site site){
      return siteService.addSite(site);
   }
@@ -306,6 +327,7 @@
   @ResponseBody
   @PostMapping("/editSite")
   @ApiOperation(value = "编辑站点", tags = {"管理后台-站点管理"})
   @Log(title = "【站点管理】编辑站点", businessType = BusinessType.UPDATE)
   public AjaxResult editSite(@RequestBody Site site){
      return siteService.editSite(site);
   }
@@ -316,6 +338,7 @@
   @ApiImplicitParams({
         @ApiImplicitParam(value = "站点id", name = "id", dataTypeClass = Integer.class, required = true)
   })
   @Log(title = "【站点管理】删除站点", businessType = BusinessType.DELETE)
   public AjaxResult delSite(@RequestParam("id") Integer[] id){
      return siteService.delSite(id);
   }
@@ -328,6 +351,7 @@
         @ApiImplicitParam(value = "站点id", name = "id", dataTypeClass = Integer.class, required = true),
         @ApiImplicitParam(value = "计费策略id", name = "accountingStrategyId", dataTypeClass = Integer.class, required = true)
   })
   @Log(title = "【站点管理】设置站点计费策略", businessType = BusinessType.OTHER)
   public AjaxResult setAccountingStrategy(@PathVariable("id") Integer id, @RequestParam("accountingStrategyId") Integer accountingStrategyId){
      siteService.setAccountingStrategy(id, accountingStrategyId);
      return AjaxResult.success();
@@ -350,7 +374,8 @@
         //非管理员需要根据角色和用户配置查询允许的站点数据
         if(!SecurityUtils.isAdmin(SecurityUtils.getUserId())){
            List<Integer> data = userSiteClient.getSiteIds(sysUser.getUserId()).getData();
            List<Integer> data1 = roleSiteClient.getSiteIds(sysUser.getRoleId()).getData();
            List<SysUserRoleVo> data2 = userRoleClient.getRoleByUserId(sysUser.getUserId()).getData();
            List<Integer> data1 = roleSiteClient.getSiteIds(data2.get(0).getRoleId()).getData();
            data.addAll(data1);
            Set<Integer> list = new HashSet<>(data);
            wrapper.in(Site::getId, list);