Created
April 19, 2016 19:33
-
-
Save sgrif/24ed31f66246d71a2f7356a806ebc57a to your computer and use it in GitHub Desktop.
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
diff --git a/diesel/src/lib.rs b/diesel/src/lib.rs | |
index 28c60a3..18225b2 100644 | |
--- a/diesel/src/lib.rs | |
+++ b/diesel/src/lib.rs | |
@@ -3,6 +3,7 @@ | |
//! found in the README. | |
#![deny(warnings)] | |
#![cfg_attr(feature = "unstable", feature(specialization))] | |
+#![cfg_attr(all(test, feature = "unstable"), feature(test))] | |
#[macro_use] | |
mod macros; | |
diff --git a/diesel/src/pg/connection/mod.rs b/diesel/src/pg/connection/mod.rs | |
index 6066ead..634d821 100644 | |
--- a/diesel/src/pg/connection/mod.rs | |
+++ b/diesel/src/pg/connection/mod.rs | |
@@ -239,4 +239,78 @@ mod tests { | |
let database_url = env::var("DATABASE_URL").unwrap(); | |
PgConnection::establish(&database_url).unwrap() | |
} | |
+ | |
+ #[cfg(feature = "unstable")] | |
+ mod benchmarks { | |
+ extern crate test; | |
+ | |
+ use backend::Backend; | |
+ use expression::{SqlLiteral, sql}; | |
+ use prelude::*; | |
+ use query_builder::*; | |
+ use super::connection; | |
+ use self::test::Bencher; | |
+ | |
+ table! { | |
+ users { | |
+ id -> Integer, | |
+ name -> VarChar, | |
+ hair_color -> Nullable<VarChar>, | |
+ } | |
+ } | |
+ | |
+ #[bench] | |
+ fn prepared_statement_lookup_query_builder(bm: &mut Bencher) { | |
+ let conn = connection(); | |
+ | |
+ bm.iter(|| { | |
+ conn.prepare_query(&users::table.as_query()).unwrap() | |
+ }) | |
+ } | |
+ | |
+ #[bench] | |
+ fn prepared_statement_lookup_raw_sql(bm: &mut Bencher) { | |
+ let conn = connection(); | |
+ | |
+ bm.iter(|| { | |
+ let sql = "SELECT users.id, users.name, users.hair_color FROM users"; | |
+ conn.prepare_query(&cached_sql::<users::SqlType>(sql)).unwrap() | |
+ }) | |
+ } | |
+ | |
+ pub struct ForceCache<T>(T); | |
+ | |
+ pub fn cached_sql<T>(query: &str) -> ForceCache<SqlLiteral<T>> { | |
+ ForceCache(sql(query)) | |
+ } | |
+ | |
+ impl<T, DB> QueryFragment<DB> for ForceCache<T> where | |
+ DB: Backend, | |
+ T: QueryFragment<DB>, | |
+ { | |
+ fn to_sql(&self, out: &mut DB::QueryBuilder) -> BuildQueryResult { | |
+ self.0.to_sql(out) | |
+ } | |
+ | |
+ fn collect_binds(&self, out: &mut DB::BindCollector) -> QueryResult<()> { | |
+ self.0.collect_binds(out) | |
+ } | |
+ | |
+ fn is_safe_to_cache_prepared(&self) -> bool { | |
+ true | |
+ } | |
+ } | |
+ | |
+ impl<T: Query> Query for ForceCache<T> { | |
+ type SqlType = T::SqlType; | |
+ } | |
+ | |
+ impl<T: QueryId> QueryId for ForceCache<T> { | |
+ type QueryId = T::QueryId; | |
+ | |
+ fn has_static_query_id() -> bool { | |
+ T::has_static_query_id() | |
+ } | |
+ } | |
+ } | |
} | |
diff --git a/diesel/src/sqlite/connection/mod.rs b/diesel/src/sqlite/connection/mod.rs | |
index d6d9d10..0fa25d2 100644 | |
--- a/diesel/src/sqlite/connection/mod.rs | |
+++ b/diesel/src/sqlite/connection/mod.rs | |
@@ -267,4 +267,84 @@ mod tests { | |
assert_eq!(Ok(true), query.get_result(&connection)); | |
assert_eq!(1, connection.statement_cache.borrow().len()); | |
} | |
+ | |
+ #[cfg(feature = "unstable")] | |
+ mod benchmarks { | |
+ extern crate test; | |
+ | |
+ use backend::Backend; | |
+ use expression::{SqlLiteral, sql}; | |
+ use prelude::*; | |
+ use query_builder::*; | |
+ use self::test::Bencher; | |
+ use sqlite::SqliteConnection; | |
+ | |
+ table! { | |
+ users { | |
+ id -> Integer, | |
+ name -> VarChar, | |
+ hair_color -> Nullable<VarChar>, | |
+ } | |
+ } | |
+ | |
+ fn connection() -> SqliteConnection { | |
+ let conn = SqliteConnection::establish(":memory:").unwrap(); | |
+ conn.execute("CREATE TABLE users (id INTEGER PRIMARY KEY NOT NULL, name VARCHAR NOT NULL, hair_color VARCHAR)").unwrap(); | |
+ conn | |
+ } | |
+ | |
+ #[bench] | |
+ fn prepared_statement_lookup_query_builder(bm: &mut Bencher) { | |
+ let conn = connection(); | |
+ | |
+ bm.iter(|| { | |
+ conn.prepare_query(&users::table.as_query()).unwrap() | |
+ }) | |
+ } | |
+ | |
+ #[bench] | |
+ fn prepared_statement_lookup_raw_sql(bm: &mut Bencher) { | |
+ let conn = connection(); | |
+ | |
+ bm.iter(|| { | |
+ let sql = "SELECT users.id, users.name, users.hair_color FROM users"; | |
+ conn.prepare_query(&cached_sql::<users::SqlType>(sql)).unwrap() | |
+ }) | |
+ } | |
+ | |
+ pub struct ForceCache<T>(T); | |
+ | |
+ pub fn cached_sql<T>(query: &str) -> ForceCache<SqlLiteral<T>> { | |
+ ForceCache(sql(query)) | |
+ } | |
+ | |
+ impl<T, DB> QueryFragment<DB> for ForceCache<T> where | |
+ DB: Backend, | |
+ T: QueryFragment<DB>, | |
+ { | |
+ fn to_sql(&self, out: &mut DB::QueryBuilder) -> BuildQueryResult { | |
+ self.0.to_sql(out) | |
+ } | |
+ | |
+ fn collect_binds(&self, out: &mut DB::BindCollector) -> QueryResult<()> { | |
+ self.0.collect_binds(out) | |
+ } | |
+ | |
+ fn is_safe_to_cache_prepared(&self) -> bool { | |
+ true | |
+ } | |
+ } | |
+ | |
+ impl<T: Query> Query for ForceCache<T> { | |
+ type SqlType = T::SqlType; | |
+ } | |
+ | |
+ impl<T: QueryId> QueryId for ForceCache<T> { | |
+ type QueryId = T::QueryId; | |
+ | |
+ fn has_static_query_id() -> bool { | |
+ T::has_static_query_id() | |
+ } | |
+ } | |
+ } | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment