From 88b54772dbcf5ecab1e2316e4e4626ac901b8908 Mon Sep 17 00:00:00 2001
From: yujian <yujian>
Date: 星期二, 22 一月 2019 15:58:24 +0800
Subject: [PATCH] 邀请码添加返回状态

---
 fanli/src/main/java/com/yeshi/fanli/service/impl/msg/UserSystemMsgServiceImpl.java |   14 ++++++++++----
 1 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/fanli/src/main/java/com/yeshi/fanli/service/impl/msg/UserSystemMsgServiceImpl.java b/fanli/src/main/java/com/yeshi/fanli/service/impl/msg/UserSystemMsgServiceImpl.java
index 5b10c5e..7330a67 100644
--- a/fanli/src/main/java/com/yeshi/fanli/service/impl/msg/UserSystemMsgServiceImpl.java
+++ b/fanli/src/main/java/com/yeshi/fanli/service/impl/msg/UserSystemMsgServiceImpl.java
@@ -13,6 +13,7 @@
 import com.yeshi.fanli.entity.bus.user.UserInfo;
 import com.yeshi.fanli.entity.system.SystemZnx;
 import com.yeshi.fanli.exception.msg.UserSystemMsgException;
+import com.yeshi.fanli.service.inter.msg.UserMsgReadStateService;
 import com.yeshi.fanli.service.inter.msg.UserSystemMsgService;
 import com.yeshi.fanli.util.StringUtil;
 
@@ -21,6 +22,9 @@
 
 	@Resource
 	private UserSystemMsgMapper userSystemMsgMapper;
+
+	@Resource
+	private UserMsgReadStateService userMsgReadStateService;
 
 	@Override
 	public void addUserSystemMsg(Long uid, UserSystemMsgTypeEnum type, String title, String content, int timeTag,
@@ -50,15 +54,17 @@
 			msg.setTimeTag(timeTag);
 		}
 		userSystemMsgMapper.insertSelective(msg);
+		userMsgReadStateService.addSystemMsgUnReadCount(uid, 1);
 	}
 
 	@Override
 	public UserSystemMsg getLatestUserSystemMsg(Long uid) {
 		UserSystemMsg msg = userSystemMsgMapper.selectLatestUserSystemMsg(uid);
-		if (msg.getSystemZNX() != null) {
-			msg.setTitle(msg.getSystemZNX().getTitle());
-			msg.setContent(msg.getSystemZNX().getContent());
-		}
+		if (msg != null)
+			if (msg.getSystemZNX() != null) {
+				msg.setTitle(msg.getSystemZNX().getTitle());
+				msg.setContent(msg.getSystemZNX().getContent());
+			}
 		return msg;
 	}
 

--
Gitblit v1.8.0