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