Add numDmlAffectedRows to com.google.cloud.bigquery.TableResult

[shrgupta28]

Hi Team,

I am using the below code to update the rows in my table using Java Cloud Bigquery.

String dmlQuery = String.format("update exploreTest.Student  set Stud_Name = \"Shruti11\", Stud_Class=\"II\" where Stud_Id = 300", datasetName, tableName);

			QueryJobConfiguration dmlQueryConfig =
					QueryJobConfiguration.newBuilder(dmlQuery).build();

			// Execute the query.
			TableResult result = bigquery.query(dmlQueryConfig);

I need to retrieve the number of rows affected/modified/updated. Request you to add the API for the same. Similar to the one we have in Bigquery V2 API as . queryResponse.getNumDmlAffectedRows()
Thanks.

[pmakani]

@shrgupta28 I tested couple of times and getting num of affected rows , could you please try this ?

public void query() throws InterruptedException {
    String dmlQuery =
        String.format(
            "update exploreTest.Student  set Stud_Name = \"Shruti11\", Stud_Class=\"II\" where Stud_Id = 300");
    QueryJobConfiguration dmlQueryConfig = QueryJobConfiguration.newBuilder(dmlQuery).build();

    // Execute the query.
    Job job = bigQuery.create(JobInfo.of(dmlQueryConfig));
    job = job.waitFor();
    JobStatistics.QueryStatistics statistics = job.getStatistics();
    TableResult result = job.getQueryResults();
    long numDmlAffectedRows = statistics.getNumDmlAffectedRows();
    System.out.println(numDmlAffectedRows);
  }

[pmakani]

Closing this issue through reference comments