Function handle_getset

nodedb/src/control/server/resp/handler_kv.rs:389–422

GETSET key value — atomically set new value and return old.

(
    cmd: &RespCommand,
    session: &RespSession,
    state: &SharedState,
)

/// GETSET key value — atomically set new value and return old.
pub(super) async fn handle_getset(
    cmd: &RespCommand,
    session: &RespSession,
    state: &SharedState,
) -> RespValue {
    if cmd.argc() < 2 {
        return RespValue::err("ERR wrong number of arguments for 'getset' command");
    }

    let key = cmd.args[0].clone();
    let new_value = cmd.args[1].clone();

    let plan = PhysicalPlan::Kv(KvOp::GetSet {
        collection: session.collection.clone(),
        key,
        new_value,
    });

    match dispatch_kv_write(state, session, plan).await {
        Ok(resp) => {
            let payload_text =
                crate::data::executor::response_codec::decode_payload_to_json(&resp.payload);
            if let Ok(json) = sonic_rs::from_str::<serde_json::Value>(&payload_text)
                && let Some(serde_json::Value::String(b64)) = json.get("old_value")
                && let Ok(data) =
                    base64::Engine::decode(&base64::engine::general_purpose::STANDARD, b64)
            {
                return RespValue::bulk(data);
            }
            RespValue::nil()
        }
        Err(e) => RespValue::err(format!("ERR {e}")),
    }
}