From 6a53233a381819bcdcc7acdc3a018db9b3b8a311 Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Sat, 06 Jan 2024 16:12:55 +0800
Subject: [PATCH] 议事报名

---
 src/main/java/org/springblade/modules/discuss/service/impl/UserTopicsServiceImpl.java |   50 ++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 42 insertions(+), 8 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 d6f00db..16cbbd8 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
@@ -23,13 +23,18 @@
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.apache.commons.lang3.StringUtils;
+import org.jetbrains.annotations.Nullable;
+import org.springblade.common.constant.CommonConstant;
 import org.springblade.core.secure.utils.AuthUtil;
+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.UserPublicEnrollEntity;
 import org.springblade.modules.discuss.entity.UserTopicsEntity;
 import org.springblade.modules.discuss.mapper.UserTopicsMapper;
+import org.springblade.modules.discuss.service.IPublicDiscussService;
 import org.springblade.modules.discuss.service.ITopicsService;
 import org.springblade.modules.discuss.service.IUserTopicsService;
+import org.springblade.modules.discuss.vo.TopicsVO;
 import org.springblade.modules.discuss.vo.UserTopicsVO;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -57,29 +62,56 @@
 
 	@Override
 	@Transactional(rollbackFor = Exception.class)
-	public Boolean batchSave(List<TopicsEntity> topics) {
+	public Boolean batchSave(List<TopicsVO> topics) throws Exception {
+		// 判断是否一户一票 还是一人一票
+		IPublicDiscussService bean = SpringUtil.getBean(IPublicDiscussService.class);
+		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()));
+			if (count > 1) {
+				throw new Exception("您的房屋已投票,不能重复投票!");
+			}
+		} else {
+			//
+			long count = count(Wrappers.<UserTopicsEntity>lambdaQuery().eq(UserTopicsEntity::getUserId, AuthUtil.getUserId()));
+			if (count > 1) {
+				throw new Exception("您的已投票,不能重复投票!");
+			}
+		}
+		Boolean userTopics = getaBoolean(topics);
+		if (userTopics != null) return userTopics;
+		return false;
+	}
+
+	@Nullable
+	private Boolean getaBoolean(List<TopicsVO> topics) {
 		List<UserTopicsEntity> objects = new ArrayList<>();
-		for (TopicsEntity topic : topics) {
+		for (TopicsVO topic : topics) {
 			UserTopicsEntity userTopicsEntity = new UserTopicsEntity();
 			userTopicsEntity.setUserId(AuthUtil.getUserId());
 			userTopicsEntity.setSelected(topic.getSelected());
 			userTopicsEntity.setTopicsId(topic.getId());
 			userTopicsEntity.setPublicDiscussId(topic.getPublicDiscussId());
 			objects.add(userTopicsEntity);
+			// 单选
 			if (topic.getOptionRange().equals(0)) {
 				if (StringUtils.isBlank(topic.getSelected())) {
 					break;
 				}
 				UserTopicsEntity userTopics = new UserTopicsEntity();
-				userTopics.setTopicsId(Integer.valueOf(topic.getSelected()));
+				userTopics.setSelected(topic.getSelected());
 				userTopics.setUserId(AuthUtil.getUserId());
 				userTopics.setPublicDiscussId(topic.getPublicDiscussId());
+				userTopics.setArticleId(topic.getArticleId());
+				userTopics.setHouseCode(topic.getHouseCode());
 				UpdateWrapper<TopicsEntity> objectUpdateWrapper = new UpdateWrapper<>();
 				objectUpdateWrapper.setSql("number = number + 1");
 				objectUpdateWrapper.eq("id", topic.getSelected());
 				topicsService.update(null, objectUpdateWrapper);
-				baseMapper.insert(userTopics);
+				return save(userTopics);
 			} else {
+				// 多选
 				if (StringUtils.isBlank(topic.getSelected())) {
 					break;
 				}
@@ -87,19 +119,21 @@
 				List<UserTopicsEntity> objectsTwo = new ArrayList<>();
 				for (Object o : objects1) {
 					UserTopicsEntity userTopics = new UserTopicsEntity();
-					userTopics.setTopicsId((Integer) o);
+					userTopics.setSelected(topic.getSelected());
 					userTopics.setUserId(AuthUtil.getUserId());
 					userTopics.setPublicDiscussId(topic.getPublicDiscussId());
+					userTopics.setArticleId(topic.getArticleId());
+					userTopics.setHouseCode(topic.getHouseCode());
 					UpdateWrapper<TopicsEntity> objectUpdateWrapper = new UpdateWrapper<>();
 					objectUpdateWrapper.setSql("number = number + 1");
 					objectUpdateWrapper.eq("id", o);
 					objectsTwo.add(userTopics);
 					topicsService.update(null, objectUpdateWrapper);
 				}
-				saveOrUpdateBatch(objectsTwo);
+				return save(objectsTwo.get(0));
 			}
 		}
-		return saveOrUpdateBatch(objects);
+		return null;
 	}
 
 

--
Gitblit v1.9.3