Module: Izolenta::ActiveRecordMigration::Helpers

Defined in:
lib/izolenta/active_record_migration.rb

Overview

helpers na

Class Method Summary collapse

Class Method Details

.create_helper_table(helper_table, column_name, column_type) ⇒ Object



30
31
32
# File 'lib/izolenta/active_record_migration.rb', line 30

def create_helper_table(helper_table, column_name, column_type)
  ActiveRecord::Base.connection.execute("CREATE TABLE #{helper_table} ( #{column_name} #{column_type} );")
end

.create_sync_trigger(table, column_name, helper_table_name, options) ⇒ Object



34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
# File 'lib/izolenta/active_record_migration.rb', line 34

def create_sync_trigger(table, column_name, helper_table_name, options)
  trg_name = "#{table}_#{column_name}_trg"
  insert_value = if options[:wrapper_function]
    "#{options[:wrapper_function]}(NEW.#{column_name})"
  else
    "NEW.#{column_name}"
  end

  trigger_condition = "WHEN( #{options[:trigger_condition]} )" if options[:trigger_condition]

  ActiveRecord::Base.connection.execute(<<~SYNC_TRIGGER)
    CREATE OR REPLACE FUNCTION #{trg_name}() RETURNS trigger AS $$
    BEGIN#{" "}
      INSERT INTO #{helper_table_name} VALUES ( #{insert_value} );
      RETURN NEW;
    END $$ LANGUAGE plpgSQL;

    CREATE TRIGGER #{trg_name} BEFORE INSERT ON #{table}#{" "}
    FOR EACH ROW
    #{trigger_condition}
    EXECUTE FUNCTION #{trg_name}();
  SYNC_TRIGGER
end

.drop_sync_trigger(table, column_name) ⇒ Object



58
59
60
61
62
63
64
# File 'lib/izolenta/active_record_migration.rb', line 58

def drop_sync_trigger(table, column_name)
  trg_name = "#{table}_#{column_name}_trg"

  ActiveRecord::Base.connection.execute(<<~SYNC_TRIGGER)
    DROP TRIGGER IF EXISTS #{trg_name} ON #{table};
  SYNC_TRIGGER
end

.get_column_type(origin_table, column) ⇒ Object



70
71
72
# File 'lib/izolenta/active_record_migration.rb', line 70

def get_column_type(origin_table, column)
  ActiveRecord::Base.connection.schema_cache.columns_hash(origin_table.to_s)[column.to_s]&.sql_type
end

.get_function_type(wrapper_function) ⇒ Object



74
75
76
77
78
79
# File 'lib/izolenta/active_record_migration.rb', line 74

def get_function_type(wrapper_function)
  ActiveRecord::Base
    .connection
    .execute("SELECT typname FROM pg_type WHERE oid=(SELECT prorettype FROM pg_proc WHERE proname ='#{wrapper_function}')") # rubocop:disable Layout/LineLength
    .first["typname"]
end

.get_new_column_type(origin_table, column, wrapper_function: nil) ⇒ Object



66
67
68
# File 'lib/izolenta/active_record_migration.rb', line 66

def get_new_column_type(origin_table, column, wrapper_function: nil, **)
  wrapper_function ? get_function_type(wrapper_function) : get_column_type(origin_table, column)
end