From a3ead770c30b310b9b1b7e3786d5d847da540e55 Mon Sep 17 00:00:00 2001
From: yujian <yujian@163.com>
Date: 星期三, 25 十二月 2019 12:21:20 +0800
Subject: [PATCH] 2.0.6

---
 fanli/src/main/java/com/yeshi/fanli/service/impl/user/TokenRecordServiceImpl.java |   19 ++++++++++++++-----
 1 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/fanli/src/main/java/com/yeshi/fanli/service/impl/user/TokenRecordServiceImpl.java b/fanli/src/main/java/com/yeshi/fanli/service/impl/user/TokenRecordServiceImpl.java
index dd59cea..84f4c8b 100644
--- a/fanli/src/main/java/com/yeshi/fanli/service/impl/user/TokenRecordServiceImpl.java
+++ b/fanli/src/main/java/com/yeshi/fanli/service/impl/user/TokenRecordServiceImpl.java
@@ -117,6 +117,11 @@
 	public TokenRecord getNearByTypeAndIdentify(String type, String identify) {
 		return tokenRecordMapper.getNearByTypeAndIdentify(type, identify);
 	}
+	
+	@Override
+	public TokenRecord getByTypeAndIdentify(String type, String identify) {
+		return tokenRecordMapper.getByTypeAndIdentify(type, identify);
+	}
 
 	@Override
 	public TokenVO discernToken(String token, Long uid, AcceptData acceptData) throws TokenRecordException {
@@ -602,10 +607,18 @@
 		try {
 			UserInfo user = userInfoService.selectByPKey(giveUid);
 			UserInfoExtra userInfoExtra = userInfoExtraService.getUserInfoExtra(uid);
+			
+			String inviteCode = null;
+			if (!StringUtil.isNullOrEmpty(userInfoExtra.getInviteCodeVip())) {
+				inviteCode = userInfoExtra.getInviteCodeVip();
+			} else {
+				inviteCode = userInfoExtra.getInviteCode();
+			}
+			
 			MsgInviteContentDTO msgInvite = new MsgInviteContentDTO();
 			msgInvite.setTitle("閭�璇锋秷鎭�");
 			msgInvite.setState("鎴愬姛婵�娲婚個璇�");
-			msgInvite.setCode("浣犵殑閭�璇风爜锛�" + userInfoExtra.getInviteCode());
+			msgInvite.setCode("浣犵殑閭�璇风爜锛�" + inviteCode);
 			msgInvite.setInviter("鏄电О锛�"+user.getNickName());
 			msgInvite.setMode("鎴愬姛棰嗗彇閭�璇蜂汉鐨�" + giftName);
 			userInviteMsgNotificationService.receiveGift(uid, "閭�璇峰叧绯讳竴鏃︾‘绔嬫棤娉曟洿鏀� ", msgInvite);
@@ -626,10 +639,6 @@
 			return;
 		Date date = new Date();
 		for (TokenRecord tokenRecord: list) {
-			Integer state = tokenRecord.getState();
-			if (state != null && state == 0) 
-				continue;
-			
 			tokenRecord.setState(1);
 			tokenRecord.setUpdateTime(date);
 			tokenRecordMapper.updateByPrimaryKeySelective(tokenRecord);

--
Gitblit v1.8.0