无关风月
2024-08-10 7f288c7515fa04cf83fadc3506317d08fb099630
ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/PartnerController.java
@@ -2,13 +2,18 @@
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.ruoyi.chargingPile.api.dto.GetPermissionConfigurationDTO;
import com.ruoyi.chargingPile.api.dto.PartnerListDTO;
import com.ruoyi.chargingPile.api.model.Partner;
import com.ruoyi.chargingPile.api.query.GetPartnerList;
import com.ruoyi.chargingPile.api.query.SetPermissionConfiguration;
import com.ruoyi.chargingPile.dto.ResetPassword;
import com.ruoyi.chargingPile.service.IPartnerService;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.common.core.web.page.PageInfo;
import com.ruoyi.common.security.utils.SecurityUtils;
import com.ruoyi.system.api.domain.SysUser;
import com.ruoyi.system.api.feignClient.SysUserClient;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
@@ -27,6 +32,10 @@
   
   @Resource
   private IPartnerService partnerService;
   @Resource
   private SysUserClient sysUserClient;
   
   
@@ -91,10 +100,38 @@
   
   
   @ResponseBody
   @PostMapping("/getPartner")
   @GetMapping("/getPartner")
   @ApiOperation(value = "获取所有合作商", tags = {"管理后台-站点管理"})
   public AjaxResult<List<Partner>> getPartner(){
      List<Partner> list = partnerService.list(new LambdaQueryWrapper<Partner>().eq(Partner::getDelFlag, 0));
      SysUser sysUser = sysUserClient.getSysUser(SecurityUtils.getUserId()).getData();
      Integer roleType = sysUser.getRoleType();
      Integer objectId = sysUser.getObjectId();
      LambdaQueryWrapper<Partner> wrapper = new LambdaQueryWrapper<Partner>().eq(Partner::getDelFlag, 0);
      if(2 == roleType){
         wrapper.eq(Partner::getId, objectId);
      }
      List<Partner> list = partnerService.list(wrapper);
      return AjaxResult.success(list);
   }
   @ResponseBody
   @GetMapping("/getPermissionConfiguration/{siteId}")
   @ApiOperation(value = "获取合作商权限设置", tags = {"管理后台-站点管理"})
   public AjaxResult<GetPermissionConfigurationDTO> getPermissionConfiguration(@PathVariable("siteId") Integer siteId, @RequestParam("partnerId") Integer partnerId){
      GetPermissionConfigurationDTO permissionConfiguration = partnerService.getPermissionConfiguration(siteId, partnerId);
      return AjaxResult.success(permissionConfiguration);
   }
   @ResponseBody
   @PostMapping("/setPermissionConfiguration/{siteId}")
   @ApiOperation(value = "设置合作商权限", tags = {"管理后台-站点管理"})
   public AjaxResult setPermissionConfiguration(@RequestBody SetPermissionConfiguration req){
      partnerService.setPermissionConfiguration(req);
      return AjaxResult.success();
   }
}