mongoid-multi-db 3.0.0

data/lib/mongoid/persistence/atomic/add_to_set.rb

@@ -0,0 +1,35 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This class provides the ability to perform an explicit $addToSet
+      # modification on a specific field.
+      class AddToSet
+        include Operation
+
+        # Sends the atomic $addToSet operation to the database.
+        #
+        # @example Persist the new values.
+        #   addToSet.persist
+        #
+        # @return [ Object ] The new array value.
+        #
+        # @since 2.0.0
+        def persist
+          prepare do
+            document[field] = [] unless document[field]
+            values = document.send(field)
+            values.push(value) unless values.include?(value)
+            values.tap do
+              if document.persisted?
+                collection.update(document.atomic_selector, operation("$addToSet"), options)
+                document.remove_change(field)
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/bit.rb

@@ -0,0 +1,37 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This operation is for performing $bit atomic operations against the
+      # database.
+      class Bit
+        include Operation
+
+        # Execute the bitwise operation. This correlates to a $bit in MongoDB.
+        #
+        # @example Execute the op.
+        #   bit.persist
+        #
+        # @return [ Integer ] The new value.
+        #
+        # @since 2.1.0
+        def persist
+          prepare do
+            current = document[field]
+            return nil unless current
+            document[field] = value.inject(current) do |result, (bit, val)|
+              result = result & val if bit.to_s == "and"
+              result = result | val if bit.to_s == "or"
+              result
+            end
+            document[field].tap do
+              collection.update(document.atomic_selector, operation("$bit"), options)
+              document.remove_change(field)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/inc.rb

@@ -0,0 +1,31 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This class provides atomic $inc behaviour.
+      class Inc
+        include Operation
+
+        # Sends the atomic $inc operation to the database.
+        #
+        # @example Persist the new values.
+        #   inc.persist
+        #
+        # @return [ Object ] The new integer value.
+        #
+        # @since 2.0.0
+        def persist
+          prepare do
+            current = document[field] || 0
+            document[field] = current + value
+            document[field].tap do
+              collection.update(document.atomic_selector, operation("$inc"), options)
+              document.remove_change(field)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/operation.rb

@@ -0,0 +1,85 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This is the included module for all atomic operation objects.
+      module Operation
+
+        attr_accessor :document, :field, :value, :options
+
+        # Get the collection to be used for persistence.
+        #
+        # @example Get the collection.
+        #   operation.collection
+        #
+        # @return [ Collection ] The root collection.
+        #
+        # @since 2.1.0
+        def collection
+          document._root.collection
+        end
+
+        # Initialize the new pullAll operation.
+        #
+        # @example Create a new pullAll operation.
+        #   PullAll.new(document, :aliases, [ "Bond" ])
+        #
+        # @param [ Document ] document The document to pullAll onto.
+        # @param [ Symbol ] field The name of the array field.
+        # @param [ Object ] value The value to pullAll.
+        # @param [ Hash ] options The persistence options.
+        #
+        # @since 2.0.0
+        def initialize(document, field, value, options = {})
+          @document, @field, @value = document, field.to_s, value
+          @options = Safety.merge_safety_options(options)
+        end
+
+        # Get the atomic operation to perform.
+        #
+        # @example Get the operation.
+        #   inc.operation
+        #
+        # @param [ String ] modifier The modifier to use.
+        #
+        # @return [ Hash ] The atomic operation for the field and addition.
+        #
+        # @since 2.0.0
+        def operation(modifier)
+          { modifier => { path => value } }
+        end
+
+        # Get the path to the field that is getting atomically updated.
+        #
+        # @example Get the path.
+        #   operation.path
+        #
+        # @return [ String, Symbol ] The path to the field.
+        #
+        # @since 2.1.0
+        def path
+          position = document.atomic_position
+          position.blank? ? field : "#{position}.#{field}"
+        end
+
+        # All atomic operations use this with a block to ensure saftey options
+        # clear out after the execution.
+        #
+        # @example Prepare the operation.
+        #   prepare do
+        #     collection.update
+        #   end
+        #
+        # @return [ Object ] The yielded value.
+        #
+        # @since 2.1.0
+        def prepare
+          yield(document).tap do
+            Threaded.clear_options!
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/pop.rb

@@ -0,0 +1,34 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This class provides the ability to perform an explicit $pop
+      # modification on a specific field.
+      class Pop
+        include Operation
+
+        # Sends the atomic $pop operation to the database.
+        #
+        # @example Persist the new values.
+        #   pop.persist
+        #
+        # @return [ Object ] The new array value.
+        #
+        # @since 2.1.0
+        def persist
+          prepare do
+            if document[field]
+              values = document.send(field)
+              value > 0 ? values.pop : values.shift
+              values.tap do
+                collection.update(document.atomic_selector, operation("$pop"), options)
+                document.remove_change(field) if document.persisted?
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/pull.rb

@@ -0,0 +1,34 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This class provides the ability to perform an explicit $pull
+      # modification on a specific field.
+      class Pull
+        include Operation
+
+        # Sends the atomic $pull operation to the database.
+        #
+        # @example Persist the new values.
+        #   pull.persist
+        #
+        # @return [ Object ] The new array value.
+        #
+        # @since 2.1.0
+        def persist
+          prepare do
+            if document[field]
+              values = document.send(field)
+              values.delete(value)
+              values.tap do
+                collection.update(document.atomic_selector, operation("$pull"), options)
+                document.remove_change(field) if document.persisted?
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/pull_all.rb

@@ -0,0 +1,34 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This class provides the ability to perform an explicit $pullAll
+      # modification on a specific field.
+      class PullAll
+        include Operation
+
+        # Sends the atomic $pullAll operation to the database.
+        #
+        # @example Persist the new values.
+        #   pull_all.persist
+        #
+        # @return [ Object ] The new array value.
+        #
+        # @since 2.0.0
+        def persist
+          prepare do
+            if document[field]
+              values = document.send(field)
+              values.delete_if { |val| value.include?(val) }
+              values.tap do
+                collection.update(document.atomic_selector, operation("$pullAll"), options)
+                document.remove_change(field) if document.persisted?
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/push.rb

@@ -0,0 +1,31 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This class provides the ability to perform an explicit $push modification
+      # on a specific field.
+      class Push
+        include Operation
+
+        # Sends the atomic $push operation to the database.
+        #
+        # @example Persist the new values.
+        #   push.persist
+        #
+        # @return [ Object ] The new array value.
+        #
+        # @since 2.0.0
+        def persist
+          prepare do
+            document[field] = [] unless document[field]
+            document.send(field).push(value).tap do |value|
+              collection.update(document.atomic_selector, operation("$push"), options)
+              document.remove_change(field)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/push_all.rb

@@ -0,0 +1,31 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This class provides the ability to perform an explicit $pushAll modification
+      # on a specific field.
+      class PushAll
+        include Operation
+
+        # Sends the atomic $pushAll operation to the database.
+        #
+        # @example Persist the new values.
+        #   pushAll.persist
+        #
+        # @return [ Object ] The new array value.
+        #
+        # @since 2.1.0
+        def persist
+          prepare do
+            document[field] = [] unless document[field]
+            document.send(field).concat(value).tap do |value|
+              collection.update(document.atomic_selector, operation("$pushAll"), options)
+              document.remove_change(field)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/rename.rb

@@ -0,0 +1,31 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # Performs an atomic rename operation.
+      class Rename
+        include Operation
+
+        # Sends the atomic $inc operation to the database.
+        #
+        # @example Persist the new values.
+        #   inc.persist
+        #
+        # @return [ Object ] The new integer value.
+        #
+        # @since 2.1.0
+        def persist
+          prepare do
+            @value = value.to_s
+            document[value] = document.attributes.delete(field)
+            document[value].tap do
+              collection.update(document.atomic_selector, operation("$rename"), options)
+              document.remove_change(value)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/sets.rb

@@ -0,0 +1,30 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # This class provides atomic $set behaviour.
+      class Sets
+        include Operation
+
+        # Sends the atomic $set operation to the database.
+        #
+        # @example Persist the new values.
+        #   set.persist
+        #
+        # @return [ Object ] The new field value.
+        #
+        # @ssete 2.0.0
+        def persist
+          prepare do
+            value ? document[field] = value : @value = document[field]
+            document[field].tap do
+              collection.update(document.atomic_selector, operation("$set"), options)
+              document.remove_change(field)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/persistence/atomic/unset.rb

@@ -0,0 +1,28 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Persistence #:nodoc:
+    module Atomic #:nodoc:
+
+      # Performs atomic $unset operations.
+      class Unset
+        include Operation
+
+        # Sends the atomic $unset operation to the database.
+        #
+        # @example Persist the new values.
+        #   unset.persist
+        #
+        # @return [ nil ] The new value.
+        #
+        # @since 2.1.0
+        def persist
+          prepare do
+            document.attributes.delete(field)
+            collection.update(document.atomic_selector, operation("$unset"), options)
+            document.remove_change(value)
+          end
+        end
+      end
+    end
+  end
+end