diff --git a/src/main/java/com/sqx/modules/taskCenter/controller/TaskCenterRewardController.java b/src/main/java/com/sqx/modules/taskCenter/controller/TaskCenterRewardController.java index 90bd8aac..d5258c93 100644 --- a/src/main/java/com/sqx/modules/taskCenter/controller/TaskCenterRewardController.java +++ b/src/main/java/com/sqx/modules/taskCenter/controller/TaskCenterRewardController.java @@ -30,6 +30,7 @@ public class TaskCenterRewardController { @ApiOperation("添加任务奖励") public Result insertTaskCenterReward(@RequestBody TaskCenterReward taskCenterReward) { taskCenterReward.setCreateTime(DateUtils.format(new Date())); + taskCenterReward.setSurplusNumber(taskCenterReward.getTotalNumber()); taskCenterRewardService.save(taskCenterReward); return Result.success(); } @@ -43,6 +44,15 @@ public class TaskCenterRewardController { @PostMapping("/updateTaskCenterReward") @ApiOperation("修改任务奖励") public Result updateTaskCenterReward(@RequestBody TaskCenterReward taskCenterReward) { + TaskCenterReward reward = taskCenterRewardService.getById(taskCenterReward.getId()); + if (taskCenterReward.getTotalNumber() != null && reward.getTotalNumber() != null && !taskCenterReward.getTotalNumber().equals(reward.getTotalNumber())) { + Integer surplusNumber = taskCenterReward.getTotalNumber() - reward.getTotalNumber(); + if (surplusNumber > 0) { + taskCenterReward.setSurplusNumber(taskCenterReward.getSurplusNumber() == null ? surplusNumber : taskCenterReward.getSurplusNumber() + surplusNumber); + } else { + taskCenterReward.setSurplusNumber(taskCenterReward.getTotalNumber()); + } + } taskCenterReward.setUpdateTime(DateUtils.format(new Date())); taskCenterRewardService.updateById(taskCenterReward); return Result.success(); @@ -57,7 +67,7 @@ public class TaskCenterRewardController { @GetMapping("/selectTaskCenterReward") @ApiOperation("查询任务奖励") - public Result selectTaskCenterReward(Integer page, Integer limit,TaskCenterReward taskCenterReward) { + public Result selectTaskCenterReward(Integer page, Integer limit, TaskCenterReward taskCenterReward) { return Result.success().put("data", taskCenterRewardService.page(new Page<>(page, limit), new QueryWrapper<>(taskCenterReward).orderByDesc("id"))); } }