Java通过命令行切换conda环境并调用Python

背景

项目需要使用Java调用Python实现实体抽取和关系抽取的功能

实现

类

使用ProcessBuilder类调用命令行

代码

public R<String> test(HttpServletRequest request){
		String output = "";
		try {
			ProcessBuilder builder = new ProcessBuilder("cmd", "/c", "conda activate deepke && " +
					"cd C:\\path && " +
					"python C:\\path\\test.py");
			builder.redirectErrorStream(true);
			Process process = builder.start();
			BufferedReader br = new BufferedReader(new InputStreamReader(process.getInputStream(), "gbk"));
			String readLine = "";
			while (null != (readLine = br.readLine()))
			{
				output += readLine;
			}
			process.waitFor();
		} catch (IOException e) {
            throw new RuntimeException(e);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }

        return R.success(output);
	}

坑

1. Python脚本中有用到关于路径的yaml配置文件，需要注意切换路径
2. getInputeStream()方法由于阻塞问题无法读取命令行内容，需要先调用ProcessBuilder.redirectErrorStream(true)。参考¹
3. 控制台返回的中文乱码问题，new InputStreamReader初始化时将编码设置为gbk

待解决的问题

• 调用Python脚本时的路径（暂时在命令行中用cd解决）
• 其他

参考

---

1. 解决Process.getInputStream()阻塞的问题 ↩︎