-
-
Save tboyko/7680960 to your computer and use it in GitHub Desktop.
added ability for script to automatically determine database name.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
desc "convert a latin1 database with utf8 data into proper utf8" | |
task :convert_to_utf8 => :environment do | |
puts Time.now | |
dryrun = ENV['DOIT'] != '1' | |
conn = ActiveRecord::Base.connection | |
if dryrun | |
def conn.run_sql(sql) | |
puts(sql) | |
end | |
else | |
def conn.run_sql(sql) | |
puts(sql) | |
execute(sql) | |
end | |
end | |
database_name = Rails.configuration.database_configuration[ Rails.env ]['database'] | |
conn.run_sql "ALTER DATABASE #{database_name} CHARACTER SET utf8 collate utf8_unicode_ci" | |
# Don't covert views | |
VIEWS = /(view|_v$)/ | |
big = [] | |
# These are table_name => model_class mappings that aren't rails standard or | |
# tables that we don't wish to convert (table_name => true). | |
mapping = { :pos => Pos, | |
:categories_products => true, | |
:delayed_jobs => Delayed::Job, | |
:schema_migrations => true | |
}.with_indifferent_access | |
tables = (conn.tables - big).select { |tbl| tbl !~ VIEWS } | |
puts "Converting #{tables.inspect}" | |
#(tables - big).each do |tbl| | |
tables.each do |tbl| | |
a = ['CHARACTER SET utf8 COLLATE utf8_unicode_ci'] | |
b = [] | |
model = mapping[tbl] || tbl.to_s.classify.constantize | |
model.columns.each do |col| | |
type = col.sql_type | |
nullable = col.null ? '' : ' NOT NULL' | |
default = col.default ? " DEFAULT '#{col.default}'" : '' | |
case type | |
when /varchar/ | |
a << "CHANGE #{conn.quote_column_name(col.name)} #{conn.quote_column_name(col.name)} VARBINARY(#{col.limit})" | |
b << "CHANGE #{conn.quote_column_name(col.name)} #{conn.quote_column_name(col.name)} VARCHAR(#{col.limit}) CHARACTER SET utf8 COLLATE utf8_unicode_ci#{nullable}#{default}" | |
when /text/ | |
a << "CHANGE #{conn.quote_column_name(col.name)} #{conn.quote_column_name(col.name)} BLOB" | |
b << "CHANGE #{conn.quote_column_name(col.name)} #{conn.quote_column_name(col.name)} TEXT CHARACTER SET utf8 COLLATE utf8_unicode_ci#{nullable}#{default}" | |
end | |
end unless model == true | |
conn.run_sql "ALTER TABLE #{tbl} #{a.join(', ')}" | |
conn.run_sql "ALTER TABLE #{tbl} #{b.join(', ')}" if b.present? | |
end | |
puts Time.now | |
puts 'Done!' | |
end |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment