package com.izouma.nineth.web; import com.izouma.nineth.domain.Recommend; import com.izouma.nineth.dto.PageQuery; import com.izouma.nineth.exception.BusinessException; import com.izouma.nineth.repo.RecommendRepo; import com.izouma.nineth.service.CacheService; import com.izouma.nineth.service.RecommendService; import com.izouma.nineth.utils.ObjUtils; import com.izouma.nineth.utils.excel.ExcelUtils; import lombok.AllArgsConstructor; import org.springframework.data.domain.Page; import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletResponse; import java.io.IOException; import java.util.List; @RestController @RequestMapping("/recommend") @AllArgsConstructor public class RecommendController extends BaseController { private RecommendService recommendService; private RecommendRepo recommendRepo; private CacheService cacheService; //@PreAuthorize("hasRole('ADMIN')") @PostMapping("/save") public Recommend save(@RequestBody Recommend record) { if (record.getId() != null) { Recommend orig = recommendRepo.findById(record.getId()).orElseThrow(new BusinessException("无记录")); ObjUtils.merge(orig, record); orig = recommendRepo.save(orig); cacheService.clearRecommend(); return orig; } record = recommendRepo.save(record); cacheService.clearRecommend(); return record; } //@PreAuthorize("hasRole('ADMIN')") @PostMapping("/all") public Page all(@RequestBody PageQuery pageQuery) { return recommendService.all(pageQuery); } @GetMapping("/get/{id}") public Recommend get(@PathVariable Long id) { return recommendRepo.findById(id).orElseThrow(new BusinessException("无记录")); } @PostMapping("/del/{id}") public void del(@PathVariable Long id) { recommendRepo.deleteById(id); cacheService.clearRecommend(); } @GetMapping("/excel") @ResponseBody public void excel(HttpServletResponse response, PageQuery pageQuery) throws IOException { List data = all(pageQuery).getContent(); ExcelUtils.export(response, data); } }