package com.thhy.pm.modules.biz.projectinfo.controller; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONObject; import com.alibaba.nacos.api.utils.StringUtils; import com.github.pagehelper.PageInfo; import com.thhy.general.common.BasicResult; import com.thhy.general.common.BasicStatus; import com.thhy.general.config.GlobalConfig; import com.thhy.general.exception.BasicException; import com.thhy.general.utils.ExcelUtils; import com.thhy.general.utils.PagingUtils; import com.thhy.pm.modules.biz.projectinfo.entity.ProInfoFile; import com.thhy.pm.modules.biz.projectinfo.entity.ProInfoFileListVo; import com.thhy.pm.modules.biz.projectinfo.entity.ProInfoListVo; import com.thhy.pm.modules.biz.projectinfo.entity.ProjectInfo; import com.thhy.pm.modules.biz.projectinfo.service.ProjectInfoService; import org.redisson.api.RBucket; import org.redisson.api.RedissonClient; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.util.List; /** * @author zhang_xiao_bo * @since 2023-04-23 17:02:12 */ @RestController @RequestMapping("projectInfo") public class ProjectInfoController { @Autowired private ProjectInfoService projectInfoService; @Autowired private RedissonClient redissonClient; @Autowired private GlobalConfig globalConfig; @RequestMapping("findList") public BasicResult findList(@RequestBody(required = false) ProjectInfo projectInfo,HttpServletRequest req){ PagingUtils.setPageParam(req); List projectInfoList = projectInfoService.findList(projectInfo); PageInfo pageInfo = new PageInfo<>(projectInfoList); return BasicResult.success(pageInfo); } @RequestMapping("fileList") public BasicResult fileList(@RequestBody ProInfoFile projectInfo, HttpServletRequest req){ PagingUtils.setPageParam(req); List projectInfoList = projectInfoService.fileList(projectInfo); PageInfo pageInfo = new PageInfo<>(projectInfoList); return BasicResult.success(pageInfo); } @RequestMapping("addProjectInfo") public BasicResult addProjectInfo(@RequestBody ProjectInfo projectInfo){ projectInfoService.addProjectInfo(projectInfo); return BasicResult.success(); } /** @RequestMapping("update") public BasicResult update(@RequestBody ProjectInfo projectInfo){ projectInfoService.update(projectInfo); return BasicResult.success(); }**/ @RequestMapping("delete") public BasicResult delete(@RequestBody ProjectInfo projectInfo){ projectInfoService.delete(projectInfo.getProInfoId()); return BasicResult.success(); } @RequestMapping("export") public void export(String authcode, HttpServletResponse response){ RBucket rBucket = redissonClient.getBucket(globalConfig.getFilePreKey()+":authcode:"+authcode); String param = rBucket.getAndDelete(); if(StringUtils.isEmpty(param)){ throw new BasicException(BasicStatus.TOKEN_IS_NULL); } JSONObject jsonObject = JSON.parseObject(param); ProjectInfo projectInfo = JSON.toJavaObject(jsonObject,ProjectInfo.class); List projectInfoList = projectInfoService.findListForExport(projectInfo,jsonObject.getString("token")); ExcelUtils.downExcel(projectInfoList,ProInfoListVo.class,response,"项目信息列表"); } }