update

src/main/java/com/qmth/exam/reserve/controller/admin/ExamSiteManageController.java

@@ -56,12 +56,19 @@ public class ExamSiteManageController extends BaseController {
     @PostMapping(value = "/save")
     public void save(@RequestBody ExamSiteSaveReq req) {
         LoginUser user = curLoginUser();
+        if(!user.getRole().equals(Role.ADMIN)) {
+            throw new StatusException("没有权限");
+        }
         examSiteService.saveExamSite(user, req);
     }
 
     @ApiOperation(value = "考点启用/禁用")
     @PostMapping(value = "/enable")
     public void enable(@ApiParam("考点ID") @RequestParam Long id, @ApiParam("启用/禁用") @RequestParam Boolean enable) {
+        LoginUser user = curLoginUser();
+        if(!user.getRole().equals(Role.ADMIN)) {
+            throw new StatusException("没有权限");
+        }
         examSiteService.enable(id, enable);
     }
 

src/main/java/com/qmth/exam/reserve/service/impl/ExamSiteServiceImpl.java

@@ -108,6 +108,7 @@ public class ExamSiteServiceImpl extends ServiceImpl<ExamSiteDao, ExamSiteEntity
         BeanUtils.copyProperties(req, site);
         site.setCategoryId(req.getTeachingId());
         if (req.getId() == null) {
+            site.setCapacity(0);
             site.setEnable(Boolean.TRUE);
             save(site);
         } else {
@@ -119,11 +120,14 @@ public class ExamSiteServiceImpl extends ServiceImpl<ExamSiteDao, ExamSiteEntity
 
     }
 
+    @Transactional
     @Override
     public void enable(Long id, Boolean enable) {
         ExamSiteEntity examSite = getById(id);
         examSite.setEnable(enable);
         baseMapper.updateById(examSite);
+        //更新教学点的容量
+        categoryService.updateTeachingCapacity(examSite.getCategoryId());
     }
 
     @Transactional
@@ -235,6 +239,7 @@ public class ExamSiteServiceImpl extends ServiceImpl<ExamSiteDao, ExamSiteEntity
             updateWrapper.eq(ExamSiteEntity::getId, examSite.getId());
             update(null, updateWrapper);
         } else {
+            examSite.setCapacity(0);
             examSite.setEnable(Boolean.TRUE);
             save(examSite);
         }