cedarling 0.0.49

The Cedarling: a high-performance local authorization service powered by the Rust Cedar Engine.
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

// This software is available under the Apache-2.0 license.
// See https://www.apache.org/licenses/LICENSE-2.0.txt for full text.
//
// Copyright (c) 2024, Gluu, Inc.

use serde_json::Value;
use std::mem::size_of;

pub(super) fn calculate_memory_usage(value: &Value) -> usize {
    let mut total = size_of::<Value>(); // Root stack size (32 bytes)
    add_heap_usage(value, &mut total);
    total
}

fn add_heap_usage(value: &Value, total: &mut usize) {
    match value {
        Value::String(s) => {
            // Use capacity for String (includes unused allocated space)
            *total += size_of::<String>();
            *total += s.capacity();
        },
        Value::Array(arr) => {
            // Use capacity for Vec (includes unused allocated space)
            *total += size_of::<Vec<Value>>() + // Vec header (24 bytes)
                    arr.capacity() * size_of::<Value>(); // Elements

            for item in arr {
                add_heap_usage(item, total);
            }
        },
        Value::Object(map) => {
            // BTreeMap: capacity is not accessible, so use length
            for (key, value) in map {
                // Use capacity for String keys
                *total += size_of::<String>() + // String header (24 bytes)
                        key.capacity(); // Key data

                // Value storage in map node
                *total += size_of::<Value>(); // Value in node

                add_heap_usage(value, total);
            }
        },
        // next values are allocated on the stack and size taken into account
        Value::Number(_) | Value::Null | Value::Bool(_) => {},
    }
}