spring_redis/
lib.rs

1//! [![spring-rs](https://img.shields.io/github/stars/spring-rs/spring-rs)](https://spring-rs.github.io/docs/plugins/spring-redis)
2#![doc(html_favicon_url = "https://spring-rs.github.io/favicon.ico")]
3#![doc(html_logo_url = "https://spring-rs.github.io/logo.svg")]
4
5pub mod config;
6
7pub use redis;
8pub use spring_macros::cache;
9
10use anyhow::Context;
11use config::RedisConfig;
12use redis::{aio::ConnectionManagerConfig, Client};
13use spring::async_trait;
14use spring::config::ConfigRegistry;
15use spring::plugin::MutableComponentRegistry;
16use spring::{app::AppBuilder, error::Result, plugin::Plugin};
17use std::time::Duration;
18
19pub type Redis = redis::aio::ConnectionManager;
20
21pub struct RedisPlugin;
22
23#[async_trait]
24impl Plugin for RedisPlugin {
25    async fn build(&self, app: &mut AppBuilder) {
26        let config = app
27            .get_config::<RedisConfig>()
28            .expect("redis plugin config load failed");
29
30        let connect: Redis = Self::connect(config).await.expect("redis connect failed");
31        app.add_component(connect);
32    }
33}
34
35impl RedisPlugin {
36    async fn connect(config: RedisConfig) -> Result<Redis> {
37        let url = config.uri;
38        let client = Client::open(url.clone())
39            .with_context(|| format!("redis connect failed:{}", url.clone()))?;
40
41        let mut conn_config = ConnectionManagerConfig::new();
42
43        if let Some(exponent_base) = config.exponent_base {
44            conn_config = conn_config.set_exponent_base(exponent_base);
45        }
46        if let Some(factor) = config.factor {
47            conn_config = conn_config.set_factor(factor);
48        }
49        if let Some(number_of_retries) = config.number_of_retries {
50            conn_config = conn_config.set_number_of_retries(number_of_retries);
51        }
52        if let Some(max_delay) = config.max_delay {
53            conn_config = conn_config.set_max_delay(max_delay);
54        }
55        if let Some(response_timeout) = config.response_timeout {
56            conn_config = conn_config.set_response_timeout(Duration::from_millis(response_timeout));
57        }
58        if let Some(connection_timeout) = config.connection_timeout {
59            conn_config =
60                conn_config.set_connection_timeout(Duration::from_millis(connection_timeout));
61        }
62
63        Ok(client
64            .get_connection_manager_with_config(conn_config)
65            .await
66            .with_context(|| format!("redis connect failed:{}", url.clone()))?)
67    }
68}
spring_redis/lib.rs

spring_redis/
lib.rs
