tgbyte-activerecord-jdbc-adapter 1.2.2.2
Sign up to get free protection for your applications and to get access to all the features.
- data/.gitignore +22 -0
- data/.travis.yml +8 -0
- data/Gemfile +14 -0
- data/Gemfile.lock +40 -0
- data/History.txt +488 -0
- data/LICENSE.txt +21 -0
- data/README.rdoc +199 -0
- data/Rakefile +60 -0
- data/activerecord-jdbc-adapter.gemspec +23 -0
- data/bench/bench_attributes.rb +13 -0
- data/bench/bench_attributes_new.rb +14 -0
- data/bench/bench_create.rb +12 -0
- data/bench/bench_find_all.rb +12 -0
- data/bench/bench_find_all_mt.rb +25 -0
- data/bench/bench_model.rb +85 -0
- data/bench/bench_new.rb +12 -0
- data/bench/bench_new_valid.rb +12 -0
- data/bench/bench_valid.rb +13 -0
- data/lib/active_record/connection_adapters/derby_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/h2_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/hsqldb_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/informix_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/jdbc_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/jndi_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/mssql_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/mysql2_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/mysql_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/oracle_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/postgresql_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/sqlite3_adapter.rb +1 -0
- data/lib/activerecord-jdbc-adapter.rb +8 -0
- data/lib/arel/engines/sql/compilers/db2_compiler.rb +9 -0
- data/lib/arel/engines/sql/compilers/derby_compiler.rb +6 -0
- data/lib/arel/engines/sql/compilers/h2_compiler.rb +6 -0
- data/lib/arel/engines/sql/compilers/hsqldb_compiler.rb +15 -0
- data/lib/arel/engines/sql/compilers/jdbc_compiler.rb +6 -0
- data/lib/arel/engines/sql/compilers/mssql_compiler.rb +46 -0
- data/lib/arel/visitors/compat.rb +13 -0
- data/lib/arel/visitors/db2.rb +17 -0
- data/lib/arel/visitors/derby.rb +32 -0
- data/lib/arel/visitors/firebird.rb +24 -0
- data/lib/arel/visitors/hsqldb.rb +26 -0
- data/lib/arel/visitors/sql_server.rb +46 -0
- data/lib/arjdbc.rb +24 -0
- data/lib/arjdbc/db2.rb +2 -0
- data/lib/arjdbc/db2/adapter.rb +510 -0
- data/lib/arjdbc/derby.rb +7 -0
- data/lib/arjdbc/derby/adapter.rb +358 -0
- data/lib/arjdbc/derby/connection_methods.rb +19 -0
- data/lib/arjdbc/discover.rb +92 -0
- data/lib/arjdbc/firebird.rb +2 -0
- data/lib/arjdbc/firebird/adapter.rb +136 -0
- data/lib/arjdbc/h2.rb +4 -0
- data/lib/arjdbc/h2/adapter.rb +54 -0
- data/lib/arjdbc/h2/connection_methods.rb +13 -0
- data/lib/arjdbc/hsqldb.rb +4 -0
- data/lib/arjdbc/hsqldb/adapter.rb +184 -0
- data/lib/arjdbc/hsqldb/connection_methods.rb +15 -0
- data/lib/arjdbc/informix.rb +3 -0
- data/lib/arjdbc/informix/adapter.rb +138 -0
- data/lib/arjdbc/informix/connection_methods.rb +11 -0
- data/lib/arjdbc/jdbc.rb +2 -0
- data/lib/arjdbc/jdbc/adapter.rb +356 -0
- data/lib/arjdbc/jdbc/adapter_java.jar +0 -0
- data/lib/arjdbc/jdbc/base_ext.rb +15 -0
- data/lib/arjdbc/jdbc/callbacks.rb +44 -0
- data/lib/arjdbc/jdbc/column.rb +47 -0
- data/lib/arjdbc/jdbc/compatibility.rb +51 -0
- data/lib/arjdbc/jdbc/connection.rb +134 -0
- data/lib/arjdbc/jdbc/connection_methods.rb +16 -0
- data/lib/arjdbc/jdbc/core_ext.rb +24 -0
- data/lib/arjdbc/jdbc/discover.rb +18 -0
- data/lib/arjdbc/jdbc/driver.rb +35 -0
- data/lib/arjdbc/jdbc/extension.rb +47 -0
- data/lib/arjdbc/jdbc/java.rb +14 -0
- data/lib/arjdbc/jdbc/jdbc.rake +131 -0
- data/lib/arjdbc/jdbc/missing_functionality_helper.rb +88 -0
- data/lib/arjdbc/jdbc/quoted_primary_key.rb +28 -0
- data/lib/arjdbc/jdbc/railtie.rb +9 -0
- data/lib/arjdbc/jdbc/rake_tasks.rb +10 -0
- data/lib/arjdbc/jdbc/require_driver.rb +16 -0
- data/lib/arjdbc/jdbc/type_converter.rb +126 -0
- data/lib/arjdbc/mimer.rb +2 -0
- data/lib/arjdbc/mimer/adapter.rb +142 -0
- data/lib/arjdbc/mssql.rb +4 -0
- data/lib/arjdbc/mssql/adapter.rb +477 -0
- data/lib/arjdbc/mssql/connection_methods.rb +31 -0
- data/lib/arjdbc/mssql/limit_helpers.rb +101 -0
- data/lib/arjdbc/mssql/lock_helpers.rb +72 -0
- data/lib/arjdbc/mssql/tsql_helper.rb +61 -0
- data/lib/arjdbc/mysql.rb +4 -0
- data/lib/arjdbc/mysql/adapter.rb +505 -0
- data/lib/arjdbc/mysql/connection_methods.rb +28 -0
- data/lib/arjdbc/oracle.rb +3 -0
- data/lib/arjdbc/oracle/adapter.rb +428 -0
- data/lib/arjdbc/oracle/connection_methods.rb +12 -0
- data/lib/arjdbc/postgresql.rb +4 -0
- data/lib/arjdbc/postgresql/adapter.rb +825 -0
- data/lib/arjdbc/postgresql/connection_methods.rb +23 -0
- data/lib/arjdbc/sqlite3.rb +4 -0
- data/lib/arjdbc/sqlite3/adapter.rb +389 -0
- data/lib/arjdbc/sqlite3/connection_methods.rb +35 -0
- data/lib/arjdbc/sybase.rb +2 -0
- data/lib/arjdbc/sybase/adapter.rb +46 -0
- data/lib/arjdbc/version.rb +8 -0
- data/lib/generators/jdbc/USAGE +10 -0
- data/lib/generators/jdbc/jdbc_generator.rb +9 -0
- data/lib/jdbc_adapter.rb +2 -0
- data/lib/jdbc_adapter/rake_tasks.rb +3 -0
- data/lib/jdbc_adapter/version.rb +3 -0
- data/lib/pg.rb +26 -0
- data/pom.xml +57 -0
- data/rails_generators/jdbc_generator.rb +15 -0
- data/rails_generators/templates/config/initializers/jdbc.rb +7 -0
- data/rails_generators/templates/lib/tasks/jdbc.rake +8 -0
- data/rakelib/bundler_ext.rb +11 -0
- data/rakelib/compile.rake +23 -0
- data/rakelib/db.rake +39 -0
- data/rakelib/rails.rake +41 -0
- data/src/java/arjdbc/db2/DB2RubyJdbcConnection.java +62 -0
- data/src/java/arjdbc/derby/DerbyModule.java +324 -0
- data/src/java/arjdbc/h2/H2RubyJdbcConnection.java +70 -0
- data/src/java/arjdbc/informix/InformixRubyJdbcConnection.java +74 -0
- data/src/java/arjdbc/jdbc/AdapterJavaService.java +68 -0
- data/src/java/arjdbc/jdbc/JdbcConnectionFactory.java +36 -0
- data/src/java/arjdbc/jdbc/RubyJdbcConnection.java +1326 -0
- data/src/java/arjdbc/jdbc/SQLBlock.java +48 -0
- data/src/java/arjdbc/mssql/MssqlRubyJdbcConnection.java +127 -0
- data/src/java/arjdbc/mysql/MySQLModule.java +134 -0
- data/src/java/arjdbc/mysql/MySQLRubyJdbcConnection.java +161 -0
- data/src/java/arjdbc/oracle/OracleRubyJdbcConnection.java +85 -0
- data/src/java/arjdbc/postgresql/PostgresqlRubyJdbcConnection.java +82 -0
- data/src/java/arjdbc/sqlite3/Sqlite3RubyJdbcConnection.java +126 -0
- data/test/abstract_db_create.rb +135 -0
- data/test/activerecord/connection_adapters/type_conversion_test.rb +31 -0
- data/test/activerecord/connections/native_jdbc_mysql/connection.rb +25 -0
- data/test/activerecord/jall.sh +7 -0
- data/test/activerecord/jtest.sh +3 -0
- data/test/db/db2.rb +11 -0
- data/test/db/derby.rb +12 -0
- data/test/db/h2.rb +11 -0
- data/test/db/hsqldb.rb +13 -0
- data/test/db/informix.rb +11 -0
- data/test/db/jdbc.rb +12 -0
- data/test/db/jndi_config.rb +40 -0
- data/test/db/logger.rb +3 -0
- data/test/db/mssql.rb +9 -0
- data/test/db/mysql.rb +10 -0
- data/test/db/oracle.rb +34 -0
- data/test/db/postgres.rb +9 -0
- data/test/db/sqlite3.rb +11 -0
- data/test/db2_reset_column_information_test.rb +8 -0
- data/test/db2_simple_test.rb +66 -0
- data/test/derby_migration_test.rb +68 -0
- data/test/derby_multibyte_test.rb +12 -0
- data/test/derby_reset_column_information_test.rb +8 -0
- data/test/derby_row_locking_test.rb +9 -0
- data/test/derby_simple_test.rb +139 -0
- data/test/generic_jdbc_connection_test.rb +29 -0
- data/test/h2_change_column_test.rb +68 -0
- data/test/h2_simple_test.rb +41 -0
- data/test/has_many_through.rb +79 -0
- data/test/helper.rb +87 -0
- data/test/hsqldb_simple_test.rb +6 -0
- data/test/informix_simple_test.rb +48 -0
- data/test/jdbc_common.rb +28 -0
- data/test/jndi_callbacks_test.rb +36 -0
- data/test/jndi_test.rb +25 -0
- data/test/manualTestDatabase.rb +191 -0
- data/test/models/add_not_null_column_to_table.rb +9 -0
- data/test/models/auto_id.rb +15 -0
- data/test/models/data_types.rb +30 -0
- data/test/models/entry.rb +40 -0
- data/test/models/mixed_case.rb +22 -0
- data/test/models/reserved_word.rb +15 -0
- data/test/models/string_id.rb +17 -0
- data/test/models/thing.rb +16 -0
- data/test/models/validates_uniqueness_of_string.rb +19 -0
- data/test/mssql_db_create_test.rb +26 -0
- data/test/mssql_identity_insert_test.rb +19 -0
- data/test/mssql_ignore_system_views_test.rb +27 -0
- data/test/mssql_legacy_types_test.rb +58 -0
- data/test/mssql_limit_offset_test.rb +136 -0
- data/test/mssql_multibyte_test.rb +18 -0
- data/test/mssql_null_test.rb +14 -0
- data/test/mssql_reset_column_information_test.rb +8 -0
- data/test/mssql_row_locking_sql_test.rb +159 -0
- data/test/mssql_row_locking_test.rb +9 -0
- data/test/mssql_simple_test.rb +55 -0
- data/test/mysql_db_create_test.rb +27 -0
- data/test/mysql_index_length_test.rb +58 -0
- data/test/mysql_info_test.rb +123 -0
- data/test/mysql_multibyte_test.rb +10 -0
- data/test/mysql_nonstandard_primary_key_test.rb +42 -0
- data/test/mysql_reset_column_information_test.rb +8 -0
- data/test/mysql_simple_test.rb +125 -0
- data/test/oracle_reset_column_information_test.rb +8 -0
- data/test/oracle_simple_test.rb +18 -0
- data/test/oracle_specific_test.rb +83 -0
- data/test/postgres_db_create_test.rb +32 -0
- data/test/postgres_drop_db_test.rb +16 -0
- data/test/postgres_information_schema_leak_test.rb +29 -0
- data/test/postgres_mixed_case_test.rb +29 -0
- data/test/postgres_native_type_mapping_test.rb +89 -0
- data/test/postgres_nonseq_pkey_test.rb +38 -0
- data/test/postgres_reserved_test.rb +22 -0
- data/test/postgres_reset_column_information_test.rb +8 -0
- data/test/postgres_schema_search_path_test.rb +48 -0
- data/test/postgres_simple_test.rb +167 -0
- data/test/postgres_table_alias_length_test.rb +15 -0
- data/test/postgres_type_conversion_test.rb +34 -0
- data/test/row_locking.rb +90 -0
- data/test/simple.rb +717 -0
- data/test/sqlite3_reset_column_information_test.rb +8 -0
- data/test/sqlite3_simple_test.rb +316 -0
- data/test/sybase_jtds_simple_test.rb +28 -0
- data/test/sybase_reset_column_information_test.rb +8 -0
- metadata +275 -0
@@ -0,0 +1,12 @@
|
|
1
|
+
class ActiveRecord::Base
|
2
|
+
class << self
|
3
|
+
def oracle_connection(config)
|
4
|
+
config[:port] ||= 1521
|
5
|
+
config[:url] ||= "jdbc:oracle:thin:@#{config[:host]}:#{config[:port]}:#{config[:database]}"
|
6
|
+
config[:driver] ||= "oracle.jdbc.driver.OracleDriver"
|
7
|
+
config[:adapter_spec] = ::ArJdbc::Oracle
|
8
|
+
jdbc_connection(config)
|
9
|
+
end
|
10
|
+
end
|
11
|
+
end
|
12
|
+
|
@@ -0,0 +1,825 @@
|
|
1
|
+
module ActiveRecord::ConnectionAdapters
|
2
|
+
PostgreSQLAdapter = Class.new(AbstractAdapter) unless const_defined?(:PostgreSQLAdapter)
|
3
|
+
end
|
4
|
+
|
5
|
+
module ::ArJdbc
|
6
|
+
module PostgreSQL
|
7
|
+
def self.extended(mod)
|
8
|
+
(class << mod; self; end).class_eval do
|
9
|
+
alias_chained_method :columns, :query_cache, :pg_columns
|
10
|
+
end
|
11
|
+
end
|
12
|
+
|
13
|
+
def self.column_selector
|
14
|
+
[/postgre/i, lambda {|cfg,col| col.extend(::ArJdbc::PostgreSQL::Column)}]
|
15
|
+
end
|
16
|
+
|
17
|
+
def self.jdbc_connection_class
|
18
|
+
::ActiveRecord::ConnectionAdapters::PostgresJdbcConnection
|
19
|
+
end
|
20
|
+
|
21
|
+
# column behavior based on postgresql_adapter in rails project
|
22
|
+
# https://github.com/rails/rails/blob/3-1-stable/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb#L41
|
23
|
+
module Column
|
24
|
+
def self.included(base)
|
25
|
+
class << base
|
26
|
+
attr_accessor :money_precision
|
27
|
+
def string_to_time(string)
|
28
|
+
return string unless String === string
|
29
|
+
|
30
|
+
case string
|
31
|
+
when 'infinity' then 1.0 / 0.0
|
32
|
+
when '-infinity' then -1.0 / 0.0
|
33
|
+
else
|
34
|
+
super
|
35
|
+
end
|
36
|
+
end
|
37
|
+
end
|
38
|
+
end
|
39
|
+
|
40
|
+
private
|
41
|
+
# Extracts the value from a Postgresql column default definition
|
42
|
+
def default_value(default)
|
43
|
+
case default
|
44
|
+
# This is a performance optimization for Ruby 1.9.2 in development.
|
45
|
+
# If the value is nil, we return nil straight away without checking
|
46
|
+
# the regular expressions. If we check each regular expression,
|
47
|
+
# Regexp#=== will call NilClass#to_str, which will trigger
|
48
|
+
# method_missing (defined by whiny nil in ActiveSupport) which
|
49
|
+
# makes this method very very slow.
|
50
|
+
when NilClass
|
51
|
+
nil
|
52
|
+
# Numeric types
|
53
|
+
when /\A\(?(-?\d+(\.\d*)?\)?)\z/
|
54
|
+
$1
|
55
|
+
# Character types
|
56
|
+
when /\A'(.*)'::(?:character varying|bpchar|text)\z/m
|
57
|
+
$1
|
58
|
+
# Character types (8.1 formatting)
|
59
|
+
when /\AE'(.*)'::(?:character varying|bpchar|text)\z/m
|
60
|
+
$1.gsub(/\\(\d\d\d)/) { $1.oct.chr }
|
61
|
+
# Binary data types
|
62
|
+
when /\A'(.*)'::bytea\z/m
|
63
|
+
$1
|
64
|
+
# Date/time types
|
65
|
+
when /\A'(.+)'::(?:time(?:stamp)? with(?:out)? time zone|date)\z/
|
66
|
+
$1
|
67
|
+
when /\A'(.*)'::interval\z/
|
68
|
+
$1
|
69
|
+
# Boolean type
|
70
|
+
when 'true'
|
71
|
+
true
|
72
|
+
when 'false'
|
73
|
+
false
|
74
|
+
# Geometric types
|
75
|
+
when /\A'(.*)'::(?:point|line|lseg|box|"?path"?|polygon|circle)\z/
|
76
|
+
$1
|
77
|
+
# Network address types
|
78
|
+
when /\A'(.*)'::(?:cidr|inet|macaddr)\z/
|
79
|
+
$1
|
80
|
+
# Bit string types
|
81
|
+
when /\AB'(.*)'::"?bit(?: varying)?"?\z/
|
82
|
+
$1
|
83
|
+
# XML type
|
84
|
+
when /\A'(.*)'::xml\z/m
|
85
|
+
$1
|
86
|
+
# Arrays
|
87
|
+
when /\A'(.*)'::"?\D+"?\[\]\z/
|
88
|
+
$1
|
89
|
+
# Object identifier types
|
90
|
+
when /\A-?\d+\z/
|
91
|
+
$1
|
92
|
+
else
|
93
|
+
# Anything else is blank, some user type, or some function
|
94
|
+
# and we can't know the value of that, so return nil.
|
95
|
+
nil
|
96
|
+
end
|
97
|
+
end
|
98
|
+
|
99
|
+
def extract_limit(sql_type)
|
100
|
+
case sql_type
|
101
|
+
when /^bigint/i then 8
|
102
|
+
when /^smallint/i then 2
|
103
|
+
else super
|
104
|
+
end
|
105
|
+
end
|
106
|
+
|
107
|
+
# Extracts the scale from PostgreSQL-specific data types.
|
108
|
+
def extract_scale(sql_type)
|
109
|
+
# Money type has a fixed scale of 2.
|
110
|
+
sql_type =~ /^money/ ? 2 : super
|
111
|
+
end
|
112
|
+
|
113
|
+
# Extracts the precision from PostgreSQL-specific data types.
|
114
|
+
def extract_precision(sql_type)
|
115
|
+
if sql_type == 'money'
|
116
|
+
self.class.money_precision
|
117
|
+
else
|
118
|
+
super
|
119
|
+
end
|
120
|
+
end
|
121
|
+
|
122
|
+
# Maps PostgreSQL-specific data types to logical Rails types.
|
123
|
+
def simplified_type(field_type)
|
124
|
+
case field_type
|
125
|
+
# Numeric and monetary types
|
126
|
+
when /^(?:real|double precision)$/ then :float
|
127
|
+
# Monetary types
|
128
|
+
when 'money' then :decimal
|
129
|
+
# Character types
|
130
|
+
when /^(?:character varying|bpchar)(?:\(\d+\))?$/ then :string
|
131
|
+
# Binary data types
|
132
|
+
when 'bytea' then :binary
|
133
|
+
# Date/time types
|
134
|
+
when /^timestamp with(?:out)? time zone$/ then :datetime
|
135
|
+
when 'interval' then :string
|
136
|
+
# Geometric types
|
137
|
+
when /^(?:point|line|lseg|box|"?path"?|polygon|circle)$/ then :string
|
138
|
+
# Network address types
|
139
|
+
when /^(?:cidr|inet|macaddr)$/ then :string
|
140
|
+
# Bit strings
|
141
|
+
when /^bit(?: varying)?(?:\(\d+\))?$/ then :string
|
142
|
+
# XML type
|
143
|
+
when 'xml' then :xml
|
144
|
+
# tsvector type
|
145
|
+
when 'tsvector' then :tsvector
|
146
|
+
# Arrays
|
147
|
+
when /^\D+\[\]$/ then :string
|
148
|
+
# Object identifier types
|
149
|
+
when 'oid' then :integer
|
150
|
+
# UUID type
|
151
|
+
when 'uuid' then :string
|
152
|
+
# Small and big integer types
|
153
|
+
when /^(?:small|big)int$/ then :integer
|
154
|
+
# Pass through all types that are not specific to PostgreSQL.
|
155
|
+
else
|
156
|
+
super
|
157
|
+
end
|
158
|
+
end
|
159
|
+
end
|
160
|
+
|
161
|
+
# constants taken from postgresql_adapter in rails project
|
162
|
+
ADAPTER_NAME = 'PostgreSQL'
|
163
|
+
|
164
|
+
NATIVE_DATABASE_TYPES = {
|
165
|
+
:primary_key => "serial primary key",
|
166
|
+
:string => { :name => "character varying", :limit => 255 },
|
167
|
+
:text => { :name => "text" },
|
168
|
+
:integer => { :name => "integer" },
|
169
|
+
:float => { :name => "float" },
|
170
|
+
:decimal => { :name => "decimal" },
|
171
|
+
:datetime => { :name => "timestamp" },
|
172
|
+
:timestamp => { :name => "timestamp" },
|
173
|
+
:time => { :name => "time" },
|
174
|
+
:date => { :name => "date" },
|
175
|
+
:binary => { :name => "bytea" },
|
176
|
+
:boolean => { :name => "boolean" },
|
177
|
+
:xml => { :name => "xml" },
|
178
|
+
:tsvector => { :name => "tsvector" }
|
179
|
+
}
|
180
|
+
|
181
|
+
def adapter_name #:nodoc:
|
182
|
+
ADAPTER_NAME
|
183
|
+
end
|
184
|
+
|
185
|
+
def self.arel2_visitors(config)
|
186
|
+
{}.tap {|v| %w(postgresql pg jdbcpostgresql).each {|a| v[a] = ::Arel::Visitors::PostgreSQL } }
|
187
|
+
end
|
188
|
+
|
189
|
+
def postgresql_version
|
190
|
+
@postgresql_version ||=
|
191
|
+
begin
|
192
|
+
value = select_value('SELECT version()')
|
193
|
+
if value =~ /PostgreSQL (\d+)\.(\d+)\.(\d+)/
|
194
|
+
($1.to_i * 10000) + ($2.to_i * 100) + $3.to_i
|
195
|
+
else
|
196
|
+
0
|
197
|
+
end
|
198
|
+
end
|
199
|
+
end
|
200
|
+
|
201
|
+
def native_database_types
|
202
|
+
NATIVE_DATABASE_TYPES
|
203
|
+
end
|
204
|
+
|
205
|
+
# Does PostgreSQL support migrations?
|
206
|
+
def supports_migrations?
|
207
|
+
true
|
208
|
+
end
|
209
|
+
|
210
|
+
# Does PostgreSQL support standard conforming strings?
|
211
|
+
def supports_standard_conforming_strings?
|
212
|
+
# Temporarily set the client message level above error to prevent unintentional
|
213
|
+
# error messages in the logs when working on a PostgreSQL database server that
|
214
|
+
# does not support standard conforming strings.
|
215
|
+
client_min_messages_old = client_min_messages
|
216
|
+
self.client_min_messages = 'panic'
|
217
|
+
|
218
|
+
# postgres-pr does not raise an exception when client_min_messages is set higher
|
219
|
+
# than error and "SHOW standard_conforming_strings" fails, but returns an empty
|
220
|
+
# PGresult instead.
|
221
|
+
has_support = select('SHOW standard_conforming_strings').to_a[0][0] rescue false
|
222
|
+
self.client_min_messages = client_min_messages_old
|
223
|
+
has_support
|
224
|
+
end
|
225
|
+
|
226
|
+
def supports_insert_with_returning?
|
227
|
+
postgresql_version >= 80200
|
228
|
+
end
|
229
|
+
|
230
|
+
def supports_ddl_transactions?
|
231
|
+
true
|
232
|
+
end
|
233
|
+
|
234
|
+
def supports_savepoints?
|
235
|
+
true
|
236
|
+
end
|
237
|
+
|
238
|
+
def supports_count_distinct? #:nodoc:
|
239
|
+
false
|
240
|
+
end
|
241
|
+
|
242
|
+
def create_savepoint
|
243
|
+
execute("SAVEPOINT #{current_savepoint_name}")
|
244
|
+
end
|
245
|
+
|
246
|
+
def rollback_to_savepoint
|
247
|
+
execute("ROLLBACK TO SAVEPOINT #{current_savepoint_name}")
|
248
|
+
end
|
249
|
+
|
250
|
+
def release_savepoint
|
251
|
+
execute("RELEASE SAVEPOINT #{current_savepoint_name}")
|
252
|
+
end
|
253
|
+
|
254
|
+
# Returns the configured supported identifier length supported by PostgreSQL,
|
255
|
+
# or report the default of 63 on PostgreSQL 7.x.
|
256
|
+
def table_alias_length
|
257
|
+
@table_alias_length ||= (postgresql_version >= 80000 ? select_one('SHOW max_identifier_length')['max_identifier_length'].to_i : 63)
|
258
|
+
end
|
259
|
+
|
260
|
+
def default_sequence_name(table_name, pk = nil)
|
261
|
+
default_pk, default_seq = pk_and_sequence_for(table_name)
|
262
|
+
default_seq || "#{table_name}_#{pk || default_pk || 'id'}_seq"
|
263
|
+
end
|
264
|
+
|
265
|
+
# Resets sequence to the max value of the table's pk if present.
|
266
|
+
def reset_pk_sequence!(table, pk = nil, sequence = nil) #:nodoc:
|
267
|
+
unless pk and sequence
|
268
|
+
default_pk, default_sequence = pk_and_sequence_for(table)
|
269
|
+
pk ||= default_pk
|
270
|
+
sequence ||= default_sequence
|
271
|
+
end
|
272
|
+
if pk
|
273
|
+
if sequence
|
274
|
+
quoted_sequence = quote_column_name(sequence)
|
275
|
+
|
276
|
+
select_value <<-end_sql, 'Reset sequence'
|
277
|
+
SELECT setval('#{quoted_sequence}', (SELECT COALESCE(MAX(#{quote_column_name pk})+(SELECT increment_by FROM #{quoted_sequence}), (SELECT min_value FROM #{quoted_sequence})) FROM #{quote_table_name(table)}), false)
|
278
|
+
end_sql
|
279
|
+
else
|
280
|
+
@logger.warn "#{table} has primary key #{pk} with no default sequence" if @logger
|
281
|
+
end
|
282
|
+
end
|
283
|
+
end
|
284
|
+
|
285
|
+
# Find a table's primary key and sequence.
|
286
|
+
def pk_and_sequence_for(table) #:nodoc:
|
287
|
+
# First try looking for a sequence with a dependency on the
|
288
|
+
# given table's primary key.
|
289
|
+
result = select(<<-end_sql, 'PK and serial sequence')[0]
|
290
|
+
SELECT attr.attname, seq.relname
|
291
|
+
FROM pg_class seq,
|
292
|
+
pg_attribute attr,
|
293
|
+
pg_depend dep,
|
294
|
+
pg_namespace name,
|
295
|
+
pg_constraint cons
|
296
|
+
WHERE seq.oid = dep.objid
|
297
|
+
AND seq.relkind = 'S'
|
298
|
+
AND attr.attrelid = dep.refobjid
|
299
|
+
AND attr.attnum = dep.refobjsubid
|
300
|
+
AND attr.attrelid = cons.conrelid
|
301
|
+
AND attr.attnum = cons.conkey[1]
|
302
|
+
AND cons.contype = 'p'
|
303
|
+
AND dep.refobjid = '#{quote_table_name(table)}'::regclass
|
304
|
+
end_sql
|
305
|
+
|
306
|
+
if result.nil? or result.empty?
|
307
|
+
# If that fails, try parsing the primary key's default value.
|
308
|
+
# Support the 7.x and 8.0 nextval('foo'::text) as well as
|
309
|
+
# the 8.1+ nextval('foo'::regclass).
|
310
|
+
result = select(<<-end_sql, 'PK and custom sequence')[0]
|
311
|
+
SELECT attr.attname,
|
312
|
+
CASE
|
313
|
+
WHEN split_part(def.adsrc, '''', 2) ~ '.' THEN
|
314
|
+
substr(split_part(def.adsrc, '''', 2),
|
315
|
+
strpos(split_part(def.adsrc, '''', 2), '.')+1)
|
316
|
+
ELSE split_part(def.adsrc, '''', 2)
|
317
|
+
END as relname
|
318
|
+
FROM pg_class t
|
319
|
+
JOIN pg_attribute attr ON (t.oid = attrelid)
|
320
|
+
JOIN pg_attrdef def ON (adrelid = attrelid AND adnum = attnum)
|
321
|
+
JOIN pg_constraint cons ON (conrelid = adrelid AND adnum = conkey[1])
|
322
|
+
WHERE t.oid = '#{quote_table_name(table)}'::regclass
|
323
|
+
AND cons.contype = 'p'
|
324
|
+
AND def.adsrc ~* 'nextval'
|
325
|
+
end_sql
|
326
|
+
end
|
327
|
+
|
328
|
+
[result["attname"], result["relname"]]
|
329
|
+
rescue
|
330
|
+
nil
|
331
|
+
end
|
332
|
+
|
333
|
+
def primary_key(table)
|
334
|
+
pk_and_sequence = pk_and_sequence_for(table)
|
335
|
+
pk_and_sequence && pk_and_sequence.first
|
336
|
+
end
|
337
|
+
|
338
|
+
# taken from rails postgresql adapter
|
339
|
+
# https://github.com/gfmurphy/rails/blob/master/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb#L611
|
340
|
+
def sql_for_insert(sql, pk, id_value, sequence_name, binds)
|
341
|
+
unless pk
|
342
|
+
table_ref = extract_table_ref_from_insert_sql(sql)
|
343
|
+
pk = primary_key(table_ref) if table_ref
|
344
|
+
end
|
345
|
+
|
346
|
+
sql = "#{sql} RETURNING #{quote_column_name(pk)}" if pk
|
347
|
+
|
348
|
+
[sql, binds]
|
349
|
+
end
|
350
|
+
|
351
|
+
def pg_columns(table_name, name=nil)
|
352
|
+
column_definitions(table_name).map do |row|
|
353
|
+
::ActiveRecord::ConnectionAdapters::PostgreSQLColumn.new(
|
354
|
+
row["column_name"], row["column_default"], row["column_type"],
|
355
|
+
row["column_not_null"] == "f")
|
356
|
+
end
|
357
|
+
end
|
358
|
+
|
359
|
+
# current database name
|
360
|
+
def current_database
|
361
|
+
exec_query("select current_database() as database").
|
362
|
+
first["database"]
|
363
|
+
end
|
364
|
+
|
365
|
+
# current database encoding
|
366
|
+
def encoding
|
367
|
+
exec_query(<<-end_sql).first["encoding"]
|
368
|
+
SELECT pg_encoding_to_char(pg_database.encoding) as encoding
|
369
|
+
FROM pg_database
|
370
|
+
WHERE pg_database.datname LIKE '#{current_database}'
|
371
|
+
end_sql
|
372
|
+
end
|
373
|
+
|
374
|
+
# Sets the maximum number columns postgres has, default 32
|
375
|
+
def multi_column_index_limit=(limit)
|
376
|
+
@multi_column_index_limit = limit
|
377
|
+
end
|
378
|
+
|
379
|
+
# Gets the maximum number columns postgres has, default 32
|
380
|
+
def multi_column_index_limit
|
381
|
+
defined?(@multi_column_index_limit) && @multi_column_index_limit || 32
|
382
|
+
end
|
383
|
+
|
384
|
+
# Based on postgresql_adapter.rb
|
385
|
+
def indexes(table_name, name = nil)
|
386
|
+
schema_search_path = @config[:schema_search_path] || select_rows('SHOW search_path')[0][0]
|
387
|
+
schemas = schema_search_path.split(/,/).map { |p| quote(p) }.join(',')
|
388
|
+
result = select_rows(<<-SQL, name)
|
389
|
+
SELECT i.relname, d.indisunique, a.attname, a.attnum, d.indkey
|
390
|
+
FROM pg_class t, pg_class i, pg_index d, pg_attribute a,
|
391
|
+
generate_series(0,#{multi_column_index_limit - 1}) AS s(i)
|
392
|
+
WHERE i.relkind = 'i'
|
393
|
+
AND d.indexrelid = i.oid
|
394
|
+
AND d.indisprimary = 'f'
|
395
|
+
AND t.oid = d.indrelid
|
396
|
+
AND t.relname = '#{table_name}'
|
397
|
+
AND i.relnamespace IN (SELECT oid FROM pg_namespace WHERE nspname IN (#{schemas}) )
|
398
|
+
AND a.attrelid = t.oid
|
399
|
+
AND d.indkey[s.i]=a.attnum
|
400
|
+
ORDER BY i.relname
|
401
|
+
SQL
|
402
|
+
|
403
|
+
current_index = nil
|
404
|
+
indexes = []
|
405
|
+
|
406
|
+
insertion_order = []
|
407
|
+
index_order = nil
|
408
|
+
|
409
|
+
result.each do |row|
|
410
|
+
if current_index != row[0]
|
411
|
+
|
412
|
+
(index_order = row[4].split(' ')).each_with_index{ |v, i| index_order[i] = v.to_i }
|
413
|
+
indexes << ::ActiveRecord::ConnectionAdapters::IndexDefinition.new(table_name, row[0], row[1] == "t", [])
|
414
|
+
current_index = row[0]
|
415
|
+
end
|
416
|
+
insertion_order = row[3]
|
417
|
+
ind = index_order.index(insertion_order)
|
418
|
+
indexes.last.columns[ind] = row[2]
|
419
|
+
end
|
420
|
+
|
421
|
+
indexes
|
422
|
+
end
|
423
|
+
|
424
|
+
# take id from result of insert query
|
425
|
+
def last_inserted_id(result)
|
426
|
+
Hash[Array(*result)].fetch("id") { result }
|
427
|
+
end
|
428
|
+
|
429
|
+
def last_insert_id(table, sequence_name)
|
430
|
+
Integer(select_value("SELECT currval('#{sequence_name}')"))
|
431
|
+
end
|
432
|
+
|
433
|
+
def recreate_database(name)
|
434
|
+
drop_database(name)
|
435
|
+
create_database(name)
|
436
|
+
end
|
437
|
+
|
438
|
+
def create_database(name, options = {})
|
439
|
+
execute "CREATE DATABASE \"#{name}\" ENCODING='#{options[:encoding] || 'utf8'}'"
|
440
|
+
end
|
441
|
+
|
442
|
+
def drop_database(name)
|
443
|
+
execute "DROP DATABASE IF EXISTS \"#{name}\""
|
444
|
+
end
|
445
|
+
|
446
|
+
def create_schema(schema_name, pg_username)
|
447
|
+
execute("CREATE SCHEMA \"#{schema_name}\" AUTHORIZATION \"#{pg_username}\"")
|
448
|
+
end
|
449
|
+
|
450
|
+
def drop_schema(schema_name)
|
451
|
+
execute("DROP SCHEMA \"#{schema_name}\"")
|
452
|
+
end
|
453
|
+
|
454
|
+
def all_schemas
|
455
|
+
select('select nspname from pg_namespace').map {|r| r["nspname"] }
|
456
|
+
end
|
457
|
+
|
458
|
+
def structure_dump
|
459
|
+
database = @config[:database]
|
460
|
+
if database.nil?
|
461
|
+
if @config[:url] =~ /\/([^\/]*)$/
|
462
|
+
database = $1
|
463
|
+
else
|
464
|
+
raise "Could not figure out what database this url is for #{@config["url"]}"
|
465
|
+
end
|
466
|
+
end
|
467
|
+
|
468
|
+
ENV['PGHOST'] = @config[:host] if @config[:host]
|
469
|
+
ENV['PGPORT'] = @config[:port].to_s if @config[:port]
|
470
|
+
ENV['PGPASSWORD'] = @config[:password].to_s if @config[:password]
|
471
|
+
search_path = @config[:schema_search_path]
|
472
|
+
search_path = "--schema=#{search_path}" if search_path
|
473
|
+
|
474
|
+
@connection.connection.close
|
475
|
+
begin
|
476
|
+
definition = `pg_dump -i -U "#{@config[:username]}" -s -x -O #{search_path} #{database}`
|
477
|
+
raise "Error dumping database" if $?.exitstatus == 1
|
478
|
+
|
479
|
+
# need to patch away any references to SQL_ASCII as it breaks the JDBC driver
|
480
|
+
definition.gsub(/SQL_ASCII/, 'UNICODE')
|
481
|
+
ensure
|
482
|
+
reconnect!
|
483
|
+
end
|
484
|
+
end
|
485
|
+
|
486
|
+
# SELECT DISTINCT clause for a given set of columns and a given ORDER BY clause.
|
487
|
+
#
|
488
|
+
# PostgreSQL requires the ORDER BY columns in the select list for distinct queries, and
|
489
|
+
# requires that the ORDER BY include the distinct column.
|
490
|
+
#
|
491
|
+
# distinct("posts.id", "posts.created_at desc")
|
492
|
+
def distinct(columns, orders) #:nodoc:
|
493
|
+
return "DISTINCT #{columns}" if orders.empty?
|
494
|
+
|
495
|
+
# Construct a clean list of column names from the ORDER BY clause, removing
|
496
|
+
# any ASC/DESC modifiers
|
497
|
+
order_columns = orders.collect { |s| s.gsub(/\s+(ASC|DESC)\s*/i, '') }.
|
498
|
+
reject(&:blank?)
|
499
|
+
order_columns = order_columns.
|
500
|
+
zip((0...order_columns.size).to_a).map { |s,i| "#{s} AS alias_#{i}" }
|
501
|
+
|
502
|
+
"DISTINCT #{columns}, #{order_columns * ', '}"
|
503
|
+
end
|
504
|
+
|
505
|
+
# ORDER BY clause for the passed order option.
|
506
|
+
#
|
507
|
+
# PostgreSQL does not allow arbitrary ordering when using DISTINCT ON, so we work around this
|
508
|
+
# by wrapping the sql as a sub-select and ordering in that query.
|
509
|
+
def add_order_by_for_association_limiting!(sql, options)
|
510
|
+
return sql if options[:order].blank?
|
511
|
+
|
512
|
+
order = options[:order].split(',').collect { |s| s.strip }.reject(&:blank?)
|
513
|
+
order.map! { |s| 'DESC' if s =~ /\bdesc$/i }
|
514
|
+
order = order.zip((0...order.size).to_a).map { |s,i| "id_list.alias_#{i} #{s}" }.join(', ')
|
515
|
+
|
516
|
+
sql.replace "SELECT * FROM (#{sql}) AS id_list ORDER BY #{order}"
|
517
|
+
end
|
518
|
+
|
519
|
+
# from postgres_adapter.rb in rails project
|
520
|
+
# https://github.com/rails/rails/blob/3-1-stable/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb#L412
|
521
|
+
# Quotes PostgreSQL-specific data types for SQL input.
|
522
|
+
def quote(value, column = nil) #:nodoc:
|
523
|
+
return super unless column
|
524
|
+
|
525
|
+
case value
|
526
|
+
when Float
|
527
|
+
return super unless value.infinite? && column.type == :datetime
|
528
|
+
"'#{value.to_s.downcase}'"
|
529
|
+
when Numeric
|
530
|
+
return super unless column.sql_type == 'money'
|
531
|
+
# Not truly string input, so doesn't require (or allow) escape string syntax.
|
532
|
+
"'#{value}'"
|
533
|
+
when String
|
534
|
+
case column.sql_type
|
535
|
+
when 'bytea' then "'#{escape_bytea(value)}'"
|
536
|
+
when 'xml' then "xml '#{quote_string(value)}'"
|
537
|
+
when /^bit/
|
538
|
+
case value
|
539
|
+
when /^[01]*$/ then "B'#{value}'" # Bit-string notation
|
540
|
+
when /^[0-9A-F]*$/i then "X'#{value}'" # Hexadecimal notation
|
541
|
+
end
|
542
|
+
else
|
543
|
+
super
|
544
|
+
end
|
545
|
+
else
|
546
|
+
super
|
547
|
+
end
|
548
|
+
end
|
549
|
+
|
550
|
+
def escape_bytea(s)
|
551
|
+
if s
|
552
|
+
result = ''
|
553
|
+
s.each_byte { |c| result << sprintf('\\\\%03o', c) }
|
554
|
+
result
|
555
|
+
end
|
556
|
+
end
|
557
|
+
|
558
|
+
def quote_table_name(name)
|
559
|
+
schema, name_part = extract_pg_identifier_from_name(name.to_s)
|
560
|
+
|
561
|
+
unless name_part
|
562
|
+
quote_column_name(schema)
|
563
|
+
else
|
564
|
+
table_name, name_part = extract_pg_identifier_from_name(name_part)
|
565
|
+
"#{quote_column_name(schema)}.#{quote_column_name(table_name)}"
|
566
|
+
end
|
567
|
+
end
|
568
|
+
|
569
|
+
def quote_column_name(name)
|
570
|
+
%("#{name.to_s.gsub("\"", "\"\"")}")
|
571
|
+
end
|
572
|
+
|
573
|
+
def quoted_date(value) #:nodoc:
|
574
|
+
if value.acts_like?(:time) && value.respond_to?(:usec)
|
575
|
+
"#{super}.#{sprintf("%06d", value.usec)}"
|
576
|
+
else
|
577
|
+
super
|
578
|
+
end
|
579
|
+
end
|
580
|
+
|
581
|
+
def disable_referential_integrity(&block) #:nodoc:
|
582
|
+
execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} DISABLE TRIGGER ALL" }.join(";"))
|
583
|
+
yield
|
584
|
+
ensure
|
585
|
+
execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} ENABLE TRIGGER ALL" }.join(";"))
|
586
|
+
end
|
587
|
+
|
588
|
+
def rename_table(name, new_name)
|
589
|
+
execute "ALTER TABLE #{name} RENAME TO #{new_name}"
|
590
|
+
end
|
591
|
+
|
592
|
+
# Adds a new column to the named table.
|
593
|
+
# See TableDefinition#column for details of the options you can use.
|
594
|
+
def add_column(table_name, column_name, type, options = {})
|
595
|
+
default = options[:default]
|
596
|
+
notnull = options[:null] == false
|
597
|
+
|
598
|
+
# Add the column.
|
599
|
+
execute("ALTER TABLE #{quote_table_name(table_name)} ADD COLUMN #{quote_column_name(column_name)} #{type_to_sql(type, options[:limit], options[:precision], options[:scale])}")
|
600
|
+
|
601
|
+
change_column_default(table_name, column_name, default) if options_include_default?(options)
|
602
|
+
change_column_null(table_name, column_name, false, default) if notnull
|
603
|
+
end
|
604
|
+
|
605
|
+
# Changes the column of a table.
|
606
|
+
def change_column(table_name, column_name, type, options = {})
|
607
|
+
quoted_table_name = quote_table_name(table_name)
|
608
|
+
|
609
|
+
begin
|
610
|
+
execute "ALTER TABLE #{quoted_table_name} ALTER COLUMN #{quote_column_name(column_name)} TYPE #{type_to_sql(type, options[:limit], options[:precision], options[:scale])}"
|
611
|
+
rescue ActiveRecord::StatementInvalid => e
|
612
|
+
raise e if postgresql_version > 80000
|
613
|
+
# This is PostgreSQL 7.x, so we have to use a more arcane way of doing it.
|
614
|
+
begin
|
615
|
+
begin_db_transaction
|
616
|
+
tmp_column_name = "#{column_name}_ar_tmp"
|
617
|
+
add_column(table_name, tmp_column_name, type, options)
|
618
|
+
execute "UPDATE #{quoted_table_name} SET #{quote_column_name(tmp_column_name)} = CAST(#{quote_column_name(column_name)} AS #{type_to_sql(type, options[:limit], options[:precision], options[:scale])})"
|
619
|
+
remove_column(table_name, column_name)
|
620
|
+
rename_column(table_name, tmp_column_name, column_name)
|
621
|
+
commit_db_transaction
|
622
|
+
rescue
|
623
|
+
rollback_db_transaction
|
624
|
+
end
|
625
|
+
end
|
626
|
+
|
627
|
+
change_column_default(table_name, column_name, options[:default]) if options_include_default?(options)
|
628
|
+
change_column_null(table_name, column_name, options[:null], options[:default]) if options.key?(:null)
|
629
|
+
end
|
630
|
+
|
631
|
+
# Changes the default value of a table column.
|
632
|
+
def change_column_default(table_name, column_name, default)
|
633
|
+
execute "ALTER TABLE #{quote_table_name(table_name)} ALTER COLUMN #{quote_column_name(column_name)} SET DEFAULT #{quote(default)}"
|
634
|
+
end
|
635
|
+
|
636
|
+
def change_column_null(table_name, column_name, null, default = nil)
|
637
|
+
unless null || default.nil?
|
638
|
+
execute("UPDATE #{quote_table_name(table_name)} SET #{quote_column_name(column_name)}=#{quote(default)} WHERE #{quote_column_name(column_name)} IS NULL")
|
639
|
+
end
|
640
|
+
execute("ALTER TABLE #{quote_table_name(table_name)} ALTER #{quote_column_name(column_name)} #{null ? 'DROP' : 'SET'} NOT NULL")
|
641
|
+
end
|
642
|
+
|
643
|
+
def rename_column(table_name, column_name, new_column_name) #:nodoc:
|
644
|
+
execute "ALTER TABLE #{quote_table_name(table_name)} RENAME COLUMN #{quote_column_name(column_name)} TO #{quote_column_name(new_column_name)}"
|
645
|
+
end
|
646
|
+
|
647
|
+
def remove_index!(table_name, index_name) #:nodoc:
|
648
|
+
execute "DROP INDEX #{quote_table_name(index_name)}"
|
649
|
+
end
|
650
|
+
|
651
|
+
def index_name_length
|
652
|
+
63
|
653
|
+
end
|
654
|
+
|
655
|
+
# Maps logical Rails types to PostgreSQL-specific data types.
|
656
|
+
def type_to_sql(type, limit = nil, precision = nil, scale = nil)
|
657
|
+
return super unless type.to_s == 'integer'
|
658
|
+
return 'integer' unless limit
|
659
|
+
|
660
|
+
case limit
|
661
|
+
when 1, 2; 'smallint'
|
662
|
+
when 3, 4; 'integer'
|
663
|
+
when 5..8; 'bigint'
|
664
|
+
else raise(ActiveRecordError, "No integer type has byte size #{limit}. Use a numeric with precision 0 instead.")
|
665
|
+
end
|
666
|
+
end
|
667
|
+
|
668
|
+
def tables(name = nil)
|
669
|
+
exec_query(<<-SQL, 'SCHEMA').map { |row| row["tablename"] }
|
670
|
+
SELECT tablename
|
671
|
+
FROM pg_tables
|
672
|
+
WHERE schemaname = ANY (current_schemas(false))
|
673
|
+
SQL
|
674
|
+
end
|
675
|
+
|
676
|
+
def table_exists?(name)
|
677
|
+
schema, table = extract_schema_and_table(name.to_s)
|
678
|
+
return false unless table # Abstract classes is having nil table name
|
679
|
+
|
680
|
+
binds = [[nil, table.gsub(/(^"|"$)/,'')]]
|
681
|
+
binds << [nil, schema] if schema
|
682
|
+
|
683
|
+
exec_query(<<-SQL, 'SCHEMA', binds).first["table_count"] > 0
|
684
|
+
SELECT COUNT(*) as table_count
|
685
|
+
FROM pg_tables
|
686
|
+
WHERE tablename = ?
|
687
|
+
AND schemaname = #{schema ? "?" : "ANY (current_schemas(false))"}
|
688
|
+
SQL
|
689
|
+
end
|
690
|
+
|
691
|
+
# Extracts the table and schema name from +name+
|
692
|
+
def extract_schema_and_table(name)
|
693
|
+
schema, table = name.split('.', 2)
|
694
|
+
|
695
|
+
unless table # A table was provided without a schema
|
696
|
+
table = schema
|
697
|
+
schema = nil
|
698
|
+
end
|
699
|
+
|
700
|
+
if name =~ /^"/ # Handle quoted table names
|
701
|
+
table = name
|
702
|
+
schema = nil
|
703
|
+
end
|
704
|
+
[schema, table]
|
705
|
+
end
|
706
|
+
|
707
|
+
private
|
708
|
+
def translate_exception(exception, message)
|
709
|
+
case exception.message
|
710
|
+
when /duplicate key value violates unique constraint/
|
711
|
+
::ActiveRecord::RecordNotUnique.new(message, exception)
|
712
|
+
when /violates foreign key constraint/
|
713
|
+
::ActiveRecord::InvalidForeignKey.new(message, exception)
|
714
|
+
else
|
715
|
+
super
|
716
|
+
end
|
717
|
+
end
|
718
|
+
|
719
|
+
# Returns the list of a table's column names, data types, and default values.
|
720
|
+
#
|
721
|
+
# The underlying query is roughly:
|
722
|
+
# SELECT column.name, column.type, default.value
|
723
|
+
# FROM column LEFT JOIN default
|
724
|
+
# ON column.table_id = default.table_id
|
725
|
+
# AND column.num = default.column_num
|
726
|
+
# WHERE column.table_id = get_table_id('table_name')
|
727
|
+
# AND column.num > 0
|
728
|
+
# AND NOT column.is_dropped
|
729
|
+
# ORDER BY column.num
|
730
|
+
#
|
731
|
+
# If the table name is not prefixed with a schema, the database will
|
732
|
+
# take the first match from the schema search path.
|
733
|
+
#
|
734
|
+
# Query implementation notes:
|
735
|
+
# - format_type includes the column size constraint, e.g. varchar(50)
|
736
|
+
# - ::regclass is a function that gives the id for a table name
|
737
|
+
def column_definitions(table_name) #:nodoc:
|
738
|
+
exec_query(<<-end_sql, 'SCHEMA')
|
739
|
+
SELECT a.attname as column_name, format_type(a.atttypid, a.atttypmod) as column_type, d.adsrc as column_default, a.attnotnull as column_not_null
|
740
|
+
FROM pg_attribute a LEFT JOIN pg_attrdef d
|
741
|
+
ON a.attrelid = d.adrelid AND a.attnum = d.adnum
|
742
|
+
WHERE a.attrelid = '#{quote_table_name(table_name)}'::regclass
|
743
|
+
AND a.attnum > 0 AND NOT a.attisdropped
|
744
|
+
ORDER BY a.attnum
|
745
|
+
end_sql
|
746
|
+
end
|
747
|
+
|
748
|
+
def extract_pg_identifier_from_name(name)
|
749
|
+
match_data = name[0,1] == '"' ? name.match(/\"([^\"]+)\"/) : name.match(/([^\.]+)/)
|
750
|
+
|
751
|
+
if match_data
|
752
|
+
rest = name[match_data[0].length..-1]
|
753
|
+
rest = rest[1..-1] if rest[0,1] == "."
|
754
|
+
[match_data[1], (rest.length > 0 ? rest : nil)]
|
755
|
+
end
|
756
|
+
end
|
757
|
+
|
758
|
+
# from rails postgresl_adapter
|
759
|
+
def extract_table_ref_from_insert_sql(sql)
|
760
|
+
sql[/into\s+([^\(]*).*values\s*\(/i]
|
761
|
+
$1.strip if $1
|
762
|
+
end
|
763
|
+
end
|
764
|
+
end
|
765
|
+
|
766
|
+
module ActiveRecord::ConnectionAdapters
|
767
|
+
remove_const(:PostgreSQLAdapter) if const_defined?(:PostgreSQLAdapter)
|
768
|
+
|
769
|
+
class PostgreSQLColumn < JdbcColumn
|
770
|
+
include ArJdbc::PostgreSQL::Column
|
771
|
+
|
772
|
+
def initialize(name, *args)
|
773
|
+
if Hash === name
|
774
|
+
super
|
775
|
+
else
|
776
|
+
super(nil, name, *args)
|
777
|
+
end
|
778
|
+
end
|
779
|
+
|
780
|
+
def call_discovered_column_callbacks(*)
|
781
|
+
end
|
782
|
+
end
|
783
|
+
|
784
|
+
class PostgresJdbcConnection < JdbcConnection
|
785
|
+
alias :java_native_database_types :set_native_database_types
|
786
|
+
|
787
|
+
# override to prevent connection from loading hash from jdbc
|
788
|
+
# metadata, which can be expensive. We can do this since
|
789
|
+
# native_database_types is defined in the adapter to use a static hash
|
790
|
+
# not relying on the driver's metadata
|
791
|
+
def set_native_database_types
|
792
|
+
@native_types = {}
|
793
|
+
end
|
794
|
+
end
|
795
|
+
|
796
|
+
class PostgreSQLAdapter < JdbcAdapter
|
797
|
+
include ArJdbc::PostgreSQL
|
798
|
+
|
799
|
+
class TableDefinition < ActiveRecord::ConnectionAdapters::TableDefinition
|
800
|
+
def xml(*args)
|
801
|
+
options = args.extract_options!
|
802
|
+
column(args[0], "xml", options)
|
803
|
+
end
|
804
|
+
|
805
|
+
def tsvector(*args)
|
806
|
+
options = args.extract_options!
|
807
|
+
column(args[0], "tsvector", options)
|
808
|
+
end
|
809
|
+
end
|
810
|
+
|
811
|
+
def table_definition
|
812
|
+
TableDefinition.new(self)
|
813
|
+
end
|
814
|
+
|
815
|
+
def jdbc_connection_class(spec)
|
816
|
+
::ArJdbc::PostgreSQL.jdbc_connection_class
|
817
|
+
end
|
818
|
+
|
819
|
+
def jdbc_column_class
|
820
|
+
ActiveRecord::ConnectionAdapters::PostgreSQLColumn
|
821
|
+
end
|
822
|
+
|
823
|
+
alias_chained_method :columns, :query_cache, :pg_columns
|
824
|
+
end
|
825
|
+
end
|