Contents

# frozen_string_literal: true

module Totoro
  class BaseWorker
    def self.setup(attrs)
      prefix = attrs[:prefix]
      queue_name = attrs[:queue_name]
      define_method('setup') do
        raise(Totoro::NeedQueueNameError) if queue_name.nil?

        @prefix = prefix
        @queue_name = queue_name
      end
    end

    def initialize
      setup
    end

    def execute
      Rails.logger.info 'Listening to the Rabbitmq'
      STDOUT.flush
      subscribe_service.subscribe(@queue_name) do |delivery_info, metadata, payload|
        Rails.logger.debug "#{@queue_name} received message"
        STDOUT.flush
        payload_hash = JSON.parse(payload).with_indifferent_access
        process(payload_hash, metadata, delivery_info)
      end

      handle_usr1_n_usr2

      subscribe_service.channel.work_pool.join
    rescue SignalException
      puts 'Terminating process ..'
      subscribe_service.channel.work_pool.shutdown(true)
      puts 'Stopped.'
    end

    def process; end

    private

    def handle_usr1_n_usr2
      %w[USR1 USR2].each do |signal|
        Signal.trap(signal) do
          puts "#{signal} received."
          handler = "on_#{signal.downcase}"
          send handler if respond_to?(handler)
        end
      end
    end

    def config
      @config ||= Totoro::Config.new(@prefix)
    end

    def subscribe_service
      @subscribe_service ||= Totoro::SubscribeService.new(config)
    end
  end
end

Version data entries

4 entries across 3 versions & 1 rubygems

Version	Path
totoro-1.0.7	lib/totoro/base_worker.rb
totoro-1.0.6	lib/totoro/base_worker.rb
totoro-1.0.6	pkg/totoro-1.0.5/lib/totoro/base_worker.rb
totoro-1.0.5	lib/totoro/base_worker.rb