fix: 增强PyContextPool Engine创建的兼容性并添加GraalPy测试

This commit is contained in:
q
2026-01-11 07:25:32 +08:00
parent 88d675fe95
commit 2e5e679cea
2 changed files with 166 additions and 25 deletions

View File

@@ -134,10 +134,30 @@ public class PyContextPool {
private PyContextPool() { private PyContextPool() {
log.info("初始化GraalPy Context池..."); log.info("初始化GraalPy Context池...");
// 创建共享Engine // 创建共享Engine - 优先使用 GraalPyResources
this.sharedEngine = Engine.newBuilder() Engine engine = null;
try {
// 使用 GraalPyResources 创建,这是推荐的方式
var tempContext = org.graalvm.python.embedding.GraalPyResources.contextBuilder()
.option("engine.WarnInterpreterOnly", "false") .option("engine.WarnInterpreterOnly", "false")
.build(); .build();
engine = tempContext.getEngine();
tempContext.close();
log.info("Engine创建成功GraalPyResources模式");
} catch (Exception e) {
log.warn("GraalPyResources模式创建Engine失败尝试标准方式: {}", e.getMessage());
try {
engine = Engine.newBuilder()
.option("engine.WarnInterpreterOnly", "false")
.build();
log.info("Engine创建成功标准模式");
} catch (Exception e2) {
log.error("标准模式创建Engine也失败: {}", e2.getMessage());
checkGraalPyAvailability();
throw new RuntimeException("无法初始化GraalPy Engine请确保GraalPy依赖正确配置", e2);
}
}
this.sharedEngine = engine;
// 创建Context池 // 创建Context池
this.contextPool = new LinkedBlockingQueue<>(MAX_POOL_SIZE); this.contextPool = new LinkedBlockingQueue<>(MAX_POOL_SIZE);
@@ -234,7 +254,10 @@ public class PyContextPool {
throw new IllegalStateException("Context池已关闭"); throw new IllegalStateException("Context池已关闭");
} }
Context context = Context.newBuilder("python") Context context;
try {
// 首先尝试使用共享Engine创建
context = Context.newBuilder("python")
.engine(sharedEngine) .engine(sharedEngine)
.allowHostAccess(HostAccess.newBuilder(HostAccess.EXPLICIT) .allowHostAccess(HostAccess.newBuilder(HostAccess.EXPLICIT)
.allowArrayAccess(true) .allowArrayAccess(true)
@@ -255,6 +278,28 @@ public class PyContextPool {
.option("python.PythonHome", "") .option("python.PythonHome", "")
.option("python.ForceImportSite", "false") .option("python.ForceImportSite", "false")
.build(); .build();
} catch (Exception e) {
log.warn("使用共享Engine创建Context失败尝试使用GraalPyResources: {}", e.getMessage());
// 使用 GraalPyResources 作为备选
context = org.graalvm.python.embedding.GraalPyResources.contextBuilder()
.allowHostAccess(HostAccess.newBuilder(HostAccess.EXPLICIT)
.allowArrayAccess(true)
.allowListAccess(true)
.allowMapAccess(true)
.allowIterableAccess(true)
.allowIteratorAccess(true)
.build())
.allowHostClassLookup(className -> false)
.allowExperimentalOptions(true)
.allowCreateThread(true)
.allowNativeAccess(false)
.allowCreateProcess(false)
.allowIO(IOAccess.newBuilder()
.allowHostFileAccess(false)
.allowHostSocketAccess(false)
.build())
.build();
}
createdCount.incrementAndGet(); createdCount.incrementAndGet();
log.debug("创建新的GraalPy Context当前总数: {}", createdCount.get()); log.debug("创建新的GraalPy Context当前总数: {}", createdCount.get());
@@ -421,6 +466,45 @@ public class PyContextPool {
return createdCount.get(); return createdCount.get();
} }
/**
* 检查GraalPy是否可用
*/
private void checkGraalPyAvailability() {
log.error("===== GraalPy 可用性检查 =====");
// 检查类路径
try {
Class.forName("org.graalvm.polyglot.Engine");
log.info("✓ org.graalvm.polyglot.Engine 类存在");
} catch (ClassNotFoundException e) {
log.error("✗ org.graalvm.polyglot.Engine 类不存在");
}
try {
Class.forName("org.graalvm.python.embedding.GraalPyResources");
log.info("✓ org.graalvm.python.embedding.GraalPyResources 类存在");
} catch (ClassNotFoundException e) {
log.error("✗ org.graalvm.python.embedding.GraalPyResources 类不存在 - 缺少 python-embedding 依赖");
}
// 尝试列出可用语言
try {
log.info("尝试使用 GraalPyResources 创建 Context...");
var context = org.graalvm.python.embedding.GraalPyResources.createContext();
log.info("✓ GraalPyResources 创建 Context 成功");
context.close();
} catch (Exception e) {
log.error("✗ GraalPyResources 创建 Context 失败: {}", e.getMessage());
}
log.error("================================");
log.error("请检查以下依赖是否正确配置:");
log.error(" 1. org.graalvm.polyglot:polyglot");
log.error(" 2. org.graalvm.polyglot:python (type=pom)");
log.error(" 3. org.graalvm.python:python-embedding");
log.error("================================");
}
/** /**
* 关闭Context池 * 关闭Context池
*/ */

View File

@@ -0,0 +1,57 @@
package cn.qaiu.parser;
import org.graalvm.polyglot.Context;
import org.graalvm.polyglot.Engine;
import org.graalvm.polyglot.Value;
/**
* GraalPy 简单测试
*/
public class GraalPyTest {
public static void main(String[] args) {
System.out.println("===== GraalPy 测试开始 =====");
try {
System.out.println("1. 检查可用语言...");
try (Engine engine = Engine.create()) {
System.out.println(" 可用语言: " + engine.getLanguages().keySet());
if (!engine.getLanguages().containsKey("python")) {
System.err.println(" ✗ Python 语言不可用!");
System.exit(1);
}
System.out.println(" ✓ Python 语言可用");
}
System.out.println("2. 尝试创建 Python Context...");
try (Context context = Context.newBuilder("python")
.option("engine.WarnInterpreterOnly", "false")
.build()) {
System.out.println(" ✓ Context 创建成功");
System.out.println("3. 执行简单 Python 代码...");
Value result = context.eval("python", "1 + 2");
System.out.println(" ✓ 计算结果: 1 + 2 = " + result.asInt());
System.out.println("4. 执行字符串操作...");
Value strResult = context.eval("python", "'Hello' + ' ' + 'GraalPy'");
System.out.println(" ✓ 字符串结果: " + strResult.asString());
System.out.println("5. 执行多行代码...");
String code = """
def greet(name):
return f"Hello, {name}!"
greet("World")
""";
Value funcResult = context.eval("python", code);
System.out.println(" ✓ 函数结果: " + funcResult.asString());
}
System.out.println("===== GraalPy 测试通过 =====");
} catch (Exception e) {
System.err.println("✗ GraalPy 测试失败: " + e.getMessage());
e.printStackTrace();
System.exit(1);
}
}
}