diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/app/TbUserBlockAppController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/app/TbUserBlockAppController.java index bf89d3b0..34f7485d 100644 --- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/app/TbUserBlockAppController.java +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/app/TbUserBlockAppController.java @@ -1,5 +1,6 @@ package com.ruoyi.web.controller.app; +import cn.hutool.core.date.DateUtil; import com.ruoyi.common.core.domain.Result; import com.ruoyi.system.domain.TbUserBlock; import com.ruoyi.system.service.ITbUserBlockService; @@ -22,7 +23,12 @@ public class TbUserBlockAppController { public Result addSave(@RequestBody TbUserBlock tbUserBlock) { tbUserBlockService.insertTbUserBlock(tbUserBlock); - return Result.success(); + // 查询当前用户今日剩余次数 + long count = tbUserBlockService.lambdaQuery() + .ge(TbUserBlock::getCreateTime, DateUtil.today()) + .eq(TbUserBlock::getTbUserId, tbUserBlock.getTbUserId()) + .count(); + return Result.success(String.valueOf(10-count), "操作成功!"); } }