From 4c96637a005891c709662ae84edd072ad9a4a57d Mon Sep 17 00:00:00 2001
From: luodangjia <luodangjia>
Date: 星期二, 08 十月 2024 15:13:07 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 ruoyi-auth/src/main/java/com/ruoyi/auth/controller/TokenController.java |   12 ++++++++++++
 1 files changed, 12 insertions(+), 0 deletions(-)

diff --git a/ruoyi-auth/src/main/java/com/ruoyi/auth/controller/TokenController.java b/ruoyi-auth/src/main/java/com/ruoyi/auth/controller/TokenController.java
index 1cc1d79..da2ac38 100644
--- a/ruoyi-auth/src/main/java/com/ruoyi/auth/controller/TokenController.java
+++ b/ruoyi-auth/src/main/java/com/ruoyi/auth/controller/TokenController.java
@@ -3,6 +3,8 @@
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 
+import com.ruoyi.chargingPile.api.feignClient.PartnerClient;
+import com.ruoyi.chargingPile.api.model.Partner;
 import com.ruoyi.system.api.domain.SysRole;
 import com.ruoyi.system.api.domain.SysUser;
 import com.ruoyi.system.api.feignClient.SysUserClient;
@@ -40,6 +42,12 @@
 	@Resource
 	private SysUserClient userClient;
 	
+	@Resource
+	private PartnerClient partnerClient;
+	
+	
+	
+	
 	
 	@PostMapping("login")
 	@GlobalTransactional(rollbackFor = Exception.class)//分布式事务
@@ -55,6 +63,10 @@
 
         map.put("roleName",roles.get(0).getRoleName());
 		map.put("info", userInfo);
+		if(userInfo.getSysUser().getRoleType() == 2){
+			Partner partner = partnerClient.getPartnerById(userInfo.getSysUser().getObjectId()).getData();
+			map.put("partnerName", partner.getName());
+		}
 		// 修改用户最后登录时间
 		SysUser sysUser = new SysUser();
 		sysUser.setUserId(userInfo.getSysUser().getUserId());

--
Gitblit v1.7.1