neva 0.3.2

MCP SDK for Rust
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

//! Utilities for handling notifications from server

use crate::types::notification::Notification;
use std::collections::HashMap;
use std::future::Future;
use std::pin::Pin;
use std::sync::Arc;
use tokio::sync::RwLock;

/// Represents a notification handler function
pub(crate) type NotificationsHandlerFunc =
    Arc<dyn Fn(Notification) -> Pin<Box<dyn Future<Output = ()> + Send + 'static>> + Send + Sync>;

/// Represents a notification handler
#[derive(Default)]
pub(super) struct NotificationsHandler {
    handlers: RwLock<HashMap<String, NotificationsHandlerFunc>>,
}

impl NotificationsHandler {
    /// Subscribes to the `event` with the `handler`
    pub(super) fn subscribe<E, F, R>(&self, event: E, handler: F)
    where
        E: Into<String>,
        F: Fn(Notification) -> R + Clone + Send + Sync + 'static,
        R: Future<Output = ()> + Send,
    {
        let handler: NotificationsHandlerFunc = Arc::new(move |params| {
            let handler = handler.clone();
            Box::pin(async move {
                handler(params).await;
            })
        });
        tokio::task::block_in_place(|| {
            self.handlers.blocking_write().insert(event.into(), handler);
        });
    }

    /// Unsubscribes from the `event`
    pub(super) fn unsubscribe(&self, event: impl AsRef<str>) {
        tokio::task::block_in_place(|| {
            self.handlers.blocking_write().remove(event.as_ref());
        });
    }

    /// Calls an appropriate notifications handler
    pub(super) async fn notify(&self, notification: Notification) {
        let guard = self.handlers.read().await;
        if let Some(handler) = guard.get(&notification.method).cloned() {
            drop(guard);
            handler(notification).await;
        }
    }
}