77 lines
3.1 KiB
Ruby
77 lines
3.1 KiB
Ruby
|
namespace :cartodb do
|
||
|
|
namespace :db do
|
||
|
|
desc 'Sets table_ids for all table metadata recors'
|
||
|
|
task :set_table_oids => :environment do
|
||
|
|
count = ::User.count
|
||
|
|
::User.all.each_with_index do |user, index|
|
||
|
|
puts "Setting table oids for #{user.username}"
|
||
|
|
begin
|
||
|
|
if %w(development staging).include?(ENV['RAILS_ENV'])
|
||
|
|
entries_in_database = Hash[
|
||
|
|
user.in_database.fetch(%Q(
|
||
|
|
SELECT table_name AS name, table_name::regclass::oid AS oid
|
||
|
|
FROM information_schema.tables
|
||
|
|
WHERE table_type = 'BASE TABLE'
|
||
|
|
AND table_catalog = 'cartodb_#{ENV['RAILS_ENV']}_user_#{user.id}_db'
|
||
|
|
AND table_schema = 'public';
|
||
|
|
)).all.map { |row| [row.fetch(:name), row.fetch(:oid)] }
|
||
|
|
]
|
||
|
|
else
|
||
|
|
entries_in_database = Hash[
|
||
|
|
user.in_database.fetch(%Q(
|
||
|
|
SELECT table_name AS name, table_name::regclass::oid AS oid
|
||
|
|
FROM information_schema.tables
|
||
|
|
WHERE table_type = 'BASE TABLE'
|
||
|
|
AND table_catalog = 'cartodb_user_#{user.id}_db'
|
||
|
|
AND table_schema = 'public';
|
||
|
|
)).all.map { |row| [row.fetch(:name), row.fetch(:oid)] }
|
||
|
|
]
|
||
|
|
end
|
||
|
|
|
||
|
|
raise 'No tables found!!!!' if entries_in_database.empty?
|
||
|
|
entries_in_metadata = Hash[
|
||
|
|
Table.fetch(%Q(
|
||
|
|
SELECT name, table_id
|
||
|
|
FROM user_tables
|
||
|
|
WHERE user_id = #{user.id}
|
||
|
|
)).map { |table| [table.name, table.table_id] }
|
||
|
|
]
|
||
|
|
entries_in_metadata
|
||
|
|
.select { |name, oid| entries_in_database[name] != oid }
|
||
|
|
.each { |name_in_metadata, oid_in_metadata|
|
||
|
|
oid = entries_in_database[name_in_metadata]
|
||
|
|
next if oid.nil? || oid.to_s.empty?
|
||
|
|
puts "Setting table oid for #{name_in_metadata}"
|
||
|
|
Table.db.run(%Q(
|
||
|
|
UPDATE user_tables
|
||
|
|
SET table_id = #{oid}
|
||
|
|
WHERE name = '#{name_in_metadata}'
|
||
|
|
AND user_id = #{user.id}
|
||
|
|
))
|
||
|
|
}
|
||
|
|
entries_in_database_by_oid = entries_in_database
|
||
|
|
.invert.delete_if { |k, v| k.nil? }
|
||
|
|
entries_in_metadata_by_oid = entries_in_metadata
|
||
|
|
.invert.delete_if { |k, v| k.nil? }
|
||
|
|
entries_in_metadata_by_oid
|
||
|
|
.select { |oid, name| entries_in_database_by_oid[oid] != name }
|
||
|
|
.each { |oid, name|
|
||
|
|
name_in_database = entries_in_database_by_oid[oid]
|
||
|
|
next unless name_in_database && name
|
||
|
|
puts "Renaming #{name} to #{name_in_database}"
|
||
|
|
table = Table.find_by_identifier(user.id, name)
|
||
|
|
table.name = name_in_database
|
||
|
|
table.save
|
||
|
|
}
|
||
|
|
|
||
|
|
printf "OK %-#{20}s (%-#{4}s/%-#{4}s)\n", user.username, index, count
|
||
|
|
rescue => exception
|
||
|
|
puts exception.backtrace.join("\n")
|
||
|
|
printf "FAIL %-#{20}s (%-#{4}s/%-#{4}s) #{exception.message}\n", user.username, index, count
|
||
|
|
end
|
||
|
|
sleep(1.0/5.0)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|