Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Browse files
Browse the repository at this point in the history
Initial attempt at flushing out how to define collections and views
- Loading branch information
0 parents
commit 43bd3a2
Showing
4 changed files
with
263 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,2 @@ | ||
/target | ||
Cargo.lock |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,14 @@ | ||
[package] | ||
name = "pliantdb" | ||
version = "0.1.0" | ||
authors = ["Jonathan Johnson <jon@khonsulabs.com>"] | ||
edition = "2018" | ||
|
||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html | ||
|
||
[dependencies] | ||
uuid = { version = "0.8", features = ["v4", "serde"] } | ||
serde = { version = "1", features = ["derive"] } | ||
serde_cbor = "0.11" | ||
thiserror = "1" | ||
anyhow = "1" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,53 @@ | ||
use std::borrow::Cow; | ||
|
||
use pliantdb::{CollectionViews, DatabaseCollections, MapResult}; | ||
use serde::{Deserialize, Serialize}; | ||
|
||
struct Basic; | ||
|
||
impl pliantdb::Database for Basic { | ||
fn add_collections(collections: &mut DatabaseCollections) { | ||
collections.push(TodoCollection); | ||
} | ||
} | ||
|
||
#[derive(Serialize, Deserialize)] | ||
struct Todo<'a> { | ||
pub completed: bool, | ||
pub task: &'a str, | ||
} | ||
|
||
struct TodoCollection; | ||
|
||
impl pliantdb::Collection for TodoCollection { | ||
fn name(&self) -> Cow<'static, str> { | ||
Cow::from("todos") | ||
} | ||
|
||
fn add_views(&self, views: &mut CollectionViews<Self>) { | ||
views.push(IncompleteTodos); | ||
} | ||
} | ||
|
||
struct IncompleteTodos; | ||
|
||
impl pliantdb::View<TodoCollection> for IncompleteTodos { | ||
type MapKey = (); | ||
type MapValue = (); | ||
type Reduce = (); | ||
|
||
fn name(&self) -> Cow<'static, str> { | ||
Cow::from("uncompleted-todos") | ||
} | ||
|
||
fn map<'d>(&self, document: &'d pliantdb::Document<TodoCollection>) -> MapResult<'d> { | ||
let todo: Todo<'d> = document.contents::<Todo>()?; | ||
if todo.completed { | ||
Ok(Some(document.emit_nothing())) | ||
} else { | ||
Ok(None) | ||
} | ||
} | ||
} | ||
|
||
fn main() {} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,194 @@ | ||
use std::borrow::Cow; | ||
|
||
use serde::{Deserialize, Serialize}; | ||
use uuid::Uuid; | ||
|
||
pub struct Document<C> { | ||
pub id: Uuid, | ||
pub revision: Revision, | ||
pub collection: Option<C>, | ||
pub contents: Vec<u8>, | ||
} | ||
|
||
impl<C> Document<C> { | ||
pub fn contents<'a, D: Deserialize<'a>>(&'a self) -> Result<D, serde_cbor::Error> { | ||
serde_cbor::from_slice(&self.contents) | ||
} | ||
|
||
pub fn set_contents<S: Serialize>(&mut self, contents: &S) -> Result<(), serde_cbor::Error> { | ||
self.contents = serde_cbor::to_vec(contents)?; | ||
Ok(()) | ||
} | ||
|
||
pub fn emit_nothing(&self) -> Map<'static, (), ()> { | ||
self.emit_with((), ()) | ||
} | ||
|
||
pub fn emit<'a, K: Into<SerializableValue<'a, Key>>, Key: Serialize>( | ||
&self, | ||
key: K, | ||
) -> Map<'a, Key, ()> { | ||
self.emit_with(key, ()) | ||
} | ||
|
||
pub fn emit_with< | ||
'a, | ||
K: Into<SerializableValue<'a, Key>>, | ||
V: Into<SerializableValue<'a, Value>>, | ||
Key: Serialize, | ||
Value: Serialize, | ||
>( | ||
&self, | ||
key: K, | ||
value: V, | ||
) -> Map<'a, Key, Value> { | ||
Map { | ||
source: self.id, | ||
key: key.into(), | ||
value: value.into(), | ||
} | ||
} | ||
} | ||
|
||
pub struct Revision { | ||
pub id: usize, | ||
pub sha256: [u8; 32], | ||
} | ||
|
||
pub trait Collection { | ||
fn name(&self) -> Cow<'static, str>; | ||
fn add_views(&self, views: &mut CollectionViews<Self>) | ||
where | ||
Self: Sized; | ||
|
||
fn boxed(self) -> Box<dyn Collection> | ||
where | ||
Self: Sized + 'static, | ||
{ | ||
Box::new(self) | ||
} | ||
} | ||
|
||
#[derive(thiserror::Error, Debug)] | ||
pub enum ViewError { | ||
#[error("error deserializing document {0}")] | ||
SerializationError(#[from] serde_cbor::Error), | ||
#[error("reduce is unimplemented")] | ||
ReduceUnimplemented, | ||
} | ||
|
||
pub type MapResult<'d, K = (), V = ()> = Result<Option<Map<'d, K, V>>, ViewError>; | ||
|
||
pub trait View<C> { | ||
type MapKey: Serialize + for<'de> Deserialize<'de>; | ||
type MapValue: Serialize + for<'de> Deserialize<'de>; | ||
type Reduce: for<'de> Deserialize<'de>; | ||
|
||
fn name(&self) -> Cow<'static, str>; | ||
|
||
fn map<'d>(&self, document: &'d Document<C>) -> MapResult<'d, Self::MapKey, Self::MapValue>; | ||
|
||
#[allow(unused_variables)] | ||
fn reduce( | ||
&self, | ||
mappings: &[Map<'_, Self::MapKey, Self::MapValue>], | ||
rereduce: bool, | ||
) -> Result<Self::Reduce, ViewError> { | ||
Err(ViewError::ReduceUnimplemented) | ||
} | ||
|
||
fn boxed(self) -> Box<dyn AnyView<C>> | ||
where | ||
Self: Sized + 'static, | ||
{ | ||
Box::new(self) | ||
} | ||
} | ||
|
||
pub enum SerializableValue<'a, T: Serialize> { | ||
Owned(T), | ||
Borrowed(&'a T), | ||
} | ||
|
||
impl<'a> Into<SerializableValue<'a, ()>> for () { | ||
fn into(self) -> SerializableValue<'a, Self> { | ||
SerializableValue::Owned(()) | ||
} | ||
} | ||
|
||
impl<'a, T> AsRef<T> for SerializableValue<'a, T> | ||
where | ||
T: Serialize, | ||
{ | ||
fn as_ref(&self) -> &T { | ||
match self { | ||
Self::Owned(value) => value, | ||
Self::Borrowed(value) => value, | ||
} | ||
} | ||
} | ||
|
||
pub struct Map<'a, K: Serialize = (), V: Serialize = ()> { | ||
pub source: Uuid, | ||
pub key: SerializableValue<'a, K>, | ||
pub value: SerializableValue<'a, V>, | ||
} | ||
|
||
pub struct SerializedMap { | ||
pub source: Uuid, | ||
pub key: serde_cbor::Value, | ||
pub value: serde_cbor::Value, | ||
} | ||
|
||
pub trait AnyView<C> { | ||
fn name(&self) -> Cow<'static, str>; | ||
fn map(&self, document: &Document<C>) -> Result<Option<SerializedMap>, ViewError>; | ||
} | ||
|
||
impl<C, T> AnyView<C> for T | ||
where | ||
T: View<C>, | ||
{ | ||
fn name(&self) -> Cow<'static, str> { | ||
View::<C>::name(self) | ||
} | ||
|
||
fn map(&self, document: &Document<C>) -> Result<Option<SerializedMap>, ViewError> { | ||
let map = View::<C>::map(self, document)?; | ||
|
||
match map { | ||
Some(map) => Ok(Some(SerializedMap { | ||
source: map.source, | ||
key: serde_cbor::value::to_value(map.key.as_ref())?, | ||
value: serde_cbor::value::to_value(map.value.as_ref())?, | ||
})), | ||
None => Ok(None), | ||
} | ||
} | ||
} | ||
|
||
pub trait Database { | ||
fn add_collections(collections: &mut DatabaseCollections); | ||
} | ||
|
||
#[derive(Default)] | ||
pub struct DatabaseCollections { | ||
collections: Vec<Box<dyn Collection>>, | ||
} | ||
|
||
impl DatabaseCollections { | ||
pub fn push<C: Collection + 'static>(&mut self, collection: C) { | ||
self.collections.push(collection.boxed()); | ||
} | ||
} | ||
|
||
#[derive(Default)] | ||
pub struct CollectionViews<C> { | ||
views: Vec<Box<dyn AnyView<C>>>, | ||
} | ||
|
||
impl<C> CollectionViews<C> { | ||
pub fn push<V: View<C> + 'static>(&mut self, view: V) { | ||
self.views.push(view.boxed()); | ||
} | ||
} |