csm-rs 0.89.1

A library for Shasta
Documentation 
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
52
53
54
55
56
57
58
59
60

use std::io::{self, Write};

use crate::{
  error::Error,
  ims::{self, job::types::Job},
};

/// Wait an IMS job to finish
pub async fn wait_ims_job_to_finish(
  shasta_token: &str,
  shasta_base_url: &str,
  shasta_root_cert: &[u8],
  ims_job_id: &str,
) -> Result<(), Error> {
  let mut i = 0;
  let max = 1800; // Max ammount of attempts to check if CFS session has ended
  loop {
    let ims_job: Job = ims::job::http_client::get(
      shasta_token,
      shasta_base_url,
      shasta_root_cert,
      Some(ims_job_id),
    )
    .await?
    .first()
    .cloned()
    .ok_or_else(|| {
      Error::Message(format!("ERROR - IMS job '{}' not found", ims_job_id))
    })?;

    log::debug!(
      "IMS job details:\n{}",
      serde_json::to_string_pretty(&ims_job).unwrap()
    );

    let ims_job_status = ims_job.status.unwrap();

    if (ims_job_status != "error" && ims_job_status != "success") && i < max {
      print!("\x1B[2K"); // Clear current line
      io::stdout().flush().unwrap();
      print!(
                "\rWaiting IMS job '{}' with job status '{}'. Checking again in 2 secs. Attempt {} of {}.",
                ims_job_id, ims_job_status, i, max
            );
      io::stdout().flush().unwrap();

      tokio::time::sleep(tokio::time::Duration::from_secs(2)).await;

      i += 1;
    } else {
      println!(
        "\nIMS job '{}' finished with job status '{}'",
        ims_job_id, ims_job_status
      );
      break;
    }
  }

  Ok(())
}