Module: Gitlab::Database::SchemaHelpers
- Included in:
- MigrationHelpers::LooseForeignKeyHelpers, Migrations::Observers::QueryStatistics, PartitioningMigrationHelpers::ForeignKeyHelpers, PartitioningMigrationHelpers::IndexHelpers, PartitioningMigrationHelpers::TableManagementHelpers
- Defined in:
- lib/gitlab/database/schema_helpers.rb
Instance Method Summary collapse
- #assert_not_in_transaction_block(scope:) ⇒ Object
- #create_comment(type, name, text) ⇒ Object
- #create_trigger(table_name, name, function_name, fires:) ⇒ Object
- #create_trigger_function(name, replace: true) ⇒ Object
- #drop_function(name, if_exists: true) ⇒ Object
- #drop_trigger(table_name, name, if_exists: true) ⇒ Object
- #function_exists?(name) ⇒ Boolean
- #object_name(table, type) ⇒ Object
- #tmp_table_name(base) ⇒ Object
- #trigger_exists?(table_name, name) ⇒ Boolean
Instance Method Details
#assert_not_in_transaction_block(scope:) ⇒ Object
76 77 78 79 80 81 82 |
# File 'lib/gitlab/database/schema_helpers.rb', line 76 def assert_not_in_transaction_block(scope:) return unless transaction_open? raise "#{scope} operations can not be run inside a transaction block, " \ "you can disable transaction blocks by calling disable_ddl_transaction! " \ "in the body of your migration class" end |
#create_comment(type, name, text) ⇒ Object
59 60 61 |
# File 'lib/gitlab/database/schema_helpers.rb', line 59 def create_comment(type, name, text) execute("COMMENT ON #{type} #{name} IS '#{text}'") end |
#create_trigger(table_name, name, function_name, fires:) ⇒ Object
23 24 25 26 27 28 29 30 31 |
# File 'lib/gitlab/database/schema_helpers.rb', line 23 def create_trigger(table_name, name, function_name, fires:) execute(<<~SQL) CREATE TRIGGER #{name} #{fires} ON #{table_name} FOR EACH ROW #{yield if block_given?} EXECUTE FUNCTION #{function_name}() SQL end |
#create_trigger_function(name, replace: true) ⇒ Object
6 7 8 9 10 11 12 13 14 15 16 17 |
# File 'lib/gitlab/database/schema_helpers.rb', line 6 def create_trigger_function(name, replace: true) replace_clause = optional_clause(replace, "OR REPLACE") execute(<<~SQL) CREATE #{replace_clause} FUNCTION #{name}() RETURNS TRIGGER AS $$ BEGIN #{yield} END $$ LANGUAGE PLPGSQL SQL end |
#drop_function(name, if_exists: true) ⇒ Object
49 50 51 52 |
# File 'lib/gitlab/database/schema_helpers.rb', line 49 def drop_function(name, if_exists: true) exists_clause = optional_clause(if_exists, "IF EXISTS") execute("DROP FUNCTION #{exists_clause} #{name}()") end |
#drop_trigger(table_name, name, if_exists: true) ⇒ Object
54 55 56 57 |
# File 'lib/gitlab/database/schema_helpers.rb', line 54 def drop_trigger(table_name, name, if_exists: true) exists_clause = optional_clause(if_exists, "IF EXISTS") execute("DROP TRIGGER #{exists_clause} #{name} ON #{table_name}") end |
#function_exists?(name) ⇒ Boolean
19 20 21 |
# File 'lib/gitlab/database/schema_helpers.rb', line 19 def function_exists?(name) connection.select_value("SELECT 1 FROM pg_proc WHERE proname = '#{name}'") end |
#object_name(table, type) ⇒ Object
69 70 71 72 73 74 |
# File 'lib/gitlab/database/schema_helpers.rb', line 69 def object_name(table, type) identifier = "#{table}_#{type}" hashed_identifier = Digest::SHA256.hexdigest(identifier).first(10) "#{type}_#{hashed_identifier}" end |
#tmp_table_name(base) ⇒ Object
63 64 65 66 67 |
# File 'lib/gitlab/database/schema_helpers.rb', line 63 def tmp_table_name(base) hashed_base = Digest::SHA256.hexdigest(base).first(10) "#{base}_#{hashed_base}" end |
#trigger_exists?(table_name, name) ⇒ Boolean
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
# File 'lib/gitlab/database/schema_helpers.rb', line 33 def trigger_exists?(table_name, name) result = connection.select_value(<<~SQL.squish) SELECT true FROM pg_catalog.pg_trigger trgr INNER JOIN pg_catalog.pg_class rel ON trgr.tgrelid = rel.oid INNER JOIN pg_catalog.pg_namespace nsp ON nsp.oid = rel.relnamespace WHERE nsp.nspname = #{connection.quote(current_schema)} AND rel.relname = #{connection.quote(table_name)} AND trgr.tgname = #{connection.quote(name)} SQL !!result end |