init

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<UserVisitorMapper,UserVi
 
     @Override
     public Page<UserVisitorListVo> pageApp(PageQuery pageQuery, UserVisitorQuery query) {
-        return baseMapper.pageApp(pageQuery.build(), query);
+        Page<UserVisitorListVo> page = baseMapper.pageApp(pageQuery.build(), query);
+        Long userId = LoginHelper.getUserId();
+        User user = userService.getById(userId);
+        if(user == null || BooleanUtils.isNotTrue(user.getOpenVip())){
+            List<UserVisitorListVo> records = page.getRecords();
+            for (UserVisitorListVo record : records) {
+                record.setUserId(null);
+                record.setNickname("***");
+            }
+        }
+        return page;
     }
 
 }