memscope-rs 0.2.0

A memory tracking library for Rust applications.
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
85
86
87
88
89
90
91
92

// Test reallocation handling in snapshot engine
use memscope_rs::event_store::{EventStore, MemoryEvent, MemoryEventType, SharedEventStore};
use memscope_rs::snapshot::SnapshotEngine;
use std::sync::Arc;

#[test]
fn test_reallocation_updates_thread_stats() {
    let event_store = EventStore::new();
    let thread_id = 1;

    // Initial allocation
    event_store.record(MemoryEvent {
        ptr: 0x1000,
        size: 1000,
        old_size: None,
        event_type: MemoryEventType::Allocate,
        timestamp: 100,
        thread_id,
        var_name: None,
        type_name: Some("Vec<i32>".to_string()),
        call_stack_hash: None,
        thread_name: None,
    });

    // Reallocation - increase size
    event_store.record(MemoryEvent {
        ptr: 0x1000,
        size: 2000,
        old_size: Some(1000),
        event_type: MemoryEventType::Reallocate,
        timestamp: 200,
        thread_id,
        var_name: None,
        type_name: Some("Vec<i32>".to_string()),
        call_stack_hash: None,
        thread_name: None,
    });

    // Build snapshot from events
    let shared_store: SharedEventStore = Arc::new(event_store);
    let snapshot = SnapshotEngine::new(shared_store).build_snapshot();

    // Check reallocation stats
    let thread_stats = snapshot.thread_stats.get(&thread_id).unwrap();
    assert_eq!(thread_stats.total_allocated, 3000); // 1000 + 2000
    assert_eq!(thread_stats.total_deallocated, 1000); // Old size from reallocation
    assert_eq!(thread_stats.current_memory, 2000); // Final size
}

#[test]
fn test_deallocation_updates_thread_stats() {
    let event_store = EventStore::new();
    let thread_id = 1;

    // Allocate
    event_store.record(MemoryEvent {
        ptr: 0x1000,
        size: 1000,
        old_size: None,
        event_type: MemoryEventType::Allocate,
        timestamp: 100,
        thread_id,
        var_name: None,
        type_name: Some("Vec<i32>".to_string()),
        call_stack_hash: None,
        thread_name: None,
    });

    // Deallocate
    event_store.record(MemoryEvent {
        ptr: 0x1000,
        size: 1000,
        old_size: None,
        event_type: MemoryEventType::Deallocate,
        timestamp: 200,
        thread_id,
        var_name: None,
        type_name: Some("Vec<i32>".to_string()),
        call_stack_hash: None,
        thread_name: None,
    });

    // Build snapshot from events
    let shared_store: SharedEventStore = Arc::new(event_store);
    let snapshot = SnapshotEngine::new(shared_store).build_snapshot();

    // Check stats
    let thread_stats = snapshot.thread_stats.get(&thread_id).unwrap();
    assert_eq!(thread_stats.total_allocated, 1000);
    assert_eq!(thread_stats.total_deallocated, 1000);
    assert_eq!(thread_stats.current_memory, 0);
}