From ef7811ead9e898f460df6830bc8e7d30ef4cfbfe Mon Sep 17 00:00:00 2001 From: Ana Rosas <a.rosas10@gmail.com> Date: Fri, 27 May 2016 16:10:16 +0200 Subject: [PATCH] Small refacto --- lib/travis/api/app/endpoint/builds.rb | 5 ++--- lib/travis/api/enqueue/services/cancel_model.rb | 9 ++++----- 2 files changed, 6 insertions(+), 8 deletions(-) diff --git a/lib/travis/api/app/endpoint/builds.rb b/lib/travis/api/app/endpoint/builds.rb index 477731ed..57ae056a 100644 --- a/lib/travis/api/app/endpoint/builds.rb +++ b/lib/travis/api/app/endpoint/builds.rb @@ -23,7 +23,7 @@ class Travis::Api::App Metriks.meter("api.request.cancel_build").mark if Travis::Features.owner_active?(:enqueue_to_hub, current_user) - service = Travis::Enqueue::Services::CancelModel.new(current_user, { build_id: params[:id], source: 'api' }) + service = Travis::Enqueue::Services::CancelModel.new(current_user, { build_id: params[:id] }) else service = self.service(:cancel_build, params.merge(source: 'api')) end @@ -47,8 +47,7 @@ class Travis::Api::App respond_with json else if service.respond_to?(:enqueue_to_hub) - payload = { build_id: params[:id], source: 'api'} - service.enqueue_to_hub(payload) + service.enqueue_to_hub else Travis::Sidekiq::BuildCancellation.perform_async(id: params[:id], user_id: current_user.id, source: 'api') end diff --git a/lib/travis/api/enqueue/services/cancel_model.rb b/lib/travis/api/enqueue/services/cancel_model.rb index d8868a50..fb434c9d 100644 --- a/lib/travis/api/enqueue/services/cancel_model.rb +++ b/lib/travis/api/enqueue/services/cancel_model.rb @@ -4,12 +4,11 @@ module Travis class CancelModel - attr_reader :current_user, :source, :target + attr_reader :current_user, :target def initialize(current_user, params) @current_user = current_user @params = params - @source = params.delete(:source) || 'unknown' target end @@ -21,16 +20,16 @@ module Travis messages end - def enqueue_to_hub(payload) + def enqueue_to_hub # target may have been retrieved with a :join query, so we need to reset the readonly status if can_cancel? target.send(:instance_variable_set, :@readonly, false) target.cancel! if type == :build - push_matrix(payload) + push_matrix(@params) else - push(payload, target) + push(@params, target) end end end