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
46
47
48
49
50
51
52
53
54
55
56
57

use bytes::BytesMut;

use crate::commands::redis::{
    define_redis_command, int, simple, write_frame, write_resp_simple_string, wrong_arity,
    wrongtype,
};
use crate::protocol::Frame;
#[cfg(feature = "server")]
use crate::server::wire::ServerWire;
use crate::storage::{EmbeddedStore, RedisObjectError};

define_redis_command!(Rename, "RENAME", true);

impl crate::commands::redis::RedisCommand for Rename {
    fn execute(store: &EmbeddedStore, args: &[&[u8]]) -> Frame {
        execute_rename(store, args, false)
    }

    #[cfg(feature = "server")]
    fn write_resp(store: &EmbeddedStore, args: &[&[u8]], out: &mut BytesMut) {
        write_rename_resp(store, args, false, out);
    }
}

pub(crate) fn execute_rename(store: &EmbeddedStore, args: &[&[u8]], nx: bool) -> Frame {
    match args {
        [source, dest] => match store.rename_key(source, dest, nx) {
            Ok(true) if nx => int(1),
            Ok(false) if nx => int(0),
            Ok(_) => simple("OK"),
            Err(RedisObjectError::MissingKey) => crate::commands::redis::error("ERR no such key"),
            Err(RedisObjectError::WrongType) => wrongtype(),
        },
        _ => wrong_arity(if nx { "RENAMENX" } else { "RENAME" }),
    }
}

#[cfg(feature = "server")]
pub(crate) fn write_rename_resp(
    store: &EmbeddedStore,
    args: &[&[u8]],
    nx: bool,
    out: &mut BytesMut,
) {
    match args {
        [source, dest] => match store.rename_key(source, dest, nx) {
            Ok(true) if nx => ServerWire::write_resp_integer(out, 1),
            Ok(false) if nx => ServerWire::write_resp_integer(out, 0),
            Ok(_) => write_resp_simple_string(out, "OK"),
            Err(RedisObjectError::MissingKey) => {
                ServerWire::write_resp_error(out, "ERR no such key")
            }
            Err(RedisObjectError::WrongType) => write_frame(out, &wrongtype()),
        },
        _ => write_frame(out, &wrong_arity(if nx { "RENAMENX" } else { "RENAME" })),
    }
}