chore: utils chnages

8shaws · Aug 3, 2024 · ebdad8c · ebdad8c
1 parent 4e90173
commit ebdad8c
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/common/Cargo.toml b/crates/common/Cargo.toml
@@ -13,3 +13,4 @@ diesel = { version = "2.2.0", features = ["postgres", "chrono", "r2d2", "uuid"]
 dotenvy = "0.15"
 chrono = { version = "0.4.19", features = ["serde"] }
 uuid = { version = "1.10.0", features = ["v4", "serde"] }
+r2d2_redis = "0.14.0"
diff --git a/crates/common/src/lib.rs b/crates/common/src/lib.rs
@@ -1 +1,3 @@
+pub mod redis;
 pub mod types;
+pub mod utils;
diff --git a/crates/common/src/redis.rs b/crates/common/src/redis.rs
@@ -0,0 +1,12 @@
+use r2d2_redis::{r2d2, RedisConnectionManager};
+use std::env;
+
+pub type RedisPool = r2d2::Pool<RedisConnectionManager>;
+
+pub fn initialize_redis_pool() -> RedisPool {
+    let redis_url = env::var("REDIS_URL").expect("REDIS_URL must be set");
+    let manager = RedisConnectionManager::new(redis_url).unwrap();
+    r2d2::Pool::builder()
+        .build(manager)
+        .expect("Failed to create pool")
+}