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
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use std::{env, sync::Arc, time::Duration};

use crate::domain::models_requests::{self, JwtKey};
use headless_lms_models as models;
use models::library::regrading;
use sqlx::{Connection, PgConnection};

/**
Starts a thread that will periodically send regrading submissions to the corresponding exercise services for regrading.
*/
pub async fn main() -> anyhow::Result<()> {
    env::set_var("RUST_LOG", "info,actix_web=info,sqlx=warn");
    dotenv::dotenv().ok();
    crate::setup_tracing()?;
    let db_url = env::var("DATABASE_URL")
        .unwrap_or_else(|_| "postgres://localhost/headless_lms_dev".to_string());
    let jwt_key = Arc::new(JwtKey::try_from_env().expect("Could not initialise JwtKey"));

    let mut interval = tokio::time::interval(Duration::from_secs(10));
    let mut ticks = 60;
    // Since this is repeating every 10 seconds we can keep the connection open.
    let mut conn = PgConnection::connect(&db_url).await?;
    loop {
        interval.tick().await;

        ticks += 1;
        // 60 10 second intervals = 10 minutes
        if ticks > 60 {
            // occasionally prints a reminder that the service is still running
            ticks = 0;
            tracing::info!("running the regrader");
        }

        let exercise_services_by_type =
            models::exercise_service_info::get_all_exercise_services_by_type(
                &mut conn,
                models_requests::fetch_service_info,
            )
            .await?;
        // do not stop the thread on error, report it and try again next tick
        if let Err(err) = regrading::regrade(
            &mut conn,
            &exercise_services_by_type,
            models_requests::make_grading_request_sender(Arc::clone(&jwt_key)),
        )
        .await
        {
            tracing::error!("Error in regrader: {}", err);
        }
    }
}