Init

src/main.rs

@@ -0,0 +1,74 @@
+use actix_web::{web, App, HttpServer};
+use actix_cors::Cors;
+use anyhow::Result;
+use sqlx::PgPool;
+use tracing::info;
+
+mod error;
+mod handlers;
+mod models;
+
+#[derive(Clone)]
+pub struct AppState {
+    db: PgPool,
+}
+
+#[actix_web::main]
+async fn main() -> Result<()> {
+    // Load environment variables from .env file
+    dotenv::dotenv().ok();
+
+    // Initialize logging
+    tracing_subscriber::fmt::init();
+
+    // Database connection string from environment
+    let database_url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set");
+
+    // Create database connection pool
+    use sqlx::postgres::PgPoolOptions;
+
+    // In main(), replace the PgPool::connect with:
+    let pool = PgPoolOptions::new()
+        .max_connections(5)
+        .acquire_timeout(std::time::Duration::from_secs(3))
+        .connect(&database_url)
+        .await?;
+
+    // Run database migrations
+    sqlx::migrate!("./migrations").run(&pool).await?;
+
+    let state = AppState { db: pool };
+
+    info!("Starting server at http://127.0.0.1:8080");
+
+    // Start HTTP server
+    HttpServer::new(move || {
+        let cors = Cors::default()
+            .allow_any_origin()
+            .allow_any_method()
+            .allow_any_header()
+            .max_age(3600);
+        
+        App::new()
+            .wrap(cors)
+            .app_data(web::Data::new(state.clone()))
+            .service(
+                web::scope("/api")
+                    .route("/shorten", web::post().to(handlers::create_short_url))
+                    .route("/links", web::get().to(handlers::get_all_links)),
+                    
+            )
+            .service(
+                web::resource("/{short_code}")
+                    .route(web::get().to(handlers::redirect_to_url))
+            )
+            .service(web::resource("/{short_code}").route(web::get().to(handlers::redirect_to_url)))
+    })
+    .workers(2) // Limit worker threads
+    .backlog(10_000)
+    .bind("127.0.0.1:8080")?
+    .run()
+    .await?;
+
+    Ok(())
+}