From 9c9aa3334c7f801f1ca9f0cd39ec375a8f58f9e2 Mon Sep 17 00:00:00 2001 From: Konstantin Haase Date: Fri, 30 Oct 2015 16:29:09 +0100 Subject: [PATCH] allow setting sidekiq worker class --- lib/travis/api/v3/queries/user.rb | 2 +- lib/travis/api/v3/query.rb | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/lib/travis/api/v3/queries/user.rb b/lib/travis/api/v3/queries/user.rb index ad903315..72db91a1 100644 --- a/lib/travis/api/v3/queries/user.rb +++ b/lib/travis/api/v3/queries/user.rb @@ -1,6 +1,6 @@ module Travis::API::V3 class Queries::User < Query - set_queue(:user_sync, :user_sync) + setup_sidekiq(:user_sync, queue: :user_sync, class_name: "Travis::GithubSync::Workers::SyncUser") params :id, :login, :email, :github_id, :is_syncing def find diff --git a/lib/travis/api/v3/query.rb b/lib/travis/api/v3/query.rb index b7ce1fad..97d803ba 100644 --- a/lib/travis/api/v3/query.rb +++ b/lib/travis/api/v3/query.rb @@ -9,8 +9,9 @@ module Travis::API::V3 ] end - def self.set_queue(identifier, queue) - sidekiq_queue(identifier)[1] = queue + def self.setup_sidekiq(identifier, queue: nil, class_name: nil) + sidekiq_queue(identifier)[0] = class_name if class_name + sidekiq_queue(identifier)[1] = queue if queue end # generate from eval to avoid additional string allocations on every params access