shardmap 0.3.0

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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

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

use crate::commands::redis::{
    bulk, define_redis_command, optional_string_value, write_resp_array_header, write_resp_null,
    write_resp_wrong_arity, wrong_arity,
};
use crate::protocol::Frame;
#[cfg(feature = "server")]
use crate::server::wire::ServerWire;
use crate::storage::{EmbeddedStore, RedisStringLookup};

define_redis_command!(MGet, "MGET", false);

impl crate::commands::redis::RedisCommand for MGet {
    fn execute(store: &EmbeddedStore, args: &[&[u8]]) -> Frame {
        if args.is_empty() {
            return wrong_arity("MGET");
        }
        Frame::Array(
            args.iter()
                .map(|key| match optional_string_value(store, key, false) {
                    Ok(Some(value)) => bulk(value),
                    _ => Frame::Null,
                })
                .collect(),
        )
    }

    #[cfg(feature = "server")]
    fn write_resp(store: &EmbeddedStore, args: &[&[u8]], out: &mut BytesMut) {
        if args.is_empty() {
            write_resp_wrong_arity(out, "MGET");
            return;
        }
        write_resp_array_header(out, args.len());
        for key in args {
            match store.get_string_value_into(key, |bytes| {
                ServerWire::write_resp_blob_string(out, bytes);
            }) {
                RedisStringLookup::Hit => {}
                RedisStringLookup::Miss | RedisStringLookup::WrongType => write_resp_null(out),
            }
        }
    }

    #[cfg(feature = "server")]
    fn write_fast(store: &EmbeddedStore, args: &[&[u8]], out: &mut BytesMut) {
        if args.is_empty() {
            ServerWire::write_fast_error(out, "ERR wrong number of arguments for 'mget' command");
            return;
        }
        let start = ServerWire::begin_fast_array(out, args.len());
        for key in args {
            match store.get_string_value_into(key, |bytes| {
                ServerWire::write_fast_array_item(out, Some(bytes));
            }) {
                RedisStringLookup::Hit => {}
                RedisStringLookup::Miss | RedisStringLookup::WrongType => {
                    ServerWire::write_fast_array_item(out, None)
                }
            }
        }
        ServerWire::finish_fast_array(out, start);
    }
}