当前位置：首页 > news >正文

springboot中测试python脚本：ProcessBuilder

news 2025/7/4 9:14:54

一.添加Jython依赖

二.使用步骤

1. 创建 ProcessBuilder 实例

2. 设置工作目录（可选）

3. 合并错误流（可选）

4. 启动进程

5. 处理输入输出流

6. 等待进程完成

7.完整案例

三.注意事项

ProcessBuilder是jdk提供的脚本执行工具类，无论是python文件还是shell脚本还是其他的指令，都可以通过此类来执行，我们来看看它是如何调用python脚本的。

ProcessBuilder 是 Java 提供的一个用于创建操作系统进程的类，它使得启动和管理外部进程变得更加简单和灵活。以下是一个详细的教程，帮助你理解如何在 Java 中使用 ProcessBuilder。

一.添加Jython依赖

<dependency>
    <groupId>org.python</groupId>
    <artifactId>jython-standalone</artifactId>
    <version>2.7.2</version> <!-- 确认版本号是最新的 -->
</dependency>

二.使用步骤

1. 创建 ProcessBuilder 实例

首先，你需要创建一个 ProcessBuilder 的实例，并指定要执行的命令及其参数。命令可以是一个简单的字符串数组，其中第一个元素是命令名称，后续元素是该命令的参数。

List<String> command = Arrays.asList("python3", "path/to/your_script.py", "arg1", "arg2");
ProcessBuilder processBuilder = new ProcessBuilder(command);

2. 设置工作目录（可选）

你可以为这个进程设置一个工作目录，如果未指定，默认的工作目录将是当前Java进程的工作目录。

processBuilder.directory(new File("/path/to/working/directory"));

3. 合并错误流（可选）

默认情况下，标准输出和标准错误是分开的。如果你希望将它们合并到一起处理，可以调用 redirectErrorStream(true) 方法。

processBuilder.redirectErrorStream(true);

4. 启动进程

通过调用 start() 方法来启动进程，并获得一个 Process 对象，你可以使用这个对象与新进程进行交互。

Process process = processBuilder.start();

5. 处理输入输出流

为了从子进程中获取输出或向子进程发送输入，你需要处理 Process 对象的标准输入、输出和错误流。

// 获取子进程的输出流
BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream(), StandardCharsets.UTF-8));
String line;
while ((line = reader.readLine()) != null) {
    System.out.println(line); // 或者做其他处理
}

// 如果需要发送数据到子进程的标准输入
OutputStream out = process.getOutputStream();
OutputStreamWriter writer = new OutputStreamWriter(out, StandardCharsets.UTF-8);
writer.write("Some input data\n");
writer.flush(); // 别忘了刷新输出流

6. 等待进程完成

最后，你可以等待子进程结束，并检查它的退出状态码。

try {
    int exitCode = process.waitFor();
    System.out.println("Exit Code: " + exitCode);
} catch (InterruptedException e) {
    e.printStackTrace();
}

7.完整案例

	@PostMapping("/process")
    public String process(@RequestBody String param) throws Exception {
        // 获取python文件所在目录地址
        ProcessBuilder pb = new ProcessBuilder("python", "D:\\DevelopWork\\my-space\\java\\pep7chiao-demos\\exe-py-script\\src\\main\\resources\\script\\spider.py", param);
        // 解决中文乱码
        pb.environment().put("PYTHONIOENCODING", "utf-8");
        // 获取进程的输出流和错误流
        pb.redirectErrorStream(true);
        Process process = pb.start();

        // 读取Python脚本的输出
        BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
        String result = "";
        String line;
        while ((line = reader.readLine()) != null) {
            System.out.println(line);
            result += line;
        }

        BufferedReader errorReader = new BufferedReader(new InputStreamReader(process.getErrorStream()));
        String errorResult;
        while ((errorResult = errorReader.readLine()) != null) {
            System.out.println(errorResult);
        }

        // 等待进程结束并检查退出状态
        int exitCode = process.waitFor();
        if (exitCode != 0) {
            throw new RuntimeException("python脚本执行失败：" + exitCode);
        }

        return result;
    }