48

所学时间：380分钟
博客量：1
代码量：几百
@RestController
@RequestMapping("/hiddendanger")
@CrossOrigin
public class HiddenDangerController {

@Autowired
private HiddenDangerService hiddenDangerService;

@GetMapping("/findByDangerCode")
public ResponseEntity<List<HiddenDanger>> findByDangerCode(@RequestParam String dangerCode) {
    return ResponseEntity.ok(hiddenDangerService.findByDangerCode(dangerCode));
}

@PostMapping
public ResponseEntity<Result> saveHiddenDanger(@RequestBody HiddenDanger hiddenDanger) {
    if (hiddenDanger.getId() == null) {
        if (hiddenDangerService.findByDangerCode(hiddenDanger.getDanger_code()).size() > 0) {
            return ResponseEntity.status(HttpStatus.BAD_REQUEST).body(Result.error("隐患编码已存在"));
        }
    }
    String operator = hiddenDanger.getOperator();
    hiddenDangerService.saveHiddenDanger(hiddenDanger, operator);
    return ResponseEntity.status(HttpStatus.CREATED).body(Result.success());
}

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

@DeleteMapping("/{id}")
public ResponseEntity<Result> deleteHiddenDanger(@PathVariable int id, @RequestBody HiddenDanger hiddenDanger) {
    String operator = hiddenDanger.getOperator();
    hiddenDangerService.deleteHiddenDanger(id, operator);
    return ResponseEntity.ok(Result.success());
}

@GetMapping("/export")
public void exportData(HttpServletResponse response) throws IOException {
    ExcelWriter writer = ExcelUtil.getWriter(true);
    writer.write(hiddenDangerService.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 (OutputStream outputStream = response.getOutputStream()) {
        writer.flush(outputStream, true);
    }
}

}