Module: MultiTenant::MigrationExtensions
- Defined in:
- lib/activerecord-multi-tenant/migrations.rb
Instance Method Summary collapse
- #citus_version ⇒ Object
- #create_distributed_table(table_name, partition_key) ⇒ Object
- #create_reference_table(table_name) ⇒ Object
- #enable_extension_on_all_nodes(extension) ⇒ Object
- #execute_on_all_nodes(sql) ⇒ Object
Instance Method Details
#citus_version ⇒ Object
30 31 32 33 34 |
# File 'lib/activerecord-multi-tenant/migrations.rb', line 30 def citus_version execute("SELECT extversion FROM pg_extension WHERE extname = 'citus'").getvalue(0,0).try(:split, '-').try(:first) rescue ArgumentError => e raise unless e. == "invalid tuple number 0" end |
#create_distributed_table(table_name, partition_key) ⇒ Object
3 4 5 6 |
# File 'lib/activerecord-multi-tenant/migrations.rb', line 3 def create_distributed_table(table_name, partition_key) return unless citus_version.present? execute "SELECT create_distributed_table($$#{table_name}$$, $$#{partition_key}$$)" end |
#create_reference_table(table_name) ⇒ Object
8 9 10 11 |
# File 'lib/activerecord-multi-tenant/migrations.rb', line 8 def create_reference_table(table_name) return unless citus_version.present? execute "SELECT create_reference_table($$#{table_name}$$)" end |
#enable_extension_on_all_nodes(extension) ⇒ Object
26 27 28 |
# File 'lib/activerecord-multi-tenant/migrations.rb', line 26 def enable_extension_on_all_nodes(extension) execute_on_all_nodes "CREATE EXTENSION IF NOT EXISTS \"#{extension}\"" end |
#execute_on_all_nodes(sql) ⇒ Object
13 14 15 16 17 18 19 20 21 22 23 24 |
# File 'lib/activerecord-multi-tenant/migrations.rb', line 13 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 |