diff --git a/blade-service/blade-desk/src/main/java/org/springblade/desk/common/controller/MesNotifyMessageController.java b/blade-service/blade-desk/src/main/java/org/springblade/desk/common/controller/MesNotifyMessageController.java index 283f6401..2b9a213a 100644 --- a/blade-service/blade-desk/src/main/java/org/springblade/desk/common/controller/MesNotifyMessageController.java +++ b/blade-service/blade-desk/src/main/java/org/springblade/desk/common/controller/MesNotifyMessageController.java @@ -25,6 +25,7 @@ */ package org.springblade.desk.common.controller; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import io.swagger.v3.oas.annotations.Operation; @@ -33,7 +34,6 @@ import io.swagger.v3.oas.annotations.tags.Tag; import lombok.AllArgsConstructor; import org.springblade.core.boot.ctrl.BladeController; import org.springblade.core.log.annotation.ApiLog; -import org.springblade.core.mp.support.Condition; import org.springblade.core.mp.support.Query; import org.springblade.core.secure.utils.AuthUtil; import org.springblade.core.tool.api.R; @@ -74,10 +74,13 @@ public class MesNotifyMessageController extends BladeController { @GetMapping("/count") @Operation(summary = "获取未读数量", description = "传入mesNotifyMessage") public R count() { - MesNotifyMessageVO mesNotifyMessage = new MesNotifyMessageVO(); - mesNotifyMessage.setReceiveUserId(AuthUtil.getUserId()); - mesNotifyMessage.setStatus(0); - long count = mesNotifyMessageService.count(Condition.getQueryWrapper(mesNotifyMessage)); + long count = mesNotifyMessageService.count(new LambdaQueryWrapper() + .eq(MesNotifyMessageEntity::getStatus, 0) + .and(wrapper -> wrapper + .eq(MesNotifyMessageEntity::getReceiveUserId, AuthUtil.getUserId()) + .or() + .eq(MesNotifyMessageEntity::getReceiveRoleName, AuthUtil.getUserRole()) + )); return R.data(count); } @@ -87,14 +90,14 @@ public class MesNotifyMessageController extends BladeController { @PostMapping("/remove") @ApiLog("删除通知消息") @Operation(summary = "逻辑删除", description = "传入ids") - public R remove(@Parameter(description = "主键集合", required = true) @RequestParam String ids) { + public R remove(@Parameter(description = "主键集合", required = true) @RequestParam String ids) { return R.status(mesNotifyMessageService.deleteLogic(Func.toLongList(ids))); } @PostMapping("/read") @ApiLog("标记已读") @Operation(summary = "标记已读", description = "传入ids") - public R read(@Parameter(description = "主键集合", required = true) String ids) { + public R read(@Parameter(description = "主键集合", required = true) String ids) { List idList = Func.toLongList(ids); return R.status(mesNotifyMessageService.updateReadStatus(idList)); }