osc-cost 0.8.0

osc-cost helps measuring OUTSCALE infrastructure costs
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
103
104
105

use std::error;

use log::{debug, warn};
use outscale_api::{
    apis::flexible_gpu_api::read_flexible_gpus,
    models::{ReadFlexibleGpusRequest, ReadFlexibleGpusResponse},
};

use crate::{
    core::{flexible_gpus::FlexibleGpu, Resource, Resources},
    VERSION,
};

use super::Input;

pub type FlexibleGpuId = String;

const RESOURCE_NAME: &str = "FlexibleGpu";

impl Input {
    pub fn fetch_flexible_gpus(&mut self) -> Result<(), Box<dyn error::Error>> {
        if self.skip_fetch(RESOURCE_NAME) {
            return Ok(());
        }
        let request = ReadFlexibleGpusRequest {
            ..Default::default()
        };
        let result: ReadFlexibleGpusResponse =
            read_flexible_gpus(&self.config, Some(request.clone()))?;
        debug!("{:#?}", result);
        let flexible_gpus = match result.flexible_gpus {
            None => {
                warn!("warning: no flexible gpu available");
                return Ok(());
            }
            Some(flexible_gpus) => flexible_gpus,
        };
        self.flexible_gpus.clear();
        for flexible_gpu in flexible_gpus {
            let flexible_gpu_id = flexible_gpu
                .flexible_gpu_id
                .clone()
                .unwrap_or_else(|| String::from(""));
            self.flexible_gpus.insert(flexible_gpu_id, flexible_gpu);
        }
        warn!("info: fetched {} flexible gpus", self.flexible_gpus.len());
        Ok(())
    }

    pub fn fill_resource_flexible_gpus(&self, resources: &mut Resources) {
        if self.flexible_gpus.is_empty() && self.need_default_resource {
            resources.resources.push(Resource::FlexibleGpu(FlexibleGpu {
                account_id: self.account_id(),
                read_date_rfc3339: self.fetch_date.map(|date| date.to_rfc3339()),
                region: self.region.clone(),
                ..Default::default()
            }));
        }
        for (flexible_gpu_id, flexible_gpu) in &self.flexible_gpus {
            let Some(model_name) = flexible_gpu.model_name.clone() else {
                warn!("warning: a flexible gpu did not have a model name");
                continue;
            };
            let Some(state) = flexible_gpu.state.clone() else {
                warn!("warning: a flexible gpu did not have a state");
                continue;
            };
            let price_per_hour = match state.as_str() {
                "attached" | "attaching" => self.catalog_entry(
                    "TinaOS-FCU",
                    format!("Gpu:attach:{model_name}").as_str(),
                    "AllocateGpu",
                ),
                "allocated" | "detaching" => self.catalog_entry(
                    "TinaOS-FCU",
                    format!("Gpu:allocate:{model_name}").as_str(),
                    "AllocateGpu",
                ),
                _ => {
                    warn!("warning: a flexible gpus does not have standard state");
                    continue;
                }
            };

            if price_per_hour.is_none() {
                warn!("warning: could not retrieve the catalog for {model_name} in state {state}");
                continue;
            }

            let core_flexible_gpu = FlexibleGpu {
                osc_cost_version: Some(String::from(VERSION)),
                account_id: self.account_id(),
                read_date_rfc3339: self.fetch_date.map(|date| date.to_rfc3339()),
                region: self.region.clone(),
                resource_id: Some(flexible_gpu_id.clone()),
                price_per_hour,
                price_per_month: None,
                model_name: Some(model_name),
            };
            resources
                .resources
                .push(Resource::FlexibleGpu(core_flexible_gpu));
        }
    }
}