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

use bitflags::bitflags;
use std::io::{Cursor, Read};

use crate::error;
use crate::frame::{FromCursor, Serialize, Version};
use crate::types::INT_LEN;

bitflags! {
    #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
    pub struct QueryFlags: u32 {
        /// Indicates that Query Params contain value.
        const VALUE = 0x001;
        /// Indicates that Query Params does not contain metadata.
        const SKIP_METADATA = 0x002;
        /// Indicates that Query Params contain page size.
        const PAGE_SIZE = 0x004;
        /// Indicates that Query Params contain paging state.
        const WITH_PAGING_STATE = 0x008;
        /// Indicates that Query Params contain serial consistency.
        const WITH_SERIAL_CONSISTENCY = 0x010;
        /// Indicates that Query Params contain default timestamp.
        const WITH_DEFAULT_TIMESTAMP = 0x020;
        /// Indicates that Query Params values are named ones.
        const WITH_NAMES_FOR_VALUES = 0x040;
        /// Indicates that Query Params contain keyspace name.
        const WITH_KEYSPACE = 0x080;
        /// Indicates that Query Params contain "now" in seconds.
        const WITH_NOW_IN_SECONDS = 0x100;
    }
}

impl Default for QueryFlags {
    #[inline]
    fn default() -> Self {
        QueryFlags::empty()
    }
}

impl Serialize for QueryFlags {
    fn serialize(&self, cursor: &mut Cursor<&mut Vec<u8>>, version: Version) {
        if version >= Version::V5 {
            self.bits().serialize(cursor, version);
        } else {
            (self.bits() as u8).serialize(cursor, version);
        }
    }
}

impl FromCursor for QueryFlags {
    fn from_cursor(cursor: &mut Cursor<&[u8]>, version: Version) -> error::Result<Self> {
        if version >= Version::V5 {
            let mut buff = [0; INT_LEN];
            cursor
                .read_exact(&mut buff)
                .map(|()| QueryFlags::from_bits_truncate(u32::from_be_bytes(buff)))
                .map_err(|error| error.into())
        } else {
            let mut buff = [0];
            cursor.read_exact(&mut buff)?;
            Ok(QueryFlags::from_bits_truncate(buff[0] as u32))
        }
    }
}