mongoid_retry 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - 1.9.3
+services: mongodb

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in mongoid_retry.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Travis Dahlke
+
+MIT License
+
+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,42 @@
+[![Build Status](https://secure.travis-ci.org/travisdahlke/mongoid_retry.png)](http://travis-ci.org/travisdahlke/mongoid_retry)
+
+# MongoidRetry
+
+Overcome duplicate key errors in MongoDB by catching the exception, finding the existing document, and updating it instead.
+This is currently only compatible with Mongoid 2.x.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'mongoid_retry'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install mongoid_retry
+
+## Usage
+
+```
+class Fruit
+  include Mongoid::Document
+  include Mongoid::MongoidRetry
+
+  field :name
+  index :type, unique: true
+end
+```
+
+    Fruit.new(type: 'apple').save_and_retry
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,7 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/lib/mongoid/mongoid_retry.rb

@@ -0,0 +1,48 @@
+require "mongoid_retry/version"
+
+module Mongoid
+  module MongoidRetry
+
+    DUPLICATE_KEY_ERROR_CODES = [11000,11001]
+
+    # Catch a duplicate key error
+    def save_and_retry
+      begin
+        save!
+      rescue Mongo::OperationFailure => e
+        if is_a_duplicate_key_error?(e)
+          keys = duplicate_key(e)
+          if (duplicate = find_duplicate(keys))
+            update_document!(duplicate)
+          end
+        else
+          raise e
+        end
+      end
+    end
+
+    private
+
+    def is_a_duplicate_key_error?(exception)
+      DUPLICATE_KEY_ERROR_CODES.include?(exception.error_code)
+    end
+
+    def find_duplicate(keys)
+      self.class.first(conditions: keys)
+    end
+
+    def duplicate_key(exception)
+      index = exception.message[/(?<=\.\$)\w*/,0]
+      fields = index.split(/\d/).map{|s| s.gsub(/^_|_-?$/,'')}
+      fields.inject({}) {|hash, key| hash[key] = send(key) if respond_to?(key); hash}
+    end
+
+    def update_document!(duplicate)
+      attributes.except("_id").each_pair do |key, value|
+        duplicate[key] = value
+      end
+      duplicate.save_and_retry
+    end
+
+  end
+end

data/lib/mongoid_retry.rb

@@ -0,0 +1 @@
+require "mongoid/mongoid_retry"

data/lib/mongoid_retry/version.rb

@@ -0,0 +1,3 @@
+module MongoidRetry
+  VERSION = "0.0.1"
+end

data/mongoid_retry.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/mongoid_retry/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Travis Dahlke"]
+  gem.email         = ["travis.dahlke@tstmedia.com"]
+  gem.description   = %q{Provides a 'save_and_retry' method that will attempt to save a document and, if a duplicate key error is thrown by mongodb, will update the existing document instead of failing.}
+  gem.summary       = %q{Catch mongo duplicate key errors and retry}
+  gem.homepage      = "http://www.github.com/travisdahlke/mongoid_retry"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "mongoid_retry"
+  gem.require_paths = ["lib"]
+  gem.version       = MongoidRetry::VERSION
+
+  gem.add_runtime_dependency "mongoid", ['~> 2.0']
+  gem.add_development_dependency "rspec"
+  gem.add_development_dependency('rake', ['>= 0.9.2'])
+end

data/spec/retry_spec.rb

@@ -0,0 +1,55 @@
+require 'spec_helper'
+
+describe Mongoid::MongoidRetry do
+
+  before(:all) do
+    Thing.create_indexes
+  end
+
+  describe "#save_and_retry" do
+    describe "when a document with the same key exists" do
+      before(:each) do
+        Thing.create(name: 'apple', color: 'red')
+      end
+
+      subject { Thing.new(name: 'apple', color: 'green') }
+
+      it "should not raise an exception" do
+        subject.save_and_retry
+      end
+
+      it "should find and update the document" do
+        subject.save_and_retry
+        Thing.all.last.color.should == 'green'
+      end
+
+      it "should not create a duplicate document" do
+        subject.save_and_retry
+        Thing.count.should == 1
+      end
+    end
+
+    describe "with compound indexes" do
+      before(:each) do
+        Thing.create(name: 'apple', color: 'red', shape: 'round')
+      end
+
+      subject { Thing.new(name: 'grapefruit', color: 'red', shape: 'round') }
+
+      it "should not raise an exception" do
+        subject.save_and_retry
+      end
+
+      it "should find and update the document" do
+        subject.save_and_retry
+        Thing.all.last.name == 'block'
+      end
+
+      it "should not create a duplicate document" do
+        subject.save_and_retry
+        Thing.count.should == 1
+      end
+    end
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,22 @@
+require 'rubygems'
+require 'bundler/setup'
+
+require 'mongoid'
+require 'mongoid/mongoid_retry'
+
+require 'rspec'
+
+Mongoid.configure do |config|
+  config.master = Mongo::Connection.new.db('mongoid_retry_test')
+  config.allow_dynamic_fields = false
+  config.persist_in_safe_mode = true
+end
+
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each { |f| require f }
+
+RSpec.configure do |config|
+  config.mock_with :rspec
+  config.after :each do
+    Mongoid.master.collections.reject { |c| c.name =~ /^system\./ }.each(&:remove)
+  end
+end

data/spec/support/models/thing.rb

@@ -0,0 +1,11 @@
+class Thing
+  include Mongoid::Document
+  include Mongoid::MongoidRetry
+
+  field :name
+  field :color
+  field :shape
+
+  index :name, unique: true
+  index ([[:color, Mongo::ASCENDING], [:shape, Mongo::DESCENDING]]), unique: true
+end

metadata

@@ -0,0 +1,112 @@
+--- !ruby/object:Gem::Specification
+name: mongoid_retry
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Travis Dahlke
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mongoid
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+description: Provides a 'save_and_retry' method that will attempt to save a document
+  and, if a duplicate key error is thrown by mongodb, will update the existing document
+  instead of failing.
+email:
+- travis.dahlke@tstmedia.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .travis.yml
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/mongoid/mongoid_retry.rb
+- lib/mongoid_retry.rb
+- lib/mongoid_retry/version.rb
+- mongoid_retry.gemspec
+- spec/retry_spec.rb
+- spec/spec_helper.rb
+- spec/support/models/thing.rb
+homepage: http://www.github.com/travisdahlke/mongoid_retry
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Catch mongo duplicate key errors and retry
+test_files:
+- spec/retry_spec.rb
+- spec/spec_helper.rb
+- spec/support/models/thing.rb
+has_rdoc: