port-sdk 0.1.0

Rust SDK for Port APIs.
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

use std::collections::{HashMap, HashSet};
use std::sync::{Arc, Mutex};

/// Tracks resources created during tests or manual runs so stragglers can be cleaned up later.
#[derive(Debug, Default)]
pub struct ResourceTracker {
    inner: Mutex<HashMap<String, HashSet<String>>>,
}

impl ResourceTracker {
    pub fn new() -> Self {
        Self::default()
    }

    pub fn record_creation(&self, resource_type: impl Into<String>, id: impl Into<String>) {
        let resource_key = resource_type.into();
        let mut guard = self.inner.lock().expect("resource tracker mutex poisoned");
        let entry = guard.entry(resource_key).or_default();
        entry.insert(id.into());
    }

    pub fn record_deletion(&self, resource_type: impl Into<String>, id: impl Into<String>) {
        let resource_key = resource_type.into();
        let resource_id = id.into();
        let mut guard = self.inner.lock().expect("resource tracker mutex poisoned");
        if let Some(entry) = guard.get_mut(&resource_key) {
            entry.remove(&resource_id);
            if entry.is_empty() {
                guard.remove(&resource_key);
            }
        }
    }

    pub fn outstanding(&self) -> HashMap<String, Vec<String>> {
        let guard = self.inner.lock().expect("resource tracker mutex poisoned");
        guard.iter().map(|(key, values)| (key.clone(), values.iter().cloned().collect())).collect()
    }
}

pub type ResourceTrackerHandle = Arc<ResourceTracker>;

pub fn new_shared_tracker() -> ResourceTrackerHandle {
    Arc::new(ResourceTracker::new())
}