diff --git a/src/main/java/top/suyiiyii/sims/controller/RecordController.java b/src/main/java/top/suyiiyii/sims/controller/RecordController.java index a0caaec..625e3ff 100644 --- a/src/main/java/top/suyiiyii/sims/controller/RecordController.java +++ b/src/main/java/top/suyiiyii/sims/controller/RecordController.java @@ -77,6 +77,10 @@ RecordController { @PutMapping("/admin/record/{id}") public Result adminUpdateRecord(@PathVariable Integer id, @RequestBody RecordDto recordDto) { Record record = modelMapper.map(recordDto, Record.class); + Integer i = recordService.IsRecord(id); + if(i==null) { + throw new RuntimeException("该记录不存在"); + } recordService.updateRecord(record, id); return Result.msg("修改成功"); } @@ -85,6 +89,10 @@ RecordController { @Operation(summary = "删除单个奖惩记录") @DeleteMapping("/admin/record/{id}") public Result adminDeleteRecord(@PathVariable Integer id) { + Integer i = recordService.IsRecord(id); + if(i==null) { + throw new RuntimeException("该记录不存在"); + } recordService.deleteRecord(id); return Result.msg("删除成功"); } diff --git a/src/main/java/top/suyiiyii/sims/mapper/RecordMapper.java b/src/main/java/top/suyiiyii/sims/mapper/RecordMapper.java index 0caff2e..2ccd16c 100644 --- a/src/main/java/top/suyiiyii/sims/mapper/RecordMapper.java +++ b/src/main/java/top/suyiiyii/sims/mapper/RecordMapper.java @@ -79,4 +79,6 @@ public interface RecordMapper { Integer getSidByCategoryId(Integer i); @Select("SELECT * FROM record WHERE student_id = #{sid} LIMIT #{page},#{size}") List getRecordsById(int page, int size, Integer sid); +@Select("SELECT id FROM record WHERE id = #{id}") + Integer IsRecord(Integer id); } diff --git a/src/main/java/top/suyiiyii/sims/service/RecordService.java b/src/main/java/top/suyiiyii/sims/service/RecordService.java index 962e39d..d8f88a2 100644 --- a/src/main/java/top/suyiiyii/sims/service/RecordService.java +++ b/src/main/java/top/suyiiyii/sims/service/RecordService.java @@ -80,4 +80,8 @@ public class RecordService { public List getRecordsById(int page, int size, Integer sid) { return recordMapper.getRecordsById(page, size, sid); } + + public Integer IsRecord(Integer id) { + return recordMapper.IsRecord(id); + } }