firecracker_http_client/
snapshot.rs

1use crate::validation::validate_existing_path;
2use crate::validation::validate_writable_path;
3use crate::FirecrackerError;
4use async_trait::async_trait;
5use serde::{Deserialize, Serialize};
6use validator::Validate;
7
8#[derive(Debug, Serialize, Deserialize, Validate)]
9pub struct SnapshotCreateParams {
10    #[validate(custom = "validate_writable_path")]
11    pub snapshot_path: String,
12    #[validate(custom = "validate_writable_path")]
13    pub mem_file_path: String,
14    #[serde(skip_serializing_if = "Option::is_none")]
15    #[validate(regex(
16        path = "SNAPSHOT_TYPE_REGEX",
17        message = "Invalid snapshot type. Must be one of: Full, Diff"
18    ))]
19    pub snapshot_type: Option<String>,
20    #[serde(skip_serializing_if = "Option::is_none")]
21    pub version: Option<String>,
22}
23
24#[derive(Debug, Serialize, Deserialize, Validate)]
25pub struct SnapshotLoadParams {
26    #[validate(custom = "validate_existing_path")]
27    pub snapshot_path: String,
28    #[validate(custom = "validate_existing_path")]
29    pub mem_file_path: String,
30    #[serde(skip_serializing_if = "Option::is_none")]
31    pub enable_diff_snapshots: Option<bool>,
32}
33
34lazy_static::lazy_static! {
35    static ref SNAPSHOT_TYPE_REGEX: regex::Regex = regex::Regex::new(r"^(Full|Diff)$").unwrap();
36}
37
38#[async_trait]
39pub trait SnapshotOperations {
40    async fn create_snapshot(&self, params: &SnapshotCreateParams) -> Result<(), FirecrackerError>;
41    async fn load_snapshot(&self, params: &SnapshotLoadParams) -> Result<(), FirecrackerError>;
42}
43
44#[async_trait]
45impl SnapshotOperations for crate::FirecrackerClient {
46    async fn create_snapshot(&self, params: &SnapshotCreateParams) -> Result<(), FirecrackerError> {
47        params.validate()?;
48
49        let url = self.url("/snapshot/create")?;
50        let response = self.client.put(url).json(params).send().await?;
51
52        if !response.status().is_success() {
53            return Err(FirecrackerError::Api {
54                status_code: response.status().as_u16(),
55                message: response.text().await?,
56            });
57        }
58
59        Ok(())
60    }
61
62    async fn load_snapshot(&self, params: &SnapshotLoadParams) -> Result<(), FirecrackerError> {
63        params.validate()?;
64
65        let url = self.url("/snapshot/load")?;
66        let response = self.client.put(url).json(params).send().await?;
67
68        if !response.status().is_success() {
69            return Err(FirecrackerError::Api {
70                status_code: response.status().as_u16(),
71                message: response.text().await?,
72            });
73        }
74
75        Ok(())
76    }
77}
firecracker_http_client/snapshot.rs

firecracker_http_client/
snapshot.rs
