Tpt
5 years ago
parent
fd3def2daf
commit
d617e72550
@ -0,0 +1,210 @@ |
|||||||
|
use crate::model::*; |
||||||
|
use crate::sparql::xml_results::{read_xml_results, write_xml_results}; |
||||||
|
use crate::Result; |
||||||
|
use failure::format_err; |
||||||
|
use std::fmt; |
||||||
|
use std::io::{BufRead, Write}; |
||||||
|
use uuid::Uuid; |
||||||
|
|
||||||
|
/// Results of a [SPARQL query](https://www.w3.org/TR/sparql11-query/)
|
||||||
|
pub enum QueryResult<'a> { |
||||||
|
/// Results of a [SELECT](https://www.w3.org/TR/sparql11-query/#select) query
|
||||||
|
Bindings(BindingsIterator<'a>), |
||||||
|
/// Result of a [ASK](https://www.w3.org/TR/sparql11-query/#ask) query
|
||||||
|
Boolean(bool), |
||||||
|
/// Results of a [CONSTRUCT](https://www.w3.org/TR/sparql11-query/#construct) or [DESCRIBE](https://www.w3.org/TR/sparql11-query/#describe) query
|
||||||
|
Graph(Box<dyn Iterator<Item = Result<Triple>> + 'a>), |
||||||
|
} |
||||||
|
|
||||||
|
impl<'a> QueryResult<'a> { |
||||||
|
pub fn read( |
||||||
|
reader: impl BufRead + 'static, |
||||||
|
syntax: QueryResultSyntax, |
||||||
|
) -> Result<QueryResult<'static>> { |
||||||
|
match syntax { |
||||||
|
QueryResultSyntax::Xml => read_xml_results(reader), |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn write<W: Write>(self, writer: W, syntax: QueryResultSyntax) -> Result<W> { |
||||||
|
match syntax { |
||||||
|
QueryResultSyntax::Xml => write_xml_results(self, writer), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// [SPARQL query](https://www.w3.org/TR/sparql11-query/) serialization formats
|
||||||
|
#[derive(Eq, PartialEq, Debug, Clone, Hash)] |
||||||
|
pub enum QueryResultSyntax { |
||||||
|
/// [SPARQL Query Results XML Format](http://www.w3.org/TR/rdf-sparql-XMLres/)
|
||||||
|
Xml, |
||||||
|
} |
||||||
|
|
||||||
|
/// An iterator over results bindings
|
||||||
|
pub struct BindingsIterator<'a> { |
||||||
|
variables: Vec<Variable>, |
||||||
|
iter: Box<dyn Iterator<Item = Result<Vec<Option<Term>>>> + 'a>, |
||||||
|
} |
||||||
|
|
||||||
|
impl<'a> BindingsIterator<'a> { |
||||||
|
pub(crate) fn new( |
||||||
|
variables: Vec<Variable>, |
||||||
|
iter: Box<dyn Iterator<Item = Result<Vec<Option<Term>>>> + 'a>, |
||||||
|
) -> Self { |
||||||
|
Self { variables, iter } |
||||||
|
} |
||||||
|
|
||||||
|
pub fn variables(&self) -> &[Variable] { |
||||||
|
&*self.variables |
||||||
|
} |
||||||
|
|
||||||
|
pub fn into_values_iter(self) -> Box<dyn Iterator<Item = Result<Vec<Option<Term>>>> + 'a> { |
||||||
|
self.iter |
||||||
|
} |
||||||
|
|
||||||
|
pub fn destruct( |
||||||
|
self, |
||||||
|
) -> ( |
||||||
|
Vec<Variable>, |
||||||
|
Box<dyn Iterator<Item = Result<Vec<Option<Term>>>> + 'a>, |
||||||
|
) { |
||||||
|
(self.variables, self.iter) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// A SPARQL query variable
|
||||||
|
#[derive(Eq, PartialEq, Ord, PartialOrd, Debug, Clone, Hash)] |
||||||
|
pub enum Variable { |
||||||
|
Variable { name: String }, |
||||||
|
BlankNode { id: Uuid }, |
||||||
|
Internal { id: Uuid }, |
||||||
|
} |
||||||
|
|
||||||
|
impl Variable { |
||||||
|
pub fn new(name: impl Into<String>) -> Self { |
||||||
|
Variable::Variable { name: name.into() } |
||||||
|
} |
||||||
|
|
||||||
|
pub fn has_name(&self) -> bool { |
||||||
|
match self { |
||||||
|
Variable::Variable { .. } => true, |
||||||
|
_ => false, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn name(&self) -> Result<&str> { |
||||||
|
match self { |
||||||
|
Variable::Variable { name } => Ok(name), |
||||||
|
_ => Err(format_err!("The variable {} has no name", self)), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl fmt::Display for Variable { |
||||||
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
||||||
|
match self { |
||||||
|
Variable::Variable { name } => write!(f, "?{}", name), |
||||||
|
Variable::BlankNode { id } => write!(f, "_:{}", id.to_simple()), |
||||||
|
Variable::Internal { id } => write!(f, "?{}", id.to_simple()), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl Default for Variable { |
||||||
|
fn default() -> Self { |
||||||
|
Variable::Internal { id: Uuid::new_v4() } |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl From<BlankNode> for Variable { |
||||||
|
fn from(blank_node: BlankNode) -> Self { |
||||||
|
Variable::BlankNode { |
||||||
|
id: *blank_node.as_uuid(), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
#[derive(Eq, PartialEq, Ord, PartialOrd, Debug, Clone, Hash)] |
||||||
|
pub enum NamedNodeOrVariable { |
||||||
|
NamedNode(NamedNode), |
||||||
|
Variable(Variable), |
||||||
|
} |
||||||
|
|
||||||
|
impl fmt::Display for NamedNodeOrVariable { |
||||||
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
||||||
|
match self { |
||||||
|
NamedNodeOrVariable::NamedNode(node) => write!(f, "{}", node), |
||||||
|
NamedNodeOrVariable::Variable(var) => write!(f, "{}", var), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl From<NamedNode> for NamedNodeOrVariable { |
||||||
|
fn from(node: NamedNode) -> Self { |
||||||
|
NamedNodeOrVariable::NamedNode(node) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl From<Variable> for NamedNodeOrVariable { |
||||||
|
fn from(var: Variable) -> Self { |
||||||
|
NamedNodeOrVariable::Variable(var) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
#[derive(Eq, PartialEq, Ord, PartialOrd, Debug, Clone, Hash)] |
||||||
|
pub enum TermOrVariable { |
||||||
|
Term(Term), |
||||||
|
Variable(Variable), |
||||||
|
} |
||||||
|
|
||||||
|
impl fmt::Display for TermOrVariable { |
||||||
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
||||||
|
match self { |
||||||
|
TermOrVariable::Term(term) => write!(f, "{}", term), |
||||||
|
TermOrVariable::Variable(var) => write!(f, "{}", var), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl From<NamedNode> for TermOrVariable { |
||||||
|
fn from(node: NamedNode) -> Self { |
||||||
|
TermOrVariable::Term(node.into()) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl From<BlankNode> for TermOrVariable { |
||||||
|
fn from(node: BlankNode) -> Self { |
||||||
|
TermOrVariable::Variable(node.into()) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl From<Literal> for TermOrVariable { |
||||||
|
fn from(literal: Literal) -> Self { |
||||||
|
TermOrVariable::Term(literal.into()) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl From<Variable> for TermOrVariable { |
||||||
|
fn from(var: Variable) -> Self { |
||||||
|
TermOrVariable::Variable(var) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl From<Term> for TermOrVariable { |
||||||
|
fn from(term: Term) -> Self { |
||||||
|
match term { |
||||||
|
Term::NamedNode(node) => TermOrVariable::Term(node.into()), |
||||||
|
Term::BlankNode(node) => TermOrVariable::Variable(node.into()), |
||||||
|
Term::Literal(literal) => TermOrVariable::Term(literal.into()), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl From<NamedNodeOrVariable> for TermOrVariable { |
||||||
|
fn from(element: NamedNodeOrVariable) -> Self { |
||||||
|
match element { |
||||||
|
NamedNodeOrVariable::NamedNode(node) => TermOrVariable::Term(node.into()), |
||||||
|
NamedNodeOrVariable::Variable(var) => TermOrVariable::Variable(var), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue