diff lib/tasks/redmine.rake @ 1517:dffacf8a6908 redmine-2.5

Update to Redmine SVN revision 13367 on 2.5-stable branch
author Chris Cannam
date Tue, 09 Sep 2014 09:29:00 +0100
parents e248c7af89ec
children
line wrap: on
line diff
--- a/lib/tasks/redmine.rake	Tue Sep 09 09:28:31 2014 +0100
+++ b/lib/tasks/redmine.rake	Tue Sep 09 09:29:00 2014 +0100
@@ -53,6 +53,58 @@
     Rake::Task["redmine:plugins:assets"].invoke
   end
 
+desc <<-DESC
+FOR EXPERIMENTAL USE ONLY, Moves Redmine data from production database to the development database.
+This task should only be used when you need to move data from one DBMS to a different one (eg. MySQL to PostgreSQL).
+WARNING: All data in the development database is deleted.
+DESC
+
+  task :migrate_dbms => :environment do
+    ActiveRecord::Base.establish_connection :development
+    target_tables = ActiveRecord::Base.connection.tables
+    ActiveRecord::Base.remove_connection
+
+    ActiveRecord::Base.establish_connection :production
+    tables = ActiveRecord::Base.connection.tables.sort - %w(schema_migrations plugin_schema_info)
+
+    if (tables - target_tables).any?
+      abort "The following table(s) are missing from the target database: #{(tables - target_tables).join(', ')}"
+    end
+
+    tables.each do |table_name|
+      Source = Class.new(ActiveRecord::Base)
+      Target = Class.new(ActiveRecord::Base)
+      Target.establish_connection(:development)
+
+      [Source, Target].each do |klass|
+        klass.table_name = table_name
+        klass.reset_column_information
+        klass.inheritance_column = "foo"
+        klass.record_timestamps = false
+      end
+      Target.primary_key = (Target.column_names.include?("id") ? "id" : nil)
+
+      source_count = Source.count
+      puts "Migrating %6d records from #{table_name}..." % source_count
+
+      Target.delete_all
+      offset = 0
+      while (objects = Source.offset(offset).limit(5000).order("1,2").to_a) && objects.any?
+        offset += objects.size
+        Target.transaction do
+          objects.each do |object|
+            new_object = Target.new(object.attributes)
+            new_object.id = object.id if Target.primary_key
+            new_object.save(:validate => false)
+          end
+        end
+      end
+      Target.connection.reset_pk_sequence!(table_name) if Target.primary_key
+      target_count = Target.count
+      abort "Some records were not migrated" unless source_count == target_count
+    end
+  end
+
   namespace :plugins do
     desc 'Migrates installed plugins.'
     task :migrate => :environment do