use api::v1::meta::ResponseHeader;
use crate::error::{IllegalServerStateSnafu, Result};
pub fn get_prefix_end_key(key: &[u8]) -> Vec<u8> {
for (i, v) in key.iter().enumerate().rev() {
if *v < 0xFF {
let mut end = Vec::from(&key[..=i]);
end[i] = *v + 1;
return end;
}
}
vec![0]
}
pub fn check_response_header(header: Option<&ResponseHeader>) -> Result<()> {
if let Some(header) = header {
if let Some(error) = &header.error {
let code = error.code;
let err_msg = &error.err_msg;
return IllegalServerStateSnafu { code, err_msg }.fail();
}
}
Ok(())
}
pub fn get_next_prefix_key(key: &[u8]) -> Vec<u8> {
let mut next = Vec::with_capacity(key.len() + 1);
next.extend_from_slice(key);
next.push(0);
next
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_get_next_prefix() {
let key = b"testa";
let mut expected = b"testa".to_vec();
expected.push(0);
assert_eq!(expected, get_next_prefix_key(key));
}
#[test]
fn test_get_prefix() {
let key = b"testa";
assert_eq!(b"testb".to_vec(), get_prefix_end_key(key));
let key = vec![0, 0, 26];
assert_eq!(vec![0, 0, 27], get_prefix_end_key(&key));
let key = vec![0, 0, 255];
assert_eq!(vec![0, 1], get_prefix_end_key(&key));
let key = vec![0, 255, 255];
assert_eq!(vec![1], get_prefix_end_key(&key));
}
}