qu-mongo 0.1.0

data/lib/qu/backend/mongo.rb

@@ -0,0 +1,110 @@
+require 'mongo'
+
+module Qu
+  module Backend
+    class Mongo < Base
+      def connection
+        @connection ||= begin
+          uri = URI.parse(ENV['MONGOHQ_URL'].to_s)
+          database = uri.path.empty? ? 'qu' : uri.path[1..-1]
+          options = {}
+          if uri.password
+            options[:auths] = [{
+              'db_name'  => database,
+              'username' => uri.user,
+              'password' => uri.password
+            }]
+          end
+          ::Mongo::Connection.new(uri.host, uri.port, options).db(database)
+        end
+      end
+      alias_method :database, :connection
+
+      def clear(queue = nil)
+        queue ||= queues + ['failed']
+        Array(queue).each do |q|
+          jobs(q).drop
+          self[:queues].remove({:name => q})
+        end
+      end
+
+      def queues
+        self[:queues].find.map {|doc| doc['name'] }
+      end
+
+      def length(queue = 'default')
+        jobs(queue).count
+      end
+
+      def enqueue(klass, *args)
+        job = Qu::Job.new(BSON::ObjectId.new, klass, args)
+        jobs(job.queue).insert({:_id => job.id, :class => job.klass.to_s, :args => job.args})
+        self[:queues].update({:name => job.queue}, {:name => job.queue}, :upsert => true)
+        job
+      end
+
+      def reserve(worker, options = {:block => true})
+        worker.queues.each do |queue|
+          begin
+            doc = jobs(queue).find_and_modify(:remove => true)
+            return Job.new(doc['_id'], doc['class'], doc['args'])
+          rescue ::Mongo::OperationFailure
+            # No jobs in the queue
+          end
+        end
+
+        if options[:block]
+          sleep 5
+          retry
+        end
+      end
+
+      def release(job)
+        jobs(job.queue).insert({:_id => job.id, :class => job.klass.to_s, :args => job.args})
+      end
+
+      def failed(job, error)
+        jobs('failed').insert(:_id => job.id, :class => job.klass.to_s, :args => job.args, :queue => job.queue)
+      end
+
+      def completed(job)
+      end
+
+      def requeue(id)
+        doc = jobs('failed').find_and_modify(:query => {:_id => id}, :remove => true)
+        jobs(doc.delete('queue')).insert(doc)
+        Job.new(doc['_id'], doc['class'], doc['args'])
+      rescue ::Mongo::OperationFailure
+        false
+      end
+
+      def register_worker(worker)
+        self[:workers].insert(worker.attributes.merge(:id => worker.id))
+      end
+
+      def unregister_worker(id)
+        self[:workers].remove(:id => id)
+      end
+
+      def workers
+        self[:workers].find.map do |doc|
+          Qu::Worker.new(doc)
+        end
+      end
+
+      def clear_workers
+        self[:workers].drop
+      end
+
+    private
+
+      def jobs(queue)
+        self["queue:#{queue}"]
+      end
+
+      def [](name)
+        database["qu.#{name}"]
+      end
+    end
+  end
+end

data/lib/qu-mongo.rb

@@ -0,0 +1,4 @@
+require 'qu'
+require 'qu/backend/mongo'
+
+Qu.backend = Qu::Backend::Mongo.new

metadata

@@ -0,0 +1,112 @@
+--- !ruby/object:Gem::Specification 
+name: qu-mongo
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Brandon Keepers
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-09-23 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mongo
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: qu
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 27
+        segments: 
+        - 0
+        - 1
+        - 0
+        version: 0.1.0
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: bson_ext
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+description: Mongo backend for qu
+email: 
+- brandon@opensoul.org
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/qu-mongo.rb
+- lib/qu/backend/mongo.rb
+has_rdoc: true
+homepage: http://github.com/bkeepers/qu
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.6.1
+signing_key: 
+specification_version: 3
+summary: Mongo backend for qu
+test_files: []
+