diff --git a/cli/src/session.rs b/cli/src/session.rs index 28772675..8075085c 100644 --- a/cli/src/session.rs +++ b/cli/src/session.rs @@ -92,8 +92,8 @@ impl Session { let rows = conn.query_iter(PROMPT_SQL).await; match rows { Ok(mut rows) => { - while let Some(row) = rows.next().await { - let name: (String,) = row.unwrap().try_into().unwrap(); + while let Some(Ok(row)) = rows.next().await { + let name: (String,) = row.try_into().unwrap(); keywords.push(name.0); } } diff --git a/cli/tests/http/05-stream.result b/cli/tests/http/05-stream.result new file mode 100644 index 00000000..d00491fd --- /dev/null +++ b/cli/tests/http/05-stream.result @@ -0,0 +1 @@ +1 diff --git a/cli/tests/http/05-stream.sh b/cli/tests/http/05-stream.sh new file mode 100644 index 00000000..e2d5dad0 --- /dev/null +++ b/cli/tests/http/05-stream.sh @@ -0,0 +1,27 @@ +#!/bin/bash +## uncomment these when QUERY_DATABEND_ENTERPRISE_LICENSE env is set + +# cat <