From 7a0183a75146de8d88a54df7b33da11bea78007e Mon Sep 17 00:00:00 2001 From: 77 <77@77.com> Date: Fri, 3 May 2024 21:24:47 +0800 Subject: [PATCH] init --- .../xq/service/impl/UserVisitorServiceImpl.java | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/ruoyi-xq/src/main/java/com/ruoyi/xq/service/impl/UserVisitorServiceImpl.java b/ruoyi-xq/src/main/java/com/ruoyi/xq/service/impl/UserVisitorServiceImpl.java index 453e23b..a0adce4 100644 --- a/ruoyi-xq/src/main/java/com/ruoyi/xq/service/impl/UserVisitorServiceImpl.java +++ b/ruoyi-xq/src/main/java/com/ruoyi/xq/service/impl/UserVisitorServiceImpl.java @@ -4,19 +4,24 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.ruoyi.common.core.domain.PageQuery; +import com.ruoyi.common.helper.LoginHelper; import com.ruoyi.xq.domain.User; import com.ruoyi.xq.domain.UserVisitor; +import com.ruoyi.xq.dto.app.userstar.vo.UserStarListVo; import com.ruoyi.xq.dto.app.uservisitor.UserVisitorQuery; import com.ruoyi.xq.dto.app.uservisitor.vo.UserVisitorListVo; import com.ruoyi.xq.lock.LockKey; import com.ruoyi.xq.mapper.UserVisitorMapper; import com.ruoyi.xq.service.UserService; import com.ruoyi.xq.service.UserVisitorService; +import org.apache.commons.lang3.BooleanUtils; import org.redisson.api.RLock; import org.redisson.api.RedissonClient; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; +import java.util.List; + /** * 用户来访记录Service业务层处理 * @@ -64,7 +69,17 @@ public class UserVisitorServiceImpl extends ServiceImpl pageApp(PageQuery pageQuery, UserVisitorQuery query) { - return baseMapper.pageApp(pageQuery.build(), query); + Page page = baseMapper.pageApp(pageQuery.build(), query); + Long userId = LoginHelper.getUserId(); + User user = userService.getById(userId); + if(user == null || BooleanUtils.isNotTrue(user.getOpenVip())){ + List records = page.getRecords(); + for (UserVisitorListVo record : records) { + record.setUserId(null); + record.setNickname("***"); + } + } + return page; } }