mcpr-core 0.4.41

Core types, traits, protocol, and proxy engine for mcpr crates
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

//! `McpHealthMiddleware` — on a successful `initialize` response, mark the MCP
//! upstream as confirmed connected.

use crate::protocol::McpMethod;
use async_trait::async_trait;

use super::ResponseMiddleware;
use crate::proxy::pipeline::context::{RequestContext, ResponseContext};
use crate::proxy::proxy_state::ProxyState;

pub struct McpHealthMiddleware;

#[async_trait]
impl ResponseMiddleware for McpHealthMiddleware {
    async fn on_response(
        &self,
        state: &ProxyState,
        req: &RequestContext,
        resp: &mut ResponseContext,
    ) {
        if req.mcp_method == Some(McpMethod::Initialize) && resp.status < 400 {
            crate::proxy::lock_health(&state.health).confirm_mcp_connected();
        }
    }
}