From 6651f6cf39bb3aadbcbbcc52db6f06fcf3c6f7b7 Mon Sep 17 00:00:00 2001
From: wuchunlei <mailcauchy@gmail.com>
Date: Wed, 29 May 2024 16:04:00 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E5=A4=9A=E5=B9=B3=E5=8F=B0to?=
 =?UTF-8?q?ken?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../peanut/common/utils/HttpContextUtil.java  | 15 +++++++
 .../sys/entity/SysUserTokenEntity.java        |  4 ++
 .../service/impl/SysUserTokenServiceImpl.java | 40 +++++++++----------
 .../resources/mapper/sys/SysUserTokenDao.xml  |  2 +-
 4 files changed, 38 insertions(+), 23 deletions(-)
 create mode 100644 src/main/java/com/peanut/common/utils/HttpContextUtil.java

diff --git a/src/main/java/com/peanut/common/utils/HttpContextUtil.java b/src/main/java/com/peanut/common/utils/HttpContextUtil.java
new file mode 100644
index 00000000..7a73ac84
--- /dev/null
+++ b/src/main/java/com/peanut/common/utils/HttpContextUtil.java
@@ -0,0 +1,15 @@
+package com.peanut.common.utils;
+
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import javax.servlet.http.HttpServletRequest;
+import java.util.Objects;
+
+public class HttpContextUtil {
+
+    private HttpContextUtil(){}
+    public static HttpServletRequest getHttpServletRequest() {
+        return ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
+    }
+}
diff --git a/src/main/java/com/peanut/modules/sys/entity/SysUserTokenEntity.java b/src/main/java/com/peanut/modules/sys/entity/SysUserTokenEntity.java
index a1fef138..06424086 100644
--- a/src/main/java/com/peanut/modules/sys/entity/SysUserTokenEntity.java
+++ b/src/main/java/com/peanut/modules/sys/entity/SysUserTokenEntity.java
@@ -32,6 +32,10 @@ public class SysUserTokenEntity implements Serializable {
 	private Long userId;
 	//token
 	private String token;
+	//token
+	private String tokenMedical;
+	//token
+	private String tokenSociology;
 	//过期时间
 	private Date expireTime;
 	//更新时间
diff --git a/src/main/java/com/peanut/modules/sys/service/impl/SysUserTokenServiceImpl.java b/src/main/java/com/peanut/modules/sys/service/impl/SysUserTokenServiceImpl.java
index b550cf6b..b1f8cff7 100644
--- a/src/main/java/com/peanut/modules/sys/service/impl/SysUserTokenServiceImpl.java
+++ b/src/main/java/com/peanut/modules/sys/service/impl/SysUserTokenServiceImpl.java
@@ -9,6 +9,7 @@
 package com.peanut.modules.sys.service.impl;
 
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.peanut.common.utils.HttpContextUtil;
 import com.peanut.common.utils.R;
 import com.peanut.modules.sys.dao.SysUserTokenDao;
 import com.peanut.modules.sys.entity.SysUserTokenEntity;
@@ -17,6 +18,7 @@ import com.peanut.modules.sys.service.SysUserTokenService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 
+import javax.servlet.http.HttpServletRequest;
 import java.util.Date;
 
 @Slf4j
@@ -30,36 +32,30 @@ public class SysUserTokenServiceImpl extends ServiceImpl<SysUserTokenDao, SysUse
 	public R createToken(long userId) {
 		//生成一个token
 		String token = TokenGenerator.generateValue();
-
 		//当前时间
 		Date now = new Date();
 		//过期时间
 		Date expireTime = new Date(now.getTime() + EXPIRE * 1000);
-
 		//判断是否生成过token
 		SysUserTokenEntity tokenEntity = this.getById(userId);
-		if(tokenEntity == null){
-			tokenEntity = new SysUserTokenEntity();
-			tokenEntity.setUserId(userId);
-			tokenEntity.setToken(token);
-			tokenEntity.setUpdateTime(now);
-			tokenEntity.setExpireTime(expireTime);
-
-			//保存token
-			this.save(tokenEntity);
-		}else{
-			if (tokenEntity.getExpireTime().getTime() > now.getTime()){
-                token = tokenEntity.getToken();
-			}
-            tokenEntity.setToken(token);
-			tokenEntity.setUpdateTime(now);
-			tokenEntity.setExpireTime(expireTime);
-			//更新token
-			this.updateById(tokenEntity);
+		if (tokenEntity == null) {
+		    tokenEntity = new SysUserTokenEntity();
 		}
-
+		//根部不同登录平台生成不同token
+		HttpServletRequest request = HttpContextUtil.getHttpServletRequest();
+		String appType = request.getHeader("appType")==null?"":request.getHeader("appType");
+		if ("".equals(appType)){
+			tokenEntity.setToken(token);
+		}else if ("medical".equals(appType)){
+			tokenEntity.setTokenMedical(token);
+		}else if ("sociology".equals(appType)){
+			tokenEntity.setTokenSociology(token);
+		}
+		tokenEntity.setUserId(userId);
+		tokenEntity.setUpdateTime(now);
+		tokenEntity.setExpireTime(expireTime);
+		this.saveOrUpdate(tokenEntity);
 		R r = R.ok().put("token", token).put("expire", EXPIRE);
-
 		return r;
 	}
 
diff --git a/src/main/resources/mapper/sys/SysUserTokenDao.xml b/src/main/resources/mapper/sys/SysUserTokenDao.xml
index f1ec85ee..7f176619 100644
--- a/src/main/resources/mapper/sys/SysUserTokenDao.xml
+++ b/src/main/resources/mapper/sys/SysUserTokenDao.xml
@@ -4,6 +4,6 @@
 <mapper namespace="com.peanut.modules.sys.dao.SysUserTokenDao">
 
 	<select id="queryByToken" resultType="com.peanut.modules.sys.entity.SysUserTokenEntity">
-		select * from sys_user_token where token = #{value}
+		select * from sys_user_token where (token = #{value} or token_medical = #{value} or token_sociology = #{value})
 	</select>
 </mapper>
\ No newline at end of file