Chicoryを使ってRustをコンパイルしたwasmをJavaから呼び出してみました。
JVMでWebAssemblyにコンパイルしたRustのコードを動かす - きしだのHatena
ただ、結構呼び出しがめんどいので、Javaインタフェースを定義したらなんかメソッド呼び出しで使える、というよく見かけるやつを作ってみます。
Rustのコードはこう。
#[no_mangle] pub fn add(left: i32, right: i32) -> i32 { left + right } #[no_mangle] pub fn sub(left: i32, right: i32) -> i32 { left - right } #[no_mangle] pub fn mul(left: i32, right: i32) -> i64 { (left as i64) * (right as i64) }
こんなJavaインタフェースで呼び出せるようにしたい。
interface RustFuncs { int add(int left, int right); int sub(int left, int right); long mul(int left, int right); }
ということで、こんな感じでProxyを作る。
static <T> T bind(String name, Class<T> type) { var wasm = CallWasm.class.getClassLoader().getResourceAsStream(name); var module = Module.builder(wasm).build(); var instance = module.instantiate(); T obj = (T) Proxy.newProxyInstance(type.getClassLoader(), new Class<?>[]{type}, (p, m, a) -> methodHandler(instance, type, p, m, a)); return obj; } static Object methodHandler(Instance ins, Class type, Object proxy, Method method, Object[] args) throws Throwable { var m = ins.export(method.getName()); // todo:use annotation name List<Value> values = new ArrayList<>(); for (int i = 0; i < args.length; ++i) { values.add(switch (args[i]) { case Integer n -> Value.i32(n); case Long l -> Value.i64(l); default ->throw new RuntimeException("unknown type " + method.getParameterTypes()[i]); }); } var result = m.apply(values.toArray(Value[]::new))[0]; var rt = method.getReturnType(); if(rt == long.class || rt == Long.class) { return result.asLong(); } else if (rt == int.class || rt == int.class) { return result.asInt(); } else { throw new RuntimeException("unknown ret type " + rt); } }
こうやって呼び出せるようになりました！やったね！
public static void main(String[] args) throws IOException { var rf = bind("hello_wasm.wasm", RustFuncs.class); System.out.println(rf.add(12, 34)); System.out.println(rf.sub(34, 12)); System.out.println(rf.mul(100_000_000, 100_000_000)); }
文字列とかはwasmでは定義されてなくて結構めんどいので、そのあたりはもう少し考えてみます。
使い物になりそうだったらasmかJavassistでバイトコード生成を。
