Work around steffengy/tiberius#47

mit-pdos · Feb 27, 2018 · 438b716 · 438b716
1 parent 2097117
commit 438b716
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/benchmarks/vote/clients/mssql.rs b/benchmarks/vote/clients/mssql.rs
@@ -57,15 +57,14 @@ impl VoteClient for Client {
         let db = args.value_of("database").unwrap();
 
         let fixconn = |conn: tiberius::SqlConnection<Box<tiberius::BoxableIo>>| {
-            conn.simple_exec(format!(
-                "USE {}; \
-                 SET NUMERIC_ROUNDABORT OFF; \
-                 SET ANSI_PADDING, ANSI_WARNINGS, \
-                 CONCAT_NULL_YIELDS_NULL, ARITHABORT, \
-                 QUOTED_IDENTIFIER, ANSI_NULLS ON; \
-                 SET TRANSACTION ISOLATION LEVEL READ UNCOMMITTED;",
-                db
-            ))
+            conn.simple_exec(format!("USE {}", db))
+                .and_then(|(_, conn)| conn.simple_exec("SET NUMERIC_ROUNDABORT OFF"))
+                .and_then(|(_, conn)| {
+                    conn.simple_exec("SET ANSI_PADDING, ANSI_WARNINGS, CONCAT_NULL_YIELDS_NULL, ARITHABORT,  QUOTED_IDENTIFIER, ANSI_NULLS ON")
+                })
+                .and_then(|(_, conn)| {
+                    conn.simple_exec("SET TRANSACTION ISOLATION LEVEL READ UNCOMMITTED")
+                })
         };
 
         // Check whether database already exists, or whether we need to create it