From 940dd1b292db4c5cc2df2f137fa46c37f2159898 Mon Sep 17 00:00:00 2001
From: Pu Zhibing <393733352@qq.com>
Date: 星期四, 05 十二月 2024 18:06:01 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/AppUserServiceImpl.java |   68 ++++++++++++++++++++++++---------
 1 files changed, 49 insertions(+), 19 deletions(-)

diff --git a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/AppUserServiceImpl.java b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/AppUserServiceImpl.java
index bfb01d7..1a219fb 100644
--- a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/AppUserServiceImpl.java
+++ b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/AppUserServiceImpl.java
@@ -10,7 +10,6 @@
 import com.ruoyi.account.util.weChat.WeChatUtil;
 import com.ruoyi.account.vo.*;
 import com.ruoyi.common.core.domain.R;
-import com.ruoyi.common.core.web.domain.AjaxResult;
 import com.ruoyi.common.redis.service.RedisService;
 import com.ruoyi.common.security.service.TokenService;
 import com.ruoyi.other.api.domain.Shop;
@@ -23,9 +22,7 @@
 import javax.annotation.Resource;
 import java.math.BigDecimal;
 import java.time.LocalDateTime;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * <p>
@@ -195,18 +192,17 @@
 //			return R.fail("验证码错误");
 //		}
 		//使用jscode获取微信openid
-//		Map<String, Object> map = weChatUtil.code2Session(registerAccount.getJscode());
-//		Integer errcode = Integer.valueOf(map.get("errcode").toString());
-//		if(0 != errcode){
-//			return R.fail(map.get("msg").toString());
-//		}
-//		String openid = map.get("openid").toString();
-//		//查询用户是否注册
-//		AppUser appUser = this.getOne(new LambdaQueryWrapper<AppUser>().eq(AppUser::getWxOpenid, openid).ne(AppUser::getStatus, 3).eq(AppUser::getDelFlag, 0));
-//		if(null != appUser){
-//			return R.fail("此微信号已注册,请直接登录!");
-//		}
-		AppUser appUser = null;
+		Map<String, Object> map = weChatUtil.code2Session(registerAccount.getJscode());
+		Integer errcode = Integer.valueOf(map.get("errcode").toString());
+		if(0 != errcode){
+			return R.fail(map.get("msg").toString());
+		}
+		String openid = map.get("openid").toString();
+		//查询用户是否注册
+		AppUser appUser = this.getOne(new LambdaQueryWrapper<AppUser>().eq(AppUser::getWxOpenid, openid).ne(AppUser::getStatus, 3).eq(AppUser::getDelFlag, 0));
+		if(null != appUser){
+			return R.fail("此微信号已注册,请直接登录!");
+		}
 		AppUser appUser1 = this.getOne(new LambdaQueryWrapper<AppUser>().eq(AppUser::getPhone, registerAccount.getPhone()).ne(AppUser::getStatus, 3).eq(AppUser::getDelFlag, 0));
 		if(null != appUser1){
 			return R.fail("手机号已注册,请直接登录!");
@@ -215,7 +211,7 @@
 			appUser = new AppUser();
 			appUser.setName(registerAccount.getName());
 			appUser.setPhone(registerAccount.getPhone());
-//			appUser.setWxOpenid(openid);
+			appUser.setWxOpenid(openid);
 			//注册默认为普通会员
 			appUser.setVipId(1);
 			appUser.setStatus(1);
@@ -308,11 +304,11 @@
 	}
 	
 	public AppUser getTopAppUser(List<AppUser> list, Long id){
-		AppUser appUser = list.stream().filter(s -> s.getInviteUserId().equals(id)).findFirst().get();
+		AppUser appUser = list.stream().filter(s -> s.getId().equals(id)).findFirst().get();
 		if(null == appUser.getInviteUserId()){
 			return appUser;
 		}
-		return getTopAppUser(list, appUser.getId());
+		return getTopAppUser(list, appUser.getInviteUserId());
 	}
 	
 	
@@ -330,4 +326,38 @@
 		List<NearbyReferrerVo> list = this.baseMapper.getNearbyReferrer(cityCode, nearbyReferrer);
 		return list;
 	}
+
+	@Override
+	public List<AppUser> getUserAncestorList(Long id, List<AppUser> list) {
+		if (list == null) {
+			list = new ArrayList<>();
+		}
+
+		Set<Long> visitedIds = new HashSet<>();
+		Long currentId = id;
+
+		while (currentId != null && !visitedIds.contains(currentId)) {
+			AppUser appUser = getById(currentId);
+			if (appUser == null) {
+				break; // 如果用户不存在,终止循环
+			}
+
+			Long inviteUserId = appUser.getInviteUserId();
+			if (inviteUserId != null) {
+				AppUser invitedUser = getById(inviteUserId);
+				if (invitedUser != null) {
+					list.add(invitedUser);
+					visitedIds.add(currentId);
+					currentId = inviteUserId;
+				} else {
+					break; // 如果邀请用户不存在,终止循环
+				}
+			} else {
+				break; // 如果没有邀请用户,终止循环
+			}
+		}
+
+		return list;
+	}
+
 }

--
Gitblit v1.7.1