1452 lines
61 KiB
Ruby
1452 lines
61 KiB
Ruby
require_relative 'thread_pool'
|
|
require_relative '../../services/dataservices-metrics/lib/geocoder_usage_metrics'
|
|
require 'timeout'
|
|
require 'date'
|
|
|
|
namespace :cartodb do
|
|
namespace :db do
|
|
|
|
#################
|
|
# LOAD TABLE OIDS
|
|
#################
|
|
desc 'Load table oids'
|
|
task :load_oids => :environment do
|
|
count = ::User.count
|
|
::User.all.each_with_index do |user, i|
|
|
begin
|
|
user.link_outdated_tables
|
|
printf "OK %-#{20}s (%-#{4}s/%-#{4}s)\n", user.username, i, count
|
|
rescue StandardError => e
|
|
printf "FAIL %-#{20}s (%-#{4}s/%-#{4}s) #{e.message}\n", user.username, i, count
|
|
end
|
|
#sleep(1.0/5.0)
|
|
end
|
|
end
|
|
|
|
desc 'Copy user api_keys from redis to postgres'
|
|
task :copy_api_keys_from_redis => :environment do
|
|
count = ::User.count
|
|
::User.all.each_with_index do |user, i|
|
|
begin
|
|
user.this.update api_key: $users_metadata.HGET(user.key, 'map_key')
|
|
raise 'No API key!!' if user.reload.api_key.blank?
|
|
puts "(#{i+1} / #{count}) OK #{user.username}"
|
|
rescue StandardError => e
|
|
puts "(#{i+1} / #{count}) FAIL #{user.username} #{e.message}"
|
|
end
|
|
end
|
|
end # copy_api_keys_from_redis
|
|
|
|
desc 'Rebuild user tables/layers join table'
|
|
task :register_table_dependencies => :environment do
|
|
count = Map.count
|
|
|
|
Map.all.each_with_index do |map, i|
|
|
begin
|
|
map.data_layers.each do |layer|
|
|
layer.register_table_dependencies
|
|
printf "OK (%-#{4}s/%-#{4}s)\n", i, count
|
|
end
|
|
rescue StandardError => e
|
|
printf "FAIL (%-#{4}s/%-#{4}s) #{e}\n", i, count
|
|
end
|
|
end
|
|
end
|
|
|
|
desc 'Removes duplicated indexes created in some accounts'
|
|
task :remove_duplicate_indexes, [:database_host, :sleep, :dryrun] => :environment do |t, args|
|
|
threads = 1
|
|
thread_sleep = 1
|
|
database_host = args[:database_host].blank? ? nil : args[:database_host]
|
|
sleep = args[:sleep].blank? ? 3 : args[:sleep].to_i
|
|
dryrun = args[:dryrun] == 'false' ? 'false' : 'true'
|
|
|
|
if database_host.nil?
|
|
count = ::User.count
|
|
else
|
|
count = ::User.where(database_host: database_host).count
|
|
end
|
|
execute_on_users_with_index(:remove_duplicate_indexes.to_s, Proc.new { |user, i|
|
|
begin
|
|
user.in_database(:as => :superuser) do |db|
|
|
db.transaction do
|
|
db.run(%Q{
|
|
CREATE OR REPLACE FUNCTION CDB_DropDupUnique(dryrun boolean DEFAULT true)
|
|
RETURNS void
|
|
LANGUAGE plpgsql
|
|
VOLATILE
|
|
AS $$
|
|
DECLARE
|
|
rec RECORD;
|
|
sql TEXT;
|
|
BEGIN
|
|
|
|
FOR rec IN SELECT
|
|
c.conname, r.oid tab
|
|
FROM
|
|
pg_constraint c,
|
|
pg_class r
|
|
WHERE c.conrelid > 0
|
|
AND c.conrelid = r.oid
|
|
AND c.contype = 'u'
|
|
AND EXISTS (
|
|
SELECT * FROM pg_constraint pc
|
|
WHERE pc.conrelid = c.conrelid -- same target table
|
|
AND pc.conkey = c.conkey -- samekey
|
|
AND pc.contype = 'p' -- index is a primary one
|
|
)
|
|
LOOP
|
|
|
|
IF NOT dryrun THEN
|
|
RAISE NOTICE 'Constraint % on table % is not needed, dropping',
|
|
rec.conname, rec.tab::regclass::text;
|
|
sql := 'ALTER TABLE ' || rec.tab::regclass::text
|
|
|| ' DROP CONSTRAINT ' || quote_ident (rec.conname);
|
|
RAISE DEBUG 'Running: %', sql;
|
|
EXECUTE sql;
|
|
ELSE
|
|
RAISE NOTICE 'Constraint % on table % is not needed (dry run)',
|
|
rec.conname, rec.tab::regclass::text;
|
|
END IF;
|
|
|
|
END LOOP;
|
|
|
|
END;
|
|
$$;
|
|
})
|
|
db.run(%Q{
|
|
SELECT CDB_DropDupUnique(#{dryrun});
|
|
})
|
|
db.run(%Q{
|
|
DROP FUNCTION IF EXISTS CDB_DropDupUnique(boolean);
|
|
})
|
|
end
|
|
end
|
|
|
|
log(sprintf("OK %-#{20}s %-#{20}s (%-#{4}s/%-#{4}s)\n", user.username, user.database_name, i+1, count), :remove_duplicate_indexes.to_s, database_host)
|
|
sleep(sleep)
|
|
rescue StandardError => e
|
|
log(sprintf("FAIL %-#{20}s (%-#{4}s/%-#{4}s) #{e.message}\n", user.username, i+1, count), :remove_duplicate_indexes.to_s, database_host)
|
|
puts "FAIL:#{i} #{e.message}"
|
|
end
|
|
}, threads, thread_sleep, database_host)
|
|
end
|
|
|
|
desc 'Unregisters extraneous members from the "cartodb" extension'
|
|
task :unregister_extraneous_cartodb_members, [:database_host, :sleep, :dryrun] => :environment do |t, args|
|
|
threads = 1
|
|
thread_sleep = 1
|
|
database_host = args[:database_host].blank? ? nil : args[:database_host]
|
|
sleep = args[:sleep].blank? ? 3 : args[:sleep].to_i
|
|
dryrun = args[:dryrun] == 'false' ? 'false' : 'true'
|
|
|
|
if database_host.nil?
|
|
count = ::User.count
|
|
else
|
|
count = ::User.where(database_host: database_host).count
|
|
end
|
|
execute_on_users_with_index(:unregister_extraneous_cartodb_members.to_s, Proc.new { |user, i|
|
|
begin
|
|
user.in_database(:as => :superuser) do |db|
|
|
db.transaction do
|
|
db.run(%Q{
|
|
CREATE OR REPLACE FUNCTION CDB_DropExtraneousExtMembers(dryrun boolean DEFAULT true)
|
|
RETURNS void
|
|
LANGUAGE plpgsql
|
|
VOLATILE
|
|
AS $$
|
|
DECLARE
|
|
rec RECORD;
|
|
sql TEXT;
|
|
BEGIN
|
|
|
|
FOR rec IN SELECT 'ALTER EXTENSION cartodb DROP '
|
|
|| pg_describe_object(d.classid, d.objid, 0) || ';' as q
|
|
FROM pg_extension e LEFT OUTER JOIN pg_depend d
|
|
ON ( d.refobjid = e.oid )
|
|
WHERE d.refclassid = 'pg_catalog.pg_extension'::pg_catalog.regclass
|
|
AND e.extname = 'cartodb'
|
|
AND d.classid = 'pg_catalog.pg_class'::pg_catalog.regclass -- relations
|
|
AND d.objid != 'cartodb.cdb_tablemetadata'::pg_catalog.regclass
|
|
AND d.deptype = 'e' -- members
|
|
LOOP
|
|
|
|
IF NOT dryrun THEN
|
|
RAISE NOTICE 'Running on %: %', current_database(), rec.q;
|
|
EXECUTE rec.q;
|
|
ELSE
|
|
RAISE NOTICE 'Would run on %: %', current_database(), rec.q;
|
|
END IF;
|
|
|
|
END LOOP;
|
|
|
|
END;
|
|
$$;
|
|
})
|
|
db.run(%Q{
|
|
SET client_min_messages TO notice;
|
|
})
|
|
db.run(%Q{
|
|
SELECT CDB_DropExtraneousExtMembers(#{dryrun});
|
|
})
|
|
db.run(%Q{
|
|
DROP FUNCTION IF EXISTS CDB_DropExtraneousExtMembers(boolean);
|
|
})
|
|
end
|
|
end
|
|
|
|
log(sprintf("OK %-#{20}s %-#{20}s (%-#{4}s/%-#{4}s)\n", user.username, user.database_name, i+1, count), :unregister_extraneous_cartodb_members.to_s, database_host)
|
|
sleep(sleep)
|
|
rescue StandardError => e
|
|
log(sprintf("FAIL %-#{20}s (%-#{4}s/%-#{4}s) #{e.message}\n", user.username, i+1, count), :unregister_extraneous_cartodb_members.to_s, database_host)
|
|
puts "FAIL:#{i} #{e.message}"
|
|
end
|
|
}, threads, thread_sleep, database_host)
|
|
end
|
|
|
|
########################
|
|
# LOAD CARTODB FUNCTIONS
|
|
########################
|
|
# e.g. bundle exec rake cartodb:db:load_functions['127.0.0.1','0.5.0']
|
|
# bundle exec rake cartodb:db:load_functions[,'0.5.0']
|
|
desc 'Install/upgrade CARTODB SQL functions'
|
|
task :load_functions, [:database_host, :version, :num_threads, :thread_sleep, :sleep, :statement_timeout] => :environment do |task_name, args|
|
|
# Send this as string, not as number
|
|
extension_version = args[:version].blank? ? nil : args[:version]
|
|
database_host = args[:database_host].blank? ? nil : args[:database_host]
|
|
threads = args[:num_threads].blank? ? 1 : args[:num_threads].to_i
|
|
thread_sleep = args[:thread_sleep].blank? ? 0.25 : args[:thread_sleep].to_f
|
|
sleep = args[:sleep].blank? ? 1 : args[:sleep].to_i
|
|
statement_timeout = args[:statement_timeout].blank? ? 180000 : args[:statement_timeout]
|
|
|
|
puts "Running extension update with following config:"
|
|
puts "extension_version: #{extension_version.nil? ? 'UNSPECIFIED/LATEST' : extension_version}"
|
|
puts "database_host: #{database_host.nil? ? 'ALL' : database_host}"
|
|
puts "threads: #{threads}"
|
|
puts "thread_sleep: #{thread_sleep}"
|
|
puts "sleep: #{sleep}"
|
|
puts "statement_timeout: #{statement_timeout}"
|
|
|
|
if database_host.nil?
|
|
count = ::User.count
|
|
else
|
|
count = ::User.where(database_host: database_host).count
|
|
end
|
|
execute_on_users_with_index(task_name, Proc.new { |user, i|
|
|
begin
|
|
log(sprintf("Trying on %-#{20}s %-#{20}s (%-#{4}s/%-#{4}s)...", user.username, user.database_name, i+1, count), task_name, database_host)
|
|
user.db_service.load_cartodb_functions(statement_timeout, extension_version)
|
|
log(sprintf("OK %-#{20}s %-#{20}s (%-#{4}s/%-#{4}s)", user.username, user.database_name, i+1, count), task_name, database_host)
|
|
sleep(sleep)
|
|
rescue StandardError => e
|
|
log(sprintf("FAIL %-#{20}s (%-#{4}s/%-#{4}s) #{e.message}", user.username, i+1, count), task_name, database_host)
|
|
puts "FAIL:#{i} #{e.message}"
|
|
end
|
|
}, threads, thread_sleep, database_host)
|
|
end
|
|
|
|
desc 'Upgrade cartodb postgresql extension'
|
|
task :upgrade_postgres_extension, [:database_host, :version, :sleep, :statement_timeout] => :environment do |task_name, args|
|
|
# Send this as string, not as number
|
|
extension_version = args[:version]
|
|
database_host = args[:database_host]
|
|
sleep = args[:sleep].blank? ? 0.5 : args[:sleep].to_i
|
|
statement_timeout = args[:statement_timeout].blank? ? 180000 : args[:statement_timeout] # 3 min by default
|
|
|
|
puts "Upgrading cartodb extension with following config:"
|
|
puts "extension_version: #{extension_version || 'LATEST'}"
|
|
puts "database_host: #{database_host || 'ALL'}"
|
|
puts "sleep: #{sleep}"
|
|
puts "statement_timeout: #{statement_timeout}"
|
|
|
|
query = User
|
|
query = query.where(database_host: database_host) if database_host
|
|
|
|
count = query.count
|
|
|
|
query.order(Sequel.asc(:created_at)).each_with_index do |user, i|
|
|
begin
|
|
# We grant 2 x statement_timeout, by default 6 min
|
|
Timeout::timeout(statement_timeout/1000 * 2) do
|
|
log(sprintf("Trying on %-#{20}s %-#{20}s (%-#{4}s/%-#{4}s)...", user.username, user.database_name, i+1, count), task_name, database_host)
|
|
user.db_service.upgrade_cartodb_postgres_extension(statement_timeout, extension_version)
|
|
log(sprintf("OK %-#{20}s %-#{20}s (%-#{4}s/%-#{4}s)", user.username, user.database_name, i+1, count), task_name, database_host)
|
|
end
|
|
rescue StandardError => e
|
|
log(sprintf("FAIL %-#{20}s (%-#{4}s/%-#{4}s) #{e.message}", user.username, i+1, count), task_name, database_host)
|
|
puts "FAIL:#{i} #{e.message}"
|
|
end
|
|
sleep(sleep)
|
|
end
|
|
end
|
|
|
|
|
|
desc 'Install/upgrade Varnish trigger for a single user'
|
|
task :load_varnish_trigger_user, [:username] => :environment do |t, args|
|
|
user = ::User.find(username: args[:username])
|
|
user.db_service.create_function_invalidate_varnish
|
|
end
|
|
|
|
desc 'Move user to its own schema'
|
|
task :move_user_to_schema, [:username] => :environment do |t, args|
|
|
user = ::User.find(username: args[:username])
|
|
user.db_service.move_to_own_schema
|
|
user.db_service.setup_organization_user_schema
|
|
user.save
|
|
end
|
|
|
|
desc 'Install/upgrade Varnish invalidation trigger'
|
|
task :load_varnish_trigger, [:num_threads, :thread_sleep, :database_host, :sleep] => :environment do |t, args|
|
|
threads = args[:num_threads].blank? ? 1 : args[:num_threads].to_i
|
|
thread_sleep = args[:thread_sleep].blank? ? 0.1 : args[:thread_sleep].to_f
|
|
database_host = args[:database_host].blank? ? nil : args[:database_host]
|
|
sleep = args[:sleep].blank? ? 5 : args[:sleep].to_i
|
|
|
|
if database_host.nil?
|
|
count = ::User.count
|
|
else
|
|
count = ::User.where(database_host: database_host).count
|
|
end
|
|
execute_on_users_with_index(:load_varnish_trigger.to_s, Proc.new { |user, i|
|
|
begin
|
|
user.db_service.create_function_invalidate_varnish
|
|
log(sprintf("OK %-#{20}s %-#{20}s (%-#{4}s/%-#{4}s)\n", user.username, user.database_name, i+1, count), :load_varnish_trigger.to_s, database_host)
|
|
sleep(sleep)
|
|
rescue StandardError => e
|
|
log(sprintf("FAIL %-#{20}s (%-#{4}s/%-#{4}s) #{e.message}\n", user.username, i+1, count), :load_varnish_trigger.to_s, database_host)
|
|
puts "FAIL:#{i} #{e.message}"
|
|
end
|
|
}, threads, thread_sleep, database_host)
|
|
end
|
|
|
|
######################################
|
|
# GRANT `publicuser` ROLE TO ALL USERS
|
|
######################################
|
|
desc 'Grant `publicuser` role to all users'
|
|
task :grant_publicuser_to_all_users => :environment do
|
|
Carto::User.pluck(:id).each do |user_id|
|
|
user = Carto::User.where(id: user_id).first
|
|
# already granted users will raise a NOTICE
|
|
grant_query = "GRANT publicuser to \"#{user.database_username}\""
|
|
begin
|
|
conn = user.in_database(as: :cluster_admin)
|
|
conn.execute(grant_query)
|
|
rescue StandardError => e
|
|
log("Failed to execute `#{grant_query}`", :grant_publicuser_to_all_users.to_s, user.database_host)
|
|
ensure
|
|
conn.close unless conn.nil?
|
|
end
|
|
end
|
|
end
|
|
|
|
##########################################
|
|
# SET ORGANIZATION GROUP ROLE TO ALL USERS
|
|
##########################################
|
|
desc 'Set organization member group role'
|
|
task :set_user_as_organization_member => :environment do
|
|
::User.all.each do |user|
|
|
next if !user.respond_to?('database_name') || user.database_name.blank?
|
|
user.db_service.set_user_as_organization_member
|
|
end
|
|
end
|
|
|
|
##############
|
|
# SET DB PERMS
|
|
##############
|
|
desc "Set/Fix DB Permissions"
|
|
task set_permissions: :environment do
|
|
::User.all.each do |user|
|
|
next if !user.respond_to?('database_name') || user.database_name.blank?
|
|
# !!! WARNING
|
|
# This will delete all database permissions, and try to recreate them from scratch.
|
|
# Use only if you know what you're doing. (or, better, don't use it)
|
|
user.db_service.reset_database_permissions
|
|
user.db_service.reset_user_schema_permissions
|
|
user.db_service.grant_publicuser_in_database
|
|
user.db_service.set_user_privileges_at_db
|
|
user.db_service.fix_table_permissions
|
|
end
|
|
end
|
|
|
|
desc 'Set user privileges in CartoDB schema and CDB_TableMetadata'
|
|
task :set_user_privileges_in_cartodb_schema, [:username] => :environment do |t, args|
|
|
user = ::User.find(username: args[:username])
|
|
user.db_service.set_user_privileges_in_cartodb_schema
|
|
end
|
|
|
|
desc 'Set all user privileges'
|
|
task :set_all_user_privileges, [:username] => :environment do |t, args|
|
|
user = ::User.find(username: args[:username])
|
|
user.db_service.grant_user_in_database
|
|
user.db_service.set_user_privileges_at_db
|
|
end
|
|
|
|
desc 'Set org role privileges in all organizations'
|
|
task :set_org_privileges_in_cartodb_schema, [:org_name] => :environment do |_t, args|
|
|
org = ::Organization.find(name: args[:org_name])
|
|
owner = org.owner
|
|
if owner
|
|
owner.db_service.setup_organization_role_permissions
|
|
else
|
|
puts 'Organization without owner'
|
|
end
|
|
end
|
|
|
|
desc 'Set org role privileges in all organizations'
|
|
task set_all_orgs_privileges_in_cartodb_schema: :environment do |_t, _args|
|
|
Organization.each do |org|
|
|
owner = org.owner
|
|
if owner
|
|
owner.db_service.setup_organization_role_permissions
|
|
else
|
|
puts "Organization without owner: #{org.name}"
|
|
end
|
|
end
|
|
end
|
|
|
|
##########################
|
|
# SET TRIGGER CHECK QUOTA
|
|
##########################
|
|
desc 'reset check quota trigger on all user tables'
|
|
task :reset_trigger_check_quota, [:start] => :environment do |_, args|
|
|
start = args[:start]
|
|
puts "Resetting check quota trigger for ##{::User.count} users"
|
|
i = 0
|
|
|
|
query = ::User.order(:id)
|
|
query = query.where('id > ?', start) if start
|
|
query.use_cursor(rows_per_fetch: 500).each do |user|
|
|
begin
|
|
puts "Setting user quota in db '#{user.database_name}' (#{user.id} #{user.username})"
|
|
user.db_service.rebuild_quota_trigger
|
|
rescue StandardError => exception
|
|
puts "\nERRORED #{user.id} (#{user.username}): #{exception.message}\n"
|
|
end
|
|
|
|
i += 1
|
|
if (i % 500).zero?
|
|
puts "\nProcessed ##{i} users"
|
|
end
|
|
end
|
|
end
|
|
|
|
desc 'reset check quota trigger for a given user'
|
|
task :reset_trigger_check_quota_for_user, [:username] => :environment do |t, args|
|
|
raise 'usage: rake cartodb:db:reset_trigger_check_quota_for_user[username]' if args[:username].blank?
|
|
puts "Resetting trigger check quota for user '#{args[:username]}'"
|
|
user = ::User.filter(:username => args[:username]).first
|
|
user.db_service.rebuild_quota_trigger
|
|
end
|
|
|
|
desc "set users quota to amount in mb"
|
|
task :set_user_quota, [:username, :quota_in_mb] => :environment do |t, args|
|
|
usage = 'usage: rake cartodb:db:set_user_quota[username,quota_in_mb]'
|
|
raise usage if args[:username].blank? || args[:quota_in_mb].blank?
|
|
|
|
user = ::User.filter(:username => args[:username]).first
|
|
quota = args[:quota_in_mb].to_i * 1024 * 1024
|
|
user.update(:quota_in_bytes => quota)
|
|
|
|
puts "Setting user quota in db '#{user.database_name}' (#{user.username})"
|
|
user.db_service.rebuild_quota_trigger
|
|
|
|
puts "User: #{user.username} quota updated to: #{args[:quota_in_mb]}MB. #{user.tables.count} tables updated."
|
|
end
|
|
|
|
|
|
###############
|
|
# SET ORG QUOTA
|
|
###############
|
|
desc "set organization quota to amount in GB"
|
|
task :set_organization_quota, [:organization_name, :quota_in_gb] => :environment do |t, args|
|
|
usage = 'usage: rake cartodb:db:set_organization_quota[organization_name,quota_in_gb]'
|
|
raise usage if args[:organization_name].blank? || args[:quota_in_gb].blank?
|
|
|
|
organization = Organization.filter(:name=> args[:organization_name]).first
|
|
quota = args[:quota_in_gb].to_i * 1024 * 1024 * 1024
|
|
organization.quota_in_bytes = quota
|
|
organization.save
|
|
|
|
puts "Organization: #{organization.name} quota updated to: #{args[:quota_in_gb]}GB."
|
|
end
|
|
|
|
desc "set organization seats"
|
|
task :set_organization_seats, [:organization_name, :seats] => :environment do |t, args|
|
|
usage = 'usage: rake cartodb:db:set_organization_seats[organization_name,seats]'
|
|
raise usage if args[:organization_name].blank? || args[:seats].blank?
|
|
|
|
organization = Organization.filter(:name=> args[:organization_name]).first
|
|
seats = args[:seats].to_i
|
|
organization.seats = seats
|
|
organization.save
|
|
|
|
puts "Organization: #{organization.name} seats updated to: #{args[:seats]}."
|
|
end
|
|
|
|
desc "set organization viewer_seats"
|
|
task :set_organization_viewer_seats, [:organization_name, :seats] => :environment do |_, args|
|
|
usage = 'usage: rake cartodb:db:set_organization_viewer_seats[organization_name,seats]'
|
|
raise usage if args[:organization_name].blank? || args[:seats].blank?
|
|
|
|
organization = Organization.filter(name: args[:organization_name]).first
|
|
seats = args[:seats].to_i
|
|
organization.viewer_seats = seats
|
|
organization.save
|
|
|
|
puts "Organization: #{organization.name} seats updated to: #{args[:seats]}."
|
|
end
|
|
|
|
#################
|
|
# SET TABLE QUOTA
|
|
#################
|
|
desc "set users table quota"
|
|
task :set_user_table_quota, [:username, :table_quota] => :environment do |t, args|
|
|
usage = "usage: rake cartodb:db:set_user_table_quota[username,table_quota]"
|
|
raise usage if args[:username].blank? || args[:table_quota].blank?
|
|
|
|
user = ::User.filter(:username => args[:username]).first
|
|
user.update(:table_quota => args[:table_quota].to_i)
|
|
|
|
puts "User: #{user.username} table quota updated to: #{args[:table_quota]}"
|
|
end
|
|
|
|
desc "set unlimited table quota"
|
|
task :set_unlimited_table_quota, [:username] => :environment do |t, args|
|
|
usage = "usage: rake cartodb:db:set_unlimited_table_quota[username]"
|
|
raise usage if args[:username].blank?
|
|
|
|
user = ::User.filter(:username => args[:username]).first
|
|
user.update(:table_quota => nil)
|
|
|
|
puts "User: #{user.username} table quota updated to: unlimited"
|
|
end
|
|
|
|
|
|
desc "reset Users table quota to 5"
|
|
task :set_all_users_to_free_table_quota => :environment do
|
|
::User.all.each do |user|
|
|
next if !user.respond_to?('database_name') || user.database_name.blank?
|
|
user.update(:table_quota => 5) if user.table_quota.blank?
|
|
end
|
|
end
|
|
|
|
|
|
##################
|
|
# SET ACCOUNT TYPE
|
|
##################
|
|
desc "Set users account type. DEDICATED or FREE"
|
|
task :set_user_account_type, [:username, :account_type] => :environment do |t, args|
|
|
usage = "usage: rake cartodb:db:set_user_account_type[username,account_type]"
|
|
raise usage if args[:username].blank? || args[:account_type].blank?
|
|
|
|
user = ::User.filter(:username => args[:username]).first
|
|
user.update(:account_type => args[:account_type])
|
|
|
|
puts "User: #{user.username} table account type updated to: #{args[:account_type]}"
|
|
end
|
|
|
|
desc "reset all Users account type to FREE"
|
|
task :set_all_users_account_type_to_free => :environment do
|
|
::User.all.each do |user|
|
|
next if !user.respond_to?('database_name') || user.database_name.blank?
|
|
user.update(:account_type => 'FREE') if user.account_type.blank?
|
|
end
|
|
end
|
|
|
|
|
|
##########################################
|
|
# SET USER PRIVATE TABLES ENABLED/DISABLED
|
|
##########################################
|
|
desc "set users private tables enabled"
|
|
task :set_user_private_tables_enabled, [:username, :private_tables_enabled] => :environment do |t, args|
|
|
usage = "usage: rake cartodb:db:set_user_private_tables_enabled[username,private_tables_enabled]"
|
|
raise usage if args[:username].blank? || args[:private_tables_enabled].blank?
|
|
|
|
user = ::User.filter(:username => args[:username]).first
|
|
user.update(:private_tables_enabled => args[:private_tables_enabled])
|
|
|
|
puts "User: #{user.username} private tables enabled: #{args[:private_tables_enabled]}"
|
|
end
|
|
|
|
desc "reset all Users privacy tables permissions type to false"
|
|
task :set_all_users_private_tables_enabled_to_false => :environment do
|
|
::User.all.each do |user|
|
|
next if !user.respond_to?('database_name') || user.database_name.blank?
|
|
user.update(:private_tables_enabled => false) if user.private_tables_enabled.blank?
|
|
end
|
|
end
|
|
|
|
|
|
##########################
|
|
# REBUILD GEOM WEBMERCATOR
|
|
##########################
|
|
desc "Add the_geom_webmercator column to every table which needs it"
|
|
task :add_the_geom_webmercator => :environment do
|
|
::User.all.each do |user|
|
|
tables = Table.filter(:user_id => user.id).all
|
|
next if tables.empty?
|
|
user.db_service.load_cartodb_functions
|
|
puts "Updating tables in db '#{user.database_name}' (#{user.username})"
|
|
tables.each do |table|
|
|
has_the_geom = false
|
|
user.in_database do |user_database|
|
|
begin
|
|
flatten_schema = user_database.schema(table.name.to_sym).flatten
|
|
rescue StandardError => e
|
|
puts " Skipping table #{table.name}: #{e}"
|
|
next
|
|
end
|
|
has_the_geom = true if flatten_schema.include?(:the_geom)
|
|
if flatten_schema.include?(:the_geom) && !flatten_schema.include?(Table::THE_GEOM_WEBMERCATOR.to_sym)
|
|
puts " Updating table #{table.name}"
|
|
geometry_type = if col = user_database["select GeometryType(the_geom) FROM #{table.name} limit 1"].first
|
|
col[:geometrytype]
|
|
end
|
|
geometry_type ||= "POINT"
|
|
user_database.run("SELECT public.AddGeometryColumn('#{user.database_schema}','#{table.name}','#{Table::THE_GEOM_WEBMERCATOR}',3857,'#{geometry_type}',2)")
|
|
user_database.run("CREATE INDEX #{table.name}_#{Table::THE_GEOM_WEBMERCATOR}_idx ON #{table.name} USING GIST(#{Table::THE_GEOM_WEBMERCATOR})")
|
|
user_database.run("ANALYZE #{table.name}")
|
|
table.save_changes
|
|
else
|
|
puts " Skipping table #{table.name}: does not have 'the_geom' or has '#{Table::THE_GEOM_WEBMERCATOR}' already"
|
|
end
|
|
end
|
|
if has_the_geom
|
|
table.set_trigger_the_geom_webmercator
|
|
|
|
user.in_database do |user_database|
|
|
user_database.run("ALTER TABLE #{table.name} DROP CONSTRAINT IF EXISTS enforce_srid_the_geom")
|
|
user_database.run("update #{table.name} set \"#{Table::THE_GEOM_WEBMERCATOR}\" = CDB_TransformToWebmercator(the_geom)")
|
|
user_database.run("ALTER TABLE #{table.name} ADD CONSTRAINT enforce_srid_the_geom CHECK (st_srid(the_geom) = #{CartoDB::SRID})")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
desc "Update test_quota trigger"
|
|
task :update_test_quota_trigger => :environment do
|
|
::User.all.each do |user|
|
|
puts "Setting user quota in db '#{user.database_name}' (#{user.username})"
|
|
user.db_service.rebuild_quota_trigger
|
|
end
|
|
end
|
|
|
|
desc "Recreates all table triggers for non-orgs users. Optionally you can pass a username param to do this only for one user. Example: cartodb:db:recreate_table_triggers['my_username']"
|
|
task :recreate_table_triggers, [:username] => :environment do |t, args|
|
|
username = args[:username]
|
|
|
|
users = username.nil? ? ::User.where('organization_id IS NOT NULL') : ::User.where(username: username)
|
|
users.each do |user|
|
|
if user.db_service.cartodb_extension_version_pre_mu? || user.database_schema=='public'
|
|
puts "SKIP: #{user.username} / #{user.id}"
|
|
else
|
|
schema_name = user.database_schema
|
|
UserTable.filter(:user_id => user.id).each do |table|
|
|
table_name = "#{user.database_schema}.#{table.name}"
|
|
begin
|
|
user.in_database do |user_db|
|
|
user_db.run(%Q{
|
|
SELECT cartodb._CDB_drop_triggers('#{table_name}'::REGCLASS);
|
|
})
|
|
user_db.run(%Q{
|
|
SELECT cartodb._CDB_create_triggers('#{schema_name}'::TEXT, '#{table_name}'::REGCLASS);
|
|
})
|
|
end
|
|
rescue StandardError => exception
|
|
puts "ERROR: #{user.username} / #{user.id} : #{table_name} #{exception}"
|
|
end
|
|
end
|
|
puts "DONE: #{user.username} / #{user.id}"
|
|
end
|
|
end
|
|
puts 'Finished'
|
|
end
|
|
|
|
desc "Update update_the_geom_webmercator_trigger"
|
|
task :update_the_geom_webmercator_trigger => :environment do
|
|
::User.all.each do |user|
|
|
user.db_service.load_cartodb_functions
|
|
|
|
tables = Table.filter(:user_id => user.id).all
|
|
next if tables.empty?
|
|
puts "Updating tables in db '#{user.database_name}' (#{user.username})"
|
|
tables.each do |table|
|
|
has_the_geom = false
|
|
user.in_database do |user_database|
|
|
begin
|
|
has_the_geom = true if user_database.schema(table.name.to_sym).flatten.include?(:the_geom)
|
|
rescue StandardError => e
|
|
puts " Skipping table #{table.name}: #{e}"
|
|
next
|
|
end
|
|
end
|
|
if has_the_geom
|
|
puts " Updating the_geom_webmercator triggers for table #{table.name}"
|
|
table.set_trigger_the_geom_webmercator
|
|
else
|
|
puts " Skipping table #{table.name}: no 'the_geom' column"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
desc "update created_at and updated_at to correct type and add the default value to now"
|
|
task :update_timestamp_fields => :environment do
|
|
::User.all.each do |user|
|
|
next if !user.respond_to?('database_name') || user.database_name.blank?
|
|
puts "user => " + user.username
|
|
user.in_database do |user_database|
|
|
user.tables.all.each do |table|
|
|
table.normalize_timestamp_field!(:created_at, user_database)
|
|
table.normalize_timestamp_field!(:updated_at, user_database)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
desc 'update the old cache trigger which was using redis to the varnish one'
|
|
task :update_cache_trigger => :environment do
|
|
::User.all.each do |user|
|
|
puts 'Update cache trigger => ' + user.username
|
|
next if !user.respond_to?('database_name') || user.database_name.blank?
|
|
user.in_database do |user_database|
|
|
user.tables.all.each do |table|
|
|
puts "\t=> #{table.name} updated"
|
|
begin
|
|
table.set_trigger_cache_timestamp
|
|
rescue StandardError => e
|
|
puts "\t=> [ERROR] #{table.name}: #{e.inspect}"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
desc 'Save users metadata in redis'
|
|
task :save_users_metadata => :environment do
|
|
::User.all.each do |u|
|
|
u.save_metadata
|
|
end
|
|
end
|
|
|
|
desc 'Create public users for users beloging to an organization'
|
|
task :create_organization_members_public_users => :environment do
|
|
::User.exclude(organization_id: nil).each do |user|
|
|
begin
|
|
user.db_service.create_public_db_user
|
|
user.save_metadata
|
|
rescue StandardError
|
|
puts "user #{user.username} already has the public user"
|
|
end
|
|
end
|
|
end
|
|
|
|
desc 'Setup default permissions on existing visualizations'
|
|
task :create_default_vis_permissions, [:page_size, :page] => :environment do |t, args|
|
|
page_size = args[:page_size].blank? ? 999999 : args[:page_size].to_i
|
|
page = args[:page].blank? ? 1 : args[:page].to_i
|
|
|
|
progress_each = (page_size > 10) ? (page_size / 10).ceil : 1
|
|
|
|
begin
|
|
items = Carto::VisualizationQueryBuilder.new.build_paged(page, page_size)
|
|
|
|
count = items.count
|
|
puts "\n>Running :create_default_vis_permissions for page #{page} (#{count} vis)" if count > 0
|
|
items.each do |vis|
|
|
if vis.permission_id.nil?
|
|
begin
|
|
raise 'No owner' if vis.user.nil?
|
|
# Just saving will trigger the permission creation
|
|
vis.save!
|
|
puts "OK #{vis.id}"
|
|
rescue StandardError => e
|
|
owner_id = vis.user.nil? ? 'nil' : vis.user.id
|
|
message = "FAIL u:#{owner_id} v:#{vis.id}: #{e.message}"
|
|
puts message
|
|
log(message, :create_default_vis_permissions.to_s)
|
|
end
|
|
end
|
|
end
|
|
page += 1
|
|
end while count > 0
|
|
|
|
puts "\n>Finished :create_default_vis_permissions"
|
|
end
|
|
|
|
# Executes a ruby code proc/block on all existing users, outputting some info
|
|
# @param task_name string
|
|
# @param block Proc
|
|
# @example:
|
|
# execute_on_users_with_index(:populate_new_fields.to_s, Proc.new { |user, i| ... })
|
|
def execute_on_users_with_index(task_name, block, num_threads=1, sleep_time=0.1, database_host=nil)
|
|
if database_host.nil?
|
|
count = ::User.count
|
|
else
|
|
count = ::User.where(database_host: database_host).count
|
|
end
|
|
|
|
start_message = ">Running #{task_name} for #{count} users"
|
|
puts start_message
|
|
log(start_message, task_name, database_host)
|
|
if database_host.nil?
|
|
puts "Detailed log stored at log/rake_db_maintenance_#{task_name}.log"
|
|
else
|
|
puts "Detailed log stored at log/rake_db_maintenance_#{task_name}_#{database_host}.log"
|
|
end
|
|
|
|
thread_pool = ThreadPool.new(num_threads, sleep_time)
|
|
|
|
if database_host.nil?
|
|
::User.order(Sequel.asc(:created_at)).each_with_index do |user, i|
|
|
thread_pool.schedule do
|
|
if i % 100 == 0
|
|
puts "PROGRESS: #{i}/#{count} users queued"
|
|
end
|
|
block.call(user, i)
|
|
end
|
|
end
|
|
else
|
|
::User.where(database_host: database_host).order(Sequel.asc(:created_at)).each_with_index do |user, i|
|
|
thread_pool.schedule do
|
|
if i % 100 == 0
|
|
puts "PROGRESS: #{i}/#{count} users queued"
|
|
end
|
|
block.call(user, i)
|
|
end
|
|
end
|
|
end
|
|
|
|
at_exit { thread_pool.shutdown }
|
|
|
|
puts "PROGRESS: #{count}/#{count} users queued"
|
|
end_message = "\n>Finished #{task_name}\n"
|
|
puts end_message
|
|
log(end_message, task_name, database_host)
|
|
end
|
|
|
|
def log(entry, task_name, filename_suffix='')
|
|
if filename_suffix.nil? || filename_suffix.empty?
|
|
log_path = Rails.root.join('log', "rake_db_maintenance_#{task_name}.log")
|
|
else
|
|
log_path = Rails.root.join('log', "rake_db_maintenance_#{task_name}_#{filename_suffix}.log")
|
|
end
|
|
File.open(log_path, 'a') do |file_handle|
|
|
file_handle.puts "[#{Time.now}] #{entry}\n"
|
|
end
|
|
end
|
|
|
|
desc 'Load api calls from ES to redis'
|
|
task :load_api_calls_from_es => :environment do
|
|
raise "You should provide a valid username" if ENV['USERNAME'].blank?
|
|
u = ::User.where(:username => ENV['USERNAME']).first
|
|
puts "Old API Calls from ES: #{u.get_es_api_calls_from_redis}"
|
|
u.set_api_calls_from_es({:force_update => true})
|
|
puts "New API Calls from ES: #{u.get_es_api_calls_from_redis}"
|
|
end
|
|
|
|
desc "Create new organization with owner"
|
|
task :create_new_organization_with_owner => :environment do
|
|
raise "You should provide a ORGANIZATION_NAME" if ENV['ORGANIZATION_NAME'].blank?
|
|
raise "You should provide a ORGANIZATION_DISPLAY_NAME" if ENV['ORGANIZATION_DISPLAY_NAME'].blank?
|
|
raise "You should provide a ORGANIZATION_SEATS" if ENV['ORGANIZATION_SEATS'].blank?
|
|
raise "You should provide a ORGANIZATION_QUOTA (in Bytes)" if ENV['ORGANIZATION_QUOTA'].blank?
|
|
raise "You should provide a USERNAME" if ENV['USERNAME'].blank?
|
|
user = ::User.where(:username => ENV['USERNAME']).first
|
|
raise "User #{ENV['USERNAME']} does not exist" if user.nil?
|
|
organization = Organization.where(:name => ENV['ORGANIZATION_NAME']).first
|
|
if organization.nil?
|
|
organization = Organization.new
|
|
organization.name = ENV['ORGANIZATION_NAME']
|
|
organization.display_name = ENV['ORGANIZATION_DISPLAY_NAME']
|
|
organization.seats = ENV['ORGANIZATION_SEATS']
|
|
organization.quota_in_bytes = ENV['ORGANIZATION_QUOTA']
|
|
if ENV['BUILDER_ENABLED'] == "true"
|
|
organization.builder_enabled = true
|
|
end
|
|
organization.save
|
|
end
|
|
uo = CartoDB::UserOrganization.new(organization.id, user.id)
|
|
uo.promote_user_to_admin
|
|
end
|
|
|
|
desc "Create new organization without owner"
|
|
task :create_new_organization_without_owner => :environment do
|
|
raise "You should provide a ORGANIZATION_NAME" if ENV['ORGANIZATION_NAME'].blank?
|
|
raise "You should provide a ORGANIZATION_DISPLAY_NAME" if ENV['ORGANIZATION_DISPLAY_NAME'].blank?
|
|
raise "You should provide a ORGANIZATION_SEATS" if ENV['ORGANIZATION_SEATS'].blank?
|
|
raise "You should provide a ORGANIZATION_QUOTA (in Bytes)" if ENV['ORGANIZATION_QUOTA'].blank?
|
|
|
|
organization = Organization.where(:name => ENV['ORGANIZATION_NAME']).first
|
|
if organization.nil?
|
|
organization = Organization.new
|
|
organization.name = ENV['ORGANIZATION_NAME']
|
|
organization.display_name = ENV['ORGANIZATION_DISPLAY_NAME']
|
|
organization.seats = ENV['ORGANIZATION_SEATS']
|
|
organization.quota_in_bytes = ENV['ORGANIZATION_QUOTA']
|
|
if ENV['BUILDER_ENABLED'] == "true"
|
|
organization.builder_enabled = true
|
|
end
|
|
organization.save
|
|
end
|
|
end
|
|
|
|
def create_user(username, organization, quota_in_bytes)
|
|
u = ::User.new
|
|
u.email = "#{username}@test-org.com"
|
|
u.password = username
|
|
u.password_confirmation = username
|
|
u.username = username
|
|
u.quota_in_bytes = quota_in_bytes
|
|
#u.database_host = ::SequelRails.configuration.environment_for(Rails.env)['host']
|
|
|
|
if organization.owner_id.nil?
|
|
u.save(raise_on_failure: true)
|
|
CartoDB::UserOrganization.new(organization.id, u.id).promote_user_to_admin
|
|
organization.reload
|
|
else
|
|
u.organization = organization
|
|
u.save(raise_on_failure: true)
|
|
end
|
|
|
|
u
|
|
end
|
|
|
|
def create_users(first_index, last_index, organization, bytes_per_user)
|
|
org_name = organization.name
|
|
|
|
(first_index..last_index).each { |i|
|
|
# See Carto::StandardPasswordStrategy
|
|
username = "user-#{org_name}-#{i}"
|
|
print "Creating user #{username}... "
|
|
user = create_user(username, organization, bytes_per_user)
|
|
puts "Done."
|
|
}
|
|
end
|
|
|
|
desc "Create an organization with an arbitrary number of users for test purposes. Owner user: <org-name>-admin. Users: user-<org-name>-<i>. You might need to set conn_validator_timeout to -1 in config/database.yml (development)"
|
|
task :create_test_organization, [:org_name, :n_users] => [:environment] do |t, args|
|
|
org_name = args[:org_name]
|
|
n_users = args[:n_users].to_i
|
|
bytes_per_user = 50 * 1024 * 1024
|
|
|
|
def create_organization(org_name, n_users, bytes_per_user)
|
|
organization = Organization.new
|
|
organization.name = org_name
|
|
organization.display_name = org_name
|
|
organization.seats = n_users * 2
|
|
organization.quota_in_bytes = bytes_per_user * organization.seats
|
|
organization.save
|
|
end
|
|
|
|
print "Creating organization #{org_name}... "
|
|
organization = create_organization(org_name, n_users, bytes_per_user)
|
|
puts "Done."
|
|
|
|
owner_name = "#{org_name}-admin"
|
|
print "Creating owner #{owner_name}... "
|
|
owner = create_user(owner_name, organization, bytes_per_user)
|
|
puts "Done."
|
|
|
|
create_users(1, n_users, organization, bytes_per_user)
|
|
end
|
|
|
|
desc "Add users to an organization. Used to resume a broken :create_test_organization task"
|
|
task :add_test_users_to_organization, [:org_name, :first_index, :last_index] => [:environment] do |t, args|
|
|
org_name = args[:org_name]
|
|
first_index = args[:first_index]
|
|
last_index = args[:last_index]
|
|
bytes_per_user = 50 * 1024 * 1024
|
|
|
|
organization = Organization.where(name: org_name).first
|
|
create_users(first_index, last_index, organization, bytes_per_user)
|
|
end
|
|
|
|
desc "Reload users avatars"
|
|
task :reload_users_avatars => :environment do
|
|
if ENV['ONLY_GRAVATAR'].blank?
|
|
users = ::User.all
|
|
else
|
|
users = ::User.where(Sequel.like(:avatar_url, '%gravatar.com%'))
|
|
end
|
|
count = users.count
|
|
users.each_with_index do |user, i|
|
|
begin
|
|
user.reload_avatar
|
|
message = "OK %-#{20}s (%-#{4}s/%-#{4}s)\n" % [user.username, i, count]
|
|
print message
|
|
log(message, :reload_users_avatars.to_s)
|
|
rescue StandardError => e
|
|
message = "FAIL %-#{20}s (%-#{4}s/%-#{4}s) #{e.message}\n" % [user.username, i, count]
|
|
print message
|
|
log(message, :reload_users_avatars.to_s)
|
|
end
|
|
end
|
|
end
|
|
|
|
desc "Grant general raster permissions"
|
|
task :grant_general_raster_permissions => :environment do
|
|
users = ::User.all
|
|
count = users.count
|
|
users.each_with_index do |user, i|
|
|
begin
|
|
user.db_service.set_raster_privileges
|
|
message = "OK %-#{20}s (%-#{4}s/%-#{4}s)\n" % [user.username, i, count]
|
|
print message
|
|
log(message, :grant_general_raster_permissions.to_s)
|
|
rescue StandardError => e
|
|
message = "FAIL %-#{20}s (%-#{4}s/%-#{4}s) MSG:#{e.message}\n" % [user.username, i, count]
|
|
print message
|
|
log(message, :grant_general_raster_permissions.to_s)
|
|
end
|
|
end
|
|
end
|
|
|
|
desc "Drop other users privileges on user schema"
|
|
task :drop_other_privileges_on_user_schema, [:username] => :environment do |t,args|
|
|
user = ::User.where(:username => args[:username].to_s).first
|
|
user.in_database({as: :superuser}) do |db|
|
|
db.transaction do
|
|
oids = db.fetch("with oids as (select (aclexplode(n.nspacl)).grantee as grantee_oid from pg_catalog.pg_namespace n
|
|
WHERE n.nspname = '#{user.database_schema}')
|
|
select distinct pg_roles.rolname from oids, pg_roles where grantee_oid = oid")
|
|
oids.each do |oid|
|
|
role = oid[:rolname]
|
|
unless [user.database_username, CartoDB::PUBLIC_DB_USER].include? role
|
|
db.run("REVOKE ALL ON SCHEMA #{user.database_schema} FROM \"#{role}\"")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
desc "Enable oracle_fdw extension in database"
|
|
task :enable_oracle_fdw_extension, [:username, :oracle_url, :remote_user, :remote_password, :remote_schema, :table_definition_json_path] => :environment do |t, args|
|
|
u = ::User.where(:username => args[:username].to_s).first
|
|
tables = JSON.parse(File.read(args['table_definition_json_path'].to_s))
|
|
u.in_database({as: :superuser, no_cartodb_in_schema: true}) do |db|
|
|
db.transaction do
|
|
name = CartoDB::Importer2::StringSanitizer.sanitize(args[:oracle_url], transliterate_cyrillic: true)
|
|
server_name = "oracle_#{name}_#{Time.now.to_i}"
|
|
db.run('CREATE EXTENSION oracle_fdw') unless db.fetch(%Q{
|
|
SELECT count(*) FROM pg_extension WHERE extname='oracle_fdw'
|
|
}).first[:count] > 0
|
|
db.run("CREATE SERVER #{server_name} FOREIGN DATA WRAPPER oracle_fdw OPTIONS (dbserver '#{args[:oracle_url].to_s}')")
|
|
db.run("GRANT USAGE ON FOREIGN SERVER #{server_name} TO \"#{u.database_username}\"")
|
|
db.run("CREATE USER MAPPING FOR \"#{u.database_username}\" SERVER #{server_name} OPTIONS (user '#{args[:remote_user].to_s}', password '#{args[:remote_password].to_s}');")
|
|
db.run("CREATE USER MAPPING FOR \"#{u.database_public_username}\" SERVER #{server_name} OPTIONS (user '#{args[:remote_user].to_s}', password '#{args[:remote_password].to_s}');")
|
|
tables["tables"].each do |table_name, th|
|
|
table_readonly = th["read_only"] ? "true" : "false"
|
|
table_columns = th["columns"].map {|name,attrs| "#{name} #{attrs['column_type']}"}
|
|
db.run("DROP FOREIGN TABLE IF EXISTS #{table_name}")
|
|
db.run("CREATE FOREIGN TABLE #{table_name} (#{table_columns.join(', ')}) SERVER #{server_name} OPTIONS (schema '#{args[:remote_schema]}', table '#{th["remote_table"]}', readonly '#{table_readonly}')")
|
|
db.run("GRANT SELECT ON #{table_name} TO \"#{u.database_username}\"")
|
|
db.run("GRANT SELECT ON #{table_name} TO \"#{CartoDB::PUBLIC_DB_USER}\"")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
|
|
desc "Get from redis the list of visualization ids viewed"
|
|
task :get_viewed_visualization_ids, [:redis_ip, :output_file] => :environment do |t, args|
|
|
if args[:redis_ip]
|
|
redis_client_config = $users_metadata.client.options
|
|
redis_client_config[:host] = args[:redis_ip]
|
|
redis_client = Redis.new(redis_client_config)
|
|
else
|
|
redis_client = $users_metadata
|
|
end
|
|
|
|
stat_tag_keys = [
|
|
"user:*:mapviews:stat_tag:*",
|
|
"user:*:mapviews_es:stat_tag:*"
|
|
]
|
|
|
|
visualization_ids = []
|
|
|
|
stat_tag_keys.each do |stat_tag_key|
|
|
redis_client.keys(stat_tag_key).each do |key|
|
|
key_parts = key.split(':')
|
|
visualization_ids << "#{key_parts[1]}:#{key_parts[4]}"
|
|
end
|
|
end
|
|
|
|
visualization_ids.uniq!
|
|
|
|
if args[:output_file]
|
|
File.write(args[:output_file], visualization_ids.join("\n"))
|
|
else
|
|
puts visualization_ids.join("\n")
|
|
end
|
|
end
|
|
|
|
|
|
desc "Populate visualization total map views from partial days"
|
|
task :populate_visualization_total_map_views, [:visualizations_list] => :environment do |t, args|
|
|
if args[:visualizations_list].blank?
|
|
raise "Missing visualization_list param which must be a plain text list of <user>:<visualization_id>"
|
|
end
|
|
File.open(args[:visualizations_list], 'r') do |f|
|
|
f.each_line do |line|
|
|
key_parts = line.strip.split(':')
|
|
username = key_parts[0]
|
|
visualization_id = key_parts[1]
|
|
stat_tag_keys = [
|
|
"user:#{username}:mapviews:stat_tag:#{visualization_id}",
|
|
"user:#{username}:mapviews_es:stat_tag:#{visualization_id}"
|
|
]
|
|
puts "Processing visualization #{visualization_id} of user #{username}"
|
|
stat_tag_keys.each do |stat_tag_key|
|
|
visualization_counter = 0
|
|
$users_metadata.zrange(stat_tag_key, 0, -1).each do |mapviews_day|
|
|
if mapviews_day =~ /[0-9]{4}(0|1)[0-9][0-3][0-9]/
|
|
count = $users_metadata.zscore(stat_tag_key, mapviews_day)
|
|
visualization_counter = visualization_counter + count unless count.nil?
|
|
end
|
|
end
|
|
$users_metadata.zadd(stat_tag_key, visualization_counter, 'total') unless visualization_counter == 0
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
desc "Assign permissions to organization shared role. See #3859 and #3881. This is used to upgrade existing organizations to new permission schema. You can optionally speciy an organization name to restrict the execution to it."
|
|
task :assign_org_permissions_to_org_role, [:organization_name] => :environment do |t, args|
|
|
organizations = args[:organization_name].present? ? Organization.where(name: args[:organization_name]).all : Organization.all
|
|
puts "Updating #{organizations.count} organizations"
|
|
organizations.each { |o|
|
|
owner = o.owner
|
|
if owner
|
|
puts "#{o.name}\t#{o.id}\tOwner: #{owner.username}\t#{owner.id}"
|
|
begin
|
|
owner.db_service.setup_organization_role_permissions
|
|
rescue StandardError => e
|
|
puts "Error: #{e.message}"
|
|
CartoDB.notify_exception(e)
|
|
end
|
|
else
|
|
puts "#{o.name}\t#{o.id}\t Has no owner, skipping"
|
|
end
|
|
}
|
|
end
|
|
|
|
def run_for_organizations_owner(organizations)
|
|
puts "Updating #{organizations.count} organizations"
|
|
organizations.each { |o|
|
|
owner = o.owner
|
|
if owner
|
|
puts "#{o.id} Owner: #{owner.id} #{owner.username}\t\tName: #{o.name}"
|
|
begin
|
|
yield owner
|
|
rescue StandardError => e
|
|
puts "Error: #{e.message}"
|
|
CartoDB.notify_exception(e)
|
|
end
|
|
else
|
|
puts "#{o.name}\t#{o.id}\t Has no owner, skipping"
|
|
end
|
|
}
|
|
end
|
|
|
|
desc "Revokes access to cdb_conf"
|
|
task :revoke_cdb_conf_access => :environment do |t, args|
|
|
execute_on_users_with_index(:revoke_cdb_conf_access.to_s, Proc.new { |user, i|
|
|
errors = user.db_service.revoke_cdb_conf_access
|
|
if errors.empty?
|
|
puts "OK #{user.username}"
|
|
else
|
|
puts "ERROR #{user.username}: #{errors.join(';')}"
|
|
end
|
|
}, 1, 0.3)
|
|
end
|
|
|
|
desc "Assign organization owner admin role at database. See CartoDB/cartodb-postgresql#104 and #5187"
|
|
task :assign_org_owner_role, [:organization_name] => :environment do |t, args|
|
|
organizations = args[:organization_name].present? ? Organization.where(name: args[:organization_name]).all : Organization.all
|
|
run_for_organizations_owner(organizations) do |owner|
|
|
begin
|
|
owner.db_service.grant_admin_permissions
|
|
rescue StandardError => e
|
|
puts "ERROR for #{owner.organization.name}: #{e.message}"
|
|
end
|
|
end
|
|
end
|
|
|
|
desc "Configure extension org metadata API endpoint, the one used by the extension to keep groups synched. See CartoDB/cartodb-postgresql#104 and CartoDB/cartodb/issues/5244"
|
|
task :configure_extension_org_metadata_api_endpoint, [:organization_name] => :environment do |t, args|
|
|
organizations = args[:organization_name].present? ? Organization.where(name: args[:organization_name]).all : Organization.all
|
|
run_for_organizations_owner(organizations) do |owner|
|
|
begin
|
|
owner.db_service.configure_extension_org_metadata_api_endpoint
|
|
rescue StandardError => e
|
|
puts "ERROR for #{owner.organization.name}: #{e.message}"
|
|
end
|
|
end
|
|
end
|
|
|
|
# e.g. bundle exec rake cartodb:db:configure_geocoder_extension_for_organizations[org-name]
|
|
# bundle exec rake cartodb:db:configure_geocoder_extension_for_organizations['',true]
|
|
desc "Configure geocoder extension configuration for organizations"
|
|
task :configure_geocoder_extension_for_organizations, [:organization_name, :all_organizations] => :environment do |t, args|
|
|
args.with_defaults(:organization_name => nil, :all_organizations => false)
|
|
if args[:organization_name].blank? and args[:all_organizations] != 'true'
|
|
# Double check before launch an update to all the orgs
|
|
raise "ERROR: You haven't passed an organization name and/or put the :all_organizations flag to true"
|
|
end
|
|
organizations = args[:organization_name].blank? ? ::Organization.all : ::Organization.where(name: args[:organization_name]).all
|
|
raise "ERROR: Organization #{args[:organization_name]} don't exists" if organizations.blank? and not args[:all_organizations]
|
|
run_for_organizations_owner(organizations) do |owner|
|
|
begin
|
|
result = owner.db_service.install_and_configure_geocoder_api_extension
|
|
puts "Owner #{owner.username}: #{result ? 'OK' : 'ERROR'}"
|
|
# TODO Improved using the execute_on_users_with_index when orgs have a lot more users
|
|
owner.organization.users.each do |u|
|
|
if not u.organization_owner?
|
|
result = u.db_service.install_and_configure_geocoder_api_extension
|
|
puts "Organization user #{u.username}: #{result ? 'OK' : 'ERROR'}"
|
|
end
|
|
end
|
|
rescue StandardError => e
|
|
puts "Error trying to configure geocoder extension for org #{owner.organization.name}: #{e.message}"
|
|
end
|
|
end
|
|
end
|
|
|
|
# e.g. bundle exec rake cartodb:db:configure_geocoder_extension_for_non_org_users[username]
|
|
# bundle exec rake cartodb:db:configure_geocoder_extension_for_non_org_users['',true]
|
|
desc "Configure geocoder extension configuration for non-organization users"
|
|
task :configure_geocoder_extension_for_non_org_users, [:username, :all_users] => :environment do |t, args|
|
|
args.with_defaults(:username => nil, :all_users => false)
|
|
if args[:username].blank? and args[:all_users] != 'true'
|
|
# Double check before launch an update to all the orgs
|
|
raise "ERROR: You haven't passed an username and/or put the :all_users flag to true"
|
|
end
|
|
if not args[:username].blank?
|
|
user = ::User.where(username: args[:username]).first
|
|
raise "ERROR: User #{args[:username]} don't exists" if user.nil?
|
|
begin
|
|
result = user.db_service.install_and_configure_geocoder_api_extension
|
|
puts "#{result ? 'OK' : 'ERROR'} #{user.username}"
|
|
rescue StandardError => e
|
|
puts "Error trying to configure geocoder extension for user #{u.name}: #{e.message}"
|
|
end
|
|
elsif args[:all_users]
|
|
# TODO Could be improved passing the query to execute_on_users_with_index function to filter by non-org-users
|
|
execute_on_users_with_index(:configure_geocoder_extension_for_non_org_users.to_s, Proc.new { |user, i|
|
|
begin
|
|
if not user.organization_user?
|
|
result = user.db_service.install_and_configure_geocoder_api_extension
|
|
puts "#{result ? 'OK' : 'ERROR'} #{user.username}"
|
|
end
|
|
rescue StandardError => e
|
|
puts "Error trying to configure geocoder extension for user #{u.name}: #{e.message}"
|
|
end
|
|
}, 1, 0.3)
|
|
end
|
|
end
|
|
|
|
# e.g. bundle exec rake cartodb:db:migrate_current_geocoder_billing_to_redis[YYYYMM,YYYYMMDD]
|
|
desc 'Migrate the current billing geocoding data to Redis'
|
|
task :migrate_current_geocoder_billing_to_redis, [:date_from, :date_to] => [:environment] do |task, args|
|
|
args.with_defaults(:date_from => nil, :date_to => nil)
|
|
if args[:date_from].blank? or args[:date_to].blank?
|
|
raise "ERROR: Is mandatory to pass a date from and to for the migration"
|
|
end
|
|
begin
|
|
date_from = DateTime.parse(args[:date_from])
|
|
date_to = DateTime.parse(args[:date_to])
|
|
rescue StandardError => e
|
|
raise "Error converting argument dates, check the arguments"
|
|
end
|
|
execute_on_users_with_index(:migrate_current_geocoder_billing_to_redis.to_s, Proc.new { |user, i|
|
|
begin
|
|
# We are working on the v2 which is only Nokia
|
|
next if user.google_maps_geocoder_enabled?
|
|
orgname = user.organization.nil? ? nil : user.organization.name
|
|
usage_metrics = CartoDB::GeocoderUsageMetrics.new(user.username, orgname)
|
|
geocoding_calls = user.get_not_aggregated_geocoding_calls({from: date_from, to: date_to})
|
|
geocoding_calls.each do |metric|
|
|
usage_metrics.incr(:geocoder_here, :success_responses, metric[:processed_rows], metric[:date])
|
|
usage_metrics.incr(:geocoder_here, :total_requests, metric[:processed_rows], metric[:date])
|
|
usage_metrics.incr(:geocoder_cache, :success_responses, metric[:cache_hits], metric[:date])
|
|
usage_metrics.incr(:geocoder_cache, :total_requests, metric[:cache_hits], metric[:date])
|
|
puts "Imported metrics for day #{metric[:date]} and user #{user.username}: #{metric}"
|
|
end
|
|
rescue StandardError => e
|
|
puts "Error trying to migrate user current billing cycle to redis #{user.username}: #{e.message}"
|
|
end
|
|
}, 1, 0.3)
|
|
end
|
|
|
|
# e.g. bundle exec rake cartodb:db:update_user_and_org_redis_metadata[username]
|
|
# bundle exec rake cartodb:db:update_user_and_org_redis_metadata
|
|
desc 'Update users and organizations metadata in Redis'
|
|
task :update_user_and_org_redis_metadata, [:username] => [:environment] do |task, args|
|
|
args.with_defaults(:username => nil)
|
|
if args[:username].blank?
|
|
execute_on_users_with_index(:update_user_and_org_redis_metadata.to_s, Proc.new { |user, i|
|
|
update_user_metadata(user)
|
|
update_organization_metadata(user)
|
|
}, 1, 0.3)
|
|
else
|
|
user = ::User.where(username: args[:username]).first
|
|
update_user_metadata(user)
|
|
update_organization_metadata(user)
|
|
end
|
|
end
|
|
|
|
# usage:
|
|
# bundle exec rake cartodb:db:connect_aggregation_fdw_tables[username]
|
|
desc 'Connect aggregation tables through FDW to user'
|
|
task :connect_aggregation_fdw_tables_to_user, [:username] => [:environment] do |task, args|
|
|
args.with_defaults(:username => nil)
|
|
raise 'Not a valid username' if args[:username].blank?
|
|
user = ::User.find(username: args[:username])
|
|
user.db_service.connect_to_aggregation_tables
|
|
end
|
|
|
|
# usage:
|
|
# bundle exec rake cartodb:db:connect_aggregation_fdw_tables[orgname]
|
|
desc 'Connect aggregation tables through FDW to orgname'
|
|
task :connect_aggregation_fdw_tables_to_org, [:orgname] => [:environment] do |task, args|
|
|
args.with_defaults(:orgname => nil)
|
|
raise 'Not a valid orgname' if args[:orgname].blank?
|
|
org = ::Organization.find(name: args[:orgname])
|
|
org.users.each do |u|
|
|
begin
|
|
u.db_service.connect_to_aggregation_tables
|
|
rescue StandardError => e
|
|
puts "Error trying to connect #{u.username}: #{e.message}"
|
|
end
|
|
end
|
|
end
|
|
|
|
desc 'Connect aggregation tables through FDW to builder users'
|
|
task :connect_aggregation_fdw_tables_to_builder_users => :environment do
|
|
unless Cartodb.get_config(:aggregation_tables).present?
|
|
raise "Aggregation tables not configured!"
|
|
end
|
|
# We're using a PostgreSQL cursor to avoid loading all the users
|
|
# in memory at once. (For ActiveRecord we'd use `find_each`)
|
|
User.where.use_cursor(rows_per_fetch: 100).each do |user|
|
|
if user.has_feature_flag?('editor-3')
|
|
begin
|
|
user.db_service.connect_to_aggregation_tables
|
|
rescue StandardError => e
|
|
puts "Error trying to connect #{user.username}: #{e.message}"
|
|
puts e.backtrace
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
desc 'Connect aggregation tables through FDW to all users'
|
|
task :connect_aggregation_fdw_tables_to_all_users => :environment do
|
|
unless Cartodb.get_config(:aggregation_tables).present?
|
|
raise "Aggregation tables not configured!"
|
|
end
|
|
# We're using a PostgreSQL cursor to avoid loading all the users
|
|
# in memory at once. (For ActiveRecord we'd use `find_each`)
|
|
User.where.use_cursor(rows_per_fetch: 100).each do |user|
|
|
begin
|
|
user.db_service.connect_to_aggregation_tables
|
|
rescue StandardError => e
|
|
puts "Error trying to connect #{user.username}: #{e.message}"
|
|
puts e.backtrace
|
|
end
|
|
end
|
|
end
|
|
|
|
# usage:
|
|
# bundle exec rake cartodb:db:obs_quota_enterprise[1000]
|
|
desc 'Give data observatory quota to all the enterprise users'
|
|
task :obs_quota_enterprise, [:quota] => [:environment] do |task, args|
|
|
args.with_defaults(:quota => 1000)
|
|
raise 'Not a valid quota' if args[:quota].blank?
|
|
do_quota = args[:quota]
|
|
users = User.where("account_type ilike 'enterprise%' or account_type ilike 'partner'").all
|
|
puts "Number of enterprise users to process: #{users.size}"
|
|
users.each do |u|
|
|
begin
|
|
if u.organization_owner? && !u.organization.nil?
|
|
u.organization.obs_general_quota = do_quota if u.organization.obs_general_quota.to_i == 0
|
|
u.organization.obs_snapshot_quota = do_quota if u.organization.obs_snapshot_quota.to_i == 0
|
|
u.organization.save
|
|
puts "Organization #{u.organization.name} processed OK"
|
|
else
|
|
u.obs_general_quota = do_quota if u.obs_general_quota.to_i == 0
|
|
u.obs_snapshot_quota = do_quota if u.obs_snapshot_quota.to_i == 0
|
|
u.save
|
|
puts "User #{u.username} processed OK"
|
|
end
|
|
rescue StandardError => e
|
|
puts "Error trying to give DO quota to #{u.username}: #{e.message}"
|
|
end
|
|
end
|
|
end
|
|
|
|
desc 'Fix analysis table the_geom type'
|
|
task fix_analysis_the_geom_type: [:environment] do
|
|
total_users = User.count
|
|
current = 0
|
|
User.where.use_cursor(rows_per_fetch: 100).each do |user|
|
|
puts "User #{current += 1} / #{total_users}"
|
|
next if user.organization && user.organization.owner != user # Filter out admin not owner users
|
|
begin
|
|
user.in_database do |db|
|
|
db.fetch("SELECT DISTINCT f_table_schema, f_table_name FROM geometry_columns WHERE f_table_name LIKE 'analysis%' AND type = 'MULTIPOLYGON'").map { |r| { schema: r[:f_table_schema], table: r[:f_table_name] } }.each do |entry|
|
|
db.execute("UPDATE \"#{entry[:schema]}\".\"#{entry[:table]}\" SET the_geom = ST_Multi(the_geom) where ST_GeometryType(the_geom) = 'ST_Polygon'")
|
|
end
|
|
end
|
|
rescue StandardError => e
|
|
puts "Error processing user #{user.username}: #{e.inspect}"
|
|
end
|
|
end
|
|
end
|
|
|
|
desc 'Fix analysis table the_geom type for batch geocoding bug (dataservices-api#538)'
|
|
task fix_batch_geocoding_the_geom_type: [:environment] do
|
|
total_users = User.count
|
|
current = 0
|
|
User.where.use_cursor(rows_per_fetch: 100).each do |user|
|
|
puts "User #{current += 1} / #{total_users}"
|
|
next if user.organization && user.organization.owner != user # Filter out admin not owner users
|
|
begin
|
|
user.in_database(as: :superuser) do |db|
|
|
db.fetch("SELECT DISTINCT f_table_schema, f_table_name FROM geometry_columns " \
|
|
"WHERE f_table_name LIKE 'analysis_1ea6dec9f3_%' AND type = 'MULTIPOLYGON'").each do |entry|
|
|
db.execute("ALTER TABLE \"#{entry[:f_table_schema]}\".\"#{entry[:f_table_name]}\" " \
|
|
"ALTER COLUMN the_geom TYPE geometry(Point, 4326)")
|
|
end
|
|
end
|
|
rescue StandardError => e
|
|
puts "Error processing user #{user.username}: #{e.inspect}"
|
|
end
|
|
end
|
|
end
|
|
|
|
def update_user_metadata(user)
|
|
begin
|
|
user.save_metadata
|
|
puts "Updated redis metadata for user #{user.username}"
|
|
rescue StandardError => e
|
|
puts "Error trying to update the user metadata for user #{user.username}: #{e.message}"
|
|
end
|
|
end
|
|
|
|
def update_organization_metadata(user)
|
|
begin
|
|
if user.organization_owner?
|
|
user.organization.save_metadata
|
|
puts "Updated redis metadata for organization #{user.organization.name}"
|
|
end
|
|
rescue StandardError => e
|
|
puts "Error trying to update the user and/or org metadata for user #{user.username}: #{e.message}"
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|