oci-util 0.2.0

云网关OCI公共库
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

pub mod build;
pub mod config;
pub mod layer;
pub mod manifest;

use crate::filesystem::FileSystem;
use anyhow::Result;
use log::warn;
use oci_distribution::Reference;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
///    "hello-world": {
//       "hello-world:latest": "sha256:feb5d9fea6a5e9606aa995e879d862b825965ba48de054caab5ef356dc6b3412",
//       "hello-world@sha256:80f31da1ac7b312ba29d65080fddf797dd76acfb870e677f390d5acba9741b17": "sha256:feb5d9fea6a5e9606aa995e879d862b825965ba48de054caab5ef356dc6b3412"
//     },
//     "repo.netfuse.cn/moss/hello-wasm": {
//       "repo.netfuse.cn/moss/hello-wasm@sha256:f04ed4a5e96eeadb8fe1146f9376f6b78e9bc7ad15a1f422bab69a10f459134d": "sha256:f40da01ab0076f721c4be69a33b3d577f2cef2bad71da6aac163e8aa0b46e737"
//     }
#[derive(Serialize, Deserialize, Default)]
pub struct Repositories {
    #[serde(default)]
    repositories: HashMap<String, HashMap<String, String>>,
}

impl Repositories {
    /// 初始本地仓库信息（从本地读取信息文件）
    pub fn init() -> Result<Self> {
        let repos_path = FileSystem.images_json()?;
        Ok(std::fs::read(&repos_path)
            .and_then(
                |x| match serde_json::from_slice::<Repositories>(x.as_slice()) {
                    Ok(ins) => Ok(ins),
                    Err(e) => {
                        warn!("初始化images.json异常：{:?}", e);
                        Err(e.into())
                    }
                },
            )
            .unwrap_or(Repositories::default()))
    }

    /// 获取本地镜像的digest
    pub fn image_digest(&self, image: &Reference) -> Option<&String> {
        let full_name = full_name(image);
        let whole_name = image.whole();
        if let Some(repo) = self.repositories.get(&full_name) {
            repo.get(&whole_name)
        } else {
            None
        }
    }
    /// 更新镜像信息
    pub fn update(&mut self, image: &Reference, digest: String) {
        let full_name = full_name(image);
        let whole_name = image.whole();

        if let Some(repo) = self.repositories.get_mut(&full_name) {
            repo.insert(whole_name, digest);
        } else {
            let mut repo = HashMap::new();
            repo.insert(whole_name, digest);
            self.repositories.insert(full_name, repo);
        }
    }
    /// 更新镜像信息、并保存至本地
    pub fn update_and_save(&mut self, image: &Reference, digest: String) -> Result<()> {
        self.update(image, digest);
        self.save()
    }
    /// 保存至本地
    pub fn save(&self) -> Result<()> {
        let repos_path = FileSystem.images_json()?;
        std::fs::write(&repos_path, serde_json::to_vec(&self)?)?;
        Ok(())
    }
}

fn full_name(image: &Reference) -> String {
    if image.registry() == "" {
        image.repository().to_string()
    } else {
        format!("{}/{}", image.registry(), image.repository())
    }
}