Use PARQUET_TEST_DATA env var

DataDog · Jan 29, 2020 · 996f2a4 · 996f2a4
1 parent 260f9ca
commit 996f2a4
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/rust/datafusion/examples/flight-server.rs b/rust/datafusion/examples/flight-server.rs
@@ -61,8 +61,15 @@ impl FlightService for FlightServiceImpl {
                 // create local execution context
                 let mut ctx = ExecutionContext::new();
 
-                ctx.register_parquet("alltypes_plain", "alltypes_plain.snappy.parquet")
-                    .unwrap();
+                let testdata = std::env::var("PARQUET_TEST_DATA")
+                    .expect("PARQUET_TEST_DATA not defined");
+
+                // register parquet file with the execution context
+                ctx.register_parquet(
+                    "alltypes_plain",
+                    &format!("{}/alltypes_plain.parquet", testdata),
+                )
+                .map_err(|e| to_tonic_err(&e))?;
 
                 // create the query plan
                 let plan = ctx