如何在Rust中获取MongoDB集群中响应查询的节点主机地址？

我完全理解你的需求——之前通过tracing日志看到了serverHost的信息，但不知道怎么在代码里直接拿到实际响应查询的MongoDB集群节点地址对吧？下面给你一个可行的解决方案，利用MongoDB Rust驱动的事件监听机制来实现：

核心思路是注册一个命令事件处理器，驱动会在命令执行的各个阶段触发事件，其中CommandSucceededEvent就包含了执行当前命令的服务器地址。我们可以通过这个事件来捕获并记录响应节点的地址。

下面是修改后的完整代码，适配你提供的ping示例：

use mongodb::{
    bson::doc,
    event::{command::CommandSucceededEvent, CommandEventHandler},
    options::{ClientOptions, ServerApi, ServerApiVersion},
    Client,
};
use std::sync::Arc;
use tokio::sync::Mutex;

// 自定义事件处理器，用来存储响应的服务器地址
#[derive(Clone)]
struct ServerAddressRecorder {
    address: Arc<Mutex<Option<String>>>,
}

// 实现CommandEventHandler trait，处理命令成功事件
impl CommandEventHandler for ServerAddressRecorder {
    fn handle_command_succeeded(&self, event: CommandSucceededEvent) {
        // 我们只关心ping命令的响应，你也可以根据需要修改为其他命令
        if event.command_name == "ping" {
            // 将服务器地址转为字符串存储
            let addr = event.server_address.to_string();
            *self.address.lock().unwrap() = Some(addr);
        }
    }
}

#[tokio::main]
async fn main() -> mongodb::error::Result<()> {
    // 替换为你的MongoDB连接字符串
    let uri = "<connection string>";

    let mut client_options = ClientOptions::parse(uri).await?;

    // 创建地址记录器，用Arc+Mutex保证线程安全（事件处理可能在后台线程执行）
    let address_recorder = ServerAddressRecorder {
        address: Arc::new(Mutex::new(None)),
    };
    // 将自定义事件处理器添加到客户端选项中
    client_options.command_event_handler = Some(Box::new(address_recorder.clone()));

    let client = Client::with_options(client_options)?;

    // 执行ping命令
    client
        .database("admin")
        .run_command(doc! {"ping": 1}, None)
        .await?;

    // 从记录器中取出响应节点的地址
    let address = address_recorder.address.lock().await
        .take()
        .unwrap_or_else(|| "unknown server".to_string());

    println!("Got ping from {}", address);

    Ok(())
}

关键说明： #

• 我们自定义了ServerAddressRecorder结构体，实现了CommandEventHandler trait，专门处理命令成功的事件。
• CommandSucceededEvent中的server_address字段就是实际响应当前命令的集群节点地址，直接转成字符串就能使用。
• 用Arc<Mutex<Option<String>>>来存储地址是因为事件处理逻辑可能在后台线程执行，需要保证线程安全的访问。
• 如果你需要监听其他命令，只需要修改handle_command_succeeded里的命令名称判断即可。

另外你提到的ServerInfo.address，那是客户端初始化时获取的集群节点列表信息，并不是针对单个查询的响应节点，所以事件监听的方式才是最准确的。

备注：内容来源于stack exchange，提问作者user22299554