use cdk::subscription::Params;
use cdk::ws::{WsResponseResult, WsSubscribeResponse};
use super::{WsContext, WsError};
pub(crate) async fn handle(
context: &mut WsContext,
params: Params,
) -> Result<WsResponseResult, WsError> {
let sub_id = params.id.clone();
if context.subscriptions.contains_key(&sub_id) {
return Err(WsError::InvalidParams);
}
let mut subscription = context
.state
.mint
.pubsub_manager()
.subscribe(params)
.map_err(|_| WsError::ParseError)?;
let publisher = context.publisher.clone();
let sub_id_for_sender = sub_id.clone();
context.subscriptions.insert(
sub_id.clone(),
tokio::spawn(async move {
while let Some(response) = subscription.recv().await {
let _ = publisher.try_send((sub_id_for_sender.clone(), response.into_inner()));
}
}),
);
Ok(WsSubscribeResponse {
status: "OK".to_string(),
sub_id,
}
.into())
}