summaryrefslogtreecommitdiff
path: root/db/migrate/20161201160452_migrate_project_statistics.rb
blob: 3ae3f2c159b57c9c09e74ae6db2cbe5693650ce4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
class MigrateProjectStatistics < ActiveRecord::Migration
  include Gitlab::Database::MigrationHelpers

  DOWNTIME = true
  DOWNTIME_REASON = 'Removes two columns from the projects table'

  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