Skip to content

Commit fe83376

Browse files
authored
fix: only return schema for query (#329)
1 parent 764be81 commit fe83376

1 file changed

Lines changed: 12 additions & 8 deletions

File tree

datafusion-postgres/src/handlers.rs

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -405,14 +405,18 @@ impl QueryParser for Parser {
405405
column_format: Option<&Format>,
406406
) -> PgWireResult<Vec<FieldInfo>> {
407407
if let (_, Some((_, plan))) = stmt {
408-
let schema = plan.schema();
409-
let fields = arrow_schema_to_pg_fields(
410-
schema.as_arrow(),
411-
column_format.unwrap_or(&Format::UnifiedText),
412-
None,
413-
)?;
414-
415-
Ok(fields)
408+
if !matches!(plan, LogicalPlan::Ddl(_) | LogicalPlan::Dml(_)) {
409+
let schema = plan.schema();
410+
let fields = arrow_schema_to_pg_fields(
411+
schema.as_arrow(),
412+
column_format.unwrap_or(&Format::UnifiedText),
413+
None,
414+
)?;
415+
416+
Ok(fields)
417+
} else {
418+
Ok(vec![])
419+
}
416420
} else {
417421
Ok(vec![])
418422
}

0 commit comments

Comments
 (0)