From 566e7986291e73051d30ee252b0ebf852b1577a7 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Tue, 02 Apr 2024 14:25:51 +0800
Subject: [PATCH] 娱乐场所标签新增180200

---
 src/main/java/org/springblade/modules/discuss/service/impl/UserTopicsServiceImpl.java |  228 +++++++++++++++++++++++++++++++++-----------------------
 1 files changed, 134 insertions(+), 94 deletions(-)

diff --git a/src/main/java/org/springblade/modules/discuss/service/impl/UserTopicsServiceImpl.java b/src/main/java/org/springblade/modules/discuss/service/impl/UserTopicsServiceImpl.java
index a432098..71be2b8 100644
--- a/src/main/java/org/springblade/modules/discuss/service/impl/UserTopicsServiceImpl.java
+++ b/src/main/java/org/springblade/modules/discuss/service/impl/UserTopicsServiceImpl.java
@@ -16,9 +16,9 @@
  */
 package org.springblade.modules.discuss.service.impl;
 
-import com.alibaba.excel.EasyExcelFactory;
+import com.alibaba.excel.EasyExcel;
 import com.alibaba.excel.ExcelWriter;
-import com.alibaba.excel.write.metadata.WriteSheet;
+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;
@@ -28,18 +28,18 @@
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.apache.commons.lang3.StringUtils;
-import org.apache.poi.ss.usermodel.HorizontalAlignment;
 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.householdExcel;
+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;
@@ -48,12 +48,9 @@
 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.HouseEntity;
 import org.springblade.modules.house.entity.HouseholdEntity;
 import org.springblade.modules.house.excel.HouseHoldExcel;
-import org.springblade.modules.house.service.IHouseService;
 import org.springblade.modules.house.service.IHouseholdService;
-import org.springblade.modules.house.vo.HouseholdVO;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -265,97 +262,140 @@
 		return save(userTopics);
 	}
 
-	@Override
-	public void handleExcel(OutputStream out, UserTopicsVO userTopics) {
-		ExcelWriter excelWriter = EasyExcelFactory.write(out).build();
-		//设置内容样式
-		WriteCellStyle contentStyle = new WriteCellStyle();
-		contentStyle.setHorizontalAlignment(HorizontalAlignment.CENTER);//居中
-		contentStyle.setWrapped(true);//自动换行
-		//设置头部样式
-		WriteCellStyle headerStyle = new WriteCellStyle();
-		headerStyle.setHorizontalAlignment(HorizontalAlignment.CENTER);
-		//设置策略
-		HorizontalCellStyleStrategy horizontalCellStyleStrategy = new HorizontalCellStyleStrategy(headerStyle, contentStyle);
-		WriteSheet proposalSheet = EasyExcelFactory.writerSheet(0, "委员提案").head(HouseHoldExcel.class).registerWriteHandler(horizontalCellStyleStrategy).build();
-		WriteSheet publicopinionSheet = EasyExcelFactory.writerSheet(1, "社情民意").head(HouseHoldExcel.class).registerWriteHandler(horizontalCellStyleStrategy).build();
-		excelWriter.write(getProposal(userTopics), proposalSheet);
-		excelWriter.write(getPublicopinion(userTopics), publicopinionSheet);
+/**
+ * 处理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();
 
 
-	}
+}
 
-	//首页-房屋记录
-	private List<householdExcel> getProposal(UserTopicsVO userTopics) {
-		List<householdExcel> list = new ArrayList<>();
-		JSONArray objects1 = JSON.parseArray(userTopics.getDistrictId());
-		List<String> collect1 = objects1.stream().map(item -> (String)item).collect(Collectors.toList());
-		IDistrictService bean3 = SpringUtils.getBean(IDistrictService.class);
-		List<DistrictEntity> list3 = bean3.list(Wrappers.<DistrictEntity>lambdaQuery().in(DistrictEntity::getId, collect1));
-		List<String> aoiCodeList = list3.stream().map(item -> item.getAoiCode() ).collect(Collectors.toList());
-		userTopics.setAoiCodeList(aoiCodeList);
-		List<householdExcel> householdList = baseMapper.getHouseholdList(userTopics);
-//
-//		householdExcel vo = new householdExcel();
-//		list.add(vo);
-//		IHouseService bean = SpringUtils.getBean(IHouseService.class);
-//		JSONArray objects = JSON.parseArray(userTopics.getDistrictId());
-//
-//		List<HouseEntity> list1 = bean.list(Wrappers.<HouseEntity>lambdaQuery().in(HouseEntity::getDistrictCode, aoiCodeList));
-//
-//		IHouseholdService bean1 = SpringUtils.getBean(IHouseholdService.class);
-//
-//		IUserTopicsService bean2 = SpringUtils.getBean(IUserTopicsService.class);
-//
-//		for (HouseEntity houseEntity : list1) {
-//			householdExcel householdExcel = new householdExcel();
-//			householdExcel.setArea(houseEntity.getArea());
-//			householdExcel.setBuilding(houseEntity.getBuilding());
-//			householdExcel.setRoom(houseEntity.getRoom());
-//			householdExcel.setRemark(houseEntity.getRemark());
-//			// 查询住户
-//			List<HouseholdEntity> list2 = bean1.list(Wrappers.<HouseholdEntity>lambdaQuery()
-//				.eq(HouseholdEntity::getHouseCode, houseEntity.getHouseCode())
-//				.eq(HouseholdEntity::getIsDeleted, 0)
-//				.ne(HouseholdEntity::getRelationship, 18));
-//			// 判断住户是否有投票
-//			for (HouseholdEntity householdEntity : list2) {
-//				householdExcel.setName(householdEntity.getName());
-//				householdExcel.setPhone(householdEntity.getPhoneNumber());
-//				long count = bean2.count(Wrappers.<UserTopicsEntity>lambdaQuery()
-//					.eq(UserTopicsEntity::getHouseholdId, householdEntity.getId()));
-//				if (count > 0) {
-//					// 查询选择的结果
-//					userTopics.setDiscussContent("管理规约");
-//					userTopics.setHouseholdId(householdEntity.getId());
-//					UserTopicsVO getresult = baseMapper.getresult(userTopics);
-//					if (getresult != null) {
-//						householdExcel.setManagementRegulations(getresult.getOptionContent());
-//					}
-//					userTopics.setDiscussContent("议事规则");
-//					UserTopicsVO getresult2 = baseMapper.getresult(userTopics);
-//					if (getresult2 != null) {
-//						householdExcel.setRulesOfProcedure(getresult2.getOptionContent());
-//					}
-//					householdExcel.setVoteFlag("是");
-//					break;
-//				} else {
-//					householdExcel.setVoteFlag("否");
-//				}
-//			}
-			// 添加
-			list.addAll(householdList);
-//		}
-		return list;
-	}
+    /**
+     * 获取首页的房屋记录
+     *
+     * @param userTopics 包含用户话题信息的对象,其中可能含有区域ID等关键信息
+     * @return 返回一个包含房屋记录信息的列表
+     */
+    private List<holdExcel> getProposal(UserTopicsVO userTopics) {
+        List<holdExcel> resultList = new ArrayList<>();
+        try {
+            // 检查区域ID字符串是否有效
+            JSONArray objects1 = validateAndParseDistrictId(userTopics.getDistrictId());
+            // 获取区域服务和话题服务的实例
+            IDistrictService districtService = SpringUtils.getBean(IDistrictService.class);
+            ITopicsService topicsService = SpringUtils.getBean(ITopicsService.class);
+            // 处理区域ID并查询对应的区域实体列表
+            List<String> aoiCodeList = processDistrictIds(districtService, objects1);
+            // 设置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 JSONArray validateAndParseDistrictId(String districtIdStr) throws IllegalArgumentException {
+        // 验证区域ID字符串是否为空
+        if (StringUtils.isBlank(districtIdStr)) {
+            throw new IllegalArgumentException("区域ID不能为空");
+        }
+        // 解析区域ID字符串为JSONArray对象
+        return JSON.parseArray(districtIdStr);
+    }
+
+    private List<String> processDistrictIds(IDistrictService districtService, JSONArray 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<>();
-		List<UserTopicsExcel> userTopicsExcels = baseMapper.getresultTwo(userTopics);
-		UserTopicsExcel vo = new UserTopicsExcel();
-		list.add(vo);
-		list.addAll(userTopicsExcels);
-		return list;
+	    // 初始化返回列表
+	    List<UserTopicsExcel> list = new ArrayList<>();
+	    // 通过基础mapper查询结果,并将结果添加到列表中
+	    List<UserTopicsExcel> userTopicsExcels = baseMapper.getresultTwo(userTopics);
+	    list.addAll(userTopicsExcels);
+	    return list;
 	}
 }

--
Gitblit v1.9.3