bench/
bench.rs

1use customizable_buddy::{AvlBuddy, BuddyAllocator, BuddyError, UsizeBuddy};
2use std::{
3    alloc::Layout,
4    ptr::{NonNull, addr_of, addr_of_mut, null_mut},
5    time::Instant,
6};
7
8type Allocator<const N: usize> = BuddyAllocator<N, UsizeBuddy, AvlBuddy>;
9
10#[repr(C, align(4096))]
11struct Page([u8; 4096]);
12
13impl Page {
14    const ZERO: Self = Self([0; 4096]);
15}
16
17/// 256 MiB
18static mut MEMORY: [Page; 65536] = [Page::ZERO; 65536];
19
20fn main() -> Result<(), BuddyError> {
21    // 创建全局分配器
22    let mut allocator = Allocator::<12>::new();
23    // 从本机操作系统获取一块内存给程序
24    let ptr = NonNull::new(addr_of_mut!(MEMORY).cast::<u8>()).unwrap();
25    let len = size_of_val(unsafe { &*addr_of!(MEMORY) });
26    // 使用最小阶数和初始地址初始化程序
27    allocator.init(12, ptr);
28    println!(
29        "MEMORY: {:#x}..{:#x}",
30        ptr.as_ptr() as usize,
31        ptr.as_ptr() as usize + len
32    );
33    // 计时
34    let t = Instant::now();
35    // 将地址空间放入分配器进行分配
36    unsafe { allocator.transfer(ptr, len) };
37    println!("transfer {:?}", t.elapsed());
38
39    assert_eq!(len, allocator.capacity());
40    assert_eq!(len, allocator.free());
41
42    println!(
43        "
44BEFORE
45{allocator:#x?}"
46    );
47
48    // 创建页面大小的指针？
49    let mut blocks = [null_mut::<Page>(); 65536];
50    let layout = Layout::new::<Page>();
51    let t = Instant::now();
52    for block in blocks.iter_mut() {
53        // 将指向对应page大小的地址指定给对应指针(将地址从buddyLine内删除)
54        let (ptr, size) = allocator.allocate_type::<Page>()?;
55        debug_assert_eq!(layout.size(), size);
56        *block = ptr.as_ptr();
57    }
58    let ta = t.elapsed();
59
60    // 由于将等同于分配空间大小的页面全部收回
61    println!(
62        "
63EMPTY
64{allocator:#x?}"
65    );
66
67    // 感觉这个地方应该 有问题, 不应该总容量不变
68    assert_eq!(len, allocator.capacity());
69    assert_eq!(len - blocks.len() * layout.size(), allocator.free());
70
71    let t = Instant::now();
72    for block in blocks.iter_mut() {
73        // 释放指针所指向的空间给分配器进行调配
74        allocator.deallocate(NonNull::new(*block).unwrap(), layout.size());
75        *block = null_mut();
76        // println!("{allocator:#x?}");
77    }
78    let td = t.elapsed();
79
80    assert_eq!(len, allocator.capacity());
81    assert_eq!(len, allocator.free());
82
83    println!(
84        "
85AFTER
86{allocator:#x?}"
87    );
88
89    println!(
90        "allocate   {:?} ({} times)",
91        ta / blocks.len() as u32,
92        blocks.len()
93    );
94    println!(
95        "deallocate {:?} ({} times)",
96        td / blocks.len() as u32,
97        blocks.len()
98    );
99
100    Ok(())
101}
bench/bench.rs

bench/
bench.rs
