pr feedback

Author: keshavdandeva

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

@@ -717,9 +717,8 @@ void handleQueryResult(String query, TableResult results, SqlType queryType, Job
       case DML:
       case DML_EXTRA:
         Long dmlRowCount;
-        if (results != null) {
-          Long affectedRows = results.getNumDmlAffectedRows();
-          dmlRowCount = affectedRows != null ? affectedRows : 0L;
+        if (results != null && results.getNumDmlAffectedRows() != null) {
+          dmlRowCount = results.getNumDmlAffectedRows();
         } else {
           QueryStatistics dmlStats = getQueryStatisticsFromJob(results, job);
           dmlRowCount =