Merge pull request poise#24 from needle-cookbooks/COOK-2611

Joshua Timberman · Joshua Timberman · commit 2f5251ffd5ee · 2013-04-22T16:33:30.000-07:00
COOK-2611 Adding queues parameter to celery LWRP
diff --git a/providers/celery.rb b/providers/celery.rb
@@ -61,7 +61,14 @@
   end
 
   cmds = {}
-  cmds[:celeryd] = "celeryd #{new_resource.celerycam ? "-E" : ""}" if new_resource.celeryd
+  if new_resource.celeryd
+    case new_resource.queue
+    when Array
+      cmds[:celeryd] = "celeryd -Q #{new_resource.queues.join(',')} #{new_resource.celerycam ? "-E" : ""}"
+    when NilClass
+      cmds[:celeryd] = "celeryd #{new_resource.celerycam ? "-E" : ""}"
+    end
+  end
   cmds[:celerybeat] = "celerybeat" if new_resource.celerybeat
   if new_resource.celerycam
     if new_resource.django
diff --git a/resources/celery.rb b/resources/celery.rb
@@ -28,6 +28,7 @@
 attribute :celerycam, :kind_of => [TrueClass, FalseClass], :default => false
 attribute :camera_class, :kind_of => [String, NilClass], :default => nil
 attribute :environment, :kind_of => [Hash], :default => {}
+attribute :queues, :kind_of => [Array,NilClass], :default => nil
 
 def config_base
   config.split(/[\\\/]/).last
