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

#![allow(unused_assignments)]
use std::collections::HashMap;
use crate::sled_json::{ TreeWrapper, JSONEncoder };
use crate::public_struct::{ ImageChainIDJSONValue,ImageChainIDItemJSONValue };


pub async fn record_image_chain_id(db: &sled::Db,image_digest:String,image_chain_id:String,image_cache_id:String,image_diff_id:String,image_parent_chain_id:String,image_size:String) -> sled::Result<()> {
    let tree = TreeWrapper::<JSONEncoder<ImageChainIDJSONValue>, JSONEncoder<ImageChainIDJSONValue>>::new(
        db.open_tree("image_chain_id")?,
    );

    let value = tree
    .get(image_digest.clone());

    let mut image_digest_json_value = ImageChainIDJSONValue{
        image_chain_id: HashMap::new()
    };
    let mut image_chain_id_item = ImageChainIDItemJSONValue::default();
    match value {
        Ok(res) => {
            match res {
                Some(res1) =>{
                    match res1.decode() {
                        Some(mut res2) => {
                            image_chain_id_item.cache_id = image_cache_id.clone();
                            image_chain_id_item.diff_id = image_diff_id.clone();
                            image_chain_id_item.parent_chain_id = image_parent_chain_id.clone();
                            image_chain_id_item.size = image_size.clone();
                            res2.image_chain_id.insert(image_chain_id.clone(),image_chain_id_item);
                            image_digest_json_value = ImageChainIDJSONValue {
                                image_chain_id: res2.image_chain_id
                            };
                        }
                        _ => {
                            image_chain_id_item.cache_id = image_cache_id.clone();
                            image_chain_id_item.diff_id = image_diff_id.clone();
                            image_chain_id_item.parent_chain_id = image_parent_chain_id.clone();
                            image_chain_id_item.size = image_size.clone();
                            let mut image_chain_id_1 = HashMap::new();
                            image_chain_id_1.insert(image_chain_id.clone(),image_chain_id_item);
                            image_digest_json_value = ImageChainIDJSONValue {
                                image_chain_id:image_chain_id_1
                            };
                        }
                    }
                },
                _ => {
                        image_chain_id_item.cache_id = image_cache_id.clone();
                        image_chain_id_item.diff_id = image_diff_id.clone();
                        image_chain_id_item.parent_chain_id = image_parent_chain_id.clone();
                        image_chain_id_item.size = image_size.clone();
                        let mut image_chain_id_1 = HashMap::new();
                        image_chain_id_1.insert(image_chain_id.clone(),image_chain_id_item);
                        image_digest_json_value = ImageChainIDJSONValue {
                            image_chain_id:image_chain_id_1
                        };
                }
            }
        },
        Err(_) => {
                    image_chain_id_item.cache_id = image_cache_id.clone();
                    image_chain_id_item.diff_id = image_diff_id.clone();
                    image_chain_id_item.parent_chain_id = image_parent_chain_id.clone();
                    image_chain_id_item.size = image_size.clone();
                    let mut image_chain_id_1 = HashMap::new();
                    image_chain_id_1.insert(image_chain_id.clone(),image_chain_id_item);
                    image_digest_json_value = ImageChainIDJSONValue {
                        image_chain_id:image_chain_id_1
                    };
        }
    };

    tree.insert(image_digest.clone(), &image_digest_json_value)?;
    // let value1 = tree
    //     .get(image_digest.clone())?
    //     .expect("Value not found")
    //     .decode()
    //     .expect("Decoding failed");
    // assert_eq!(value, image_repositories_json_value);
    // println!("image_digest:{:?}",value1);
    // println!("image_version:{:?}",value.image_version["nginx:latest"]);
    Ok(())
}


pub async fn remove_image_chain_id(db: &sled::Db,image_digest:String) -> sled::Result<()> {
    let tree = db.open_tree("image_chain_id")?;
    tree.remove(image_digest.clone()).unwrap();
    Ok(())
}