error handling

pprudhvi · Sep 7, 2019 · 63c88d8 · 63c88d8
1 parent 42566c3
commit 63c88d8
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/rust/datafusion/src/execution/context.rs b/rust/datafusion/src/execution/context.rs
@@ -279,10 +279,18 @@ impl ExecutionContext {
                     let it = p.execute().unwrap();
                     let mut it = it.lock().unwrap();
                     let mut results: Vec<RecordBatch> = vec![];
-                    while let Ok(Some(batch)) = it.next() {
-                        results.push(batch);
+                    loop {
+                        match it.next() {
+                            Ok(Some(batch)) => {
+                                results.push(batch);
+                            }
+                            Ok(None) => {
+                                // end of result set
+                                return Ok(results);
+                            }
+                            Err(e) => return Err(e),
+                        }
                     }
-                    Ok(results)
                 })
             })
             .collect();