fix: handle EXPORT DATA, EXPORT MODEL, and LOAD DATA statements

Author: keshavdandeva

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

@@ -72,8 +72,10 @@ static SqlType getSqlTypeFromStatementType(StatementType statementType) {
       case "ROLLBACK_TRANSACTION":
         return SqlType.TCL;
       case "EXPORT_DATA":
+        return SqlType.EXPORT;
       case "EXPORT_MODEL":
       case "LOAD_DATA":
+        return SqlType.DDL;
       default:
         return SqlType.OTHER;
     }

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

@@ -725,6 +725,20 @@ void handleQueryResult(String query, TableResult results, SqlType queryType)
           throw new BigQueryJdbcException(ex);
         }
         break;
+      case EXPORT:
+        try {
+          Job completedJob = this.bigQuery.getJob(results.getJobId()).waitFor();
+          JobStatistics.QueryStatistics statistics =
+              (JobStatistics.QueryStatistics) completedJob.getStatistics();
+          if (statistics.getExportDataStats() != null) {
+            updateAffectedRowCount(statistics.getExportDataStats().getRowCount());
+          } else {
+            updateAffectedRowCount(0L);
+          }
+        } catch (InterruptedException ex) {
+          throw new BigQueryJdbcRuntimeException(ex);
+        }
+        break;
       case OTHER:
         throw new BigQueryJdbcException(String.format("Unexpected value: " + queryType));
     }
@@ -1610,6 +1624,7 @@ enum SqlType {
     DDL,
     SCRIPT,
     TCL,
+    EXPORT,
     OTHER
   }