From 0b977fb16e8272739bbe25973947f45a55ebe842 Mon Sep 17 00:00:00 2001
From: 唐耀东 <18861537@qq.com>
Date: 星期五, 31 十二月 2021 11:30:49 +0800
Subject: [PATCH] 提交代码

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TokenServiceImpl.java |   22 ++++++++++++++++------
 1 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TokenServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TokenServiceImpl.java
index 15b9759..40a1fe3 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TokenServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TokenServiceImpl.java
@@ -138,12 +138,22 @@
      */
     @Override
     public void setUserAgent(LoginUser loginUser) {
-        UserAgent userAgent = UserAgentUtil.parse(ServletUtils.getRequest().getHeader("User-Agent"));
-        String ip = ServletUtils.getClientIP();
-        loginUser.setIpaddr(ip);
-        loginUser.setLoginLocation(AddressUtils.getRealAddressByIP(ip));
-        loginUser.setBrowser(userAgent.getBrowser().getName());
-        loginUser.setOs(userAgent.getOs().getName());
+        HttpServletRequest request = ServletUtils.getRequest();
+        // 鐢ㄦ埛绫诲瀷锛圖EVICE锛氳澶囩锛岀┖锛歱c绯荤粺锛�
+        String userType = request.getHeader("User-Type");
+        if (StringUtils.isEmpty(userType)) {
+            UserAgent userAgent = UserAgentUtil.parse(ServletUtils.getRequest().getHeader("User-Agent"));
+            String ip = ServletUtils.getClientIP();
+            loginUser.setIpaddr(ip);
+            loginUser.setLoginLocation(AddressUtils.getRealAddressByIP(ip));
+            loginUser.setBrowser(userAgent.getBrowser().getName());
+            loginUser.setOs(userAgent.getOs().getName());
+        }
+        if ("DEVICE".equals(userType)) {
+            String ip = ServletUtils.getClientIP();
+            loginUser.setIpaddr(ip);
+            loginUser.setLoginLocation(AddressUtils.getRealAddressByIP(ip));
+        }
     }
 
     /**

--
Gitblit v1.9.1