firecracker-http-client 0.1.3

A comprehensive HTTP client for the Firecracker VMM API
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

use crate::models::{Balloon, BalloonStats};
use crate::FirecrackerError;
use async_trait::async_trait;
use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize)]
pub struct BalloonUpdate {
    pub amount_mib: u32,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct BalloonStatsUpdate {
    pub stats_polling_interval_s: u32,
}

#[async_trait]
pub trait BalloonOperations {
    async fn get_balloon_config(&self) -> Result<Balloon, FirecrackerError>;
    async fn put_balloon_config(&self, config: &Balloon) -> Result<(), FirecrackerError>;
    async fn patch_balloon_config(&self, update: &BalloonUpdate) -> Result<(), FirecrackerError>;
    async fn get_balloon_stats(&self) -> Result<BalloonStats, FirecrackerError>;
    async fn patch_balloon_stats(
        &self,
        update: &BalloonStatsUpdate,
    ) -> Result<(), FirecrackerError>;
}

#[async_trait]
impl BalloonOperations for crate::FirecrackerClient {
    async fn get_balloon_config(&self) -> Result<Balloon, FirecrackerError> {
        let url = self.url("balloon")?;
        let response = self.client.get(url).send().await?;

        if !response.status().is_success() {
            return Err(FirecrackerError::Api {
                status_code: response.status().as_u16(),
                message: response.text().await?,
            });
        }

        Ok(response.json().await?)
    }

    async fn put_balloon_config(&self, config: &Balloon) -> Result<(), FirecrackerError> {
        let url = self.url("balloon")?;
        let response = self.client.put(url).json(config).send().await?;

        if !response.status().is_success() {
            return Err(FirecrackerError::Api {
                status_code: response.status().as_u16(),
                message: response.text().await?,
            });
        }

        Ok(())
    }

    async fn patch_balloon_config(&self, update: &BalloonUpdate) -> Result<(), FirecrackerError> {
        let url = self.url("balloon")?;
        let response = self.client.patch(url).json(update).send().await?;

        if !response.status().is_success() {
            return Err(FirecrackerError::Api {
                status_code: response.status().as_u16(),
                message: response.text().await?,
            });
        }

        Ok(())
    }

    async fn get_balloon_stats(&self) -> Result<BalloonStats, FirecrackerError> {
        let url = self.url("balloon/statistics")?;
        let response = self.client.get(url).send().await?;

        if !response.status().is_success() {
            return Err(FirecrackerError::Api {
                status_code: response.status().as_u16(),
                message: response.text().await?,
            });
        }

        Ok(response.json().await?)
    }

    async fn patch_balloon_stats(
        &self,
        update: &BalloonStatsUpdate,
    ) -> Result<(), FirecrackerError> {
        let url = self.url("balloon/statistics")?;
        let response = self.client.patch(url).json(update).send().await?;

        if !response.status().is_success() {
            return Err(FirecrackerError::Api {
                status_code: response.status().as_u16(),
                message: response.text().await?,
            });
        }

        Ok(())
    }
}