物业公司只查询自己公司发布的公告

linwei

2024-01-19 2550a534393ffbed3744a2e9877ad53f5357b1dd

 src/main/java/org/springblade/modules/article/service/impl/ArticleServiceImpl.java

@@ -1,11 +1,16 @@
package org.springblade.modules.article.service.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.common.utils.SpringUtils;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.modules.article.entity.Article;
import org.springblade.modules.article.mapper.ArticleMapper;
import org.springblade.modules.article.service.ArticleService;
import org.springblade.modules.article.vo.ArticleVO;
import org.springblade.modules.property.entity.PropertyCompanyDistrictEntity;
import org.springblade.modules.property.service.IPropertyCompanyDistrictService;
import org.springframework.stereotype.Service;

import java.util.List;
@@ -28,6 +33,15 @@
    */
   @Override
   public IPage<Article> selectArticlePage(IPage<Article> page, ArticleVO article) {
      String userRole = AuthUtil.getUserRole();
      // 物业身份，只查询该物业公司下的公告
      if (userRole.contains("wygly")) {
         IPropertyCompanyDistrictService bean = SpringUtils.getBean(IPropertyCompanyDistrictService.class);
         PropertyCompanyDistrictEntity one = bean.getOne(Wrappers.<PropertyCompanyDistrictEntity>lambdaQuery()
            .eq(PropertyCompanyDistrictEntity::getUserId, AuthUtil.getUserId()).last("limit 1"));
         article.setDistrictId(one.getDistrictId());
      }

      return page.setRecords(baseMapper.selectArticlePage(page, article));
   }