fix waiting for job to be done

airbytehq · Oct 12, 2023 · f9059ab · f9059ab
1 parent bd4b9f2
commit f9059ab
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...rc/main/java/io/airbyte/integrations/destination/bigquery/writer/BigQueryTableWriter.java b/...rc/main/java/io/airbyte/integrations/destination/bigquery/writer/BigQueryTableWriter.java
@@ -51,10 +51,10 @@ public void close(boolean hasFailed) throws IOException {
     this.writeChannel.close();
     try {
       Job job = writeChannel.getJob();
-      /* while (!JobStatus.State.DONE.equals(job.getStatus())) {
+      while (job == null) {
         Thread.sleep(1000L);
-        job = job.reload();
-      } */
+        job = writeChannel.getJob();
+      }
       if (job.getStatus().getError() != null) {
         throw new RuntimeException("Fail to complete a load job in big query, Job id: " + writeChannel.getJob().getJobId() +
                 ", with error: " + writeChannel.getJob().getStatus().getError());