20161201160452_migrate_project_statistics.rb 865 Bytes
Newer Older
1 2 3 4
class MigrateProjectStatistics < ActiveRecord::Migration
  include Gitlab::Database::MigrationHelpers

  DOWNTIME = true
Douwe Maan's avatar
Douwe Maan committed
5
  DOWNTIME_REASON = 'Removes two columns from the projects table'.freeze
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23

  def up
    # convert repository_size in float (megabytes) to integer (bytes),
    # initialize total storage_size with repository_size
    execute <<-EOF
      INSERT INTO project_statistics (project_id, namespace_id, commit_count, storage_size, repository_size)
        SELECT id, namespace_id, commit_count, (repository_size * 1024 * 1024), (repository_size * 1024 * 1024) FROM projects
    EOF

    remove_column :projects, :repository_size
    remove_column :projects, :commit_count
  end

  def down
    add_column_with_default :projects, :repository_size, :float, default: 0.0
    add_column_with_default :projects, :commit_count, :integer, default: 0
  end
end