delayed_job_mongoid 1.0.0.rc

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright © 2010 Collective Idea
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,10 @@
+# delayed_job Mongoid backend
+
+## Installation
+
+Add the gems to you Gemfile:
+
+  gem 'delayed_job', '2.1.0.pre2'
+  gem 'delayed_job_mongoid', '1.0.0.rc'
+  
+That's it. Use [delayed_job as normal](http://github.com/collectiveidea/delayed_job).

data/lib/delayed/backend/mongoid.rb

@@ -0,0 +1,82 @@
+module Delayed
+  module Backend
+    module Mongoid
+      class Job
+        include ::Mongoid::Document
+        include ::Mongoid::Timestamps
+        include Delayed::Backend::Base
+        field :priority,    :type=> Integer, :default => 0
+        field :attempts,    :type=> Integer, :default => 0
+        field :handler,     :type=> String
+        field :run_at,      :type=> Time
+        field :locked_at,   :type=> Time
+        field :locked_by,   :type=> String, :index => true
+        field :failed_at,   :type=> Time
+        field :last_error,  :type=> String
+      
+      
+        before_save :set_default_run_at
+        
+        def self.before_fork
+          ::Mongoid.master.connection.close
+        end
+
+        def self.after_fork
+          ::Mongoid.master.connection.connect_to_master
+        end
+        
+        def self.db_time_now
+          Time.now.utc
+        end
+        
+        def self.find_available(worker_name, limit = 5, max_run_time = Worker.max_run_time)
+          right_now = db_time_now
+
+          conditions = {:run_at  => {"$lte" => right_now}, :failed_at => nil}
+          (conditions[:priority] ||= {})['$gte'] = Worker.min_priority.to_i if Worker.min_priority
+          (conditions[:priority] ||= {})['$lte'] = Worker.max_priority.to_i if Worker.max_priority
+         
+          
+          where = "this.locked_at == null || this.locked_at < #{make_date(right_now - max_run_time)}"
+          results = self.where(conditions.merge(:locked_by => worker_name)).limit(-limit).order_by([['priority', 1], ['run_at', 1]]).to_a
+          results += self.where(conditions.merge('$where' => where)).limit(-limit+results.size).order_by([['priority', 1], ['run_at', 1]]).to_a if results.size < limit
+          results
+        end
+        # When a worker is exiting, make sure we don't have any locked jobs.
+        def self.clear_locks!(worker_name)
+          self.collection.update({:locked_by => worker_name}, {"$set" => {:locked_at => nil, :locked_by => nil}}, :multi => true)
+        end
+        
+        # Lock this job for this worker.
+        # Returns true if we have the lock, false otherwise.
+        def lock_exclusively!(max_run_time, worker = worker_name)
+          right_now = self.class.db_time_now
+          overtime = right_now - max_run_time.to_i
+          
+          query = "this.locked_at == null || this.locked_at < #{make_date(overtime)} || this.locked_by == #{worker.to_json}"
+          conditions = {:_id => id, :run_at => {"$lte" => right_now}, "$where" => query}
+          
+          self.collection.update(conditions, {"$set" => {:locked_at => right_now, :locked_by => worker}})
+          affected_rows = self.collection.find({:_id => id, :locked_by => worker}).count
+          if affected_rows == 1
+            self.locked_at = right_now
+            self.locked_by = worker
+            return true
+          else
+            return false
+          end
+        end
+        
+        private
+
+        def self.make_date(date_or_seconds)
+        "new Date(#{date_or_seconds.to_f * 1000})"
+        end
+
+        def make_date(date)
+          self.class.make_date(date)
+        end
+      end
+    end
+  end
+end

data/lib/delayed/serialization/mongoid.rb

@@ -0,0 +1,16 @@
+Mongoid::Document.class_eval do
+  yaml_as "tag:ruby.yaml.org,2002:Mongoid"
+  
+  def self.yaml_new(klass, tag, val)
+    begin
+      klass.find(val['attributes']['_id'])
+    rescue Mongoid::Errors::DocumentNotFound
+      nil
+    end
+  end
+  
+  def to_yaml_properties
+    ['@attributes']
+  end
+end
+

data/lib/delayed_job_mongoid.rb

@@ -0,0 +1,6 @@
+require 'mongoid'
+require 'delayed_job'
+require 'delayed/serialization/mongoid'
+require 'delayed/backend/mongoid'
+
+Delayed::Worker.backend = :mongoid

data/spec/delayed_job_mongoid_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+
+describe Delayed::Backend::Mongoid::Job do
+  it_should_behave_like 'a delayed_job backend'
+  
+  describe "before_fork" do
+    after do
+      ::Mongoid.master.connection.close
+    end
+    
+    it "should disconnect" do
+      lambda do
+        Delayed::Backend::Mongoid::Job.before_fork
+      end.should change { !!Mongoid.master.connection.connected? }.from(true).to(false)
+    end
+  end
+
+  describe "after_fork" do
+    before do
+      ::Mongoid.master.connection.close
+    end
+    
+    it "should call reconnect" do
+      lambda do
+        Delayed::Backend::Mongoid::Job.after_fork
+      end.should change { !!Mongoid.master.connection.connected? }.from(false).to(true)
+    end
+  end
+end

data/spec/spec.opts

@@ -0,0 +1 @@
+--color

data/spec/spec_helper.rb

@@ -0,0 +1,26 @@
+require 'rubygems'
+require 'bundler/setup'
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'spec'
+require 'spec/autorun'
+require 'delayed_job_mongoid'
+require 'delayed/backend/shared_spec'
+
+Spec::Runner.configure do |config|
+  
+end
+
+Mongoid.configure do |config|
+  config.master = config.master = Mongo::Connection.new.db('dl_spec') 
+end
+
+class Story
+  include ::Mongoid::Document
+  def tell; text; end       
+  def whatever(n, _); tell*n; end
+  def self.count; end
+
+  handle_asynchronously :whatever
+end
+

metadata

@@ -0,0 +1,114 @@
+--- !ruby/object:Gem::Specification 
+name: delayed_job_mongoid
+version: !ruby/object:Gem::Version 
+  prerelease: true
+  segments: 
+  - 1
+  - 0
+  - 0
+  - rc
+  version: 1.0.0.rc
+platform: ruby
+authors: 
+- Brandon Keepers
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-09-05 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mongoid
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 2
+        - 0
+        version: "2.0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: delayed_job
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 2
+        - 1
+        version: "2.1"
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 2
+        - 9
+        version: 1.2.9
+  type: :development
+  version_requirements: *id003
+description: 
+email: brandon@collectiveidea.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.md
+files: 
+- lib/delayed/backend/mongoid.rb
+- lib/delayed/serialization/mongoid.rb
+- lib/delayed_job_mongoid.rb
+- spec/delayed_job_mongoid_spec.rb
+- spec/spec.opts
+- spec/spec_helper.rb
+- LICENSE
+- README.md
+has_rdoc: true
+homepage: http://github.com/collectiveidea/delayed_job_mongoid
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 1
+      - 3
+      - 1
+      version: 1.3.1
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Mongoid backend for delayed_job
+test_files: 
+- spec/delayed_job_mongoid_spec.rb
+- spec/spec.opts
+- spec/spec_helper.rb