Module: AssociateJsonb::ConnectionAdapters::SchemaStatements

Defined in:
lib/associate_jsonb/connection_adapters/schema_statements.rb

Instance Method Summary collapse

Instance Method Details

#add_constraint(table_name, **options) ⇒ Object 



57
58
59
60
61
 
# File 'lib/associate_jsonb/connection_adapters/schema_statements.rb', line 57

def add_constraint(table_name, **options)
  at = create_alter_table table_name
  at.add_constraint(**options)
  execute schema_creation.accept at
end

#add_jsonb_foreign_key_function ⇒ Object 



11
12
13
 
# File 'lib/associate_jsonb/connection_adapters/schema_statements.rb', line 11

def add_jsonb_foreign_key_function
  execute schema_creation.accept(AddJsonbForeignKeyFunction.new)
end

#add_jsonb_nested_set_function ⇒ Object 



7
8
9
 
# File 'lib/associate_jsonb/connection_adapters/schema_statements.rb', line 7

def add_jsonb_nested_set_function
  execute schema_creation.accept(AddJsonbNestedSetFunction.new)
end

#constraints(table_name) ⇒ Object

:nodoc:



63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
 
# File 'lib/associate_jsonb/connection_adapters/schema_statements.rb', line 63

def constraints(table_name) # :nodoc:
  scope = quoted_scope(table_name)

  result = query(<<~SQL, "SCHEMA")
    SELECT
      con.oid,
      con.conname,
      con.connamespace,
      con.contype,
      con.condeferrable,
      con.condeferred,
      con.convalidated,
      pg_get_constraintdef(con.oid) as consrc
    FROM pg_catalog.pg_constraint con
    INNER JOIN pg_catalog.pg_class rel
               ON rel.oid = con.conrelid
    INNER JOIN pg_catalog.pg_namespace nsp
               ON nsp.oid = connamespace
    WHERE nsp.nspname = #{scope[:schema]}
    AND rel.relname = #{scope[:name]}
    ORDER BY rel.relname
  SQL

  result.map do |row|
    {
      oid: row[0],
      name: row[1],
      deferrable: row[4],
      deferred: row[5],
      validated: row[6],
      definition: row[7],
      type:
        case row[3].to_s.downcase
        when "c"
          "CHECK"
        when "f"
          "FOREIGN KEY"
        when "p"
          "PRIMARY KEY"
        when "u"
          "UNIQUE"
        when "t"
          "TRIGGER"
        when "x"
          "EXCLUDE"
        else
          "UNKNOWN"
        end
    }
  end
end

#create_table(table_name, **options) {|td| ... } ⇒ Object

Yields:

  • (td) 


15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/associate_jsonb/connection_adapters/schema_statements.rb', line 15

def create_table(table_name, **options)
  td = create_table_definition(table_name, **options)

  if options[:id] != false && !options[:as]
    pk = options.fetch(:primary_key) do
      ActiveRecord::Base.get_primary_key table_name.to_s.singularize
    end

    if pk.is_a?(Array)
      td.primary_keys pk
    else
      td.primary_key pk, options.fetch(:id, :primary_key), **options.except(:comment)
    end
  end

  yield td if block_given?

  if options[:force]
    drop_table(table_name, **options, if_exists: true)
  end

  result = execute schema_creation.accept td

  td.indexes.each do |column_name, index_options|
    add_index(table_name, column_name, index_options)
  end

  td.constraints.each do |ct|
    add_constraint(table_name, **ct)
  end

  if table_comment = options[:comment].presence
    change_table_comment(table_name, table_comment)
  end

  td.columns.each do |column|
    change_column_comment(table_name, column.name, column.comment) if column.comment.present?
  end

  result
end