linkletter 0.1.0

URL replacement and analytics service
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

mod database;
mod service;
mod settings;

use axum::{Router, routing::{get, post}};

use crate::settings::Settings;

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    // Initialize service state
    let settings = Settings::new()?;
    let pool = database::create_pool(settings.database).await?;
    let state = service::State {
        settings: settings.api,
        pool: pool,
    };

    // Set up service endpoints
    let app = Router::new()
        .route("/documents", post(service::documents::process))
        .route("/documents/{doc_id}/urls", get(service::documents::urls))
        .route("/documents/{doc_id}/visits", get(service::documents::visits))
        .route("/documents/{doc_id}/analytics", get(service::documents::analytics))
        .route("/urls/{url_id}", get(service::urls::get))
        .route("/urls/{url_id}/visits", get(service::urls::visits))
        .route("/visit/{url_id}", get(service::visit::process))
        .with_state(state)
        .into_make_service_with_connect_info::<std::net::SocketAddr>();

    // Start service
    let service_address = (settings.address.ip, settings.address.port);
    let listener = tokio::net::TcpListener::bind(service_address).await?;
    axum::serve(listener, app).await?;

    Ok(())
}