[go: up one dir, main page]

worker/
rate_limit.rs

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
55
56
57
58
59
60
61
62
63
64
65
66
use crate::{send::SendFuture, EnvBinding, Result};
use serde::{Deserialize, Serialize};
use wasm_bindgen::{JsCast, JsValue};
use wasm_bindgen_futures::JsFuture;
use worker_sys::RateLimiter as RateLimiterSys;

pub struct RateLimiter(RateLimiterSys);

#[derive(Serialize, Deserialize)]
struct RateLimitOptions {
    key: String,
}

#[derive(Serialize, Deserialize)]
pub struct RateLimitOutcome {
    pub success: bool,
}

unsafe impl Send for RateLimiter {}
unsafe impl Sync for RateLimiter {}

impl EnvBinding for RateLimiter {
    const TYPE_NAME: &'static str = "RateLimiter";
}
impl RateLimiter {
    pub async fn limit(&self, key: String) -> Result<RateLimitOutcome> {
        let arg = serde_wasm_bindgen::to_value(&RateLimitOptions { key })?;
        let promise = self.0.limit(arg.into())?;
        let fut = SendFuture::new(JsFuture::from(promise));
        let result = fut.await?;
        let outcome = serde_wasm_bindgen::from_value(result)?;
        Ok(outcome)
    }
}

impl JsCast for RateLimiter {
    fn instanceof(val: &JsValue) -> bool {
        val.is_instance_of::<RateLimiterSys>()
    }

    fn unchecked_from_js(val: JsValue) -> Self {
        Self(val.into())
    }

    fn unchecked_from_js_ref(val: &JsValue) -> &Self {
        unsafe { &*(val as *const JsValue as *const Self) }
    }
}

impl From<RateLimiter> for JsValue {
    fn from(limiter: RateLimiter) -> Self {
        JsValue::from(limiter.0)
    }
}

impl AsRef<JsValue> for RateLimiter {
    fn as_ref(&self) -> &JsValue {
        &self.0
    }
}

impl From<RateLimiterSys> for RateLimiter {
    fn from(inner: RateLimiterSys) -> Self {
        Self(inner)
    }
}