Sha256: 6ab5d2c56c647822510354ba4552b5d14c8a8452c41e45401e9a73c02b37b8f2
Contents?: true
Size: 1.57 KB
Versions: 8
Compression:
Stored size: 1.57 KB
Contents
module MultiTenant
module MigrationExtensions
def create_distributed_table(table_name, partition_key)
return unless citus_version.present?
execute "SELECT create_distributed_table($$#{table_name}$$, $$#{partition_key}$$)"
end
def execute_on_all_nodes(sql)
execute sql
case citus_version
when '6.0'
execute "SELECT citus_run_on_all_workers($$#{sql}$$)" # initial citus_tools.sql with different names
when nil
# Do nothing, this is regular Postgres
else # 6.1 and newer
execute "SELECT run_command_on_workers($$#{sql}$$)"
end
end
def enable_extension_on_all_nodes(extension)
execute_on_all_nodes "CREATE EXTENSION IF NOT EXISTS \"#{extension}\""
end
def citus_version
execute("SELECT extversion FROM pg_extension WHERE extname = 'citus'").getvalue(0,0).try(:split, '-').try(:first)
end
end
end
if defined?(ActiveRecord::Migration)
ActiveRecord::Migration.send(:include, MultiTenant::MigrationExtensions)
end
module ActiveRecord
module ConnectionAdapters # :nodoc:
module SchemaStatements
alias :orig_create_table :create_table
def create_table(table_name, options = {}, &block)
ret = orig_create_table(table_name, options.except(:partition_key), &block)
if options[:partition_key] && options[:partition_key].to_s != 'id'
execute "ALTER TABLE #{table_name} DROP CONSTRAINT #{table_name}_pkey"
execute "ALTER TABLE #{table_name} ADD PRIMARY KEY(id, \"#{options[:partition_key]}\")"
end
ret
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems