lin
2024-04-19 73ceaaa73dd6b73808aac425c70e0d9184f9f415
src/main/java/org/springblade/modules/discuss/service/impl/UserTopicsServiceImpl.java
@@ -16,6 +16,11 @@
 */
package org.springblade.modules.discuss.service.impl;
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.write.builder.ExcelWriterSheetBuilder;
import com.alibaba.excel.write.metadata.style.WriteCellStyle;
import com.alibaba.excel.write.style.HorizontalCellStyleStrategy;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
@@ -26,12 +31,15 @@
import org.jetbrains.annotations.Nullable;
import org.springblade.common.constant.CommonConstant;
import org.springblade.common.utils.SpringUtils;
import org.springblade.core.redis.cache.BladeRedis;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.core.tool.utils.SpringUtil;
import org.springblade.modules.discuss.entity.PublicDiscussEntity;
import org.springblade.modules.discuss.entity.TopicsEntity;
import org.springblade.modules.discuss.entity.UserTopicsEntity;
import org.springblade.modules.discuss.excel.UserTopicsExcel;
import org.springblade.modules.discuss.excel.holdExcel;
import org.springblade.modules.discuss.mapper.UserTopicsMapper;
import org.springblade.modules.discuss.service.IPublicDiscussService;
import org.springblade.modules.discuss.service.ITopicsService;
@@ -40,11 +48,17 @@
import org.springblade.modules.discuss.vo.UserTopicsVO;
import org.springblade.modules.district.entity.DistrictEntity;
import org.springblade.modules.district.service.IDistrictService;
import org.springblade.modules.house.entity.HouseholdEntity;
import org.springblade.modules.house.excel.HouseHoldExcel;
import org.springblade.modules.house.service.IHouseholdService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import javax.annotation.Resource;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
@@ -58,11 +72,15 @@
public class UserTopicsServiceImpl extends ServiceImpl<UserTopicsMapper, UserTopicsEntity> implements IUserTopicsService {
   @Resource
   private ITopicsService topicsService;
   public static final String SMS_VALIDATE_PHONE = "sms:validate:code:";
   @Autowired
   private BladeRedis redisTemplate;
   @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);
         List<String> longs = Arrays.asList(userTopics.getDistrictId().split(","));
         IDistrictService bean = SpringUtils.getBean(IDistrictService.class);
         List<DistrictEntity> list = bean.list(Wrappers.<DistrictEntity>lambdaQuery().in(DistrictEntity::getId, longs));
         List<String> collect = list.stream().map(item ->
@@ -72,44 +90,96 @@
            userTopics.setAoiCodeList(collect);
         }
      }
      return page.setRecords(baseMapper.selectUserTopicsPage(page, userTopics));
      IHouseholdService householdService = SpringUtils.getBean(IHouseholdService.class);
      HouseholdEntity householdEntity = householdService.getOne(Wrappers.<HouseholdEntity>lambdaQuery()
         .eq(HouseholdEntity::getHouseCode, userTopics.getHouseCode())
         .eq(HouseholdEntity::getAssociatedUserId, AuthUtil.getUserId())
         .eq(HouseholdEntity::getIsDeleted, 0)
         .last("limit 1"));
      if (householdEntity != null) {
         userTopics.setHouseholdId(householdEntity.getId());
      }
      List<UserTopicsVO> userTopicsVOS = baseMapper.selectUserTopicsPage(page, userTopics);
      return page.setRecords(userTopicsVOS);
   }
   @Override
   @Transactional(rollbackFor = Exception.class)
   public String batchSave(List<TopicsVO> topics) throws Exception {
   public String batchSave(TopicsVO topics) throws Exception {
      if (StringUtils.isNotBlank(topics.getPhone())) {
         if (StringUtils.isBlank(topics.getCode())) {
            return "验证码不能为空!";
         }
         Object validateCode = redisTemplate.get(SMS_VALIDATE_PHONE + topics.getPhone());
         if (validateCode == null) {
            return "验证码已过期";
         }
         if (!validateCode.toString().equals(topics.getCode())) {
            return "验证码错误";
         }
         //删除验证码
         redisTemplate.del(SMS_VALIDATE_PHONE + topics.getPhone());
      }
      // 判断是否一户一票 还是一人一票
      List<TopicsVO> topicsList = topics.getChildren();
      IHouseholdService householdService = SpringUtils.getBean(IHouseholdService.class);
      HouseholdEntity householdEntity = householdService.getOne(Wrappers.<HouseholdEntity>lambdaQuery()
         .eq(HouseholdEntity::getHouseCode, topicsList.get(0).getHouseCode())
         .eq(HouseholdEntity::getAssociatedUserId, AuthUtil.getUserId())
         .eq(HouseholdEntity::getIsDeleted, 0)
         .last("limit 1"));
      IPublicDiscussService bean = SpringUtil.getBean(IPublicDiscussService.class);
      PublicDiscussEntity one = bean.getOne(Wrappers.<PublicDiscussEntity>lambdaQuery().eq(PublicDiscussEntity::getArticleId, topics.get(0).getArticleId()));
      PublicDiscussEntity discussEntity = bean.getOne(Wrappers.<PublicDiscussEntity>lambdaQuery().eq(PublicDiscussEntity::getArticleId, topicsList.get(0).getArticleId()));
      // 判断是否在开始和结束时间内
      if (discussEntity.getStartTime().getTime() > System.currentTimeMillis() || discussEntity.getEndTime().getTime() < System.currentTimeMillis()) {
         return "目前不在投票时间内";
      }
      // 一户一票
      if (one.getVoteRestrictions().equals(CommonConstant.NUMBER_ONE)) {
      if (discussEntity.getVoteRestrictions().equals(CommonConstant.NUMBER_ONE)) {
         long count = count(Wrappers.<UserTopicsEntity>lambdaQuery()
            .eq(UserTopicsEntity::getHouseCode, topics.get(0).getHouseCode())
            .eq(UserTopicsEntity::getArticleId, topics.get(0).getArticleId()));
            .eq(UserTopicsEntity::getHouseCode, topicsList.get(0).getHouseCode())
            .eq(UserTopicsEntity::getArticleId, topicsList.get(0).getArticleId()));
         if (count > 0) {
            return "您的房屋已投票,不能重复投票!";
         }
      } else {
         //
         long count = count(Wrappers.<UserTopicsEntity>lambdaQuery()
            .eq(UserTopicsEntity::getUserId, AuthUtil.getUserId())
            .eq(UserTopicsEntity::getArticleId, topics.get(0).getArticleId()));
            .eq(UserTopicsEntity::getHouseholdId, householdEntity.getId())
            .eq(UserTopicsEntity::getArticleId, topicsList.get(0).getArticleId()));
         if (count > 0) {
            return "您的已投票,不能重复投票!";
            return "您已投票,不能重复投票!";
         }
      }
      Boolean userTopics = getaBoolean(topics);
      for (TopicsVO topic : topicsList) {
         if (topic.getMandatoryFlag().equals(2)) {
            if (topic.getOptionNumberMin() > 1) {
               JSONArray objects1 = JSON.parseArray(topic.getSelected());
               if (objects1.size() < topic.getOptionNumberMin()) {
                  return topic.getDiscussContent() + ":议题最少需要选择" + topic.getOptionNumberMin() + "项";
               }
            }
            if (topic.getOptionNumber() > 1) {
               JSONArray objects1 = JSON.parseArray(topic.getSelected());
               if (objects1.size() > topic.getOptionNumber()) {
                  return topic.getDiscussContent() + ":议题最多需要选择" + topic.getOptionNumber() + "项";
               }
            }
         }
      }
      Boolean userTopics = getaBoolean(topicsList, householdEntity);
      if (userTopics) return "200";
      return "操作失败!";
   }
   @Nullable
   private Boolean getaBoolean(List<TopicsVO> topics) {
   private Boolean getaBoolean(List<TopicsVO> topics, HouseholdEntity householdEntity) {
      List<UserTopicsEntity> objects = new ArrayList<>();
      Boolean result = false;
      for (TopicsVO topic : topics) {
         UserTopicsEntity userTopicsEntity = new UserTopicsEntity();
         userTopicsEntity.setUserId(AuthUtil.getUserId());
         userTopicsEntity.setHouseholdId(householdEntity.getId());
         userTopicsEntity.setSelected(topic.getSelected());
         userTopicsEntity.setTopicsId(topic.getId());
         userTopicsEntity.setPublicDiscussId(topic.getPublicDiscussId());
@@ -121,7 +191,7 @@
            }
            UserTopicsEntity userTopics = new UserTopicsEntity();
            userTopics.setSelected(topic.getSelected());
            userTopics.setUserId(AuthUtil.getUserId());
            userTopics.setHouseholdId(householdEntity.getId());
            userTopics.setPublicDiscussId(topic.getPublicDiscussId());
            userTopics.setTopicsId(Integer.valueOf(topic.getSelected()));
            userTopics.setArticleId(topic.getArticleId());
@@ -131,7 +201,7 @@
            objectUpdateWrapper.setSql("number = number + 1");
            objectUpdateWrapper.eq("id", topic.getSelected());
            topicsService.update(null, objectUpdateWrapper);
            return save(userTopics);
            result = save(userTopics);
         } else {
            // 多选
            if (StringUtils.isBlank(topic.getSelected())) {
@@ -142,7 +212,7 @@
            for (Object o : objects1) {
               UserTopicsEntity userTopics = new UserTopicsEntity();
               userTopics.setSelected(topic.getSelected());
               userTopics.setUserId(AuthUtil.getUserId());
               userTopics.setHouseholdId(householdEntity.getId());
               userTopics.setPublicDiscussId(topic.getPublicDiscussId());
               userTopics.setArticleId(topic.getArticleId());
               userTopics.setHouseCode(topic.getHouseCode());
@@ -154,10 +224,10 @@
               objectUpdateWrapper.eq("id", o);
               topicsService.update(null, objectUpdateWrapper);
            }
            return saveBatch(objectsTwo);
            result = saveBatch(objectsTwo);
         }
      }
      return false;
      return result;
   }
@@ -165,14 +235,14 @@
   public Integer getCount(Integer id) {
      List<UserTopicsEntity> list = list(Wrappers.<UserTopicsEntity>lambdaQuery()
         .eq(UserTopicsEntity::getPublicDiscussId, id)
         .groupBy(UserTopicsEntity::getUserId));
         .groupBy(UserTopicsEntity::getHouseholdId));
      return list.size();
   }
   @Override
   public List<UserTopicsExcel> exportUser(UserTopicsVO userTopics) {
      if (StringUtils.isNotBlank(userTopics.getDistrictId())) {
         List<String> longs = JSON.parseArray(userTopics.getDistrictId()).toJavaList(String.class);
         List<String> longs = Arrays.asList(userTopics.getDistrictId().split(","));
         IDistrictService bean = SpringUtils.getBean(IDistrictService.class);
         List<DistrictEntity> list = bean.list(Wrappers.<DistrictEntity>lambdaQuery().in(DistrictEntity::getId, longs));
         List<String> collect = list.stream().map(item ->
@@ -185,4 +255,154 @@
      List<UserTopicsExcel> userTopicsExcels = baseMapper.exportUser(userTopics);
      return userTopicsExcels;
   }
   @Override
   public Boolean saveUserTopicsEntity(UserTopicsEntity userTopics) {
      IHouseholdService householdService = SpringUtils.getBean(IHouseholdService.class);
      HouseholdEntity householdEntity = householdService.getOne(Wrappers.<HouseholdEntity>lambdaQuery()
         .eq(HouseholdEntity::getHouseCode, userTopics.getHouseCode())
         .eq(HouseholdEntity::getAssociatedUserId, AuthUtil.getUserId())
         .eq(HouseholdEntity::getIsDeleted, 0)
         .last("limit 1"));
      if (householdEntity != null) {
         userTopics.setHouseholdId(householdEntity.getId());
      }
      return save(userTopics);
   }
   /**
    * 处理Excel文件的生成,将用户主题数据写入Excel并输出到流中。
    *
    * @param out        输出流,用于保存生成的Excel文件。
    * @param userTopics 用户主题数据,包含提案和投票记录信息。
    */
   @Override
   public void handleExcel(OutputStream out, UserTopicsVO userTopics) {
      // 初始化Excel写入器
      ExcelWriter excelWriter = EasyExcel.write(out)
         .build();
      // 设置内容自动换行的样式
      WriteCellStyle contentWriteCellStyle = new WriteCellStyle();
      contentWriteCellStyle.setWrapped(true);
      // 处理并写入提案数据
      List<holdExcel> proposal = getProposal(userTopics);
      ExcelWriterSheetBuilder userExcelSheet = new ExcelWriterSheetBuilder();
      userExcelSheet.registerWriteHandler(new HorizontalCellStyleStrategy(null, contentWriteCellStyle));
      userExcelSheet.sheetName("小区投票人员");
      userExcelSheet.sheetNo(0);
      userExcelSheet.head(holdExcel.class);
      excelWriter.write(proposal, userExcelSheet.build());
      // 处理并写入投票记录数据
      List<UserTopicsExcel> publicopinion = getPublicopinion(userTopics);
      ExcelWriterSheetBuilder classExcelSheet = new ExcelWriterSheetBuilder();
      classExcelSheet.registerWriteHandler(new HorizontalCellStyleStrategy(null, contentWriteCellStyle));
      classExcelSheet.sheetName("投票记录");
      classExcelSheet.sheetNo(1);
      classExcelSheet.head(UserTopicsExcel.class);
      excelWriter.write(publicopinion, classExcelSheet.build());
      // 完成Excel写入并释放资源
      excelWriter.finish();
   }
   /**
    * 获取首页的房屋记录
    *
    * @param userTopics 包含用户话题信息的对象,其中可能含有区域ID等关键信息
    * @return 返回一个包含房屋记录信息的列表
    */
   private List<holdExcel> getProposal(UserTopicsVO userTopics) {
      List<holdExcel> resultList = new ArrayList<>();
      try {
         // 检查区域ID字符串是否有效
         List<String> strings = validateAndParseDistrictId(userTopics.getDistrictId());
         // 获取区域服务和话题服务的实例
         IDistrictService districtService = SpringUtils.getBean(IDistrictService.class);
         ITopicsService topicsService = SpringUtils.getBean(ITopicsService.class);
         // 处理区域ID并查询对应的区域实体列表
         List<String> aoiCodeList = processDistrictIds(districtService, strings);
         // 设置AOI代码列表到用户话题对象中
         userTopics.setAoiCodeList(aoiCodeList);
         // 查询并处理房屋列表信息
         List<holdExcel> householdList = processHouseholdList(topicsService, userTopics);
         resultList.addAll(householdList);
      } catch (Exception e) {
         // 适当的异常处理逻辑
         // 例如:记录日志、返回错误信息等
         e.printStackTrace();
      }
      return resultList;
   }
   /**
    * 验证并解析区域ID字符串。
    *
    * @param districtIdStr 区域ID的字符串表示。不能为空或空白字符串。
    * @return 返回一个JSONArray对象,包含解析后的区域ID。
    * @throws IllegalArgumentException 如果区域ID字符串为空或空白字符串,抛出此异常。
    */
   private List<String>  validateAndParseDistrictId(String districtIdStr) throws IllegalArgumentException {
      // 验证区域ID字符串是否为空
      if (StringUtils.isBlank(districtIdStr)) {
         throw new IllegalArgumentException("区域ID不能为空");
      }
      // 解析区域ID字符串为JSONArray对象
      return  Arrays.asList(districtIdStr.split(","));
   }
   private List<String> processDistrictIds(IDistrictService districtService, List<String> districtIds) {
      List<String> collect1 = districtIds.stream().map(item -> (String) item).collect(Collectors.toList());
      List<DistrictEntity> list3 = districtService.list(Wrappers.<DistrictEntity>lambdaQuery().in(DistrictEntity::getId, collect1));
      return list3.stream().map(item -> item.getAoiCode()).collect(Collectors.toList());
   }
   private List<holdExcel> processHouseholdList(ITopicsService topicsService, UserTopicsVO userTopics) {
      List<holdExcel> householdList = baseMapper.getHouseholdList(userTopics);
      householdList.forEach(item -> processItem(topicsService, item));
      return householdList;
   }
   private void processItem(ITopicsService topicsService, holdExcel item) {
      if (StringUtils.isBlank(item.getOptionContent())) {
         item.setVoteFlag("否");
      } else {
         List<Long> longs = Func.toLongList(item.getOptionContent());
         for (Long aLong : longs) {
            TopicsEntity topics = topicsService.getById(aLong);
            setManagementRegulationsOrRulesOfProcedure(item, topics);
         }
         item.setVoteFlag("是");
      }
   }
   private void setManagementRegulationsOrRulesOfProcedure(holdExcel item, TopicsEntity topics) {
      if (StringUtils.isNotBlank(topics.getDiscussContent())) {
         if (topics.getDiscussContent().equals("管理公约")) {
            item.setManagementRegulations(topics.getOptionContent());
         } else if (topics.getDiscussContent().equals("议事规则")) {
            item.setRulesOfProcedure(topics.getOptionContent());
         }
      }
   }
   /**
    * 获取公开意见信息
    *
    * @param userTopics 用户话题VO对象,封装了用户话题的查询条件
    * @return 返回用户话题的Excel格式列表,包含所有符合条件的意见信息
    */
   private List<UserTopicsExcel> getPublicopinion(UserTopicsVO userTopics) {
      // 初始化返回列表
      List<UserTopicsExcel> list = new ArrayList<>();
      // 通过基础mapper查询结果,并将结果添加到列表中
      List<UserTopicsExcel> userTopicsExcels = baseMapper.getresultTwo(userTopics);
      list.addAll(userTopicsExcels);
      return list;
   }
}