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

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

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

bitflags! {
    pub struct PrepareFlags: u32 {
        /// The prepare request contains explicit keyspace.
        const WITH_KEYSPACE = 0x01;
    }
}

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

impl Serialize for PrepareFlags {
    #[inline]
    fn serialize(&self, cursor: &mut Cursor<&mut Vec<u8>>, version: Version) {
        self.bits().serialize(cursor, version);
    }
}

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