mongoid_monkey 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 713cd432161aff92379202903badf76643f5c345
-  data.tar.gz: 2a6da7db3fd0fdb03a8828bd761fd93dac98500b
+  metadata.gz: e5814ac5f6752cc4afee52c517ed3996b7d4373f
+  data.tar.gz: 5c42d28b26b8a8c5aa7d1ca71c29397d3392560c
 SHA512:
-  metadata.gz: 417de743fe275b6dc78d88eec28ef8120505972e08e5096810f973cf862d9d2df9d150eb308f81c50041643fd4290fe90eae133ef47099a529a0c345cd6b6ca0
-  data.tar.gz: 3cbddb3acdd988d96c61a1ed1557057895520288cee81f29fce9ad657bb4e973bae50174ca4f3c0c9fe5db8d6a82ed8b909f8a3f929539e054724834bb250852
+  metadata.gz: 858f9e0888815dcebe7974e7ddd3737e3c5ce9a5117d47dc0436c9e87626e68ad639722207760dda09202f56ec7a8180325799ba7bc39cd7aaf1dc76824bc40b
+  data.tar.gz: d0c827a916d331114e8b36a75a12d118c83284d491fb055ec80d670c3924856765621a514f1b4acfe7df63505ec42239488704e782bf5d4a9145158aa0b6d275

data/lib/mongoid_monkey.rb

@@ -1,15 +1,7 @@
 require 'version'
 
-if Mongoid::VERSION =~ /\A3\./
-  require 'patches/atomic'
-  require 'patches/reorder'
-end
-
-if Mongoid::VERSION =~ /\A[345]\./
-  require 'patches/big_decimal'
-end
-
-if defined?(Moped)
-  require 'patches/instrument' if Moped::VERSION =~ /\A1\./
-  require 'patches/db_commands'
-end
+require 'patches/atomic'
+require 'patches/big_decimal'
+require 'patches/db_commands'
+require 'patches/instrument'
+require 'patches/reorder'

data/lib/patches/atomic.rb

@@ -1,3 +1,7 @@
+# Backport Mongoid 4 hash-based atomic method syntax to Mongoid 3.
+
+if Mongoid::VERSION =~ /\A3\./
+
 module Mongoid
 module Contextual
 module Atomic
@@ -107,3 +111,4 @@ module Atomic
 end
 end
 end
+end

data/lib/patches/big_decimal.rb

@@ -1,38 +1,41 @@
 # Fixes inconsistent behavior of BigDecimal. This can be removed after
 # https://github.com/mongodb/mongoid/pull/4164 is merged, planned for Mongoid 6.
 
-module MongoidMonkey
-  module Mongoid
-    module Extensions
-      module BigDecimal
+if Mongoid::VERSION =~ /\A[345]\./
 
-        def numeric?
-          true
-        end
-
-        module ClassMethods
+  module MongoidMonkey
+    module Mongoid
+      module Extensions
+        module BigDecimal
 
-          def demongoize(object)
-            object && object.numeric? ? ::BigDecimal.new(object.to_s) : nil
+          def numeric?
+            true
           end
 
-          def mongoize(object)
-            object && object.numeric? ? object.to_s : nil
+          module ClassMethods
+
+            def demongoize(object)
+              object && object.numeric? ? ::BigDecimal.new(object.to_s) : nil
+            end
+
+            def mongoize(object)
+              object && object.numeric? ? object.to_s : nil
+            end
           end
         end
-      end
 
-      module String
+        module String
 
-        def numeric?
-          true if Float(self) rescue (self =~ /^NaN|\-?Infinity$/)
+          def numeric?
+            true if Float(self) rescue (self =~ /^NaN|\-?Infinity$/)
+          end
         end
       end
     end
   end
-end
 
-::BigDecimal.__send__(:include, MongoidMonkey::Mongoid::Extensions::BigDecimal)
-::BigDecimal.extend(MongoidMonkey::Mongoid::Extensions::BigDecimal::ClassMethods)
+  ::BigDecimal.__send__(:include, MongoidMonkey::Mongoid::Extensions::BigDecimal)
+  ::BigDecimal.extend(MongoidMonkey::Mongoid::Extensions::BigDecimal::ClassMethods)
 
-::String.__send__(:include, MongoidMonkey::Mongoid::Extensions::String)
+  ::String.__send__(:include, MongoidMonkey::Mongoid::Extensions::String)
+end

data/lib/patches/db_commands.rb

@@ -4,49 +4,119 @@
 # - listIndexes
 # - createIndexes
 
-module Moped
-  class Database
+if defined?(Moped)
 
-    def collection_names
-      namespaces = command(listCollections: 1, filter: { name: { "$not" => /system\.|\$/ } })
-      namespaces["cursor"]["firstBatch"].map do |doc|
-        doc["name"]
+  module Moped
+    class Database
+
+      def collection_names
+        namespaces = command(listCollections: 1, filter: { name: { "$not" => /system\.|\$/ } })
+        namespaces["cursor"]["firstBatch"].map do |doc|
+          doc["name"]
+        end
       end
     end
   end
-end
 
-module Moped
-  class Indexes
+  module Moped
+    class Indexes
 
-    def [](key)
-      list_indexes_command.detect do |index|
-        (index['name'] == key) || (index['key'] == normalize_keys(key))
+      def [](key)
+        list_indexes_command.detect do |index|
+          (index['name'] == key) || (index['key'] == normalize_keys(key))
+        end
       end
-    end
 
-    def create(key, options = {})
-      spec = options.merge(ns: namespace, key: key)
-      spec[:name] ||= key.to_a.join("_")
-      database.command(createIndexes: collection_name, indexes: [spec])
-    end
+      def create(key, options = {})
+        spec = options.merge(ns: namespace, key: key)
+        spec[:name] ||= key.to_a.join("_")
+        database.command(createIndexes: collection_name, indexes: [spec])
+      end
 
-    def each(&block)
-      list_indexes_command.each(&block)
+      def each(&block)
+        list_indexes_command.each(&block)
+      end
+
+      protected
+
+      def list_indexes_command
+        database.command(listIndexes: collection_name)["cursor"]["firstBatch"]
+      end
+
+      def normalize_keys(spec)
+        return false if spec.is_a?(String)
+        spec.reduce({}) do |transformed, (key, value)|
+          transformed[key.to_s] = value
+          transformed
+        end
+      end
     end
+  end
+end
+
+if Mongoid::VERSION =~ /\A3\./ && defined?(Rails::Mongoid)
 
-    protected
+  module Rails::Mongoid
 
-    def list_indexes_command
-      database.command(listIndexes: collection_name)["cursor"]["firstBatch"]
+    def remove_indexes(*globs)
+      models(*globs).each do |model|
+        next if model.embedded?
+        begin
+          indexes = model.collection.indexes.map{ |doc| doc["name"] }
+          indexes.delete_one("_id_")
+          model.remove_indexes
+        rescue Moped::Errors::OperationFailure
+          next
+        end
+        logger.info("MONGOID: Removing indexes on: #{model} for: #{indexes.join(', ')}.")
+        model
+      end.compact
     end
+  end
+end
+
+if Mongoid::VERSION =~ /\A4\./
 
-    def normalize_keys(spec)
-      return false if spec.is_a?(String)
-      spec.reduce({}) do |transformed, (key, value)|
-        transformed[key.to_s] = value
-        transformed
+  module Mongoid::Tasks::Database
+
+    def undefined_indexes(models = ::Mongoid.models)
+      undefined_by_model = {}
+
+      models.each do |model|
+        unless model.embedded?
+          begin
+            model.collection.indexes.each do |index|
+                # ignore default index
+                unless index['name'] == '_id_'
+                  key = index['key'].symbolize_keys
+                  spec = model.index_specification(key)
+                  unless spec
+                    # index not specified
+                    undefined_by_model[model] ||= []
+                    undefined_by_model[model] << index
+                  end
+                end
+            end
+          rescue Moped::Errors::OperationFailure; end
+        end
       end
+
+      undefined_by_model
+    end
+
+    def remove_indexes(models = ::Mongoid.models)
+      models.each do |model|
+        next if model.embedded?
+        begin
+          indexes = model.collection.indexes.map{ |doc| doc["name"] }
+          indexes.delete_one("_id_")
+          model.remove_indexes
+        rescue Moped::Errors::OperationFailure
+          next
+        end
+        logger.info("MONGOID: Removing indexes on: #{model} for: #{indexes.join(', ')}.")
+        model
+      end.compact
     end
   end
 end

data/lib/patches/instrument.rb

@@ -1,45 +1,48 @@
 # Instrument Moped 1.x same as Moped 2.x.
 # Useful for integration with third-party services.
 
-module Moped
-  module Instrumentable
-    class Noop
+if defined?(Moped) && Moped::VERSION =~ /\A1\./
 
-      class << self
+  module Moped
+    module Instrumentable
+      class Noop
 
-        # Do not instrument anything.
-        def instrument(name, payload = {})
-          yield payload if block_given?
+        class << self
+
+          # Do not instrument anything.
+          def instrument(name, payload = {})
+            yield payload if block_given?
+          end
         end
       end
     end
   end
-end
 
-module Moped
-  module Instrumentable
+  module Moped
+    module Instrumentable
 
-    TOPIC = "query.moped"
+      TOPIC = "query.moped"
 
-    def instrumenter
-      @instrumenter ||= Moped::Instrumentable::Noop
-    end
+      def instrumenter
+        @instrumenter ||= Moped::Instrumentable::Noop
+      end
 
-    def instrument(name, payload = {}, &block)
-      instrumenter.instrument(name, payload, &block)
+      def instrument(name, payload = {}, &block)
+        instrumenter.instrument(name, payload, &block)
+      end
     end
   end
-end
 
-module Moped
-  class Node
-    include Moped::Instrumentable
+  module Moped
+    class Node
+      include Moped::Instrumentable
 
-    def logging_with_instrument(operations, &block)
-      instrument(TOPIC, prefix: "  MOPED: #{resolved_address}", ops: operations) do
-        logging_without_instrument(operations, &block)
+      def logging_with_instrument(operations, &block)
+        instrument(TOPIC, prefix: "  MOPED: #{resolved_address}", ops: operations) do
+          logging_without_instrument(operations, &block)
+        end
       end
+      alias_method_chain :logging, :instrument
     end
-    alias_method_chain :logging, :instrument
   end
 end

data/lib/patches/reorder.rb

@@ -1,11 +1,14 @@
 # Backport of Criteria#reorder method from Mongoid 4 to Mongoid 3.
 
-module Origin
-  module Optional
+if Mongoid::VERSION =~ /\A3\./
 
-    def reorder(*spec)
-      options.delete(:sort)
-      order_by(*spec)
+  module Origin
+    module Optional
+
+      def reorder(*spec)
+        options.delete(:sort)
+        order_by(*spec)
+      end
     end
   end
 end

data/lib/version.rb

@@ -1,3 +1,3 @@
 module MongoidMonkey
-  VERSION = '0.1.2'
+  VERSION = '0.1.3'
 end

data/spec/app/models/account.rb

@@ -0,0 +1,4 @@
+
+class Account
+  include Mongoid::Document
+end

data/spec/app/models/address.rb

@@ -0,0 +1,7 @@
+
+class Address
+  include Mongoid::Document
+  field :street
+  field :name, localize: true
+  embedded_in :addressable, polymorphic: true
+end

data/spec/app/models/draft.rb

@@ -0,0 +1,7 @@
+
+class Draft
+  include Mongoid::Document
+  field :text
+  recursively_embeds_one
+  index text: 1
+end

data/spec/app/models/person.rb

@@ -0,0 +1,11 @@
+
+class Person
+  include Mongoid::Document
+
+  index age: 1
+  index addresses: 1
+  index dob: 1
+  index name: 1
+  index title: 1
+  index({ ssn: 1 }, { unique: true })
+end

data/spec/app/models/user.rb

@@ -0,0 +1,6 @@
+
+class User
+  include Mongoid::Document
+  field :name
+  index name: 1
+end

data/spec/spec_helper.rb

@@ -1,8 +1,13 @@
 $LOAD_PATH.unshift(File.dirname(__FILE__))
 $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+MODELS = File.join(File.dirname(__FILE__), "app/models")
+$LOAD_PATH.unshift(MODELS)
+
 $database_name = 'mongoid_monkey_test'
 
 require 'mongoid'
+require 'rails/mongoid' if Mongoid::VERSION =~ /\A3\./
 require 'mongoid_monkey'
 require 'rspec'
 
@@ -13,6 +18,12 @@ end
 Mongoid.logger.level = Logger::INFO
 Mongo::Logger.logger.level = Logger::INFO unless Mongoid::VERSION =~ /\A[34]\./
 
+# Autoload every model for the test suite that sits in spec/app/models.
+Dir[ File.join(MODELS, "*.rb") ].sort.each do |file|
+  name = File.basename(file, ".rb")
+  autoload name.camelize.to_sym, name
+end
+
 RSpec.configure do |config|
   config.after(:all) do
     if Mongoid::VERSION =~ /\A[34]\./