shardmap 0.3.2

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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

use crate::storage::RedisStringStore;
use bytes::BytesMut;

use crate::commands::redis::{error, parse_i64, write_frame, wrongtype};
use crate::protocol::Frame;
#[cfg(feature = "server")]
use crate::server::wire::ServerWire;
use crate::storage::{EmbeddedStore, RedisStringLookup};

pub(crate) fn append_value(
    store: &EmbeddedStore,
    key: &[u8],
    suffix: &[u8],
) -> std::result::Result<i64, Frame> {
    store.transform_string_value_no_ttl(
        key,
        |existing| {
            let existing = existing.unwrap_or_default();
            let mut value = Vec::with_capacity(existing.len().saturating_add(suffix.len()));
            value.extend_from_slice(existing);
            value.extend_from_slice(suffix);
            Ok((value.len() as i64, value))
        },
        wrongtype,
    )
}

pub(crate) fn strlen_value(store: &EmbeddedStore, key: &[u8]) -> std::result::Result<i64, ()> {
    let mut len = 0_i64;
    match store.get_string_value_into(key, |bytes| len = bytes.len() as i64) {
        RedisStringLookup::Hit => Ok(len),
        RedisStringLookup::Miss => Ok(0),
        RedisStringLookup::WrongType => Err(()),
    }
}

pub(crate) fn incrby_value(
    store: &EmbeddedStore,
    key: &[u8],
    delta: i64,
) -> std::result::Result<i64, Frame> {
    integer_transform_value(store, key, |current| current.checked_add(delta))
}

pub(crate) fn decrby_value(
    store: &EmbeddedStore,
    key: &[u8],
    decrement: i64,
) -> std::result::Result<i64, Frame> {
    integer_transform_value(store, key, |current| current.checked_sub(decrement))
}

fn integer_transform_value(
    store: &EmbeddedStore,
    key: &[u8],
    apply: impl FnOnce(i64) -> Option<i64>,
) -> std::result::Result<i64, Frame> {
    store.transform_string_value_no_ttl(
        key,
        |existing| {
            let current = match existing {
                None | Some(b"") => 0,
                Some(value) => match parse_i64(value) {
                    Ok(value) => value,
                    Err(_) => return Err(error("ERR value is not an integer or out of range")),
                },
            };
            let Some(next) = apply(current) else {
                return Err(error("ERR increment or decrement would overflow"));
            };
            let mut buffer = itoa::Buffer::new();
            Ok((next, buffer.format(next).as_bytes().to_vec()))
        },
        wrongtype,
    )
}

#[cfg(feature = "server")]
pub(crate) fn write_integer_result_resp(
    out: &mut BytesMut,
    result: std::result::Result<i64, Frame>,
) {
    match result {
        Ok(value) => ServerWire::write_resp_integer(out, value),
        Err(frame) => write_frame(out, &frame),
    }
}

#[cfg(not(feature = "server"))]
pub(crate) fn write_integer_result_resp(
    _out: &mut BytesMut,
    _result: std::result::Result<i64, Frame>,
) {
    unreachable!("RESP integer writers are only called by the server feature")
}