fix infinite error stream issue

It was possible for an SQL error to trigger an infinite loop where SQLPage would stream error messages to the browsers forever
sqlpage · Jul 26, 2023 · 18cdb63 · 18cdb63
1 parent dea7485
commit 18cdb63
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/webserver/database/mod.rs b/src/webserver/database/mod.rs
@@ -121,7 +121,11 @@ pub async fn stream_query_results<'a>(
                     };
                     let mut stream = query.fetch_many(connection);
                     while let Some(elem) = stream.next().await {
-                        yield parse_single_sql_result(elem)
+                        let is_err = elem.is_err();
+                        yield parse_single_sql_result(elem);
+                        if is_err {
+                            break;
+                        }
                     }
                 },
                 ParsedSQLStatement::StaticSimpleSelect(value) => {