harmos-storage 0.1.6

Harmos storage: durable journals, snapshots and Arrow-backed stream storage.
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

/* ***
 *
 * Harmos
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * ***
 */

/* === Modules === */

/* === Re-Exports === */

/* === Imports === */

// Standard Library Imports
use std::num::NonZeroUsize;

// Third Party Imports
use sha2::{Digest, Sha256};

// Local Imports
use super::admissions::Mutation;
use super::objects::invalid;
use super::{ObjectReceipt, ObjectValue, StorageManager};
use crate::Result;
use crate::storage::MutationId;

/* === Constants === */

/* === Macros === */

/* === Type Aliases === */

/* === Structs === */

/* === Enums === */

/* === Traits === */

/* === Implementations === */

impl<C> StorageManager<C> {
    #[tracing::instrument(name = "storage.object_admissions.write_objects_once", level = "debug", skip_all)]
    pub async fn write_objects_once(&self, mutation: &MutationId, objects: Vec<ObjectValue>) -> Result<Vec<ObjectReceipt>> {
        let count = NonZeroUsize::new(objects.len()).ok_or_else(|| invalid("object batch must not be empty"))?;
        let mut hash = Sha256::new();
        part(&mut hash, b"storage.write_objects:v1");
        hash.update((objects.len() as u64).to_le_bytes());
        for object in &objects {
            part(&mut hash, object.schema.id.as_str().as_bytes());
            hash.update(object.schema.version.get().to_le_bytes());
            part(&mut hash, object.item.as_str().as_bytes());
            part(&mut hash, &object.payload);
        }
        self.admit_object_group(
            self.object_admission_bytes(),
            count,
            Some(Mutation { id: mutation, fingerprint: hash.finalize().into() }),
            |group| {
                for object in objects {
                    group.write_value(object)?;
                }
                Ok(())
            },
        )
        .await
    }
}

/* === Public Functions === */

/* === Private Functions === */

fn part(hash: &mut Sha256, bytes: &[u8]) {
    hash.update((bytes.len() as u64).to_le_bytes());
    hash.update(bytes);
}

/* === Tests === */