kevy 6.4.0

kevy — a pure-Rust, zero-dependency, Redis-compatible KV server.
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

//! `XSETID key last-id [ENTRIESADDED n] [MAXDELETEDID id]` — overwrite a
//! stream's scalar state (Redis 7 shape). The AOF rewrite leans on this
//! to restore `last_id` / `entries_added` / `max_deleted_id` exactly when
//! a bare XADD replay wouldn't (deleted tail, deleted-only stream).

use kevy_resp::CmdError;
use kevy_resp::{ArgvView, encode_error, encode_simple_string};
use kevy_store::{Store, StreamId, parse_explicit_id};

use crate::cmd::{store_err, wrong_args};

pub(super) fn cmd_xsetid<A: ArgvView + ?Sized>(store: &mut Store, args: &A, out: &mut Vec<u8>) {
    if !matches!(args.len(), 3 | 5 | 7) {
        return wrong_args(out, "xsetid");
    }
    let last_id = match parse_id(&args[2]) {
        Ok(id) => id,
        Err(msg) => return encode_error(out, msg.as_wire()),
    };
    let mut entries_added: Option<u64> = None;
    let mut max_deleted_id: Option<StreamId> = None;
    let mut i = 3;
    while i < args.len() {
        let tok = args[i].to_ascii_uppercase();
        match tok.as_slice() {
            b"ENTRIESADDED" => {
                let n = std::str::from_utf8(&args[i + 1]).ok().and_then(|s| s.parse().ok());
                let Some(n) = n else {
                    return encode_error(out, "ERR value is not an integer or out of range");
                };
                entries_added = Some(n);
            }
            b"MAXDELETEDID" => {
                max_deleted_id = match parse_id(&args[i + 1]) {
                    Ok(id) => Some(id),
                    Err(msg) => return encode_error(out, msg.as_wire()),
                };
            }
            _ => return encode_error(out, "ERR syntax error"),
        }
        i += 2;
    }
    match store.xsetid(&args[1], last_id, entries_added, max_deleted_id) {
        Ok(()) => encode_simple_string(out, "OK"),
        Err(kevy_store::StoreError::NoSuchKey) => {
            encode_error(out, "ERR The XSETID command requires the key to exist.")
        }
        Err(kevy_store::StoreError::OutOfRange) => encode_error(
            out,
            "ERR The ID specified in XSETID is smaller than the target stream top item",
        ),
        Err(e) => store_err(out, e),
    }
}

fn parse_id(s: &[u8]) -> Result<StreamId, CmdError> {
    parse_explicit_id(s, /*end=*/ false)
        .map_err(|_| CmdError::Wire("ERR Invalid stream ID specified as stream command argument"))
}