57 lines
1.9 KiB
Rust
57 lines
1.9 KiB
Rust
use crate::DHTServer;
|
|
use crate::jni_bindings::env::JavaCallback;
|
|
use crate::jni_bindings::types::torrent_info_to_java;
|
|
use jni::objects::JValue;
|
|
use std::sync::Arc;
|
|
|
|
/// 向 `DHTServer` 注册所有 Java 回调（on_torrent、on_error）。
|
|
///
|
|
/// `callback` 持有 Java listener 的全局引用，可跨线程安全使用。
|
|
pub fn register_callbacks(server: &Arc<DHTServer>, callback: JavaCallback) {
|
|
register_on_torrent(server, callback.clone());
|
|
register_on_error(server, callback);
|
|
}
|
|
|
|
/// 注册 on_torrent 回调：Rust TorrentInfo → Java listener.onTorrent(TorrentInfo)
|
|
fn register_on_torrent(server: &Arc<DHTServer>, callback: JavaCallback) {
|
|
server.on_torrent(move |info| {
|
|
let result = callback.with_env(|env, listener| {
|
|
// 映射为 Java TorrentInfo 对象
|
|
let j_info = torrent_info_to_java(env, &info).map_err(|e| {
|
|
log::error!("TorrentInfo 转换 Java 对象失败: {e}");
|
|
e
|
|
})?;
|
|
env.call_method(
|
|
listener,
|
|
"onTorrent",
|
|
"(Lcn/lmcw/dht/model/TorrentInfo;)V",
|
|
&[JValue::Object(&j_info)],
|
|
)?;
|
|
Ok(())
|
|
});
|
|
if let Err(e) = result {
|
|
log::error!("回调 onTorrent 失败: {e}");
|
|
}
|
|
});
|
|
}
|
|
|
|
/// 注册 on_error 回调：Rust DHTError → Java listener.onError(String)
|
|
fn register_on_error(server: &Arc<DHTServer>, callback: JavaCallback) {
|
|
server.on_error(move |err| {
|
|
let msg = err.to_string();
|
|
let result = callback.with_env(|env, listener| {
|
|
let j_msg = env.new_string(&msg)?;
|
|
env.call_method(
|
|
listener,
|
|
"onError",
|
|
"(Ljava/lang/String;)V",
|
|
&[JValue::Object(&j_msg.into())],
|
|
)?;
|
|
Ok(())
|
|
});
|
|
if let Err(e) = result {
|
|
log::error!("回调 onError 失败: {e}");
|
|
}
|
|
});
|
|
}
|