Function discord_request_retry

components/runtime/src/extensions/discord.rs:205–271

(
    state: &Rc<RefCell<OpState>>,
    f: impl Fn(Arc<DiscordConfig>) -> Fut,
)

}

pub async fn discord_request_retry<T: Send, Fut>(
    state: &Rc<RefCell<OpState>>,
    f: impl Fn(Arc<DiscordConfig>) -> Fut,
) -> Result<T, AnyError>
where
    Fut: Future<Output = Result<T, AnyError>>,
{
    let rt_handle = {
        let state = state.borrow();
        let rt_ctx: &RuntimeContext = state.borrow();
        rt_ctx.main_tokio_runtime.clone()
    };

    let mut retry_sleep = Duration::from_secs(1);
    let max_sleep = Duration::from_secs(60);
    let mut retry_count = 1;

    let res = loop {
        let discord_config = {
            let state = state.borrow();
            let rt_ctx: &RuntimeContext = state.borrow();
            rt_ctx.discord_config.clone()
        };

        // We run the request on the main botloader tokio runtime, the vm runtime is disposed of on shutdown
        // and any futures cancelled, which twilights ratelimit handling does not like
        let inner_fut = f(discord_config);
        let fut = WrappedFuture {
            inner: inner_fut,
            rt: rt_handle.clone(),
        };

        let retry_reason = match fut.await {
            Ok(v) => break Ok(v),
            Err(err) => match err.downcast::<twilight_http::Error>() {
                Ok(discord_err) => match discord_err.kind() {
                    ErrorType::RequestCanceled => "REQUEST_CANCELED",
                    ErrorType::RequestError => "REQUEST_ERROR",

                    // I've disabled this for now, from experience this could lead to requests being processed twice.
                    // ErrorType::RequestTimedOut => "REQUEST_TIMEOUT",
                    ErrorType::Response { status, .. } if status.get() == 429 => {
                        if check_bad_request(state, status.get()).is_err() {
                            break Err(anyhow::anyhow!(
                                "Hit maximum number of bad requests within a time period"
                            ));
                        }

                        "RATELIMIT_429"
                    }
                    _ => break Err(handle_discord_error(state, discord_err)),
                },
                Err(err) => break Err(err),
            },
        };
        tokio::time::sleep(retry_sleep).await;
        retry_sleep = retry_sleep * 2;
        if retry_sleep > max_sleep {