shardmap 0.2.1

Sharded embedded in-memory map with optional cache, protocol, and server internals
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

#[cfg(feature = "server")]
use bytes::BytesMut;

use crate::commands::redis::{array_bulk, define_redis_command, filter_key_pattern, wrong_arity};
#[cfg(feature = "server")]
use crate::commands::redis::{key_pattern_matches, write_frame, write_resp_array_header};
use crate::protocol::Frame;
#[cfg(feature = "server")]
use crate::server::wire::ServerWire;
use crate::storage::EmbeddedStore;

define_redis_command!(Keys, "KEYS", false);

impl crate::commands::redis::RedisCommand for Keys {
    fn execute(store: &EmbeddedStore, args: &[&[u8]]) -> Frame {
        match args {
            [pattern] => array_bulk(filter_key_pattern(store.key_snapshot_unsorted(), pattern)),
            _ => wrong_arity("KEYS"),
        }
    }

    #[cfg(feature = "server")]
    fn write_resp(store: &EmbeddedStore, args: &[&[u8]], out: &mut BytesMut) {
        match args {
            [pattern] => write_keys_resp(store, pattern, out),
            _ => write_frame(out, &wrong_arity("KEYS")),
        }
    }
}

#[cfg(feature = "server")]
fn write_keys_resp(store: &EmbeddedStore, pattern: &[u8], out: &mut BytesMut) {
    let mut items = BytesMut::new();
    let mut count = 0usize;
    store.visit_redis_keys(&mut |key| {
        if key_pattern_matches(key, pattern) {
            ServerWire::write_resp_blob_string(&mut items, key);
            count += 1;
        }
        true
    });

    write_resp_array_header(out, count);
    out.extend_from_slice(&items);
}