Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 10 additions & 8 deletions lib/activerecord-multi-tenant/migrations.rb
Original file line number Diff line number Diff line change
Expand Up @@ -82,14 +82,12 @@ def create_table(table_name, options = {}, &block)
end
ActiveRecord::ConnectionAdapters::SchemaStatements.prepend(MultiTenant::SchemaStatementsExtensions)

module ActiveRecord
class SchemaDumper
module MultiTenant
module SchemaDumperExtensions
private

alias initialize_without_citus initialize

def initialize(connection, options = {})
initialize_without_citus(connection, options)
super

citus_version =
begin
Expand All @@ -114,10 +112,8 @@ def initialize(connection, options = {})
end

# Support for create_distributed_table & create_reference_table
alias table_without_citus table

def table(table, stream)
table_without_citus(table, stream)
super
table_name = remove_prefix_and_suffix(table)
distribution_column = @distribution_columns[table_name]
if distribution_column
Expand All @@ -130,3 +126,9 @@ def table(table, stream)
end
end
end

module ActiveRecord
class SchemaDumper
prepend MultiTenant::SchemaDumperExtensions
end
end