Skip to content

Commit 79940e9

Browse files
chore(bigquery-jdbc): eliminate dry run to resolve statement type
1 parent f8a7bda commit 79940e9

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

java-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/BigQueryStatement.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -717,8 +717,9 @@ void handleQueryResult(String query, TableResult results, SqlType queryType, Job
717717
case DML:
718718
case DML_EXTRA:
719719
Long dmlRowCount;
720-
if (results != null && results.getNumDmlAffectedRows() != null) {
721-
dmlRowCount = results.getNumDmlAffectedRows();
720+
if (results != null) {
721+
Long affectedRows = results.getNumDmlAffectedRows();
722+
dmlRowCount = affectedRows != null ? affectedRows : 0L;
722723
} else {
723724
QueryStatistics dmlStats = getQueryStatisticsFromJob(results, job);
724725
dmlRowCount =

0 commit comments

Comments
 (0)