2017-02-13 18:30:02 +00:00
|
|
|
// Copyright 2016 Mozilla
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use
|
|
|
|
// this file except in compliance with the License. You may obtain a copy of the
|
|
|
|
// License at http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
// Unless required by applicable law or agreed to in writing, software distributed
|
|
|
|
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
|
|
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
|
|
// specific language governing permissions and limitations under the License.
|
|
|
|
|
|
|
|
use std::collections::HashMap;
|
|
|
|
|
2017-03-06 22:40:10 +00:00
|
|
|
use rusqlite;
|
|
|
|
use rusqlite::types::ToSql;
|
|
|
|
|
2017-02-13 18:30:02 +00:00
|
|
|
use mentat_core::{
|
2017-02-24 05:16:19 +00:00
|
|
|
Schema,
|
2017-02-13 18:30:02 +00:00
|
|
|
TypedValue,
|
|
|
|
};
|
|
|
|
|
2017-02-24 05:16:19 +00:00
|
|
|
use mentat_query_algebrizer::algebrize;
|
2017-02-13 18:30:02 +00:00
|
|
|
|
2017-03-06 22:40:10 +00:00
|
|
|
pub use mentat_query::{
|
|
|
|
NamespacedKeyword,
|
|
|
|
PlainSymbol,
|
|
|
|
};
|
|
|
|
|
2017-02-13 18:30:02 +00:00
|
|
|
use mentat_query_parser::{
|
|
|
|
parse_find_string,
|
Extract partial storage abstraction; use error-chain throughout. Fixes #328. r=rnewman (#341)
* Pre: Drop unneeded tx0 from search results.
* Pre: Don't require a schema in some of the DB code.
The idea is to separate the transaction applying code, which is
schema-aware, from the concrete storage code, which is just concerned
with getting bits onto disk.
* Pre: Only reference Schema, not DB, in debug module.
This is part of a larger separation of the volatile PartitionMap,
which is modified every transaction, from the stable Schema, which is
infrequently modified.
* Pre: Fix indentation.
* Extract part of DB to new SchemaTypeChecking trait.
* Extract part of DB to new PartitionMapping trait.
* Pre: Don't expect :db.part/tx partition to advance when tx fails.
This fails right now, because we allocate tx IDs even when we shouldn't.
* Sketch a db interface without DB.
* Add ValueParseError; use error-chain in tx-parser.
This can be simplified when
https://github.com/Marwes/combine/issues/86 makes it to a published
release, but this unblocks us for now. This converts the `combine`
error type `ParseError<&'a [edn::Value]>` to a type with owned
`Vec<edn::Value>` collections, re-using `edn::Value::Vector` for
making them `Display`.
* Pre: Accept Borrow<Schema> instead of just &Schema in debug module.
This makes it easy to use Rc<Schema> or Arc<Schema> without inserting
&* sigils throughout the code.
* Use error-chain in query-parser.
There are a few things to point out here:
- the fine grained error types have been flattened into one crate-wide
error type; it's pretty easy to regain the granularity as needed.
- edn::ParseError is automatically lifted to
mentat_query_parser::errors::Error;
- we use mentat_parser_utils::ValueParser to maintain parsing error
information from `combine`.
* Patch up top-level.
* Review comment: Only `borrow()` once.
2017-02-24 23:32:41 +00:00
|
|
|
};
|
|
|
|
|
2017-02-24 05:16:19 +00:00
|
|
|
use mentat_sql::{
|
|
|
|
SQLQuery,
|
2017-02-13 18:30:02 +00:00
|
|
|
};
|
|
|
|
|
2017-02-24 05:16:19 +00:00
|
|
|
use mentat_query_translator::{
|
2017-03-06 22:40:10 +00:00
|
|
|
query_to_select,
|
2017-02-24 05:16:19 +00:00
|
|
|
};
|
|
|
|
|
2017-03-06 22:40:10 +00:00
|
|
|
pub use mentat_query_projector::{
|
|
|
|
QueryResults,
|
|
|
|
};
|
2017-02-13 18:30:02 +00:00
|
|
|
|
2017-03-06 22:40:10 +00:00
|
|
|
use errors::Result;
|
2017-02-13 18:30:02 +00:00
|
|
|
|
2017-02-24 05:16:19 +00:00
|
|
|
pub type QueryExecutionResult = Result<QueryResults>;
|
|
|
|
|
2017-02-13 18:30:02 +00:00
|
|
|
/// Take an EDN query string, a reference to a open SQLite connection, a Mentat DB, and an optional
|
|
|
|
/// collection of input bindings (which should be keyed by `"?varname"`), and execute the query
|
|
|
|
/// immediately, blocking the current thread.
|
|
|
|
/// Returns a structure that corresponds to the kind of input query, populated with `TypedValue`
|
|
|
|
/// instances.
|
2017-02-24 05:16:19 +00:00
|
|
|
/// The caller is responsible for ensuring that the SQLite connection is in a transaction if
|
|
|
|
/// isolation is required.
|
2017-02-13 18:30:02 +00:00
|
|
|
#[allow(unused_variables)]
|
2017-03-07 00:27:13 +00:00
|
|
|
pub fn q_once<'sqlite, 'schema, 'query, T, U>
|
2017-02-24 05:16:19 +00:00
|
|
|
(sqlite: &'sqlite rusqlite::Connection,
|
|
|
|
schema: &'schema Schema,
|
|
|
|
query: &'query str,
|
2017-03-07 00:27:13 +00:00
|
|
|
inputs: T,
|
|
|
|
limit: U) -> QueryExecutionResult
|
|
|
|
where T: Into<Option<HashMap<String, TypedValue>>>,
|
|
|
|
U: Into<Option<u64>>
|
|
|
|
{
|
2017-02-13 18:30:02 +00:00
|
|
|
// TODO: validate inputs.
|
2017-03-07 00:27:13 +00:00
|
|
|
|
2017-02-13 18:30:02 +00:00
|
|
|
let parsed = parse_find_string(query)?;
|
2017-03-07 00:27:13 +00:00
|
|
|
let mut algebrized = algebrize(schema, parsed);
|
2017-03-07 04:18:38 +00:00
|
|
|
|
|
|
|
if algebrized.is_known_empty() {
|
|
|
|
// We don't need to do any SQL work at all.
|
|
|
|
return Ok(QueryResults::empty(&algebrized.find_spec));
|
|
|
|
}
|
|
|
|
|
2017-03-07 00:27:13 +00:00
|
|
|
algebrized.apply_limit(limit.into());
|
|
|
|
|
2017-03-06 22:40:10 +00:00
|
|
|
let select = query_to_select(algebrized);
|
|
|
|
let SQLQuery { sql, args } = select.query.to_sql_query()?;
|
2017-02-24 05:16:19 +00:00
|
|
|
|
|
|
|
let mut statement = sqlite.prepare(sql.as_str())?;
|
|
|
|
|
2017-03-06 22:40:10 +00:00
|
|
|
let rows = if args.is_empty() {
|
2017-02-24 05:16:19 +00:00
|
|
|
statement.query(&[])?
|
|
|
|
} else {
|
2017-03-06 22:40:10 +00:00
|
|
|
let refs: Vec<(&str, &ToSql)> =
|
|
|
|
args.iter()
|
|
|
|
.map(|&(ref k, ref v)| (k.as_str(), v as &ToSql))
|
|
|
|
.collect();
|
|
|
|
statement.query_named(refs.as_slice())?
|
2017-02-24 05:16:19 +00:00
|
|
|
};
|
|
|
|
|
2017-03-06 22:40:10 +00:00
|
|
|
select.projector
|
|
|
|
.project(rows)
|
|
|
|
.map_err(|e| e.into())
|
2017-02-13 18:30:02 +00:00
|
|
|
}
|