Contents

module ActsAsTenant::Sidekiq
  # Get the current tenant and store in the message to be sent to Sidekiq.
  class Client
    def call(worker_class, msg, queue, redis_pool)
      msg['acts_as_tenant'] ||=
        {
            'class' => ActsAsTenant.current_tenant.class.name,
            'id' => ActsAsTenant.current_tenant.id
        } if ActsAsTenant.current_tenant.present?

      yield
    end
  end

  # Pull the tenant out and run the current thread with it.
  class Server
    def call(worker_class, msg, queue)
      if msg.has_key?('acts_as_tenant')
        account = msg['acts_as_tenant']['class'].constantize.find msg['acts_as_tenant']['id']
        ActsAsTenant.with_tenant account do
          yield
        end
      else
        yield
      end
    end
  end
end

Sidekiq.configure_client do |config|
  config.client_middleware do |chain|
    chain.add ActsAsTenant::Sidekiq::Client
  end
end

Sidekiq.configure_server do |config|
  config.client_middleware do |chain|
    chain.add ActsAsTenant::Sidekiq::Client
  end
  config.server_middleware do |chain|
    chain.insert_before Sidekiq::Middleware::Server::RetryJobs, ActsAsTenant::Sidekiq::Server
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
acts_as_tenant-0.4.2	lib/acts_as_tenant/sidekiq.rb
acts_as_tenant-0.4.1	lib/acts_as_tenant/sidekiq.rb
acts_as_tenant-0.4.0	lib/acts_as_tenant/sidekiq.rb
acts_as_tenant-0.3.9	lib/acts_as_tenant/sidekiq.rb
acts_as_tenant-0.3.8	lib/acts_as_tenant/sidekiq.rb