10903: support need_transaction for job and pipeline_instance cancel methods.
authorradhika <radhika@curoverse.com>
Fri, 10 Feb 2017 19:08:50 +0000 (14:08 -0500)
committerradhika <radhika@curoverse.com>
Fri, 10 Feb 2017 19:08:50 +0000 (14:08 -0500)
services/api/app/controllers/arvados/v1/jobs_controller.rb
services/api/app/controllers/arvados/v1/pipeline_instances_controller.rb
services/api/app/models/job.rb
services/api/app/models/pipeline_instance.rb
services/api/test/unit/job_test.rb

index c38e41908476ea19c84ff18028e782d6205ff6ea..bca74cd55fecddd469a000eb4d41bcb666f344f9 100644 (file)
@@ -46,7 +46,7 @@ class Arvados::V1::JobsController < ApplicationController
 
   def cancel
     reload_object_before_update
-    @object.cancel params[:cascade]
+    @object.cancel cascade: params[:cascade]
     show
   end
 
index 476fc0abb59bae84841369cbfb44b45597ec63c2..c44c81a733d1f432f378cea58c25d3eb359bf32d 100644 (file)
@@ -5,7 +5,7 @@ class Arvados::V1::PipelineInstancesController < ApplicationController
 
   def cancel
     reload_object_before_update
-    @object.cancel params[:cascade]
+    @object.cancel cascade: params[:cascade]
     show
   end
 end
index 5ce5ff20f203eeb0ba8b95055c62d2de10db0b0b..d73d629905d7f6dcb4eda083bfc20fa7cbf5e437 100644 (file)
@@ -284,7 +284,14 @@ class Job < ArvadosModel
     end
   end
 
-  def cancel cascade=nil
+  def cancel(cascade: false, need_transaction: true)
+    if need_transaction
+      ActiveRecord::Base.transaction do
+        cancel(cascade: cascade, need_transaction: false)
+      end
+      return
+    end
+
     if self.state.in?([Queued, Running])
       self.state = Cancelled
       self.save!
@@ -300,13 +307,13 @@ class Job < ArvadosModel
     return if children.empty?
 
     # cancel any child jobs
-    Job.where(uuid: children).each do |job|
-      job.cancel cascade if job.state.in?([Queued, Running])
+    Job.where(uuid: children, state: [Queued, Running]).each do |job|
+      job.cancel(cascade: cascade, need_transaction: false)
     end
 
     # cancel any child pipelines
-    PipelineInstance.where(uuid: children).each do |pi|
-      pi.cancel cascade if pi.state.in?([PipelineInstance::RunningOnServer, PipelineInstance::RunningOnClient])
+    PipelineInstance.where(uuid: children, state: [PipelineInstance::RunningOnServer, PipelineInstance::RunningOnClient]).each do |pi|
+      pi.cancel(cascade: cascade, need_transaction: false)
     end
   end
 
index 651f36d30f16350f2dee5fde0b47bca824184d8f..75903ca7d05551d0b5486ed33c1e0c4ee9225cb3 100644 (file)
@@ -100,7 +100,14 @@ class PipelineInstance < ArvadosModel
     self.where("state = 'RunningOnServer'")
   end
 
-  def cancel cascade=nil
+  def cancel(cascade: false, need_transaction: true)
+    if need_transaction
+      ActiveRecord::Base.transaction do
+        cancel(cascade: cascade, need_transaction: false)
+      end
+      return
+    end
+
     if self.state.in?([RunningOnServer, RunningOnClient])
       self.state = Paused
       self.save!
@@ -115,8 +122,8 @@ class PipelineInstance < ArvadosModel
 
     return if children.empty?
 
-    Job.where(uuid: children).each do |job|
-      job.cancel cascade if job.state.in?([Job::Queued, Job::Running])
+    Job.where(uuid: children, state: [Job::Queued, Job::Running]).each do |job|
+      job.cancel(cascade: cascade, need_transaction: false)
     end
   end
 
index 72b7689817ed219d74a06c89f9faa0b9ca09d8a5..8355ccf8973cd3e7b88f29af7a1140a6f9dbb528 100644 (file)
@@ -509,7 +509,7 @@ class JobTest < ActiveSupport::TestCase
   ].each do |cascade|
     test "cancel job with cascade #{cascade}" do
       job = Job.find_by_uuid jobs(:running_job_with_components_at_level_1).uuid
-      job.cancel cascade
+      job.cancel cascade: cascade
       assert_equal Job::Cancelled, job.state
 
       descendents = ['zzzzz-8i9sb-jobcomponentsl2',
@@ -542,7 +542,7 @@ class JobTest < ActiveSupport::TestCase
   test 'cancelling a job with circular relationship with another does not result in an infinite loop' do
     job = Job.find_by_uuid jobs(:running_job_2_with_circular_component_relationship).uuid
 
-    job.cancel true
+    job.cancel cascade: true
 
     assert_equal Job::Cancelled, job.state