Commit 45a105b1 authored by Kamil Trzcinski's avatar Kamil Trzcinski

Use pure SQL queries to migrate CI tags

parent a6f2caf7
...@@ -4,45 +4,37 @@ module Ci ...@@ -4,45 +4,37 @@ module Ci
module Migrate module Migrate
class Tags class Tags
def restore def restore
puts 'Migrating tags for Runners... ' ActiveRecord::Base.transaction do
list_objects('Runner').each do |id| puts 'Inserting tags...'
putc '.' connection.execute(
runner = Ci::Runner.find_by_id(id) 'INSERT INTO tags (name) ' +
if runner 'SELECT ci_tags.name FROM ci_tags ' +
tags = list_tags('Runner', id) 'WHERE (SELECT COUNT(*) FROM tags WHERE tags.name = ci_tags.name)=0'
runner.update_attributes(tag_list: tags) )
end
end
puts ''
puts 'Migrating tags for Builds... ' puts 'Deleting old records'
list_objects('Build').each do |id| connection.execute "DELETE FROM taggings WHERE context = 'tags' AND taggable_type LIKE 'Ci::%'"
putc '.'
build = Ci::Build.find_by_id(id)
if build
tags = list_tags('Build', id)
build.update_attributes(tag_list: tags)
end
end
puts ''
end
protected puts 'Inserting tags...'
connection.execute(
'INSERT INTO taggings (taggable_type, taggable_id, tag_id, context) ' +
"SELECT CONCAT('Ci::', ci_taggings.taggable_type), ci_taggings.taggable_id, tags.id, 'tags' FROM ci_taggings " +
'JOIN ci_tags ON ci_tags.id = ci_taggings.tag_id ' +
'JOIN tags ON tags.name = ci_tags.name '
)
def list_objects(type) puts 'Resetting counters... '
ids = ActiveRecord::Base.connection.select_all( connection.execute(
"select distinct taggable_id from ci_taggings where taggable_type = #{ActiveRecord::Base::sanitize(type)}" 'UPDATE tags SET ' +
'taggings_count = (SELECT COUNT(*) FROM taggings WHERE tags.id = taggings.tag_id)'
) )
ids.map { |id| id['taggable_id'] } end
end end
def list_tags(type, id) protected
tags = ActiveRecord::Base.connection.select_all(
'select ci_tags.name from ci_tags ' + def connection
'join ci_taggings on ci_tags.id = ci_taggings.tag_id ' + ActiveRecord::Base.connection
"where taggable_type = #{ActiveRecord::Base::sanitize(type)} and taggable_id = #{ActiveRecord::Base::sanitize(id)} and context = 'tags'"
)
tags.map { |tag| tag['name'] }
end end
end end
end end
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment