74 lines
1.9 KiB
Rust
74 lines
1.9 KiB
Rust
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(())
|
|
}
|