Tpt
4 years ago
parent
fcb3a33606
commit
5038d95a00
@ -0,0 +1,173 @@ |
||||
use crate::sparql::EvaluationError; |
||||
use crate::store::numeric_encoder::{ |
||||
EncodedQuad, EncodedTerm, MemoryStrStore, StrContainer, StrHash, StrId, StrLookup, |
||||
WithStoreError, |
||||
}; |
||||
use crate::store::ReadableEncodedStore; |
||||
use std::cell::RefCell; |
||||
use std::iter::empty; |
||||
|
||||
pub(crate) struct DatasetView<S: ReadableEncodedStore> { |
||||
store: S, |
||||
extra: RefCell<MemoryStrStore>, |
||||
default_graph_as_union: bool, |
||||
} |
||||
|
||||
impl<S: ReadableEncodedStore> DatasetView<S> { |
||||
pub fn new(store: S, default_graph_as_union: bool) -> Self { |
||||
Self { |
||||
store, |
||||
extra: RefCell::new(MemoryStrStore::default()), |
||||
default_graph_as_union, |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl<S: ReadableEncodedStore> WithStoreError for DatasetView<S> { |
||||
type Error = EvaluationError; |
||||
type StrId = DatasetStrId<S::StrId>; |
||||
} |
||||
|
||||
impl<S: ReadableEncodedStore> StrLookup for DatasetView<S> { |
||||
fn get_str(&self, id: DatasetStrId<S::StrId>) -> Result<Option<String>, EvaluationError> { |
||||
match id { |
||||
DatasetStrId::Store(id) => self.store.get_str(id).map_err(|e| e.into()), |
||||
DatasetStrId::Temporary(id) => Ok(self.extra.borrow().get_str(id)?), |
||||
} |
||||
} |
||||
|
||||
fn get_str_id(&self, value: &str) -> Result<Option<DatasetStrId<S::StrId>>, EvaluationError> { |
||||
if let Some(id) = self.extra.borrow().get_str_id(value)? { |
||||
Ok(Some(DatasetStrId::Temporary(id))) |
||||
} else { |
||||
Ok(self |
||||
.store |
||||
.get_str_id(value) |
||||
.map_err(|e| e.into())? |
||||
.map(DatasetStrId::Store)) |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl<S: ReadableEncodedStore> ReadableEncodedStore for DatasetView<S> { |
||||
type QuadsIter = |
||||
Box<dyn Iterator<Item = Result<EncodedQuad<DatasetStrId<S::StrId>>, EvaluationError>>>; |
||||
|
||||
fn encoded_quads_for_pattern( |
||||
&self, |
||||
subject: Option<EncodedTerm<Self::StrId>>, |
||||
predicate: Option<EncodedTerm<Self::StrId>>, |
||||
object: Option<EncodedTerm<Self::StrId>>, |
||||
graph_name: Option<EncodedTerm<Self::StrId>>, |
||||
) -> Box<dyn Iterator<Item = Result<EncodedQuad<DatasetStrId<S::StrId>>, EvaluationError>>> |
||||
{ |
||||
if let Some((subject, predicate, object, graph_name)) = |
||||
try_map_quad_pattern(subject, predicate, object, graph_name) |
||||
{ |
||||
if graph_name == None { |
||||
Box::new( |
||||
map_iter( |
||||
self.store |
||||
.encoded_quads_for_pattern(subject, predicate, object, None), |
||||
) |
||||
.filter(|quad| match quad { |
||||
Err(_) => true, |
||||
Ok(quad) => quad.graph_name != EncodedTerm::DefaultGraph, |
||||
}), |
||||
) |
||||
} else if graph_name == Some(EncodedTerm::DefaultGraph) && self.default_graph_as_union { |
||||
Box::new( |
||||
map_iter( |
||||
self.store |
||||
.encoded_quads_for_pattern(subject, predicate, object, None), |
||||
) |
||||
.map(|quad| { |
||||
let quad = quad?; |
||||
Ok(EncodedQuad::new( |
||||
quad.subject, |
||||
quad.predicate, |
||||
quad.object, |
||||
EncodedTerm::DefaultGraph, |
||||
)) |
||||
}), |
||||
) |
||||
} else { |
||||
Box::new(map_iter(self.store.encoded_quads_for_pattern( |
||||
subject, predicate, object, graph_name, |
||||
))) |
||||
} |
||||
} else { |
||||
Box::new(empty()) |
||||
} |
||||
} |
||||
} |
||||
|
||||
fn map_iter<'a, I: StrId>( |
||||
iter: impl Iterator<Item = Result<EncodedQuad<I>, impl Into<EvaluationError>>> + 'a, |
||||
) -> impl Iterator<Item = Result<EncodedQuad<DatasetStrId<I>>, EvaluationError>> + 'a { |
||||
iter.map(|t| { |
||||
t.map(|q| EncodedQuad { |
||||
subject: q.subject.map_id(DatasetStrId::Store), |
||||
predicate: q.predicate.map_id(DatasetStrId::Store), |
||||
object: q.object.map_id(DatasetStrId::Store), |
||||
graph_name: q.graph_name.map_id(DatasetStrId::Store), |
||||
}) |
||||
.map_err(|e| e.into()) |
||||
}) |
||||
} |
||||
|
||||
type QuadPattern<I> = ( |
||||
Option<EncodedTerm<I>>, |
||||
Option<EncodedTerm<I>>, |
||||
Option<EncodedTerm<I>>, |
||||
Option<EncodedTerm<I>>, |
||||
); |
||||
|
||||
fn try_map_quad_pattern<I: StrId>( |
||||
subject: Option<EncodedTerm<DatasetStrId<I>>>, |
||||
predicate: Option<EncodedTerm<DatasetStrId<I>>>, |
||||
object: Option<EncodedTerm<DatasetStrId<I>>>, |
||||
graph_name: Option<EncodedTerm<DatasetStrId<I>>>, |
||||
) -> Option<QuadPattern<I>> { |
||||
Some(( |
||||
transpose(subject.map(|t| t.try_map_id(unwrap_store_id)))?, |
||||
transpose(predicate.map(|t| t.try_map_id(unwrap_store_id)))?, |
||||
transpose(object.map(|t| t.try_map_id(unwrap_store_id)))?, |
||||
transpose(graph_name.map(|t| t.try_map_id(unwrap_store_id)))?, |
||||
)) |
||||
} |
||||
|
||||
fn transpose<T>(o: Option<Option<T>>) -> Option<Option<T>> { |
||||
match o { |
||||
Some(Some(v)) => Some(Some(v)), |
||||
Some(None) => None, |
||||
None => Some(None), |
||||
} |
||||
} |
||||
|
||||
fn unwrap_store_id<I: StrId>(id: DatasetStrId<I>) -> Option<I> { |
||||
match id { |
||||
DatasetStrId::Store(id) => Some(id), |
||||
DatasetStrId::Temporary(_) => None, |
||||
} |
||||
} |
||||
|
||||
impl<'a, S: ReadableEncodedStore> StrContainer for &'a DatasetView<S> { |
||||
fn insert_str(&mut self, value: &str) -> Result<Self::StrId, EvaluationError> { |
||||
if let Some(id) = self.store.get_str_id(value).map_err(|e| e.into())? { |
||||
Ok(DatasetStrId::Store(id)) |
||||
} else { |
||||
Ok(DatasetStrId::Temporary( |
||||
self.extra.borrow_mut().insert_str(value)?, |
||||
)) |
||||
} |
||||
} |
||||
} |
||||
|
||||
#[derive(Eq, PartialEq, Debug, Copy, Clone, Hash)] |
||||
pub enum DatasetStrId<I: StrId> { |
||||
Store(I), |
||||
Temporary(StrHash), |
||||
} |
||||
|
||||
impl<I: StrId> StrId for DatasetStrId<I> {} |
||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
Loading…
Reference in new issue