saturnflyer-activerecord-jdbc-adapter 0.9.3
Sign up to get free protection for your applications and to get access to all the features.
- data/.gitignore +16 -0
- data/History.txt +225 -0
- data/LICENSE.txt +21 -0
- data/README.txt +161 -0
- data/Rakefile +28 -0
- data/VERSION +1 -0
- data/activerecord-jdbc-adapter.gemspec +277 -0
- data/adapters/derby/LICENSE.txt +21 -0
- data/adapters/derby/README.txt +5 -0
- data/adapters/derby/Rakefile +31 -0
- data/adapters/derby/lib/active_record/connection_adapters/jdbcderby_adapter.rb +30 -0
- data/adapters/h2/History.txt +0 -0
- data/adapters/h2/LICENSE.txt +21 -0
- data/adapters/h2/README.txt +5 -0
- data/adapters/h2/Rakefile +31 -0
- data/adapters/h2/lib/active_record/connection_adapters/jdbch2_adapter.rb +30 -0
- data/adapters/hsqldb/History.txt +0 -0
- data/adapters/hsqldb/LICENSE.txt +21 -0
- data/adapters/hsqldb/README.txt +5 -0
- data/adapters/hsqldb/Rakefile +31 -0
- data/adapters/hsqldb/lib/active_record/connection_adapters/jdbchsqldb_adapter.rb +30 -0
- data/adapters/mysql/History.txt +0 -0
- data/adapters/mysql/LICENSE.txt +21 -0
- data/adapters/mysql/README.txt +5 -0
- data/adapters/mysql/Rakefile +31 -0
- data/adapters/mysql/lib/active_record/connection_adapters/jdbcmysql_adapter.rb +30 -0
- data/adapters/postgresql/History.txt +0 -0
- data/adapters/postgresql/LICENSE.txt +21 -0
- data/adapters/postgresql/README.txt +5 -0
- data/adapters/postgresql/Rakefile +31 -0
- data/adapters/postgresql/lib/active_record/connection_adapters/jdbcpostgresql_adapter.rb +30 -0
- data/adapters/sqlite3/LICENSE.txt +21 -0
- data/adapters/sqlite3/README.txt +5 -0
- data/adapters/sqlite3/Rakefile +31 -0
- data/adapters/sqlite3/lib/active_record/connection_adapters/jdbcsqlite3_adapter.rb +30 -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/drivers/derby/LICENSE.txt +1 -0
- data/drivers/derby/README.txt +6 -0
- data/drivers/derby/Rakefile +26 -0
- data/drivers/derby/lib/derby-10.5.3.0.jar +0 -0
- data/drivers/derby/lib/jdbc/derby.rb +11 -0
- data/drivers/h2/History.txt +0 -0
- data/drivers/h2/LICENSE.txt +2 -0
- data/drivers/h2/README.txt +6 -0
- data/drivers/h2/Rakefile +26 -0
- data/drivers/h2/lib/h2-1.1.107.jar +0 -0
- data/drivers/h2/lib/jdbc/h2.rb +10 -0
- data/drivers/hsqldb/History.txt +0 -0
- data/drivers/hsqldb/LICENSE.txt +2 -0
- data/drivers/hsqldb/README.txt +6 -0
- data/drivers/hsqldb/Rakefile +26 -0
- data/drivers/hsqldb/lib/hsqldb-1.8.0.7.jar +0 -0
- data/drivers/hsqldb/lib/jdbc/hsqldb.rb +10 -0
- data/drivers/mysql/History.txt +0 -0
- data/drivers/mysql/LICENSE.txt +1 -0
- data/drivers/mysql/README.txt +6 -0
- data/drivers/mysql/Rakefile +26 -0
- data/drivers/mysql/lib/jdbc/mysql.rb +10 -0
- data/drivers/mysql/lib/mysql-connector-java-5.0.4-bin.jar +0 -0
- data/drivers/postgres/History.txt +7 -0
- data/drivers/postgres/LICENSE.txt +12 -0
- data/drivers/postgres/README.txt +5 -0
- data/drivers/postgres/Rakefile +27 -0
- data/drivers/postgres/lib/jdbc/postgres.rb +25 -0
- data/drivers/postgres/lib/postgresql-8.3-604.jdbc3.jar +0 -0
- data/drivers/postgres/lib/postgresql-8.3-604.jdbc4.jar +0 -0
- data/drivers/sqlite3/LICENSE.txt +13 -0
- data/drivers/sqlite3/README.txt +6 -0
- data/drivers/sqlite3/Rakefile +26 -0
- data/drivers/sqlite3/lib/jdbc/sqlite3.rb +10 -0
- data/drivers/sqlite3/lib/sqlitejdbc-3.6.3.054.jar +0 -0
- data/lib/active_record/connection_adapters/cachedb_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/derby_adapter.rb +13 -0
- data/lib/active_record/connection_adapters/h2_adapter.rb +13 -0
- data/lib/active_record/connection_adapters/hsqldb_adapter.rb +13 -0
- data/lib/active_record/connection_adapters/informix_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/jdbc_adapter.rb +637 -0
- data/lib/active_record/connection_adapters/jdbc_adapter_spec.rb +14 -0
- data/lib/active_record/connection_adapters/jndi_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/mysql_adapter.rb +13 -0
- data/lib/active_record/connection_adapters/oracle_adapter.rb +1 -0
- data/lib/active_record/connection_adapters/postgresql_adapter.rb +13 -0
- data/lib/active_record/connection_adapters/sqlite3_adapter.rb +13 -0
- data/lib/jdbc_adapter.rb +27 -0
- data/lib/jdbc_adapter/jdbc.rake +104 -0
- data/lib/jdbc_adapter/jdbc_cachedb.rb +33 -0
- data/lib/jdbc_adapter/jdbc_db2.rb +191 -0
- data/lib/jdbc_adapter/jdbc_derby.rb +421 -0
- data/lib/jdbc_adapter/jdbc_firebird.rb +109 -0
- data/lib/jdbc_adapter/jdbc_hsqldb.rb +210 -0
- data/lib/jdbc_adapter/jdbc_informix.rb +147 -0
- data/lib/jdbc_adapter/jdbc_mimer.rb +141 -0
- data/lib/jdbc_adapter/jdbc_mssql.rb +333 -0
- data/lib/jdbc_adapter/jdbc_mysql.rb +233 -0
- data/lib/jdbc_adapter/jdbc_oracle.rb +374 -0
- data/lib/jdbc_adapter/jdbc_postgre.rb +483 -0
- data/lib/jdbc_adapter/jdbc_sqlite3.rb +360 -0
- data/lib/jdbc_adapter/jdbc_sybase.rb +50 -0
- data/lib/jdbc_adapter/missing_functionality_helper.rb +85 -0
- data/lib/jdbc_adapter/rake_tasks.rb +10 -0
- data/lib/jdbc_adapter/tsql_helper.rb +60 -0
- data/lib/jdbc_adapter/version.rb +5 -0
- data/pom.xml +57 -0
- data/rakelib/compile.rake +23 -0
- data/rakelib/package.rake +73 -0
- data/rakelib/rails.rake +41 -0
- data/src/java/jdbc_adapter/JdbcAdapterInternalService.java +53 -0
- data/src/java/jdbc_adapter/JdbcConnectionFactory.java +36 -0
- data/src/java/jdbc_adapter/JdbcDerbySpec.java +293 -0
- data/src/java/jdbc_adapter/JdbcMySQLSpec.java +133 -0
- data/src/java/jdbc_adapter/MssqlRubyJdbcConnection.java +71 -0
- data/src/java/jdbc_adapter/PostgresRubyJdbcConnection.java +55 -0
- data/src/java/jdbc_adapter/RubyJdbcConnection.java +1163 -0
- data/src/java/jdbc_adapter/SQLBlock.java +27 -0
- data/src/java/jdbc_adapter/Sqlite3RubyJdbcConnection.java +41 -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/cachedb_simple_test.rb +6 -0
- data/test/db/cachedb.rb +9 -0
- data/test/db/db2.rb +9 -0
- data/test/db/derby.rb +14 -0
- data/test/db/h2.rb +11 -0
- data/test/db/hsqldb.rb +12 -0
- data/test/db/informix.rb +11 -0
- data/test/db/jdbc.rb +11 -0
- data/test/db/jndi_config.rb +30 -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 +15 -0
- data/test/db2_simple_test.rb +10 -0
- data/test/derby_migration_test.rb +21 -0
- data/test/derby_multibyte_test.rb +12 -0
- data/test/derby_simple_test.rb +12 -0
- data/test/generic_jdbc_connection_test.rb +9 -0
- data/test/h2_simple_test.rb +6 -0
- data/test/has_many_through.rb +79 -0
- data/test/hsqldb_simple_test.rb +6 -0
- data/test/informix_simple_test.rb +48 -0
- data/test/jdbc_adapter/jdbc_db2_test.rb +26 -0
- data/test/jdbc_adapter/jdbc_sybase_test.rb +33 -0
- data/test/jdbc_common.rb +24 -0
- data/test/jndi_callbacks_test.rb +38 -0
- data/test/jndi_test.rb +35 -0
- data/test/manualTestDatabase.rb +191 -0
- data/test/minirunit.rb +109 -0
- data/test/minirunit/testConnect.rb +14 -0
- data/test/minirunit/testH2.rb +73 -0
- data/test/minirunit/testHsqldb.rb +73 -0
- data/test/minirunit/testLoadActiveRecord.rb +3 -0
- data/test/minirunit/testMysql.rb +83 -0
- data/test/minirunit/testRawSelect.rb +24 -0
- data/test/models/add_not_null_column_to_table.rb +12 -0
- data/test/models/auto_id.rb +18 -0
- data/test/models/data_types.rb +28 -0
- data/test/models/entry.rb +23 -0
- data/test/models/mixed_case.rb +20 -0
- data/test/models/reserved_word.rb +18 -0
- data/test/mssql_simple_test.rb +6 -0
- data/test/mysql_multibyte_test.rb +10 -0
- data/test/mysql_nonstandard_primary_key_test.rb +43 -0
- data/test/mysql_simple_test.rb +26 -0
- data/test/oracle_simple_test.rb +29 -0
- data/test/postgres_mixed_case_test.rb +19 -0
- data/test/postgres_nonseq_pkey_test.rb +40 -0
- data/test/postgres_reserved_test.rb +22 -0
- data/test/postgres_simple_test.rb +12 -0
- data/test/simple.rb +397 -0
- data/test/sqlite3_simple_test.rb +179 -0
- data/test/sybase_jtds_simple_test.rb +6 -0
- metadata +294 -0
@@ -0,0 +1,483 @@
|
|
1
|
+
|
2
|
+
module ::JdbcSpec
|
3
|
+
# Don't need to load native postgres adapter
|
4
|
+
$LOADED_FEATURES << "active_record/connection_adapters/postgresql_adapter.rb"
|
5
|
+
|
6
|
+
module ActiveRecordExtensions
|
7
|
+
def postgresql_connection(config)
|
8
|
+
config[:host] ||= "localhost"
|
9
|
+
config[:port] ||= 5432
|
10
|
+
config[:url] ||= "jdbc:postgresql://#{config[:host]}:#{config[:port]}/#{config[:database]}"
|
11
|
+
config[:url] << config[:pg_params] if config[:pg_params]
|
12
|
+
config[:driver] ||= "org.postgresql.Driver"
|
13
|
+
jdbc_connection(config)
|
14
|
+
end
|
15
|
+
end
|
16
|
+
|
17
|
+
module PostgreSQL
|
18
|
+
def self.adapter_matcher(name, *)
|
19
|
+
name =~ /postgre/i ? self : false
|
20
|
+
end
|
21
|
+
|
22
|
+
def self.column_selector
|
23
|
+
[/postgre/i, lambda {|cfg,col| col.extend(::JdbcSpec::PostgreSQL::Column)}]
|
24
|
+
end
|
25
|
+
|
26
|
+
def self.jdbc_connection_class
|
27
|
+
::ActiveRecord::ConnectionAdapters::PostgresJdbcConnection
|
28
|
+
end
|
29
|
+
|
30
|
+
module Column
|
31
|
+
def type_cast(value)
|
32
|
+
case type
|
33
|
+
when :boolean then cast_to_boolean(value)
|
34
|
+
else super
|
35
|
+
end
|
36
|
+
end
|
37
|
+
|
38
|
+
def simplified_type(field_type)
|
39
|
+
return :integer if field_type =~ /^serial/i
|
40
|
+
return :string if field_type =~ /\[\]$/i || field_type =~ /^interval/i
|
41
|
+
return :string if field_type =~ /^(?:point|lseg|box|"?path"?|polygon|circle)/i
|
42
|
+
return :datetime if field_type =~ /^timestamp/i
|
43
|
+
return :float if field_type =~ /^real|^money/i
|
44
|
+
return :binary if field_type =~ /^bytea/i
|
45
|
+
return :boolean if field_type =~ /^bool/i
|
46
|
+
super
|
47
|
+
end
|
48
|
+
|
49
|
+
def cast_to_boolean(value)
|
50
|
+
return nil if value.nil?
|
51
|
+
if value == true || value == false
|
52
|
+
value
|
53
|
+
else
|
54
|
+
%w(true t 1).include?(value.to_s.downcase)
|
55
|
+
end
|
56
|
+
end
|
57
|
+
|
58
|
+
# Post process default value from JDBC into a Rails-friendly format (columns{-internal})
|
59
|
+
def default_value(value)
|
60
|
+
# Boolean types
|
61
|
+
return "t" if value =~ /true/i
|
62
|
+
return "f" if value =~ /false/i
|
63
|
+
|
64
|
+
# Char/String/Bytea type values
|
65
|
+
return $1 if value =~ /^'(.*)'::(bpchar|text|character varying|bytea)$/
|
66
|
+
|
67
|
+
# Numeric values
|
68
|
+
return value if value =~ /^-?[0-9]+(\.[0-9]*)?/
|
69
|
+
|
70
|
+
# Fixed dates / timestamp
|
71
|
+
return $1 if value =~ /^'(.+)'::(date|timestamp)/
|
72
|
+
|
73
|
+
# Anything else is blank, some user type, or some function
|
74
|
+
# and we can't know the value of that, so return nil.
|
75
|
+
return nil
|
76
|
+
end
|
77
|
+
end
|
78
|
+
|
79
|
+
def modify_types(tp)
|
80
|
+
tp[:primary_key] = "serial primary key"
|
81
|
+
tp[:string][:limit] = 255
|
82
|
+
tp[:integer][:limit] = nil
|
83
|
+
tp[:boolean][:limit] = nil
|
84
|
+
tp
|
85
|
+
end
|
86
|
+
|
87
|
+
def postgresql_version
|
88
|
+
@postgresql_version ||=
|
89
|
+
begin
|
90
|
+
value = select_value('SELECT version()')
|
91
|
+
if value =~ /PostgreSQL (\d+)\.(\d+)\.(\d+)/
|
92
|
+
($1.to_i * 10000) + ($2.to_i * 100) + $3.to_i
|
93
|
+
else
|
94
|
+
0
|
95
|
+
end
|
96
|
+
end
|
97
|
+
end
|
98
|
+
|
99
|
+
# Does PostgreSQL support migrations?
|
100
|
+
def supports_migrations?
|
101
|
+
true
|
102
|
+
end
|
103
|
+
|
104
|
+
# Does PostgreSQL support standard conforming strings?
|
105
|
+
def supports_standard_conforming_strings?
|
106
|
+
# Temporarily set the client message level above error to prevent unintentional
|
107
|
+
# error messages in the logs when working on a PostgreSQL database server that
|
108
|
+
# does not support standard conforming strings.
|
109
|
+
client_min_messages_old = client_min_messages
|
110
|
+
self.client_min_messages = 'panic'
|
111
|
+
|
112
|
+
# postgres-pr does not raise an exception when client_min_messages is set higher
|
113
|
+
# than error and "SHOW standard_conforming_strings" fails, but returns an empty
|
114
|
+
# PGresult instead.
|
115
|
+
has_support = select('SHOW standard_conforming_strings').to_a[0][0] rescue false
|
116
|
+
self.client_min_messages = client_min_messages_old
|
117
|
+
has_support
|
118
|
+
end
|
119
|
+
|
120
|
+
def supports_insert_with_returning?
|
121
|
+
postgresql_version >= 80200
|
122
|
+
end
|
123
|
+
|
124
|
+
def supports_ddl_transactions?
|
125
|
+
true
|
126
|
+
end
|
127
|
+
|
128
|
+
def supports_savepoints?
|
129
|
+
true
|
130
|
+
end
|
131
|
+
|
132
|
+
# Returns the configured supported identifier length supported by PostgreSQL,
|
133
|
+
# or report the default of 63 on PostgreSQL 7.x.
|
134
|
+
def table_alias_length
|
135
|
+
@table_alias_length ||= (postgresql_version >= 80000 ? select('SHOW max_identifier_length').to_a[0][0].to_i : 63)
|
136
|
+
end
|
137
|
+
|
138
|
+
def default_sequence_name(table_name, pk = nil)
|
139
|
+
default_pk, default_seq = pk_and_sequence_for(table_name)
|
140
|
+
default_seq || "#{table_name}_#{pk || default_pk || 'id'}_seq"
|
141
|
+
end
|
142
|
+
|
143
|
+
# Resets sequence to the max value of the table's pk if present.
|
144
|
+
def reset_pk_sequence!(table, pk = nil, sequence = nil) #:nodoc:
|
145
|
+
unless pk and sequence
|
146
|
+
default_pk, default_sequence = pk_and_sequence_for(table)
|
147
|
+
pk ||= default_pk
|
148
|
+
sequence ||= default_sequence
|
149
|
+
end
|
150
|
+
if pk
|
151
|
+
if sequence
|
152
|
+
quoted_sequence = quote_column_name(sequence)
|
153
|
+
|
154
|
+
select_value <<-end_sql, 'Reset sequence'
|
155
|
+
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)
|
156
|
+
end_sql
|
157
|
+
else
|
158
|
+
@logger.warn "#{table} has primary key #{pk} with no default sequence" if @logger
|
159
|
+
end
|
160
|
+
end
|
161
|
+
end
|
162
|
+
|
163
|
+
def quote_regclass(table_name)
|
164
|
+
table_name.to_s.split('.').map do |part|
|
165
|
+
part =~ /".*"/i ? part : quote_table_name(part)
|
166
|
+
end.join('.')
|
167
|
+
end
|
168
|
+
|
169
|
+
# Find a table's primary key and sequence.
|
170
|
+
def pk_and_sequence_for(table) #:nodoc:
|
171
|
+
# First try looking for a sequence with a dependency on the
|
172
|
+
# given table's primary key.
|
173
|
+
result = select(<<-end_sql, 'PK and serial sequence')[0]
|
174
|
+
SELECT attr.attname, seq.relname
|
175
|
+
FROM pg_class seq,
|
176
|
+
pg_attribute attr,
|
177
|
+
pg_depend dep,
|
178
|
+
pg_namespace name,
|
179
|
+
pg_constraint cons
|
180
|
+
WHERE seq.oid = dep.objid
|
181
|
+
AND seq.relkind = 'S'
|
182
|
+
AND attr.attrelid = dep.refobjid
|
183
|
+
AND attr.attnum = dep.refobjsubid
|
184
|
+
AND attr.attrelid = cons.conrelid
|
185
|
+
AND attr.attnum = cons.conkey[1]
|
186
|
+
AND cons.contype = 'p'
|
187
|
+
AND dep.refobjid = '#{table}'::regclass
|
188
|
+
end_sql
|
189
|
+
|
190
|
+
if result.nil? or result.empty?
|
191
|
+
# If that fails, try parsing the primary key's default value.
|
192
|
+
# Support the 7.x and 8.0 nextval('foo'::text) as well as
|
193
|
+
# the 8.1+ nextval('foo'::regclass).
|
194
|
+
result = query(<<-end_sql, 'PK and custom sequence')[0]
|
195
|
+
SELECT attr.attname,
|
196
|
+
CASE
|
197
|
+
WHEN split_part(def.adsrc, '''', 2) ~ '.' THEN
|
198
|
+
substr(split_part(def.adsrc, '''', 2),
|
199
|
+
strpos(split_part(def.adsrc, '''', 2), '.')+1)
|
200
|
+
ELSE split_part(def.adsrc, '''', 2)
|
201
|
+
END as relname
|
202
|
+
FROM pg_class t
|
203
|
+
JOIN pg_attribute attr ON (t.oid = attrelid)
|
204
|
+
JOIN pg_attrdef def ON (adrelid = attrelid AND adnum = attnum)
|
205
|
+
JOIN pg_constraint cons ON (conrelid = adrelid AND adnum = conkey[1])
|
206
|
+
WHERE t.oid = '#{table}'::regclass
|
207
|
+
AND cons.contype = 'p'
|
208
|
+
AND def.adsrc ~* 'nextval'
|
209
|
+
end_sql
|
210
|
+
end
|
211
|
+
|
212
|
+
[result["attname"], result["relname"]]
|
213
|
+
rescue
|
214
|
+
nil
|
215
|
+
end
|
216
|
+
|
217
|
+
def insert(sql, name = nil, pk = nil, id_value = nil, sequence_name = nil)
|
218
|
+
# Extract the table from the insert sql. Yuck.
|
219
|
+
table = sql.split(" ", 4)[2].gsub('"', '')
|
220
|
+
|
221
|
+
# Try an insert with 'returning id' if available (PG >= 8.2)
|
222
|
+
if supports_insert_with_returning? && id_value.nil? && false # FIXME:
|
223
|
+
# Disabled, as it causes:
|
224
|
+
# ActiveRecord::ActiveRecordError: A result was returned when none was expected
|
225
|
+
# This was previously disabled because postgresql_version returned 0
|
226
|
+
pk, sequence_name = *pk_and_sequence_for(table) unless pk
|
227
|
+
if pk
|
228
|
+
id_value = select_value("#{sql} RETURNING #{quote_column_name(pk)}")
|
229
|
+
clear_query_cache #FIXME: Why now?
|
230
|
+
return id_value
|
231
|
+
end
|
232
|
+
end
|
233
|
+
|
234
|
+
# Otherwise, plain insert
|
235
|
+
execute(sql, name)
|
236
|
+
|
237
|
+
# Don't need to look up id_value if we already have it.
|
238
|
+
# (and can't in case of non-sequence PK)
|
239
|
+
unless id_value
|
240
|
+
# If neither pk nor sequence name is given, look them up.
|
241
|
+
unless pk || sequence_name
|
242
|
+
pk, sequence_name = *pk_and_sequence_for(table)
|
243
|
+
end
|
244
|
+
|
245
|
+
# If a pk is given, fallback to default sequence name.
|
246
|
+
# Don't fetch last insert id for a table without a pk.
|
247
|
+
if pk && sequence_name ||= default_sequence_name(table, pk)
|
248
|
+
id_value = last_insert_id(table, sequence_name)
|
249
|
+
end
|
250
|
+
end
|
251
|
+
id_value
|
252
|
+
end
|
253
|
+
|
254
|
+
def columns(table_name, name=nil)
|
255
|
+
schema_name = "public"
|
256
|
+
if table_name =~ /\./
|
257
|
+
parts = table_name.split(/\./)
|
258
|
+
table_name = parts.pop
|
259
|
+
schema_name = parts.join(".")
|
260
|
+
end
|
261
|
+
@connection.columns_internal(table_name, name, schema_name)
|
262
|
+
end
|
263
|
+
|
264
|
+
# From postgresql_adapter.rb
|
265
|
+
def indexes(table_name, name = nil)
|
266
|
+
result = select_rows(<<-SQL, name)
|
267
|
+
SELECT i.relname, d.indisunique, a.attname
|
268
|
+
FROM pg_class t, pg_class i, pg_index d, pg_attribute a
|
269
|
+
WHERE i.relkind = 'i'
|
270
|
+
AND d.indexrelid = i.oid
|
271
|
+
AND d.indisprimary = 'f'
|
272
|
+
AND t.oid = d.indrelid
|
273
|
+
AND t.relname = '#{table_name}'
|
274
|
+
AND a.attrelid = t.oid
|
275
|
+
AND ( d.indkey[0]=a.attnum OR d.indkey[1]=a.attnum
|
276
|
+
OR d.indkey[2]=a.attnum OR d.indkey[3]=a.attnum
|
277
|
+
OR d.indkey[4]=a.attnum OR d.indkey[5]=a.attnum
|
278
|
+
OR d.indkey[6]=a.attnum OR d.indkey[7]=a.attnum
|
279
|
+
OR d.indkey[8]=a.attnum OR d.indkey[9]=a.attnum )
|
280
|
+
ORDER BY i.relname
|
281
|
+
SQL
|
282
|
+
|
283
|
+
current_index = nil
|
284
|
+
indexes = []
|
285
|
+
|
286
|
+
result.each do |row|
|
287
|
+
if current_index != row[0]
|
288
|
+
indexes << ::ActiveRecord::ConnectionAdapters::IndexDefinition.new(table_name, row[0], row[1] == "t", [])
|
289
|
+
current_index = row[0]
|
290
|
+
end
|
291
|
+
|
292
|
+
indexes.last.columns << row[2]
|
293
|
+
end
|
294
|
+
|
295
|
+
indexes
|
296
|
+
end
|
297
|
+
|
298
|
+
def last_insert_id(table, sequence_name)
|
299
|
+
Integer(select_value("SELECT currval('#{sequence_name}')"))
|
300
|
+
end
|
301
|
+
|
302
|
+
def recreate_database(name)
|
303
|
+
drop_database(name)
|
304
|
+
create_database(name)
|
305
|
+
end
|
306
|
+
|
307
|
+
def create_database(name, options = {})
|
308
|
+
execute "CREATE DATABASE \"#{name}\" ENCODING='#{options[:encoding] || 'utf8'}'"
|
309
|
+
end
|
310
|
+
|
311
|
+
def drop_database(name)
|
312
|
+
execute "DROP DATABASE \"#{name}\""
|
313
|
+
end
|
314
|
+
|
315
|
+
def structure_dump
|
316
|
+
database = @config[:database]
|
317
|
+
if database.nil?
|
318
|
+
if @config[:url] =~ /\/([^\/]*)$/
|
319
|
+
database = $1
|
320
|
+
else
|
321
|
+
raise "Could not figure out what database this url is for #{@config["url"]}"
|
322
|
+
end
|
323
|
+
end
|
324
|
+
|
325
|
+
ENV['PGHOST'] = @config[:host] if @config[:host]
|
326
|
+
ENV['PGPORT'] = @config[:port].to_s if @config[:port]
|
327
|
+
ENV['PGPASSWORD'] = @config[:password].to_s if @config[:password]
|
328
|
+
search_path = @config[:schema_search_path]
|
329
|
+
search_path = "--schema=#{search_path}" if search_path
|
330
|
+
|
331
|
+
@connection.connection.close
|
332
|
+
begin
|
333
|
+
definition = `pg_dump -i -U "#{@config[:username]}" -s -x -O #{search_path} #{database}`
|
334
|
+
raise "Error dumping database" if $?.exitstatus == 1
|
335
|
+
|
336
|
+
# need to patch away any references to SQL_ASCII as it breaks the JDBC driver
|
337
|
+
definition.gsub(/SQL_ASCII/, 'UNICODE')
|
338
|
+
ensure
|
339
|
+
reconnect!
|
340
|
+
end
|
341
|
+
end
|
342
|
+
|
343
|
+
def _execute(sql, name = nil)
|
344
|
+
case sql.strip
|
345
|
+
when /\A\(?\s*(select|show)/i then
|
346
|
+
@connection.execute_query(sql)
|
347
|
+
else
|
348
|
+
@connection.execute_update(sql)
|
349
|
+
end
|
350
|
+
end
|
351
|
+
|
352
|
+
# SELECT DISTINCT clause for a given set of columns and a given ORDER BY clause.
|
353
|
+
#
|
354
|
+
# PostgreSQL requires the ORDER BY columns in the select list for distinct queries, and
|
355
|
+
# requires that the ORDER BY include the distinct column.
|
356
|
+
#
|
357
|
+
# distinct("posts.id", "posts.created_at desc")
|
358
|
+
def distinct(columns, order_by)
|
359
|
+
return "DISTINCT #{columns}" if order_by.blank?
|
360
|
+
|
361
|
+
# construct a clean list of column names from the ORDER BY clause, removing
|
362
|
+
# any asc/desc modifiers
|
363
|
+
order_columns = order_by.split(',').collect { |s| s.split.first }
|
364
|
+
order_columns.delete_if(&:blank?)
|
365
|
+
order_columns = order_columns.zip((0...order_columns.size).to_a).map { |s,i| "#{s} AS alias_#{i}" }
|
366
|
+
|
367
|
+
# return a DISTINCT ON() clause that's distinct on the columns we want but includes
|
368
|
+
# all the required columns for the ORDER BY to work properly
|
369
|
+
sql = "DISTINCT ON (#{columns}) #{columns}, "
|
370
|
+
sql << order_columns * ', '
|
371
|
+
end
|
372
|
+
|
373
|
+
# ORDER BY clause for the passed order option.
|
374
|
+
#
|
375
|
+
# PostgreSQL does not allow arbitrary ordering when using DISTINCT ON, so we work around this
|
376
|
+
# by wrapping the sql as a sub-select and ordering in that query.
|
377
|
+
def add_order_by_for_association_limiting!(sql, options)
|
378
|
+
return sql if options[:order].blank?
|
379
|
+
|
380
|
+
order = options[:order].split(',').collect { |s| s.strip }.reject(&:blank?)
|
381
|
+
order.map! { |s| 'DESC' if s =~ /\bdesc$/i }
|
382
|
+
order = order.zip((0...order.size).to_a).map { |s,i| "id_list.alias_#{i} #{s}" }.join(', ')
|
383
|
+
|
384
|
+
sql.replace "SELECT * FROM (#{sql}) AS id_list ORDER BY #{order}"
|
385
|
+
end
|
386
|
+
|
387
|
+
def quote(value, column = nil)
|
388
|
+
return value.quoted_id if value.respond_to?(:quoted_id)
|
389
|
+
|
390
|
+
if value.kind_of?(String) && column && column.type == :binary
|
391
|
+
"'#{escape_bytea(value)}'"
|
392
|
+
elsif column && column.type == :primary_key
|
393
|
+
return value.to_s
|
394
|
+
else
|
395
|
+
super
|
396
|
+
end
|
397
|
+
end
|
398
|
+
|
399
|
+
def escape_bytea(s)
|
400
|
+
if s
|
401
|
+
result = ''
|
402
|
+
s.each_byte { |c| result << sprintf('\\\\%03o', c) }
|
403
|
+
result
|
404
|
+
end
|
405
|
+
end
|
406
|
+
|
407
|
+
def quote_column_name(name)
|
408
|
+
%("#{name}")
|
409
|
+
end
|
410
|
+
|
411
|
+
def quoted_date(value) #:nodoc:
|
412
|
+
if value.acts_like?(:time) && value.respond_to?(:usec)
|
413
|
+
"#{super}.#{sprintf("%06d", value.usec)}"
|
414
|
+
else
|
415
|
+
super
|
416
|
+
end
|
417
|
+
end
|
418
|
+
|
419
|
+
def disable_referential_integrity(&block) #:nodoc:
|
420
|
+
execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} DISABLE TRIGGER ALL" }.join(";"))
|
421
|
+
yield
|
422
|
+
ensure
|
423
|
+
execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} ENABLE TRIGGER ALL" }.join(";"))
|
424
|
+
end
|
425
|
+
|
426
|
+
def rename_table(name, new_name)
|
427
|
+
execute "ALTER TABLE #{name} RENAME TO #{new_name}"
|
428
|
+
end
|
429
|
+
|
430
|
+
def add_column(table_name, column_name, type, options = {})
|
431
|
+
execute("ALTER TABLE #{quote_table_name(table_name)} ADD #{quote_column_name(column_name)} #{type_to_sql(type, options[:limit])}")
|
432
|
+
change_column_default(table_name, column_name, options[:default]) unless options[:default].nil?
|
433
|
+
if options[:null] == false
|
434
|
+
execute("UPDATE #{quote_table_name(table_name)} SET #{quote_column_name(column_name)} = '#{options[:default]}'") if options[:default]
|
435
|
+
execute("ALTER TABLE #{quote_table_name(table_name)} ALTER #{quote_column_name(column_name)} SET NOT NULL")
|
436
|
+
end
|
437
|
+
end
|
438
|
+
|
439
|
+
def change_column(table_name, column_name, type, options = {}) #:nodoc:
|
440
|
+
begin
|
441
|
+
execute "ALTER TABLE #{quote_table_name(table_name)} ALTER #{quote_column_name(column_name)} TYPE #{type_to_sql(type, options[:limit])}"
|
442
|
+
rescue ActiveRecord::StatementInvalid
|
443
|
+
# This is PG7, so we use a more arcane way of doing it.
|
444
|
+
begin_db_transaction
|
445
|
+
add_column(table_name, "#{column_name}_ar_tmp", type, options)
|
446
|
+
execute "UPDATE #{table_name} SET #{column_name}_ar_tmp = CAST(#{column_name} AS #{type_to_sql(type, options[:limit])})"
|
447
|
+
remove_column(table_name, column_name)
|
448
|
+
rename_column(table_name, "#{column_name}_ar_tmp", column_name)
|
449
|
+
commit_db_transaction
|
450
|
+
end
|
451
|
+
change_column_default(table_name, column_name, options[:default]) unless options[:default].nil?
|
452
|
+
end
|
453
|
+
|
454
|
+
def change_column_default(table_name, column_name, default) #:nodoc:
|
455
|
+
execute "ALTER TABLE #{quote_table_name(table_name)} ALTER COLUMN #{quote_column_name(column_name)} SET DEFAULT '#{default}'"
|
456
|
+
end
|
457
|
+
|
458
|
+
def rename_column(table_name, column_name, new_column_name) #:nodoc:
|
459
|
+
execute "ALTER TABLE #{quote_table_name(table_name)} RENAME COLUMN #{quote_column_name(column_name)} TO #{quote_column_name(new_column_name)}"
|
460
|
+
end
|
461
|
+
|
462
|
+
def remove_index(table_name, options) #:nodoc:
|
463
|
+
execute "DROP INDEX #{index_name(table_name, options)}"
|
464
|
+
end
|
465
|
+
|
466
|
+
def type_to_sql(type, limit = nil, precision = nil, scale = nil) #:nodoc:
|
467
|
+
return super unless type.to_s == 'integer'
|
468
|
+
|
469
|
+
if limit.nil? || limit == 4
|
470
|
+
'integer'
|
471
|
+
elsif limit < 4
|
472
|
+
'smallint'
|
473
|
+
else
|
474
|
+
'bigint'
|
475
|
+
end
|
476
|
+
end
|
477
|
+
|
478
|
+
def tables
|
479
|
+
@connection.tables(database_name, nil, nil, ["TABLE"])
|
480
|
+
end
|
481
|
+
end
|
482
|
+
end
|
483
|
+
|