Class: Mongoid::Migrator

Inherits:
Object
  • Object
show all
Defined in:
lib/mongoid_migrations/active_record_ext/migrations.rb

Overview

:nodoc:

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(direction, migrations_path, target_version = nil) ⇒ Migrator

Returns a new instance of Migrator.



276
277
278
279
280
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 276

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
end

Class Method Details

.current_version ⇒ Object 



246
247
248
249
250
251
252
253
254
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 246

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

.down(migrations_path, target_version = nil) ⇒ Object 



212
213
214
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 212

def down(migrations_path, target_version = nil)
  self.new(:down, migrations_path, target_version).migrate
end

.forward(migrations_path, steps = 1) ⇒ Object 



204
205
206
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 204

def forward(migrations_path, steps=1)
  move(:up, migrations_path, steps)
end

.generate(migration_name) ⇒ Object 



224
225
226
227
228
229
230
231
232
233
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 224

def generate(migration_name)
  file_name = "#{Time.now.utc.strftime("%Y%m%d%H%M%S")}_#{migration_name.underscore}.rb"

  File.open("#{migrations_path}/#{file_name}", 'w') do |f|
    f << "class #{migration_name.camelize} < Mongoid::Migration\n"
    f << "  def self.up\n  end\n\n"
    f << "  def self.down\n  end\n"
    f << "end\n"
  end
end

.get_all_versions ⇒ Object

def schema_migrations_table_name

# Base.table_name_prefix + 'schema_migrations' + Base.table_name_suffix
'data_migrations'

end



240
241
242
243
244
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 240

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
end

.migrate(migrations_path, target_version = nil) ⇒ Object 



192
193
194
195
196
197
198
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 192

def migrate(migrations_path, target_version = nil)
  case
    when target_version.nil?              then up(migrations_path, target_version)
    when current_version > target_version then down(migrations_path, target_version)
    else                                       up(migrations_path, target_version)
  end
end

.migrations_path ⇒ Object 



220
221
222
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 220

def migrations_path
  'db/migrate'
end

.proper_table_name(name) ⇒ Object 



256
257
258
259
260
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 256

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
end

.rollback(migrations_path, steps = 1) ⇒ Object 



200
201
202
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 200

def rollback(migrations_path, steps=1)
  move(:down, migrations_path, steps)
end

.run(direction, migrations_path, target_version) ⇒ Object 



216
217
218
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 216

def run(direction, migrations_path, target_version)
  self.new(direction, migrations_path, target_version).run
end

.up(migrations_path, target_version = nil) ⇒ Object 



208
209
210
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 208

def up(migrations_path, target_version = nil)
  self.new(:up, migrations_path, target_version).migrate
end

Instance Method Details

#current_migration ⇒ Object 



286
287
288
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 286

def current_migration
  migrations.detect { |m| m.version == current_version }
end

#current_version ⇒ Object 



282
283
284
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 282

def current_version
  migrated.last || 0
end

#migrate ⇒ Object 



299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 299

def migrate
  current = migrations.detect { |m| m.version == current_version }
  target = migrations.detect { |m| m.version == @target_version }

  if target.nil? && !@target_version.nil? && @target_version > 0
    raise UnknownMigrationVersionError.new(@target_version)
  end

  start = up? ? 0 : (migrations.index(current) || 0)
  finish = migrations.index(target) || migrations.size - 1
  runnable = migrations[start..finish]

  # skip the last migration if we're headed down, but not ALL the way down
  runnable.pop if down? && !target.nil?

  runnable.each do |migration|
    # puts "Migrating to #{migration.name} (#{migration.version})"

    # On our way up, we skip migrating the ones we've already migrated
    next if up? && migrated.include?(migration.version.to_i)

    # On our way down, we skip reverting the ones we've never migrated
    if down? && !migrated.include?(migration.version.to_i)
      migration.announce 'never migrated, skipping'; migration.write
      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)
    rescue => e
      raise StandardError, "An error has occurred, #{migration.version} and all later migrations canceled:\n\n#{e}", e.backtrace
    end
  end
end

#migrated ⇒ Object 



379
380
381
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 379

def migrated
  @migrated_versions ||= self.class.get_all_versions
end

#migrations ⇒ Object 



344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 344

def migrations
  @migrations ||= begin
    files = Dir["#{@migrations_path}/[0-9]*_*.rb"]

    migrations = files.inject([]) do |klasses, file|
      version, name = file.scan(/([0-9]+)_([_a-z0-9]*).rb/).first

      raise IllegalMigrationNameError.new(file) unless version
      version = version.to_i

      if klasses.detect { |m| m.version == version }
        raise DuplicateMigrationVersionError.new(version)
      end

      if klasses.detect { |m| m.name == name.camelize }
        raise DuplicateMigrationNameError.new(name.camelize)
      end

      migration = MigrationProxy.new
      migration.name     = name.camelize
      migration.version  = version
      migration.filename = file
      klasses << migration
    end

    migrations = migrations.sort_by(&:version)
    down? ? migrations.reverse : migrations
  end
end

#pending_migrations ⇒ Object 



374
375
376
377
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 374

def pending_migrations
  already_migrated = migrated
  migrations.reject { |m| already_migrated.include?(m.version.to_i) }
end

#run ⇒ Object

Raises:



290
291
292
293
294
295
296
297
 
# File 'lib/mongoid_migrations/active_record_ext/migrations.rb', line 290

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)
  end
end