Merge branch 'model/tsymbol' into next
Sayan Nandan
3 years ago
commit
3bd34ae5c9
@ -0,0 +1,179 @@
|
||||
/*
|
||||
* Created on Thu Aug 12 2021
|
||||
*
|
||||
* This file is a part of Skytable
|
||||
* Skytable (formerly known as TerrabaseDB or Skybase) is a free and open-source
|
||||
* NoSQL database written by Sayan Nandan ("the Author") with the
|
||||
* vision to provide flexibility in data modelling without compromising
|
||||
* on performance, queryability or scalability.
|
||||
*
|
||||
* Copyright (c) 2021, Sayan Nandan <ohsayan@outlook.com>
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*
|
||||
*/
|
||||
|
||||
use crate::corestore::buffers::Integer64;
|
||||
use crate::corestore::Data;
|
||||
use crate::dbnet::connection::ProtocolConnectionExt;
|
||||
use crate::protocol::responses::groups;
|
||||
use crate::IoResult;
|
||||
use core::marker::PhantomData;
|
||||
use tokio::io::AsyncReadExt;
|
||||
use tokio::io::AsyncWriteExt;
|
||||
|
||||
/// Write a raw mono group with a custom tsymbol
|
||||
pub async unsafe fn write_raw_mono<T, Strm>(
|
||||
con: &mut T,
|
||||
tsymbol: u8,
|
||||
payload: &Data,
|
||||
) -> IoResult<()>
|
||||
where
|
||||
T: ProtocolConnectionExt<Strm>,
|
||||
Strm: AsyncReadExt + AsyncWriteExt + Unpin + Send + Sync,
|
||||
{
|
||||
let raw_stream = unsafe { con.raw_stream() };
|
||||
raw_stream.write_all(&[tsymbol; 1]).await?; // first write tsymbol
|
||||
let bytes = Integer64::from(payload.len());
|
||||
raw_stream.write_all(&bytes).await?; // then len
|
||||
raw_stream.write_all(&[b'\n']).await?; // LF
|
||||
raw_stream.write_all(payload).await?; // payload
|
||||
raw_stream.write_all(&[b'\n']).await?; // final LF
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
/// A writer for a flat array, which is a multi-typed non-recursive array
|
||||
pub struct FlatArrayWriter<'a, T, Strm> {
|
||||
tsymbol: u8,
|
||||
con: &'a mut T,
|
||||
_owned: PhantomData<Strm>,
|
||||
}
|
||||
|
||||
impl<'a, T, Strm> FlatArrayWriter<'a, T, Strm>
|
||||
where
|
||||
T: ProtocolConnectionExt<Strm>,
|
||||
Strm: AsyncReadExt + AsyncWriteExt + Unpin + Send + Sync,
|
||||
{
|
||||
/// Intialize a new flat array writer. This will write out the tsymbol
|
||||
/// and length for the flat array
|
||||
pub async unsafe fn new(
|
||||
con: &'a mut T,
|
||||
tsymbol: u8,
|
||||
len: usize,
|
||||
) -> IoResult<FlatArrayWriter<'a, T, Strm>> {
|
||||
{
|
||||
let stream = unsafe { con.raw_stream() };
|
||||
// first write _
|
||||
stream.write_all(&[b'_']).await?;
|
||||
let bytes = Integer64::from(len);
|
||||
// now write len
|
||||
stream.write_all(&bytes).await?;
|
||||
// first LF
|
||||
stream.write_all(&[b'\n']).await?;
|
||||
}
|
||||
Ok(Self {
|
||||
con,
|
||||
tsymbol,
|
||||
_owned: PhantomData,
|
||||
})
|
||||
}
|
||||
/// Write an element
|
||||
pub async fn write_element(&mut self, bytes: impl AsRef<[u8]>) -> IoResult<()> {
|
||||
let stream = unsafe { self.con.raw_stream() };
|
||||
let bytes = bytes.as_ref();
|
||||
// first write <tsymbol>
|
||||
stream.write_all(&[self.tsymbol]).await?;
|
||||
// now len
|
||||
let len = Integer64::from(bytes.len());
|
||||
stream.write_all(&len).await?;
|
||||
// now LF
|
||||
stream.write_all(&[b'\n']).await?;
|
||||
// now element
|
||||
stream.write_all(bytes).await?;
|
||||
// now final LF
|
||||
stream.write_all(&[b'\n']).await?;
|
||||
Ok(())
|
||||
}
|
||||
/// Write the NIL response code
|
||||
pub async fn write_nil(&mut self) -> IoResult<()> {
|
||||
let stream = unsafe { self.con.raw_stream() };
|
||||
stream.write_all(groups::NIL).await?;
|
||||
Ok(())
|
||||
}
|
||||
/// Write the SERVER_ERR (5) response code
|
||||
pub async fn write_server_error(&mut self) -> IoResult<()> {
|
||||
let stream = unsafe { self.con.raw_stream() };
|
||||
stream.write_all(groups::NIL).await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
/// A writer for a typed array, which is a singly-typed array which either
|
||||
/// has a typed element or a `NULL`
|
||||
pub struct TypedArrayWriter<'a, T, Strm> {
|
||||
con: &'a mut T,
|
||||
_owned: PhantomData<Strm>,
|
||||
}
|
||||
|
||||
impl<'a, T, Strm> TypedArrayWriter<'a, T, Strm>
|
||||
where
|
||||
T: ProtocolConnectionExt<Strm>,
|
||||
Strm: AsyncReadExt + AsyncWriteExt + Unpin + Send + Sync,
|
||||
{
|
||||
/// Create a new `typedarraywriter`. This will write the tsymbol and
|
||||
/// the array length
|
||||
pub async unsafe fn new(
|
||||
con: &'a mut T,
|
||||
tsymbol: u8,
|
||||
len: usize,
|
||||
) -> IoResult<TypedArrayWriter<'a, T, Strm>> {
|
||||
{
|
||||
let stream = unsafe { con.raw_stream() };
|
||||
// first write @<tsymbol>
|
||||
stream.write_all(&[b'@', tsymbol]).await?;
|
||||
let bytes = Integer64::from(len);
|
||||
// now write len
|
||||
stream.write_all(&bytes).await?;
|
||||
// first LF
|
||||
stream.write_all(&[b'\n']).await?;
|
||||
}
|
||||
Ok(Self {
|
||||
con,
|
||||
_owned: PhantomData,
|
||||
})
|
||||
}
|
||||
/// Write an element
|
||||
pub async fn write_element(&mut self, bytes: impl AsRef<[u8]>) -> IoResult<()> {
|
||||
let stream = unsafe { self.con.raw_stream() };
|
||||
let bytes = bytes.as_ref();
|
||||
// write len
|
||||
let len = Integer64::from(bytes.len());
|
||||
stream.write_all(&len).await?;
|
||||
// now LF
|
||||
stream.write_all(&[b'\n']).await?;
|
||||
// now element
|
||||
stream.write_all(bytes).await?;
|
||||
// now final LF
|
||||
stream.write_all(&[b'\n']).await?;
|
||||
Ok(())
|
||||
}
|
||||
/// Write a null
|
||||
pub async fn write_null(&mut self) -> IoResult<()> {
|
||||
let stream = unsafe { self.con.raw_stream() };
|
||||
stream.write_all(&[b'\0', b'\n']).await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue