delayed_job_mongo_mapper 1.0.0

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright © 2010 Andrew Timberlake
+
+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,14 @@
+# delayed_job MongoMapper backend
+
+## Installation
+
+Add the gems to your Gemfile:
+
+    gem 'delayed_job'
+    gem 'delayed_job_mongo_mapper'
+
+Create the indexes:
+
+    script/rails runner 'Delayed::Backend::MongoMapper::Job.create_indexes'
+
+That's it. Use [delayed_job as normal](http://github.com/collectiveidea/delayed_job).

data/lib/delayed/backend/mongo_mapper.rb

@@ -0,0 +1,91 @@
+module Delayed
+  module Backend
+    module MongoMapper
+      class Job
+        include ::MongoMapper::Document
+        include Delayed::Backend::Base
+        set_collection_name 'delayed_jobs'
+
+        key :priority,    Integer, :default => 0
+        key :attempts,    Integer, :default => 0
+        key :handler,     String
+        key :run_at,      Time
+        key :locked_at,   Time
+        key :locked_by,   String    #,  :index => true
+        key :failed_at,   Time
+        key :last_error,  String
+        timestamps!
+
+        #ensure_index [[:priority, 1], [:run_at, 1]]
+
+        before_save :set_default_run_at
+
+        def self.before_fork
+          ::MongoMapper.connection.close
+        end
+
+        def self.after_fork
+          ::MongoMapper.connection.connect
+        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},
+            :limit => -limit, # In mongo, positive limits are 'soft' and negative are 'hard'
+            :failed_at => nil,
+            :sort => [[:priority, 1], [:run_at, 1]]
+          }
+
+          where = "this.locked_at == null || this.locked_at < #{make_date(right_now - max_run_time)}"
+
+          (conditions[:priority] ||= {})['$gte'] = Worker.min_priority.to_i if Worker.min_priority
+          (conditions[:priority] ||= {})['$lte'] = Worker.max_priority.to_i if Worker.max_priority
+
+          results = all(conditions.merge(:locked_by => worker_name))
+          results += all(conditions.merge('$where' => where)) 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)
+          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}
+
+          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/mongo_mapper.rb

@@ -0,0 +1,14 @@
+MongoMapper::Document.class_eval do
+  yaml_as "tag:ruby.yaml.org,2002:MongoMapper"
+
+  def self.yaml_new(klass, tag, val)
+    klass.find!(val['_id'])
+  rescue MongoMapper::DocumentNotFound
+    raise Delayed::DeserializationError
+  end
+
+  def to_yaml_properties
+    ['@_id']
+  end
+end
+

data/lib/delayed_job_mongo_mapper.rb

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

data/spec/delayed_job_mongo_mapper_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+
+describe Delayed::Backend::MongoMapper::Job do
+  it_should_behave_like 'a delayed_job backend'
+
+  describe "before_fork" do
+    after do
+      ::MongoMapper.connection.close
+    end
+
+    it "should disconnect" do
+      lambda do
+        Delayed::Backend::MongoMapper::Job.before_fork
+      end.should change { !!MongoMapper.connection.connected? }.from(true).to(false)
+    end
+  end
+
+  describe "after_fork" do
+    before do
+      ::MongoMapper.connection.close
+    end
+
+    it "should call reconnect" do
+      lambda do
+        Delayed::Backend::MongoMapper::Job.after_fork
+      end.should change { !!MongoMapper.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,20 @@
+require 'rubygems'
+require 'bundler/setup'
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'rspec'
+require 'delayed_job_mongo_mapper'
+require 'delayed/backend/shared_spec'
+
+MongoMapper.database = 'dl_spec'
+
+class Story
+  include ::MongoMapper::Document
+
+  def tell; text; end
+  def whatever(n, _); tell*n; end
+  def self.count; end
+
+  handle_asynchronously :whatever
+end
+

metadata

@@ -0,0 +1,123 @@
+--- !ruby/object:Gem::Specification 
+name: delayed_job_mongo_mapper
+version: !ruby/object:Gem::Version 
+  hash: 23
+  prerelease: false
+  segments: 
+  - 1
+  - 0
+  - 0
+  version: 1.0.0
+platform: ruby
+authors: 
+- Andrew Timberlake
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-11-24 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mongo_mapper
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 51
+        segments: 
+        - 0
+        - 8
+        - 6
+        version: 0.8.6
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: delayed_job
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 9
+        segments: 
+        - 2
+        - 1
+        - 1
+        version: 2.1.1
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 2
+        - 0
+        version: "2.0"
+  type: :development
+  version_requirements: *id003
+description: 
+email: andrew@andrewtimberlake.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.md
+files: 
+- lib/delayed_job_mongo_mapper.rb
+- lib/delayed/backend/mongo_mapper.rb
+- lib/delayed/serialization/mongo_mapper.rb
+- spec/spec.opts
+- spec/delayed_job_mongo_mapper_spec.rb
+- spec/spec_helper.rb
+- LICENSE
+- README.md
+has_rdoc: true
+homepage: http://github.com/andrewtimberlake/delayed_job_mongo_mapper
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+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.3.7
+signing_key: 
+specification_version: 3
+summary: MongoMapper backend for delayed_job
+test_files: 
+- spec/spec.opts
+- spec/delayed_job_mongo_mapper_spec.rb
+- spec/spec_helper.rb