harmos_storage/
object_admissions.rs

1/* ***
2 *
3 * Harmos
4 *
5 * SPDX-License-Identifier: Apache-2.0
6 *
7 * ***
8 */
9
10/* === Modules === */
11
12/* === Re-Exports === */
13
14/* === Imports === */
15
16// Standard Library Imports
17use std::num::NonZeroUsize;
18
19// Third Party Imports
20use sha2::{Digest, Sha256};
21
22// Local Imports
23use super::admissions::Mutation;
24use super::objects::invalid;
25use super::{ObjectReceipt, ObjectValue, StorageManager};
26use crate::Result;
27use crate::storage::MutationId;
28
29/* === Constants === */
30
31/* === Macros === */
32
33/* === Type Aliases === */
34
35/* === Structs === */
36
37/* === Enums === */
38
39/* === Traits === */
40
41/* === Implementations === */
42
43impl<C> StorageManager<C> {
44    pub async fn write_objects_once(&self, mutation: &MutationId, objects: &[ObjectValue]) -> Result<Vec<ObjectReceipt>> {
45        let count = NonZeroUsize::new(objects.len()).ok_or_else(|| invalid("object batch must not be empty"))?;
46        let mut hash = Sha256::new();
47        part(&mut hash, b"storage.write_objects:v1");
48        hash.update((objects.len() as u64).to_le_bytes());
49        for object in objects {
50            part(&mut hash, object.schema.id.as_str().as_bytes());
51            hash.update(object.schema.version.get().to_le_bytes());
52            part(&mut hash, object.item.as_str().as_bytes());
53            part(&mut hash, &object.payload);
54        }
55        self.admit_object_group(
56            self.object_admission_bytes(),
57            count,
58            Some(Mutation { id: mutation, fingerprint: hash.finalize().into() }),
59            |group| {
60                for object in objects {
61                    group.write_value(object)?;
62                }
63                Ok(())
64            },
65        )
66        .await
67    }
68}
69
70/* === Public Functions === */
71
72/* === Private Functions === */
73
74fn part(hash: &mut Sha256, bytes: &[u8]) {
75    hash.update((bytes.len() as u64).to_le_bytes());
76    hash.update(bytes);
77}
78
79/* === Tests === */
harmos_storage/object_admissions.rs

harmos_storage/
object_admissions.rs
