hirefire-resource 0.3.6 → 0.3.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7a837ad85215ea4c9fb9db5d9df59e9d2efb9682
-  data.tar.gz: 019bb5a936269518290899332c41fc1b4c831e5c
+  metadata.gz: 871488e23cf718f9e13eb8bddd89098fb9951b82
+  data.tar.gz: ee697e01f7d2456d7f9d803878ddce0f8a8d35f1
 SHA512:
-  metadata.gz: 8f79e026eea2e841a67faeec56bceb7acf835dd08f31100447782de70c7715505cda00437fed0a72d33ffe6917dc944e0e9f11bcc73f422b2d3232b57bc0de22
-  data.tar.gz: fc1cb09f2e55ec48999f0a3a4ac01eec489696c44241d7f03babf6afbf93a20bd76d6f94daf8a33f2f23f92fb5866dd97ae835730c2c9ac96acd57f7549b9f59
+  metadata.gz: afd1042bac8694192d66e19cc2058236cd26bff7796979a83beea0c56fa269e2116ce2b94e2ecefd10dd2a08c5fa919a977caa14ef75aa915b57276728395c03
+  data.tar.gz: ac30b7c4403cab11501e67f75f12b9bd168ee296a36f681a6ca7da22b9815ff8243b2d672c2e792478c21a194aa881cb4b5732cb46eadd1da50448b5e8b71519

data/hirefire-resource.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |gem|
   gem.name = "hirefire-resource"
-  gem.version = "0.3.6"
+  gem.version = "0.3.7"
   gem.platform = Gem::Platform::RUBY
   gem.authors = "Michael van Rooijen"
   gem.email = "michael@hirefire.io"

data/lib/hirefire-resource.rb

@@ -6,7 +6,7 @@ HIREFIRE_PATH = File.expand_path("../hirefire", __FILE__)
   require "#{HIREFIRE_PATH}/#{file}"
 end
 
-%w[delayed_job resque sidekiq qu qc].each do |file|
+%w[delayed_job resque sidekiq qu qc bunny].each do |file|
   require "#{HIREFIRE_PATH}/macro/#{file}"
 end
 

data/lib/hirefire/macro/bunny.rb

@@ -0,0 +1,80 @@
+# encoding: utf-8
+
+module HireFire
+  module Macro
+    module Bunny
+      extend self
+
+      # Returns the job quantity for the provided queue(s).
+      #
+      # @example Bunny Macro Usage
+      #
+      #   # all queues using existing RabbitMQ connection.
+      #   HireFire::Macro::Bunny.queue("queue1", "queue2", :connection => connection)
+      #
+      #   # all queues using new RabbitMQ connection.
+      #   HireFire::Macro::Bunny.queue("queue1", "queue2", :amqp_url => url)
+      #
+      #   # all non-durable queues using new RabbitMQ connection.
+      #   HireFire::Macro::Bunny.queue("queue1", "queue2", :amqp_url => url, :durable => false)
+      #
+      # @param [Array] queues provide one or more queue names, or none for "all".
+      #   Last argument can pass in a Hash containing :connection => rabbitmq_connection or :amqp => :rabbitmq_url
+      # @return [Integer] the number of jobs in the queue(s).
+      #
+      def queue(*queues)
+        require "bunny"
+
+        queues.flatten!
+
+        if queues.last.is_a?(Hash)
+          options = queues.pop
+        else
+          options = {}
+        end
+
+        if options[:durable].nil?
+          options[:durable] = true
+        end
+
+        if options[:connection]
+          connection = options[:connection]
+
+          channel = nil
+          begin
+            channel = connection.create_channel
+            return count_messages(channel, queues, options)
+          ensure
+            if channel
+              channel.close
+            end
+          end
+        elsif options[:amqp_url]
+          connection = ::Bunny.new(options[:amqp_url])
+          begin
+            connection.start
+            channel = connection.create_channel
+            return count_messages(channel, queues, options)
+          ensure
+            if channel
+              channel.close
+            end
+
+            connection.close
+          end
+        else
+          raise %{Must pass in :connection => rabbitmq_connection or :amqp_url => url\n} +
+                  %{For example: HireFire::Macro::Bunny.queue("queue1", :connection => rabbitmq_connection}
+        end
+      end
+
+      def count_messages(channel, queue_names, options)
+        queue_names.inject(0) do |sum, queue_name|
+          queue = channel.queue(queue_name, :durable => options[:durable])
+          sum + queue.message_count
+        end
+      end
+    end
+  end
+end
+

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hirefire-resource
 version: !ruby/object:Gem::Version
-  version: 0.3.6
+  version: 0.3.7
 platform: ruby
 authors:
 - Michael van Rooijen
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-04-24 00:00:00.000000000 Z
+date: 2015-04-30 00:00:00.000000000 Z
 dependencies: []
 description: Load-based scaling, schedule-based scaling, dyno crash recovery, for
   web- and worker dynos.
@@ -28,6 +28,7 @@ files:
 - hirefire-resource.gemspec
 - lib/hirefire-resource.rb
 - lib/hirefire/cli.rb
+- lib/hirefire/macro/bunny.rb
 - lib/hirefire/macro/delayed_job.rb
 - lib/hirefire/macro/qc.rb
 - lib/hirefire/macro/qu.rb