From 27e5e0716b9e576100515058fd589cd98bd9bb72 Mon Sep 17 00:00:00 2001
From: lin <sbla5888@163.com>
Date: Wed, 27 Mar 2024 13:21:27 +0800
Subject: [PATCH] 校验

---
 src/main/java/org/springblade/modules/discuss/service/impl/PublicDiscussServiceImpl.java |   39 +++++++++++++++++++++++++++++++++++++--
 1 files changed, 37 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/springblade/modules/discuss/service/impl/PublicDiscussServiceImpl.java b/src/main/java/org/springblade/modules/discuss/service/impl/PublicDiscussServiceImpl.java
index 1a63151..3187b36 100644
--- a/src/main/java/org/springblade/modules/discuss/service/impl/PublicDiscussServiceImpl.java
+++ b/src/main/java/org/springblade/modules/discuss/service/impl/PublicDiscussServiceImpl.java
@@ -17,9 +17,13 @@
 package org.springblade.modules.discuss.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.constant.CommonConstant;
+import org.springblade.common.utils.SpringUtils;
 import org.springblade.core.mp.support.Condition;
 import org.springblade.core.mp.support.Query;
+import org.springblade.core.secure.utils.AuthUtil;
 import org.springblade.modules.discuss.entity.PublicDiscussEntity;
 import org.springblade.modules.discuss.entity.UserTopicsEntity;
 import org.springblade.modules.discuss.mapper.PublicDiscussMapper;
@@ -28,6 +32,8 @@
 import org.springblade.modules.discuss.service.IUserTopicsService;
 import org.springblade.modules.discuss.vo.PublicDiscussVO;
 import org.springblade.modules.discuss.vo.UserTopicsVO;
+import org.springblade.modules.house.entity.HouseholdEntity;
+import org.springblade.modules.house.service.IHouseholdService;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
@@ -53,7 +59,7 @@
 		List<PublicDiscussVO> publicDiscussVOS = baseMapper.selectPublicDiscussPage(page, publicDiscuss);
 		for (PublicDiscussVO publicDiscussVO : publicDiscussVOS) {
 			Long result = iUserPublicEnrollService.getCount(publicDiscussVO.getId());
-			if(result != null){
+			if (result != null) {
 				publicDiscussVO.setEnrollCount(result.intValue());
 			}
 			UserTopicsVO userTopicsVo = new UserTopicsVO();
@@ -73,6 +79,35 @@
 
 	@Override
 	public PublicDiscussVO getDetail(PublicDiscussVO publicDiscuss) {
-		return baseMapper.selectPublicDiscuss(publicDiscuss);
+		IHouseholdService householdService = SpringUtils.getBean(IHouseholdService.class);
+		HouseholdEntity householdEntity = householdService.getOne(Wrappers.<HouseholdEntity>lambdaQuery()
+			.eq(HouseholdEntity::getHouseCode, publicDiscuss.getHouseCode())
+			.eq(HouseholdEntity::getAssociatedUserId, AuthUtil.getUserId())
+			.eq(HouseholdEntity::getIsDeleted, 0)
+			.last("limit 1"));
+		if (householdEntity != null) {
+			publicDiscuss.setHouseholdId(householdEntity.getId());
+		}
+		PublicDiscussVO publicDiscussVO = baseMapper.selectPublicDiscuss(publicDiscuss);
+		publicDiscussVO.setDisabled(false);
+		// 一户一票
+		IUserTopicsService bean = SpringUtils.getBean(IUserTopicsService.class);
+		if (publicDiscussVO.getVoteRestrictions().equals(CommonConstant.NUMBER_ONE)) {
+			long count = bean.count(Wrappers.<UserTopicsEntity>lambdaQuery()
+				.eq(UserTopicsEntity::getHouseCode, publicDiscuss.getHouseCode())
+				.eq(UserTopicsEntity::getArticleId, publicDiscussVO.getArticleId()));
+			if (count > 0) {
+				publicDiscussVO.setDisabled(true);
+			}
+		} else {
+			// 一个一票
+			long count = bean.count(Wrappers.<UserTopicsEntity>lambdaQuery()
+				.eq(UserTopicsEntity::getHouseholdId, householdEntity.getId())
+				.eq(UserTopicsEntity::getArticleId, publicDiscussVO.getArticleId()));
+			if (count > 0) {
+				publicDiscussVO.setDisabled(true);
+			}
+		}
+		return publicDiscussVO;
 	}
 }

--
Gitblit v1.9.3