Function run_actix_server_with_context

src/web_server.rs:26–81

(ctx: Data<AppContext>)

}

pub async fn run_actix_server_with_context(ctx: Data<AppContext>) -> std::io::Result<()> {
    let config = &ctx.config.clone();

    let tls = match &config.server.mode {
        ServerMode::Https => Some(load_rustls_config(&config.server.ssl)),
        ServerMode::Http => {
            log::warn!("server.mode=http is not recommended in production");
            None
        }
    };

    log::info!(
        "starting {} server at {}://{}:{}",
        &config.server.mode.protocol().to_uppercase(),
        &config.server.mode.protocol(),
        &config.server.host,
        &config.server.port,
    );

    let server = HttpServer::new(move || {
        App::new()
            // enable logger
            .wrap(middleware::Logger::default())
            // ensure the CORS middleware is wrapped around the httpauth middleware
            // so it is able to add headers to error responses
            .wrap(Cors::permissive())
            // register simple handler, handle all methods
            .app_data(web::JsonConfig::default().limit(8_388_608)) // 8 MB)
            .app_data(ctx.clone())
            .service(api_docs_module())
            .service(auth_module())
            .service(
                web::scope("/vectordb")
                    .wrap(AuthenticationMiddleware(
                        ctx.ain_env.active_sessions.clone(),
                    ))
                    // vectors module must be registered before collections module
                    // as its scope path is more specific than collections module
                    .service(search_module())
                    .service(indexes_module())
                    .service(vectors_module())
                    .service(transactions_module())
                    .service(streaming_module())
                    .service(version_module())
                    .service(collections_module()),
            )
    })
    .keep_alive(std::time::Duration::from_secs(10));

    let addr = config.server.listen_address();
    let server = match tls {
        Some(tls_config) => server.bind_rustls_0_23(addr, tls_config),
        None => server.bind(addr),
    };
    server?.run().await
}

fn load_rustls_config(ssl_config: &Ssl) -> rustls::ServerConfig {