投票人员查询优化

linwei

2024-02-22 7bcd0d2d7510cb2ba67099cd768e1b2b6d047dc2

 src/main/java/org/springblade/modules/discuss/service/impl/UserTopicsServiceImpl.java

@@ -45,6 +45,7 @@
import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 用户议题报表 服务实现类
@@ -60,10 +61,14 @@
   @Override
   public IPage<UserTopicsVO> selectUserTopicsPage(IPage<UserTopicsVO> page, UserTopicsVO userTopics) {
      if (StringUtils.isNotBlank(userTopics.getDistrictId())) {
         List<String> longs = JSON.parseArray(userTopics.getDistrictId()).toJavaList(String.class);
         IDistrictService bean = SpringUtils.getBean(IDistrictService.class);
         DistrictEntity one = bean.getOne(Wrappers.<DistrictEntity>lambdaQuery().eq(DistrictEntity::getId, userTopics.getDistrictId()));
         if (one != null) {
            userTopics.setAoiCode(one.getAoiCode());
         List<DistrictEntity> list = bean.list(Wrappers.<DistrictEntity>lambdaQuery().in(DistrictEntity::getId, longs));
         List<String> collect = list.stream().map(item ->
            item.getAoiCode()
         ).collect(Collectors.toList());
         if (collect != null) {
            userTopics.setAoiCodeList(collect);
         }
      }
      return page.setRecords(baseMapper.selectUserTopicsPage(page, userTopics));
@@ -78,13 +83,17 @@
      PublicDiscussEntity one = bean.getOne(Wrappers.<PublicDiscussEntity>lambdaQuery().eq(PublicDiscussEntity::getArticleId, topics.get(0).getArticleId()));
      // 一户一票
      if (one.getVoteRestrictions().equals(CommonConstant.NUMBER_ONE)) {
         long count = count(Wrappers.<UserTopicsEntity>lambdaQuery().eq(UserTopicsEntity::getHouseCode, topics.get(0).getHouseCode()));
         long count = count(Wrappers.<UserTopicsEntity>lambdaQuery()
            .eq(UserTopicsEntity::getHouseCode, topics.get(0).getHouseCode())
            .eq(UserTopicsEntity::getArticleId,topics.get(0).getArticleId()));
         if (count > 1) {
            throw new Exception("您的房屋已投票，不能重复投票！");
         }
      } else {
         //
         long count = count(Wrappers.<UserTopicsEntity>lambdaQuery().eq(UserTopicsEntity::getUserId, AuthUtil.getUserId()));
         long count = count(Wrappers.<UserTopicsEntity>lambdaQuery()
            .eq(UserTopicsEntity::getUserId, AuthUtil.getUserId())
            .eq(UserTopicsEntity::getArticleId,topics.get(0).getArticleId()));
         if (count > 1) {
            throw new Exception("您的已投票，不能重复投票！");
         }