49

所学时间：300分钟
博客量：1
代码量：几百
所学知识：个人作业安卓端。计算机网络。
@RestController
@RequestMapping("/safetyrisk2")
@CrossOrigin
public class SafetyriskController2 {

@Autowired
private SafetyRiskService2 safetyRiskService;

@GetMapping("/findByRiskCode")
public ResponseEntity<List<safetyrisk2>> findByRiskCode(@RequestParam String riskCode) {
    return ResponseEntity.ok(safetyRiskService.findByRiskCode(riskCode));
}

@PostMapping
public ResponseEntity<Result> saveSafetyRisk(@RequestBody safetyrisk2 safetyRisk) {
    if (safetyRisk.getId() == null) {
        if (safetyRiskService.findByRiskCode(safetyRisk.getRisk_code()).size() > 0) {
            return ResponseEntity.status(HttpStatus.BAD_REQUEST).body(Result.error("风险编码已存在"));
        }
    }
    String operator = safetyRisk.getOperator();
    safetyRiskService.saveSafetyRisk(safetyRisk, operator);
    return ResponseEntity.status(HttpStatus.CREATED).body(Result.success());
}

@GetMapping("/all")
public ResponseEntity<Page<safetyrisk2>> getAllSafetyRisks(
        @RequestParam(defaultValue = "1") int page,
        @RequestParam(defaultValue = "10") int size
) {
    return ResponseEntity.ok(safetyRiskService.getAllSafetyRisks(page, size));
}

@DeleteMapping("/{id}")
public ResponseEntity<Result> deleteSafetyRisk(@PathVariable int id, @RequestBody safetyrisk2 safetyRisk) {
    String operator = safetyRisk.getOperator();
    safetyRiskService.deleteSafetyRisk(id, operator);
    return ResponseEntity.ok(Result.success());
}

@GetMapping("/export")
public void exportData(HttpServletResponse response) throws IOException {
    ExcelWriter writer = ExcelUtil.getWriter(true);
    writer.write(safetyRiskService.list(), true);
    response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
    response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("安全风险表", "UTF-8") + ".xlsx");
    try (ServletOutputStream outputStream = response.getOutputStream()) {
        writer.flush(outputStream, true);
    }
}

}