From e623e633f1a28b66632d9c17dcfe2f6e2a78073b Mon Sep 17 00:00:00 2001
From: liusheng <337615773@qq.com>
Date: 星期四, 10 八月 2023 16:24:21 +0800
Subject: [PATCH] 代码修改

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java |   54 +++++++++++++++++++++++++++++++++---------------------
 1 files changed, 33 insertions(+), 21 deletions(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java
index 5894b48..b2ac01f 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java
@@ -1,6 +1,5 @@
 package com.ruoyi.web.controller.common;
 
-import org.apache.coyote.AbstractProtocol;
 import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.common.constant.Constants;
 import com.ruoyi.common.core.domain.AjaxResult;
@@ -10,6 +9,7 @@
 import com.ruoyi.framework.config.ServerConfig;
 import org.apache.catalina.connector.Connector;
 import org.apache.catalina.startup.Tomcat;
+import org.apache.coyote.AbstractProtocol;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -20,16 +20,10 @@
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.multipart.MultipartFile;
 
-import javax.management.MBeanServer;
-import javax.management.MalformedObjectNameException;
-import javax.management.ObjectName;
-import javax.management.Query;
+import javax.management.*;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
-import java.io.IOException;
 import java.lang.management.ManagementFactory;
-import java.net.InetAddress;
-import java.net.ServerSocket;
 import java.util.Set;
 
 /**
@@ -77,13 +71,22 @@
      */
     @PostMapping("/common/upload")
     public AjaxResult uploadFile(MultipartFile file) throws Exception {
-        int tomcatPort = getTomcatPort();
+
+        int tomcatPort1 = getTomcatPort();
+        log.info("tomcatPort1鐨勫�间负锛� {}", tomcatPort1);
+        //杩欎釜鐪佷汉姘戞槸8099锛屾墍浠ョ洿鎺ュ啓姝�
+        int tomcatPort = 8099;
         try {
             // 涓婁紶鏂囦欢璺緞
             String filePath = RuoYiConfig.getUploadPath();
             // 涓婁紶骞惰繑鍥炴柊鏂囦欢鍚嶇О
             String fileName = FileUploadUtils.upload(filePath, file);
-            String url = serverConfig.getUrl() + fileName;
+            String url1 = serverConfig.getUrl();
+            log.info("url1鐨勫�紀ld :{}", url1);
+            //杩欓噷鐨�8032涓嶆槸tomcat鐨勶紝鎵�浠ヨ杞垚tomcat鐨�
+            url1 = url1.replace("8032", "8099");
+            log.info("url1鐨勫�糿ew :{}", url1);
+            String url = url1 + fileName;
             AjaxResult ajax = AjaxResult.success();
             ajax.put("fileName", fileName);
             ajax.put("url", url);
@@ -93,16 +96,6 @@
         }
     }
 
-    private static int getTomcatPort() {
-        int port = 0;
-        Tomcat tomcat = new Tomcat();
-        Connector connector = tomcat.getConnector();
-        if (connector != null) {
-            AbstractProtocol protocol = (AbstractProtocol) connector.getProtocolHandler();
-            port = protocol.getPort();
-        }
-        return port;
-    }
 
     /**
      * 鏈湴璧勬簮閫氱敤涓嬭浇
@@ -126,4 +119,23 @@
             log.error("涓嬭浇鏂囦欢澶辫触", e);
         }
     }
-}
+
+
+    public int getTomcatPort() {
+        MBeanServer beanServer = ManagementFactory.getPlatformMBeanServer();
+        try {
+            QueryExp protocol = Query.match(Query.attr("protocol"), Query.value("HTTP/1.1"));
+            ObjectName name = new ObjectName("*:type=Connector,*");
+            Set<ObjectName> objectNames = beanServer.queryNames(name, protocol);
+            for (ObjectName objectName : objectNames) {
+                String catalina = objectName.getDomain();
+                if ("Catalina".equals(catalina)) {
+                    return Integer.parseInt(objectName.getKeyProperty("port"));
+                }
+            }
+        } catch (MalformedObjectNameException e) {
+            e.printStackTrace();
+        }
+        return 0;
+    }
+}
\ No newline at end of file

--
Gitblit v1.9.3