Merge pull request #16 from trocco-io/patch_table_with_retry

Author: giwa

lib/embulk/output/bigquery/bigquery_client.rb

@@ -514,26 +514,28 @@ def get_table_or_partition(table, dataset: nil)
 
         # update only column.description
         def patch_table
-          table = get_table(@task['table'])
-
-          def patch_description(fields, column_options)
-            fields.map do |field|
-              column_option = column_options.select{|col_opt| col_opt['name'] == field.name}.first
-              if column_option
-                field.update!(description: column_option['description']) if column_option['description']
-                if field.fields && column_option['fields']
-                  nested_fields = patch_description(field.fields, column_option['fields'])
-                  field.update!(fields: nested_fields)
+          with_job_retry do
+            table = get_table(@task['table'])
+
+            def patch_description(fields, column_options)
+              fields.map do |field|
+                column_option = column_options.select{|col_opt| col_opt['name'] == field.name}.first
+                if column_option
+                  field.update!(description: column_option['description']) if column_option['description']
+                  if field.fields && column_option['fields']
+                    nested_fields = patch_description(field.fields, column_option['fields'])
+                    field.update!(fields: nested_fields)
+                  end
                 end
+                field
               end
-              field
             end
-          end
 
-          fields = patch_description(table.schema.fields, @task['column_options'])
-          table.schema.update!(fields: fields)
-          table_id = Helper.chomp_partition_decorator(@task['table'])
-          client.patch_table(@project, @dataset, table_id, table)
+            fields = patch_description(table.schema.fields, @task['column_options'])
+            table.schema.update!(fields: fields)
+            table_id = Helper.chomp_partition_decorator(@task['table'])
+            with_network_retry { client.patch_table(@project, @dataset, table_id, table) }
+          end
         end
       end
     end