Line data Source code
1 : use std::collections::HashMap;
2 :
3 : const BLOCK_SIZE: usize = 8192;
4 :
5 : /// A simple in-memory implementation of a block storage. Can be used to implement external
6 : /// storage in tests.
7 : pub struct BlockStorage {
8 : blocks: HashMap<u64, [u8; BLOCK_SIZE]>,
9 : }
10 :
11 : impl Default for BlockStorage {
12 0 : fn default() -> Self {
13 0 : Self::new()
14 0 : }
15 : }
16 :
17 : impl BlockStorage {
18 85192 : pub fn new() -> Self {
19 85192 : BlockStorage {
20 85192 : blocks: HashMap::new(),
21 85192 : }
22 85192 : }
23 :
24 31604 : pub fn read(&self, pos: u64, buf: &mut [u8]) {
25 31604 : let mut buf_offset = 0;
26 31604 : let mut storage_pos = pos;
27 47129 : while buf_offset < buf.len() {
28 15525 : let block_id = storage_pos / BLOCK_SIZE as u64;
29 15525 : let block = self.blocks.get(&block_id).unwrap_or(&[0; BLOCK_SIZE]);
30 15525 : let block_offset = storage_pos % BLOCK_SIZE as u64;
31 15525 : let block_len = BLOCK_SIZE as u64 - block_offset;
32 15525 : let buf_len = buf.len() - buf_offset;
33 15525 : let copy_len = std::cmp::min(block_len as usize, buf_len);
34 15525 : buf[buf_offset..buf_offset + copy_len]
35 15525 : .copy_from_slice(&block[block_offset as usize..block_offset as usize + copy_len]);
36 15525 : buf_offset += copy_len;
37 15525 : storage_pos += copy_len as u64;
38 15525 : }
39 31604 : }
40 :
41 134001 : pub fn write(&mut self, pos: u64, buf: &[u8]) {
42 134001 : let mut buf_offset = 0;
43 134001 : let mut storage_pos = pos;
44 268377 : while buf_offset < buf.len() {
45 134376 : let block_id = storage_pos / BLOCK_SIZE as u64;
46 134376 : let block = self.blocks.entry(block_id).or_insert([0; BLOCK_SIZE]);
47 134376 : let block_offset = storage_pos % BLOCK_SIZE as u64;
48 134376 : let block_len = BLOCK_SIZE as u64 - block_offset;
49 134376 : let buf_len = buf.len() - buf_offset;
50 134376 : let copy_len = std::cmp::min(block_len as usize, buf_len);
51 134376 : block[block_offset as usize..block_offset as usize + copy_len]
52 134376 : .copy_from_slice(&buf[buf_offset..buf_offset + copy_len]);
53 134376 : buf_offset += copy_len;
54 134376 : storage_pos += copy_len as u64
55 : }
56 134001 : }
57 : }
|