Skip to content

Commit a366b4b

Browse files
committed
fix: hook return for multi statement in simple query
1 parent 2b9a11c commit a366b4b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

datafusion-postgres/src/handlers.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -471,7 +471,7 @@ impl SimpleQueryHandler for DfSessionService {
471471
}
472472

473473
let mut results = vec![];
474-
for statement in statements {
474+
'stmt: for statement in statements {
475475
// TODO: improve statement check by using statement directly
476476
let query = statement.to_string();
477477
let query_lower = query.to_lowercase().trim().to_string();
@@ -495,7 +495,8 @@ impl SimpleQueryHandler for DfSessionService {
495495
.handle_query(&statement, &self.session_context, client)
496496
.await
497497
{
498-
return result.map(|response| vec![response]);
498+
results.push(result?);
499+
break 'stmt;
499500
}
500501
}
501502

0 commit comments

Comments
 (0)