resuma 0.4.2

Resuma - SSR + Resumability + Islands + Server Actions + JS Bridge 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

//! Global registries for Resuma Flow `#[load]` and `#[submit]` handlers.

use std::collections::HashMap;
use std::future::Future;
use std::pin::Pin;

use crate::core::{Result, ResumaError};
use once_cell::sync::Lazy;
use parking_lot::RwLock;
use serde_json::Value;

use super::request::FlowRequest;

pub type LoadFuture = Pin<Box<dyn Future<Output = Result<Value>> + Send>>;
pub type LoadFn = fn(FlowRequest) -> LoadFuture;

pub type SubmitFuture = Pin<Box<dyn Future<Output = Result<Value>> + Send>>;
pub type SubmitFn = fn(Value, FlowRequest) -> SubmitFuture;

static LOADS: Lazy<RwLock<HashMap<String, LoadFn>>> = Lazy::new(|| RwLock::new(HashMap::new()));
static SUBMITS: Lazy<RwLock<HashMap<String, SubmitFn>>> = Lazy::new(|| RwLock::new(HashMap::new()));

pub fn register_loader(name: &str, f: LoadFn) {
    LOADS.write().insert(name.to_string(), f);
}

pub fn register_submit(name: &str, f: SubmitFn) {
    SUBMITS.write().insert(name.to_string(), f);
}

pub fn get_loader(name: &str) -> Option<LoadFn> {
    LOADS.read().get(name).copied()
}

pub fn get_submit(name: &str) -> Option<SubmitFn> {
    SUBMITS.read().get(name).copied()
}

pub async fn dispatch_load(name: &str, req: FlowRequest) -> Result<Value> {
    match get_loader(name) {
        Some(f) => f(req).await,
        None => Err(ResumaError::UnknownLoader(name.to_string())),
    }
}

pub async fn dispatch_submit(name: &str, data: Value, req: FlowRequest) -> Result<Value> {
    match get_submit(name) {
        Some(f) => f(data, req).await,
        None => Err(ResumaError::UnknownSubmit(name.to_string())),
    }
}