mongoid_rails_migrations 1.2.1 → 1.6.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: c8fc6474257b2d2e0046a797c410b91e3ccac8e1
-  data.tar.gz: aaed74adb2274149fe6dae16912cfa6f97dcf739
+SHA256:
+  metadata.gz: 72576a321b2015fb560f410a5909cce262d4ce35b8773e4359fde3120e44e1eb
+  data.tar.gz: da19409b30015ec555bee75b273f36800d278daca8567d9bbf16e0d7b82b5c00
 SHA512:
-  metadata.gz: 9606b16b2eab7e1e73e072eabc5bc96de07cc920d0f2c4c88352d21af7f3719ed180879fc6ce0ca6a817b2b3228d1abcac99ee98687ae9f6d040b97f7bf38780
-  data.tar.gz: 6b1c82473b629467e43239a74ed7de02ec4704fdd3c0a9ec2aadc3c8e6f3d51e5468bcb62f849d52c15892ccb8f28baa0b9c054a2e312af0a45b6df2b68088bb
+  metadata.gz: 13653010703d967705e3c67cc7d69992ef2e40426a1c84e87042dd0239a261d7750b0436e74d4854f76cf0ea207460fe3da7edb3d99ca956fa70de57dea50990
+  data.tar.gz: a355aa9a0a3cf6b2838d1d9c95088bd78f5bd2ed80ba46f99ea93d015149cae56c2eea58e93c3c86e0a8786a7bfa226ef4e9e80fc1305b115ed66de308db5971

data/lib/mongoid_rails_migrations/active_record_ext/migrations.rb

@@ -55,13 +55,13 @@ module Mongoid #:nodoc
   # If you'd prefer to use numeric prefixes, you can turn timestamped migrations
   # off by setting:
   #
-  #    Mongoid.config.timestamped_migrations = false
+  #    Mongoid.configure.timestamped_migrations = false
   #
   # In environment.rb.
   #
   class Migration
     @@verbose = true
-    cattr_accessor :verbose
+    cattr_accessor :verbose, :after_migrate, :buffer_output
 
     class << self
       def up_with_benchmarks #:nodoc:
@@ -89,6 +89,12 @@ module Mongoid #:nodoc
           when :down then announce "reverted (%.4fs)" % time.real; write
         end
 
+        begin
+          @@after_migrate.call(@@buffer_output, name, direction, false) if @@after_migrate
+          @@buffer_output = nil
+        rescue => e
+          say("Error in after_migrate hook: #{e}")
+        end
         result
       end
 
@@ -97,7 +103,6 @@ module Mongoid #:nodoc
       # it is safe for the call to proceed.
       def singleton_method_added(sym) #:nodoc:
         return if defined?(@ignore_new_methods) && @ignore_new_methods
-
         begin
           @ignore_new_methods = true
 
@@ -112,6 +117,8 @@ module Mongoid #:nodoc
       end
 
       def write(text="")
+        @@buffer_output ||=  ""
+        @@buffer_output += text + "\n"
         puts(text) if verbose
       end
 
@@ -144,22 +151,10 @@ module Mongoid #:nodoc
       end
 
       def connection
-        # ActiveRecord::Base.connection
-        if ::Mongoid.respond_to?(:default_client)
-          ::Mongoid.default_client
+        if ENV['MONGOID_CLIENT_NAME']
+          Mongoid.client(ENV['MONGOID_CLIENT_NAME'])
         else
-          ::Mongoid.default_session
-        end
-      end
-
-      def method_missing(method, *arguments, &block)
-        arg_list = arguments.map(&:inspect) * ', '
-
-        say_with_time "#{method}(#{arg_list})" do
-          # unless arguments.empty? || method == :execute
-          #   arguments[0] = Migrator.proper_table_name(arguments.first)
-          # end
-          connection.send(method, *arguments, &block)
+          Mongoid.default_client
         end
       end
     end
@@ -168,25 +163,25 @@ module Mongoid #:nodoc
   # MigrationProxy is used to defer loading of the actual migration classes
   # until they are needed
   class MigrationProxy
-
-    attr_accessor :name, :version, :filename
+    attr_accessor :name, :version, :filename, :sharded
 
     delegate :migrate, :announce, :write, :to=>:migration
 
     private
 
-      def migration
-        @migration ||= load_migration
-      end
-
-      def load_migration
-        require(File.expand_path(filename))
-        name.constantize
-      end
+    def migration
+      @migration ||= load_migration
+    end
 
+    def load_migration
+      require(File.expand_path(filename))
+      name.constantize
+    end
   end
 
   class Migrator#:nodoc:
+    delegate :with_mongoid_client, :to => "self.class"
+
     class << self
       attr_writer :migrations_path
 
@@ -208,7 +203,10 @@ module Mongoid #:nodoc
 
       def rollback_to(migrations_path, target_version)
         all_versions = get_all_versions
-        rollback_to = all_versions.index(target_version.to_i) + 1
+        target_version_index = all_versions.index(target_version.to_i)
+        raise UnknownMigrationVersionError.new(target_version) if target_version_index.nil?
+
+        rollback_to = target_version_index + 1
         rollback_steps = all_versions.size - rollback_to
         rollback migrations_path, rollback_steps
       end
@@ -233,31 +231,22 @@ module Mongoid #:nodoc
         @migrations_path ||= ['db/migrate']
       end
 
-      # def schema_migrations_table_name
-      #   # Base.table_name_prefix + 'schema_migrations' + Base.table_name_suffix
-      #   'data_migrations'
-      # end
-
       def get_all_versions
-        # table = Arel::Table.new(schema_migrations_table_name)
-        #         Base.connection.select_values(table.project(table['version']).to_sql).map(&:to_i).sort
-        DataMigration.all.map { |datamigration| datamigration.version.to_i }.sort
+        with_mongoid_client(ENV['MONGOID_CLIENT_NAME']) do
+          DataMigration.all.map { |datamigration| datamigration.version.to_i }.sort
+        end
       end
 
       def current_version
-        # sm_table = schema_migrations_table_name
-        # if Base.connection.table_exists?(sm_table)
-        #   get_all_versions.max || 0
-        # else
-        #   0
-        # end
         get_all_versions.max || 0
       end
 
-      def proper_table_name(name)
-        # Use the Active Record objects own table_name, or pre/suffix from ActiveRecord::Base if name is a symbol/string
-        # name.table_name rescue "#{ActiveRecord::Base.table_name_prefix}#{name}#{ActiveRecord::Base.table_name_suffix}"
-        name
+      def with_mongoid_client(mongoid_client_name, &block)
+        previous_mongoid_client_name = Mongoid::Threaded.client_override
+        Mongoid.override_client(mongoid_client_name)
+        block.call
+      ensure
+        Mongoid.override_client(previous_mongoid_client_name)
       end
 
       private
@@ -275,9 +264,12 @@ module Mongoid #:nodoc
     end
 
     def initialize(direction, migrations_path, target_version = nil)
-      # raise StandardError.new("This database does not yet support migrations") unless Base.connection.supports_migrations?
-      # Base.connection.initialize_schema_migrations_table
       @direction, @migrations_path, @target_version = direction, migrations_path, target_version
+
+      @mongoid_client_name = ENV["MONGOID_CLIENT_NAME"]
+      if @mongoid_client_name && !Mongoid.clients.has_key?(@mongoid_client_name)
+        raise Mongoid::Errors::NoClientConfig.new(@mongoid_client_name)
+      end
     end
 
     def current_version
@@ -291,9 +283,12 @@ module Mongoid #:nodoc
     def run
       target = migrations.detect { |m| m.version == @target_version }
       raise UnknownMigrationVersionError.new(@target_version) if target.nil?
+
       unless (up? && migrated.include?(target.version.to_i)) || (down? && !migrated.include?(target.version.to_i))
         target.migrate(@direction)
-        record_version_state_after_migrating(target.version)
+        with_mongoid_client(@mongoid_client_name) do
+          record_version_state_after_migrating(target.version)
+        end
       end
     end
 
@@ -312,19 +307,19 @@ module Mongoid #:nodoc
           next
         end
 
-        # begin
-        #   ddl_transaction do
-        #     migration.migrate(@direction)
-        #     record_version_state_after_migrating(migration.version)
-        #   end
-        # rescue => e
-        #   canceled_msg = Base.connection.supports_ddl_transactions? ? "this and " : ""
-        #   raise StandardError, "An error has occurred, #{canceled_msg}all later migrations canceled:\n\n#{e}", e.backtrace
-        # end
         begin
           migration.migrate(@direction)
-          record_version_state_after_migrating(migration.version)
+          with_mongoid_client(@mongoid_client_name) do
+            record_version_state_after_migrating(migration.version)
+          end
         rescue => e
+          output = Migration.buffer_output + "An error has occurred, #{migration.version} and all later migrations canceled:\n\n#{e}\n#{e.backtrace.join("\n")}"
+          begin
+            Migration.after_migrate.call(output, migration.name, @direction, true) if Migration.after_migrate
+            Migration.buffer_output = nil
+          rescue => error
+            puts("Error in after_migrate hook: #{error}")
+          end
           raise StandardError, "An error has occurred, #{migration.version} and all later migrations canceled:\n\n#{e}", e.backtrace
         end
       end
@@ -345,7 +340,7 @@ module Mongoid #:nodoc
     def migrations
       @migrations ||= begin
         files = Array(@migrations_path).inject([]) do |files, path|
-          files += Dir["#{path}/[0-9]*_*.rb"]
+          files += Dir["#{path}/**/[0-9]*_*.rb"]
         end
 
         migrations = files.inject([]) do |klasses, file|
@@ -363,10 +358,15 @@ module Mongoid #:nodoc
           end
 
           migration = MigrationProxy.new
+          migration.sharded  = file.match?(/\/shards\/#{version}_#{name}\.rb/)
           migration.name     = name.camelize
           migration.version  = version
           migration.filename = file
-          klasses << migration
+
+          if (@mongoid_client_name && migration.sharded) || (!@mongoid_client_name && !migration.sharded)
+            klasses << migration
+          end
+          klasses
         end
 
         migrations = migrations.sort_by(&:version)
@@ -402,42 +402,24 @@ module Mongoid #:nodoc
     end
 
     private
-      def record_version_state_after_migrating(version)
-        # table = Arel::Table.new(self.class.schema_migrations_table_name)
-
-        @migrated_versions ||= []
-        # if down?
-        #   @migrated_versions.delete(version)
-        #   table.where(table["version"].eq(version.to_s)).delete
-        # else
-        #   @migrated_versions.push(version).sort!
-        #   table.insert table["version"] => version.to_s
-        # end
-        if down?
-          @migrated_versions.delete(version)
-          DataMigration.where(:version => version.to_s).first.destroy
-        else
-          @migrated_versions.push(version).sort!
-          DataMigration.create(:version => version.to_s)
-        end
-      end
 
-      def up?
-        @direction == :up
+    def record_version_state_after_migrating(version)
+      @migrated_versions ||= []
+      if down?
+        @migrated_versions.delete(version)
+        DataMigration.where(:version => version.to_s).destroy_all
+      else
+        @migrated_versions.push(version).sort!
+        DataMigration.find_or_create_by(:version => version.to_s)
       end
+    end
 
-      def down?
-        @direction == :down
-      end
+    def up?
+      @direction == :up
+    end
 
-      # Wrap the migration in a transaction only if supported by the adapter.
-      def ddl_transaction(&block)
-        # if Base.connection.supports_ddl_transactions?
-        #   Base.transaction { block.call }
-        # else
-        #   block.call
-        # end
-        block.call
-      end
+    def down?
+      @direction == :down
+    end
   end
 end

data/lib/mongoid_rails_migrations/models/data_migration.rb

@@ -2,4 +2,4 @@ class DataMigration
   include Mongoid::Document
 
   field :version
-end
+end

data/lib/mongoid_rails_migrations/mongoid_ext/mongoid.rb

@@ -1,21 +1,7 @@
 # encoding: utf-8
 module Mongoid
   # Specify whether or not to use timestamps for migration versions
-  # NOTE: newer style is a module
-  # Config.module_eval would work for both but still need to determine the type
-  # that's why we do it the ug way.
-  if Config.is_a? Class
-    # older mongoid style; pre 2.0.0.rc.1
-    Config.module_eval do
-      cattr_accessor :timestamped_migrations
-      class_variable_set(:@@timestamped_migrations, true) unless class_variable_get(:@@timestamped_migrations)
-
-      def self.reset
-        @@timestamped_migrations = true
-      end
-    end
-  else # module
-    # newer mongoid style; >= 2.0.0.rc.1
-    Config.option :timestamped_migrations, :default => true
-  end
-end
+  Config.option :timestamped_migrations, default: true
+  # Specify whether or not to use shards migrations as default type
+  Config.option :shards_migration_as_default, default: false
+end

data/lib/mongoid_rails_migrations/mongoid_ext/railties/database.rake

@@ -1,34 +1,12 @@
 namespace :db do
-  unless Rake::Task.task_defined?("db:drop")
-    desc 'Drops all the collections for the database for the current Rails.env'
-    task :drop => :environment do
-      Mongoid.master.collections.each {|col| col.drop_indexes && col.drop unless ['system.indexes', 'system.users'].include?(col.name) }
-    end
-  end
-
-  unless Rake::Task.task_defined?("db:seed")
-    # if another ORM has defined db:seed, don't run it twice.
-    desc 'Load the seed data from db/seeds.rb'
-    task :seed => :environment do
-      seed_file = File.join(Rails.application.root, 'db', 'seeds.rb')
-      load(seed_file) if File.exist?(seed_file)
-    end
+  if Rake::Task.task_defined?("db:drop")
+    Rake::Task["db:drop"].clear
   end
 
-  unless Rake::Task.task_defined?("db:setup")
-    desc 'Create the database, and initialize with the seed data'
-    task :setup => [ 'db:create', 'db:seed' ]
-  end
-
-  unless Rake::Task.task_defined?("db:reseed")
-    desc 'Delete data and seed'
-    task :reseed => [ 'db:drop', 'db:seed' ]
-  end
-
-  unless Rake::Task.task_defined?("db:create")
-    task :create => :environment do
-      # noop
-    end
+  desc 'Drops the database for the current Mongoid client'
+  task :drop => :environment do
+    # Unlike Mongoid's default, this implementation supports the MONGOID_CLIENT_NAME override
+    Mongoid::Migration.connection.database.drop
   end
 
   desc 'Current database version'
@@ -43,7 +21,7 @@ namespace :db do
   end
 
   namespace :migrate do
-    desc  'Rollback the database one migration and re migrate up. If you want to rollback more than one step, define STEP=x. Target specific version with VERSION=x.'
+    desc 'Rollback the database one migration and re migrate up. If you want to rollback more than one step, define STEP=x. Target specific version with VERSION=x.'
     task :redo => :environment do
       if ENV["VERSION"]
         Rake::Task["db:migrate:down"].invoke
@@ -74,7 +52,7 @@ namespace :db do
 
     desc 'Display status of migrations'
     task :status => :environment do
-      Mongoid::Migrator.status("db/migrate/")
+      Mongoid::Migrator.status(Mongoid::Migrator.migrations_path)
     end
   end
 
@@ -88,18 +66,6 @@ namespace :db do
   task :rollback_to => :environment do
     version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
     raise "VERSION is required" unless version
-    Mongoid::Migrator.rollback_to('db/migrate/', version)
-  end
-
-  namespace :schema do
-    task :load do
-      # noop
-    end
-  end
-
-  namespace :test do
-    task :prepare do
-      # Stub out for MongoDB
-    end
+    Mongoid::Migrator.rollback_to(Mongoid::Migrator.migrations_path, version)
   end
 end

data/lib/mongoid_rails_migrations/version.rb

@@ -1,3 +1,3 @@
 module MongoidRailsMigrations #:nodoc:
-  VERSION = '1.2.1'
+  VERSION = '1.6.1'
 end

data/lib/rails/generators/mongoid/migration/migration_generator.rb

@@ -3,9 +3,16 @@ require 'rails/generators/mongoid/mongoid_generator'
 module Mongoid
   module Generators
     class MigrationGenerator < Base
+      class_option :shards, type: :boolean, optional: true, desc: "Create migration in shards subfolder"
 
       def create_migration_file
-        migration_template "migration.rb", "db/migrate/#{file_name}.rb"
+        destination_folder = "db/migrate"
+        if options.fetch(:shards, Config.shards_migration_as_default)
+          destination_folder = "#{destination_folder}/shards"
+          FileUtils.mkdir_p("#{Rails.root}/#{destination_folder}")
+        end
+
+        migration_template "migration.rb", "#{destination_folder}/#{file_name}.rb"
       end
 
       protected
@@ -13,4 +20,4 @@ module Mongoid
 
     end
   end
-end
+end

data/lib/rails/generators/mongoid/migration/templates/migration.rb

@@ -4,4 +4,4 @@ class <%= migration_class_name %> < Mongoid::Migration
 
   def self.down
   end
-end
+end

data/mongoid_rails_migrations.gemspec

@@ -5,8 +5,8 @@ require 'mongoid_rails_migrations/version'
 Gem::Specification.new do |spec|
   spec.name        = 'mongoid_rails_migrations'
   spec.version     = MongoidRailsMigrations::VERSION
-  spec.authors     = ['Alan Da Costa']
-  spec.email      = ['alandacosta@gmail.com']
+  spec.authors     = ['Alan Da Costa', 'Adrien Rey-Jarthon']
+  spec.email      = ['alandacosta@gmail.com', 'jobs@adrienjarthon.com']
 
   spec.summary     = 'Data migrations for Mongoid.'
   spec.description = 'Data migrations for Mongoid in Active Record style, minus column input.'
@@ -19,7 +19,9 @@ Gem::Specification.new do |spec|
   rails_version = '>= 4.2.0'
 
   spec.add_runtime_dependency('bundler', '>= 1.0.0')
-  spec.add_runtime_dependency('mongoid', '>= 4.0.0')
+  # 9.0.0 broke client override isolation: https://jira.mongodb.org/browse/MONGOID-5815
+  # got fixed in 9.0.3 so rejectings versions in between to prevent bad surprises:
+  spec.add_runtime_dependency('mongoid', '>= 5.0.0', '!= 9.0.0', '!= 9.0.1', '!= 9.0.2')
   spec.add_runtime_dependency('rails',  rails_version)
   spec.add_runtime_dependency('railties',  rails_version)
   spec.add_runtime_dependency('activesupport',  rails_version)

metadata

@@ -1,14 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: mongoid_rails_migrations
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.6.1
 platform: ruby
 authors:
 - Alan Da Costa
-autorequire: 
+- Adrien Rey-Jarthon
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2019-02-04 00:00:00.000000000 Z
+date: 2024-11-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -30,14 +31,32 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 4.0.0
+        version: 5.0.0
+    - - "!="
+      - !ruby/object:Gem::Version
+        version: 9.0.0
+    - - "!="
+      - !ruby/object:Gem::Version
+        version: 9.0.1
+    - - "!="
+      - !ruby/object:Gem::Version
+        version: 9.0.2
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 4.0.0
+        version: 5.0.0
+    - - "!="
+      - !ruby/object:Gem::Version
+        version: 9.0.0
+    - - "!="
+      - !ruby/object:Gem::Version
+        version: 9.0.1
+    - - "!="
+      - !ruby/object:Gem::Version
+        version: 9.0.2
 - !ruby/object:Gem::Dependency
   name: rails
   requirement: !ruby/object:Gem::Requirement
@@ -111,6 +130,7 @@ dependencies:
 description: Data migrations for Mongoid in Active Record style, minus column input.
 email:
 - alandacosta@gmail.com
+- jobs@adrienjarthon.com
 executables: []
 extensions: []
 extra_rdoc_files: []
@@ -130,7 +150,7 @@ homepage: http://github.com/adacosta/mongoid_rails_migrations
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -145,10 +165,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.5.1
-signing_key: 
+rubygems_version: 3.5.16
+signing_key:
 specification_version: 4
 summary: Data migrations for Mongoid.
 test_files: []
-has_rdoc: