From 7183e066def4e42afaa70288963ea19d6e9a0899 Mon Sep 17 00:00:00 2001
From: shiziyuan9527 <yuhao.li@fit2cloud.com>
Date: Fri, 21 Oct 2022 11:50:51 +0800
Subject: [PATCH] =?UTF-8?q?fix(=E9=A1=B9=E7=9B=AE=E8=AE=BE=E7=BD=AE):=20?=
 =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BB=A3=E7=A0=81=E7=89=87=E6=AE=B5=E6=89=A7?=
 =?UTF-8?q?=E8=A1=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 project-management/backend/pom.xml            | 47 +++++++++++++++++++
 .../listener/ProjectAppStartListener.java     | 25 ++++++++++
 2 files changed, 72 insertions(+)

diff --git a/project-management/backend/pom.xml b/project-management/backend/pom.xml
index ecf239a272..aa20ba6e02 100644
--- a/project-management/backend/pom.xml
+++ b/project-management/backend/pom.xml
@@ -117,6 +117,53 @@
                     <release>${java.version}</release>
                 </configuration>
             </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-dependency-plugin</artifactId>
+                <executions>
+                    <execution>
+                        <id>copy</id>
+                        <phase>generate-resources</phase>
+                        <goals>
+                            <goal>copy</goal>
+                        </goals>
+                    </execution>
+                </executions>
+                <configuration>
+                    <artifactItems>
+                        <artifactItem>
+                            <groupId>org.apache.jmeter</groupId>
+                            <artifactId>ApacheJMeter_functions</artifactId>
+                            <version>${jmeter.version}</version>
+                            <type>jar</type>
+                            <overWrite>true</overWrite>
+                            <outputDirectory>src/main/resources/jmeter/lib/ext</outputDirectory>
+                            <destFileName>ApacheJMeter_functions.jar</destFileName>
+                        </artifactItem>
+                        <artifactItem>
+                            <groupId>io.metersphere</groupId>
+                            <artifactId>metersphere-jmeter-functions</artifactId>
+                            <version>${metersphere-jmeter-functions.version}</version>
+                            <type>jar</type>
+                            <overWrite>true</overWrite>
+                            <outputDirectory>src/main/resources/jmeter/lib/ext</outputDirectory>
+                            <destFileName>metersphere-jmeter-functions.jar</destFileName>
+                        </artifactItem>
+                        <artifactItem>
+                            <groupId>org.python</groupId>
+                            <artifactId>jython-standalone</artifactId>
+                            <version>${jython.version}</version>
+                            <type>jar</type>
+                            <overWrite>true</overWrite>
+                            <outputDirectory>src/main/resources/jmeter/lib/ext</outputDirectory>
+                            <destFileName>jython-standalone.jar</destFileName>
+                        </artifactItem>
+                    </artifactItems>
+                    <outputDirectory>${project.build.directory}/wars</outputDirectory>
+                    <overWriteReleases>false</overWriteReleases>
+                    <overWriteSnapshots>true</overWriteSnapshots>
+                </configuration>
+            </plugin>
         </plugins>
     </build>
 
diff --git a/project-management/backend/src/main/java/io/metersphere/listener/ProjectAppStartListener.java b/project-management/backend/src/main/java/io/metersphere/listener/ProjectAppStartListener.java
index db8a2baafe..ce5b335750 100644
--- a/project-management/backend/src/main/java/io/metersphere/listener/ProjectAppStartListener.java
+++ b/project-management/backend/src/main/java/io/metersphere/listener/ProjectAppStartListener.java
@@ -2,6 +2,8 @@ package io.metersphere.listener;
 
 import io.metersphere.commons.utils.LogUtil;
 import org.apache.jmeter.util.JMeterUtils;
+import org.python.core.Options;
+import org.python.util.PythonInterpreter;
 import org.springframework.boot.context.event.ApplicationReadyEvent;
 import org.springframework.context.ApplicationListener;
 import org.springframework.context.i18n.LocaleContextHolder;
@@ -14,8 +16,13 @@ public class ProjectAppStartListener implements ApplicationListener<ApplicationR
     @Override
     public void onApplicationEvent(ApplicationReadyEvent event) {
         LogUtil.info("================= PROJECT 应用启动 =================");
+
         this.initJmeterHome();
+
+        LogUtil.info("导入内置python包处理");
+        this.initPythonEnv();
     }
+
     private void initJmeterHome() {
         String JMETER_HOME = Objects.requireNonNull(getClass().getResource("/")).getPath() + "jmeter";
         String JMETER_PROPERTIES = JMETER_HOME + "/bin/jmeter.properties";
@@ -23,4 +30,22 @@ public class ProjectAppStartListener implements ApplicationListener<ApplicationR
         JMeterUtils.setJMeterHome(JMETER_HOME);
         JMeterUtils.setLocale(LocaleContextHolder.getLocale());
     }
+
+    /**
+     * 解决接口测试-无法导入内置python包
+     */
+    private void initPythonEnv() {
+        //解决无法加载 PyScriptEngineFactory
+        try {
+            Options.importSite = false;
+            PythonInterpreter interpreter = new PythonInterpreter();
+            StringBuffer pathBuffer = new StringBuffer(getClass().getResource("/").getPath() + "jmeter");
+            pathBuffer.append("/lib/ext/jython-standalone.jar/Lib");
+            interpreter.exec("import sys");
+            interpreter.exec("sys.path.append(\"" + pathBuffer.toString() + "\")");
+            LogUtil.info("sys.path: ", pathBuffer.toString());
+        } catch (Exception e) {
+            LogUtil.error("导入内置python包处理异常 ", e);
+        }
+    }
 }