Module: Switchman::ActiveRecord::PostgreSQLAdapter

Defined in:
lib/switchman/active_record/postgresql_adapter.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.prepended(klass) ⇒ Object 



4
5
6
 
# File 'lib/switchman/active_record/postgresql_adapter.rb', line 4

def self.prepended(klass)
  klass::NATIVE_DATABASE_TYPES[:primary_key] = "bigserial primary key".freeze
end

Instance Method Details

#current_schemas ⇒ Object 



8
9
10
 
# File 'lib/switchman/active_record/postgresql_adapter.rb', line 8

def current_schemas
  select_values("SELECT * FROM unnest(current_schemas(false))")
end

#index_name_exists?(table_name, index_name, default) ⇒ Boolean

Returns:

  • (Boolean) 


109
110
111
112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/switchman/active_record/postgresql_adapter.rb', line 109

def index_name_exists?(table_name, index_name, default)
  schema = shard.name if @config[:use_qualified_names]

  exec_query("      SELECT COUNT(*)\n      FROM pg_class t\n      INNER JOIN pg_index d ON t.oid = d.indrelid\n      INNER JOIN pg_class i ON d.indexrelid = i.oid\n      WHERE i.relkind = 'i'\n        AND i.relname = '\#{index_name}'\n        AND t.relname = '\#{table_name}'\n        AND i.relnamespace IN (SELECT oid FROM pg_namespace WHERE nspname = \#{schema ? \"'\#{schema}'\" : 'ANY (current_schemas(false))'} )\n  SQL\nend\n", 'SCHEMA').rows.first[0].to_i > 0

#indexes(table_name) ⇒ Object 



57
58
59
60
61
62
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
 
# File 'lib/switchman/active_record/postgresql_adapter.rb', line 57

def indexes(table_name)
  schema = shard.name if @config[:use_qualified_names]

  result = query("     SELECT distinct i.relname, d.indisunique, d.indkey, pg_get_indexdef(d.indexrelid), t.oid\n     FROM pg_class t\n     INNER JOIN pg_index d ON t.oid = d.indrelid\n     INNER JOIN pg_class i ON d.indexrelid = i.oid\n     WHERE i.relkind = 'i'\n       AND d.indisprimary = 'f'\n       AND t.relname = '\#{table_name}'\n       AND i.relnamespace IN (SELECT oid FROM pg_namespace WHERE nspname = \#{schema ? \"'\#{schema}'\" : 'ANY (current_schemas(false))'} )\n    ORDER BY i.relname\n  SQL\n\n\n  result.map do |row|\n    index_name = row[0]\n    unique = row[1] == 't'\n    indkey = row[2].split(\" \")\n    inddef = row[3]\n    oid = row[4]\n\n    columns = Hash[query(<<-SQL, \"SCHEMA\")]\n    SELECT a.attnum, a.attname\n    FROM pg_attribute a\n    WHERE a.attrelid = \#{oid}\n    AND a.attnum IN (\#{indkey.join(\",\")})\n    SQL\n\n    column_names = columns.values_at(*indkey).compact\n\n    if ::Rails.version >= '4'\n      unless column_names.empty?\n        # add info on sort order for columns (only desc order is explicitly specified, asc is the default)\n        desc_order_columns = inddef.scan(/(\\w+) DESC/).flatten\n        orders = desc_order_columns.any? ? Hash[desc_order_columns.map {|order_column| [order_column, :desc]}] : {}\n        where = inddef.scan(/WHERE (.+)$/).flatten[0]\n        using = inddef.scan(/USING (.+?) /).flatten[0].to_sym\n\n        ::ActiveRecord::ConnectionAdapters::IndexDefinition.new(table_name, index_name, unique, column_names, [], orders, where, nil, using)\n      end\n    else\n      # add info on sort order for columns (only desc order is explicitly specified, asc is the default)\n      desc_order_columns = inddef.scan(/(\\w+) DESC/).flatten\n      orders = desc_order_columns.any? ? Hash[desc_order_columns.map {|order_column| [order_column, :desc]}] : {}\n\n      column_names.empty? ? nil : ::ActiveRecord::ConnectionAdapters::IndexDefinition.new(table_name, index_name, unique, column_names, [], orders)\n    end\n  end.compact\nend\n", 'SCHEMA')

#quote_table_name(name) ⇒ Object 



124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
 
# File 'lib/switchman/active_record/postgresql_adapter.rb', line 124

def quote_table_name name
  if ::Rails.version < '4.2'
    schema, name_part = extract_pg_identifier_from_name(name.to_s)

    if !name_part && @config[:use_qualified_names] && shard.name
      schema, name_part = shard.name, schema
    end

    unless name_part
      quote_column_name(schema)
    else
      table_name, name_part = extract_pg_identifier_from_name(name_part)
      "#{quote_column_name(schema)}.#{quote_column_name(table_name)}"
    end
  else
    name = Utils.extract_schema_qualified_name(name.to_s)
    if !name.schema && @config[:use_qualified_names]
      name.instance_variable_set(:@schema, shard.name)
    end
    name.quoted
  end
end

#table_exists?(name) ⇒ Boolean

Returns:

  • (Boolean) 


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/switchman/active_record/postgresql_adapter.rb', line 22

def table_exists?(name)
  if ::Rails.version < '4.2'
    schema, table = ::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter::Utils.extract_schema_and_table(name.to_s)
    return false unless table
    schema ||= shard.name if @config[:use_qualified_names]

    binds = [[nil, table]]
    binds << [nil, schema] if schema

    exec_query("        SELECT COUNT(*)\n        FROM pg_class c\n        LEFT JOIN pg_namespace n ON n.oid = c.relnamespace\n        WHERE c.relkind in ('v','r')\n        AND c.relname = '\#{table.gsub(/(^\"|\"$)/,'')}'\n        AND n.nspname = \#{schema ? \"'\#{schema}'\" : 'ANY (current_schemas(false))'}\n    SQL\n  else\n    name = Utils.extract_schema_qualified_name(name.to_s)\n    return false unless name.identifier\n    if !name.schema && @config[:use_qualified_names]\n      name.instance_variable_set(:@schema, shard.name)\n    end\n\n    exec_query(<<-SQL, 'SCHEMA').rows.first[0].to_i > 0\n        SELECT COUNT(*)\n        FROM pg_class c\n        LEFT JOIN pg_namespace n ON n.oid = c.relnamespace\n        WHERE c.relkind IN ('r','v','m') -- (r)elation/table, (v)iew, (m)aterialized view\n        AND c.relname = '\#{name.identifier}'\n        AND n.nspname = \#{name.schema ? \"'\#{name.schema}'\" : 'ANY (current_schemas(false))'}\n    SQL\n  end\nend\n", 'SCHEMA').rows.first[0].to_i > 0

#tables(name = nil) ⇒ Object 



12
13
14
15
16
17
18
19
20
 
# File 'lib/switchman/active_record/postgresql_adapter.rb', line 12

def tables(name = nil)
  schema = shard.name if @config[:use_qualified_names]

  query("    SELECT tablename\n    FROM pg_tables\n    WHERE schemaname = \#{schema ? \"'\#{schema}'\" : 'ANY (current_schemas(false))'}\n  SQL\nend\n", 'SCHEMA').map { |row| row[0] }