package com.thhy.materials.modules.biz.spare.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.materials.modules.biz.spare.entity.Spare; import com.thhy.materials.modules.biz.spare.entity.SpareListVo; import com.thhy.materials.modules.biz.spare.service.SpareService; 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-11 16:07:53 */ @RestController @RequestMapping("spare") public class SpareController { @Autowired private SpareService spareService; @Autowired private RedissonClient redissonClient; @Autowired private GlobalConfig globalConfig; @RequestMapping("findList") public BasicResult findList(@RequestBody(required = false) Spare spare,HttpServletRequest req){ PagingUtils.setPageParam(req); List spareList = spareService.findList(spare); PageInfo pageInfo = new PageInfo<>(spareList); return BasicResult.success(pageInfo); } @RequestMapping("export") public void export(String authcode, HttpServletRequest request,HttpServletResponse resp){ 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); Spare spare = JSON.toJavaObject(jsonObject,Spare.class); List spareList = spareService.findListForExport(spare,jsonObject.getString("token")); ExcelUtils.downExcel(spareList,SpareListVo.class,resp,"备品备件列表"); } @RequestMapping("addSpare") public BasicResult addSpare(@RequestBody Spare spare){ spareService.addSpare(spare); return BasicResult.success(); } /** @RequestMapping("update") public BasicResult update(@RequestBody Spare spare){ spareService.update(spare); return BasicResult.success(); }**/ @RequestMapping("delete") public BasicResult delete(@RequestBody Spare spare){ spareService.delete(spare.getSpareId()); return BasicResult.success(); } }