first commit

src/db/mod.rs

@@ -0,0 +1,11 @@
+pub mod postgres;
+pub mod redis_helper;
+
+pub type Db = sqlx::PgPool;
+pub type Redis = redis::aio::MultiplexedConnection;
+
+#[derive(Clone)]
+pub struct AppState {
+    pub db: Db,
+    pub redis: Redis,
+}

src/db/postgres.rs

@@ -0,0 +1,27 @@
+use sqlx::{postgres::PgPoolOptions, PgPool};
+use std::env;
+
+pub async fn connect() -> PgPool {
+    let db_url = env::var("DATABASE_URL").expect("DATABASE_URL is required");
+
+    let pool = PgPoolOptions::new()
+        .max_connections(5)
+        .connect(&db_url)
+        .await
+        .expect("Failed to connect to Postgres");
+
+    sqlx::query(
+        r#"
+        CREATE TABLE IF NOT EXISTS timezones (
+            user_id TEXT PRIMARY KEY,
+            username TEXT NOT NULL,
+            timezone TEXT NOT NULL
+        );
+        "#,
+    )
+    .execute(&pool)
+    .await
+    .expect("Failed to create timezones table");
+
+    pool
+}

src/db/redis_helper.rs

@@ -0,0 +1,12 @@
+use redis::Client;
+use redis::aio::MultiplexedConnection;
+use std::env;
+
+pub async fn connect() -> MultiplexedConnection {
+    let url = env::var("REDIS_URL").expect("REDIS_URL is required");
+    let client = Client::open(url).expect("Failed to create Redis client");
+    client
+        .get_multiplexed_tokio_connection()
+        .await
+        .expect("Failed to connect to Redis")
+}