From 89a0dfd28a77b7acea5e5f54edac352535141f69 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 01 Jul 2020 16:28:18 +0800
Subject: [PATCH] modify regist need inviteCode

---
 src/main/java/com/xcong/excoin/modules/member/service/impl/MemberServiceImpl.java |   12 ++++++++----
 1 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/member/service/impl/MemberServiceImpl.java b/src/main/java/com/xcong/excoin/modules/member/service/impl/MemberServiceImpl.java
index 364b1b3..89dc55e 100644
--- a/src/main/java/com/xcong/excoin/modules/member/service/impl/MemberServiceImpl.java
+++ b/src/main/java/com/xcong/excoin/modules/member/service/impl/MemberServiceImpl.java
@@ -122,17 +122,21 @@
         member.setPassword(SecureUtil.md5(registerDto.getPassword()));
 
         // 判断账号类型
-        if (MemberEntity.ACCOUNT_TYPE_PHONE == registerDto.getType()) {
+        if (MemberEntity.ACCOUNT_TYPE_PHONE.equals(registerDto.getType())) {
             member.setPhone(registerDto.getAccount());
-        } else if (MemberEntity.ACCOUNT_TYPE_EMAIL == registerDto.getType()) {
+        } else if (MemberEntity.ACCOUNT_TYPE_EMAIL.equals(registerDto.getType())) {
             member.setEmail(registerDto.getAccount());
         } else {
             return Result.fail("账号类型错误");
         }
 
         // 判断是否拥有推荐人,若为空则默认系统
-        if (StrUtil.isBlank(registerDto.getRefererId())) {
-            registerDto.setRefererId(AppContants.SYSTEM_REFERER);
+//        if (StrUtil.isBlank(registerDto.getRefererId())) {
+//            registerDto.setRefererId(AppContants.SYSTEM_REFERER);
+//        }
+        MemberEntity isExist = memberDao.selectMemberInfoByInviteId(registerDto.getRefererId());
+        if (isExist == null) {
+            return Result.fail("推荐人不存在");
         }
 
         member.setRefererId(registerDto.getRefererId());

--
Gitblit v1.9.1