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,141 @@
|
|
1
|
+
module JdbcSpec
|
2
|
+
module Mimer
|
3
|
+
def self.adapter_matcher(name, *)
|
4
|
+
name =~ /mimer/i ? self : false
|
5
|
+
end
|
6
|
+
|
7
|
+
def modify_types(tp)
|
8
|
+
tp[:primary_key] = "INTEGER NOT NULL PRIMARY KEY"
|
9
|
+
tp[:boolean][:limit] = nil
|
10
|
+
tp[:string][:limit] = 255
|
11
|
+
tp[:binary] = {:name => "BINARY VARYING", :limit => 4096}
|
12
|
+
tp[:text] = {:name => "VARCHAR", :limit => 4096}
|
13
|
+
tp[:datetime] = { :name => "TIMESTAMP" }
|
14
|
+
tp[:timestamp] = { :name => "TIMESTAMP" }
|
15
|
+
tp[:time] = { :name => "TIMESTAMP" }
|
16
|
+
tp[:date] = { :name => "TIMESTAMP" }
|
17
|
+
tp
|
18
|
+
end
|
19
|
+
|
20
|
+
def default_sequence_name(table, column) #:nodoc:
|
21
|
+
"#{table}_seq"
|
22
|
+
end
|
23
|
+
|
24
|
+
def create_table(name, options = {}) #:nodoc:
|
25
|
+
super(name, options)
|
26
|
+
execute "CREATE SEQUENCE #{name}_seq" unless options[:id] == false
|
27
|
+
end
|
28
|
+
|
29
|
+
def drop_table(name, options = {}) #:nodoc:
|
30
|
+
super(name) rescue nil
|
31
|
+
execute "DROP SEQUENCE #{name}_seq" rescue nil
|
32
|
+
end
|
33
|
+
|
34
|
+
def change_column(table_name, column_name, type, options = {}) #:nodoc:
|
35
|
+
execute "ALTER TABLE #{table_name} ALTER COLUMN #{column_name} #{type_to_sql(type, options[:limit])}"
|
36
|
+
end
|
37
|
+
|
38
|
+
def change_column_default(table_name, column_name, default) #:nodoc:
|
39
|
+
execute "ALTER TABLE #{table_name} ALTER COLUMN #{column_name} SET DEFAULT #{quote(default)}"
|
40
|
+
end
|
41
|
+
|
42
|
+
def remove_index(table_name, options = {}) #:nodoc:
|
43
|
+
execute "DROP INDEX #{index_name(table_name, options)}"
|
44
|
+
end
|
45
|
+
|
46
|
+
def insert(sql, name = nil, pk = nil, id_value = nil, sequence_name = nil) #:nodoc:
|
47
|
+
if pk.nil? # Who called us? What does the sql look like? No idea!
|
48
|
+
execute sql, name
|
49
|
+
elsif id_value # Pre-assigned id
|
50
|
+
log(sql, name) { @connection.execute_insert sql,pk }
|
51
|
+
else # Assume the sql contains a bind-variable for the id
|
52
|
+
id_value = select_one("SELECT NEXT_VALUE OF #{sequence_name} AS val FROM MIMER.ONEROW")['val']
|
53
|
+
log(sql, name) {
|
54
|
+
execute_prepared_insert(sql,id_value)
|
55
|
+
}
|
56
|
+
end
|
57
|
+
id_value
|
58
|
+
end
|
59
|
+
|
60
|
+
def execute_prepared_insert(sql, id)
|
61
|
+
@stmts ||= {}
|
62
|
+
@stmts[sql] ||= @connection.ps(sql)
|
63
|
+
stmt = @stmts[sql]
|
64
|
+
stmt.setLong(1,id)
|
65
|
+
stmt.executeUpdate
|
66
|
+
id
|
67
|
+
end
|
68
|
+
|
69
|
+
def quote(value, column = nil) #:nodoc:
|
70
|
+
return value.quoted_id if value.respond_to?(:quoted_id)
|
71
|
+
|
72
|
+
if String === value && column && column.type == :binary
|
73
|
+
return "X'#{quote_string(value.unpack("C*").collect {|v| v.to_s(16)}.join)}'"
|
74
|
+
end
|
75
|
+
case value
|
76
|
+
when String
|
77
|
+
%Q{'#{quote_string(value)}'}
|
78
|
+
when NilClass
|
79
|
+
'NULL'
|
80
|
+
when TrueClass
|
81
|
+
'1'
|
82
|
+
when FalseClass
|
83
|
+
'0'
|
84
|
+
when Numeric
|
85
|
+
value.to_s
|
86
|
+
when Date, Time
|
87
|
+
%Q{TIMESTAMP '#{value.strftime("%Y-%m-%d %H:%M:%S")}'}
|
88
|
+
else
|
89
|
+
%Q{'#{quote_string(value.to_yaml)}'}
|
90
|
+
end
|
91
|
+
end
|
92
|
+
|
93
|
+
def quoted_true
|
94
|
+
'1'
|
95
|
+
end
|
96
|
+
|
97
|
+
def quoted_false
|
98
|
+
'0'
|
99
|
+
end
|
100
|
+
|
101
|
+
def add_limit_offset!(sql, options) # :nodoc:
|
102
|
+
@limit = options[:limit]
|
103
|
+
@offset = options[:offset]
|
104
|
+
end
|
105
|
+
|
106
|
+
def select_all(sql, name = nil)
|
107
|
+
@offset ||= 0
|
108
|
+
if !@limit || @limit == -1
|
109
|
+
range = @offset..-1
|
110
|
+
else
|
111
|
+
range = @offset...(@offset+@limit)
|
112
|
+
end
|
113
|
+
select(sql, name)[range]
|
114
|
+
ensure
|
115
|
+
@limit = @offset = nil
|
116
|
+
end
|
117
|
+
|
118
|
+
def select_one(sql, name = nil)
|
119
|
+
@offset ||= 0
|
120
|
+
select(sql, name)[@offset]
|
121
|
+
ensure
|
122
|
+
@limit = @offset = nil
|
123
|
+
end
|
124
|
+
|
125
|
+
def _execute(sql, name = nil)
|
126
|
+
if sql =~ /^select/i
|
127
|
+
@offset ||= 0
|
128
|
+
if !@limit || @limit == -1
|
129
|
+
range = @offset..-1
|
130
|
+
else
|
131
|
+
range = @offset...(@offset+@limit)
|
132
|
+
end
|
133
|
+
@connection.execute_query(sql)[range]
|
134
|
+
else
|
135
|
+
@connection.execute_update(sql)
|
136
|
+
end
|
137
|
+
ensure
|
138
|
+
@limit = @offset = nil
|
139
|
+
end
|
140
|
+
end
|
141
|
+
end
|
@@ -0,0 +1,333 @@
|
|
1
|
+
require 'jdbc_adapter/tsql_helper'
|
2
|
+
|
3
|
+
module ::JdbcSpec
|
4
|
+
module MsSQL
|
5
|
+
include TSqlMethods
|
6
|
+
|
7
|
+
def self.extended(mod)
|
8
|
+
unless @lob_callback_added
|
9
|
+
ActiveRecord::Base.class_eval do
|
10
|
+
def after_save_with_mssql_lob
|
11
|
+
self.class.columns.select { |c| c.sql_type =~ /image/i }.each do |c|
|
12
|
+
value = self[c.name]
|
13
|
+
value = value.to_yaml if unserializable_attribute?(c.name, c)
|
14
|
+
next if value.nil? || (value == '')
|
15
|
+
|
16
|
+
connection.write_large_object(c.type == :binary, c.name, self.class.table_name, self.class.primary_key, quote_value(id), value)
|
17
|
+
end
|
18
|
+
end
|
19
|
+
end
|
20
|
+
|
21
|
+
ActiveRecord::Base.after_save :after_save_with_mssql_lob
|
22
|
+
@lob_callback_added = true
|
23
|
+
end
|
24
|
+
end
|
25
|
+
|
26
|
+
def self.adapter_matcher(name, *)
|
27
|
+
name =~ /sqlserver|tds/i ? self : false
|
28
|
+
end
|
29
|
+
|
30
|
+
def self.column_selector
|
31
|
+
[/sqlserver|tds/i, lambda {|cfg,col| col.extend(::JdbcSpec::MsSQL::Column)}]
|
32
|
+
end
|
33
|
+
|
34
|
+
def self.jdbc_connection_class
|
35
|
+
::ActiveRecord::ConnectionAdapters::MssqlJdbcConnection
|
36
|
+
end
|
37
|
+
|
38
|
+
module Column
|
39
|
+
attr_accessor :identity, :is_special
|
40
|
+
|
41
|
+
def simplified_type(field_type)
|
42
|
+
case field_type
|
43
|
+
when /int|bigint|smallint|tinyint/i then :integer
|
44
|
+
when /numeric/i then (@scale.nil? || @scale == 0) ? :integer : :decimal
|
45
|
+
when /float|double|decimal|money|real|smallmoney/i then :decimal
|
46
|
+
when /datetime|smalldatetime/i then :datetime
|
47
|
+
when /timestamp/i then :timestamp
|
48
|
+
when /time/i then :time
|
49
|
+
when /text|ntext/i then :text
|
50
|
+
when /binary|image|varbinary/i then :binary
|
51
|
+
when /char|nchar|nvarchar|string|varchar/i then :string
|
52
|
+
when /bit/i then :boolean
|
53
|
+
when /uniqueidentifier/i then :string
|
54
|
+
end
|
55
|
+
end
|
56
|
+
|
57
|
+
def type_cast(value)
|
58
|
+
return nil if value.nil? || value == "(null)" || value == "(NULL)"
|
59
|
+
case type
|
60
|
+
when :string then unquote_string value
|
61
|
+
when :integer then unquote(value).to_i rescue value ? 1 : 0
|
62
|
+
when :primary_key then value == true || value == false ? value == true ? 1 : 0 : value.to_i
|
63
|
+
when :decimal then self.class.value_to_decimal(unquote(value))
|
64
|
+
when :datetime then cast_to_datetime(value)
|
65
|
+
when :timestamp then cast_to_time(value)
|
66
|
+
when :time then cast_to_time(value)
|
67
|
+
when :date then cast_to_datetime(value)
|
68
|
+
when :boolean then value == true or (value =~ /^t(rue)?$/i) == 0 or unquote(value)=="1"
|
69
|
+
when :binary then unquote value
|
70
|
+
else value
|
71
|
+
end
|
72
|
+
end
|
73
|
+
|
74
|
+
# JRUBY-2011: Match balanced quotes and parenthesis - 'text',('text') or (text)
|
75
|
+
def unquote_string(value)
|
76
|
+
value.to_s.sub(/^\((.*)\)$/,'\1').sub(/^'(.*)'$/,'\1')
|
77
|
+
end
|
78
|
+
|
79
|
+
def unquote(value)
|
80
|
+
value.to_s.sub(/\A\([\(\']?/, "").sub(/[\'\)]?\)\Z/, "")
|
81
|
+
end
|
82
|
+
|
83
|
+
def cast_to_time(value)
|
84
|
+
return value if value.is_a?(Time)
|
85
|
+
time_array = ParseDate.parsedate(value)
|
86
|
+
time_array[0] ||= 2000
|
87
|
+
time_array[1] ||= 1
|
88
|
+
time_array[2] ||= 1
|
89
|
+
Time.send(ActiveRecord::Base.default_timezone, *time_array) rescue nil
|
90
|
+
end
|
91
|
+
|
92
|
+
def cast_to_datetime(value)
|
93
|
+
if value.is_a?(Time)
|
94
|
+
if value.year != 0 and value.month != 0 and value.day != 0
|
95
|
+
return value
|
96
|
+
else
|
97
|
+
return Time.mktime(2000, 1, 1, value.hour, value.min, value.sec) rescue nil
|
98
|
+
end
|
99
|
+
end
|
100
|
+
return cast_to_time(value) if value.is_a?(Date) or value.is_a?(String) rescue nil
|
101
|
+
value
|
102
|
+
end
|
103
|
+
|
104
|
+
# These methods will only allow the adapter to insert binary data with a length of 7K or less
|
105
|
+
# because of a SQL Server statement length policy.
|
106
|
+
def self.string_to_binary(value)
|
107
|
+
''
|
108
|
+
end
|
109
|
+
end
|
110
|
+
|
111
|
+
def quote(value, column = nil)
|
112
|
+
return value.quoted_id if value.respond_to?(:quoted_id)
|
113
|
+
|
114
|
+
case value
|
115
|
+
when String, ActiveSupport::Multibyte::Chars
|
116
|
+
value = value.to_s
|
117
|
+
if column && column.type == :binary
|
118
|
+
"'#{quote_string(JdbcSpec::MsSQL::Column.string_to_binary(value))}'" # ' (for ruby-mode)
|
119
|
+
elsif column && [:integer, :float].include?(column.type)
|
120
|
+
value = column.type == :integer ? value.to_i : value.to_f
|
121
|
+
value.to_s
|
122
|
+
else
|
123
|
+
"N'#{quote_string(value)}'" # ' (for ruby-mode)
|
124
|
+
end
|
125
|
+
when TrueClass then '1'
|
126
|
+
when FalseClass then '0'
|
127
|
+
when Time, DateTime then "'#{value.strftime("%Y%m%d %H:%M:%S")}'"
|
128
|
+
when Date then "'#{value.strftime("%Y%m%d")}'"
|
129
|
+
else super
|
130
|
+
end
|
131
|
+
end
|
132
|
+
|
133
|
+
def quote_string(string)
|
134
|
+
string.gsub(/\'/, "''")
|
135
|
+
end
|
136
|
+
|
137
|
+
def quote_table_name(name)
|
138
|
+
name
|
139
|
+
end
|
140
|
+
|
141
|
+
def quote_column_name(name)
|
142
|
+
"[#{name}]"
|
143
|
+
end
|
144
|
+
|
145
|
+
def quoted_true
|
146
|
+
quote true
|
147
|
+
end
|
148
|
+
|
149
|
+
def quoted_false
|
150
|
+
quote false
|
151
|
+
end
|
152
|
+
|
153
|
+
def change_order_direction(order)
|
154
|
+
order.split(",").collect {|fragment|
|
155
|
+
case fragment
|
156
|
+
when /\bDESC\b/i then fragment.gsub(/\bDESC\b/i, "ASC")
|
157
|
+
when /\bASC\b/i then fragment.gsub(/\bASC\b/i, "DESC")
|
158
|
+
else String.new(fragment).split(',').join(' DESC,') + ' DESC'
|
159
|
+
end
|
160
|
+
}.join(",")
|
161
|
+
end
|
162
|
+
|
163
|
+
def recreate_database(name)
|
164
|
+
drop_database(name)
|
165
|
+
create_database(name)
|
166
|
+
end
|
167
|
+
|
168
|
+
def drop_database(name)
|
169
|
+
execute "USE master"
|
170
|
+
execute "DROP DATABASE #{name}"
|
171
|
+
end
|
172
|
+
|
173
|
+
def create_database(name)
|
174
|
+
execute "CREATE DATABASE #{name}"
|
175
|
+
execute "USE #{name}"
|
176
|
+
end
|
177
|
+
|
178
|
+
def rename_table(name, new_name)
|
179
|
+
execute "EXEC sp_rename '#{name}', '#{new_name}'"
|
180
|
+
end
|
181
|
+
|
182
|
+
# Adds a new column to the named table.
|
183
|
+
# See TableDefinition#column for details of the options you can use.
|
184
|
+
def add_column(table_name, column_name, type, options = {})
|
185
|
+
add_column_sql = "ALTER TABLE #{table_name} ADD #{quote_column_name(column_name)} #{type_to_sql(type, options[:limit], options[:precision], options[:scale])}"
|
186
|
+
add_column_options!(add_column_sql, options)
|
187
|
+
# TODO: Add support to mimic date columns, using constraints to mark them as such in the database
|
188
|
+
# add_column_sql << " CONSTRAINT ck__#{table_name}__#{column_name}__date_only CHECK ( CONVERT(CHAR(12), #{quote_column_name(column_name)}, 14)='00:00:00:000' )" if type == :date
|
189
|
+
execute(add_column_sql)
|
190
|
+
end
|
191
|
+
|
192
|
+
def rename_column(table, column, new_column_name)
|
193
|
+
execute "EXEC sp_rename '#{table}.#{column}', '#{new_column_name}'"
|
194
|
+
end
|
195
|
+
|
196
|
+
def change_column(table_name, column_name, type, options = {}) #:nodoc:
|
197
|
+
sql_commands = ["ALTER TABLE #{table_name} ALTER COLUMN #{column_name} #{type_to_sql(type, options[:limit], options[:precision], options[:scale])}"]
|
198
|
+
if options_include_default?(options)
|
199
|
+
remove_default_constraint(table_name, column_name)
|
200
|
+
sql_commands << "ALTER TABLE #{table_name} ADD CONSTRAINT DF_#{table_name}_#{column_name} DEFAULT #{quote(options[:default], options[:column])} FOR #{column_name}"
|
201
|
+
end
|
202
|
+
sql_commands.each {|c|
|
203
|
+
execute(c)
|
204
|
+
}
|
205
|
+
end
|
206
|
+
def change_column_default(table_name, column_name, default) #:nodoc:
|
207
|
+
remove_default_constraint(table_name, column_name)
|
208
|
+
execute "ALTER TABLE #{table_name} ADD CONSTRAINT DF_#{table_name}_#{column_name} DEFAULT #{quote(default, column_name)} FOR #{column_name}"
|
209
|
+
end
|
210
|
+
def remove_column(table_name, column_name)
|
211
|
+
remove_check_constraints(table_name, column_name)
|
212
|
+
remove_default_constraint(table_name, column_name)
|
213
|
+
execute "ALTER TABLE #{table_name} DROP COLUMN [#{column_name}]"
|
214
|
+
end
|
215
|
+
|
216
|
+
def remove_default_constraint(table_name, column_name)
|
217
|
+
defaults = select "select def.name from sysobjects def, syscolumns col, sysobjects tab where col.cdefault = def.id and col.name = '#{column_name}' and tab.name = '#{table_name}' and col.id = tab.id"
|
218
|
+
defaults.each {|constraint|
|
219
|
+
execute "ALTER TABLE #{table_name} DROP CONSTRAINT #{constraint["name"]}"
|
220
|
+
}
|
221
|
+
end
|
222
|
+
|
223
|
+
def remove_check_constraints(table_name, column_name)
|
224
|
+
# TODO remove all constraints in single method
|
225
|
+
constraints = select "SELECT CONSTRAINT_NAME FROM INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE where TABLE_NAME = '#{table_name}' and COLUMN_NAME = '#{column_name}'"
|
226
|
+
constraints.each do |constraint|
|
227
|
+
execute "ALTER TABLE #{table_name} DROP CONSTRAINT #{constraint["CONSTRAINT_NAME"]}"
|
228
|
+
end
|
229
|
+
end
|
230
|
+
|
231
|
+
def remove_index(table_name, options = {})
|
232
|
+
execute "DROP INDEX #{table_name}.#{index_name(table_name, options)}"
|
233
|
+
end
|
234
|
+
|
235
|
+
|
236
|
+
def columns(table_name, name = nil)
|
237
|
+
return [] if table_name =~ /^information_schema\./i
|
238
|
+
cc = super
|
239
|
+
cc.each do |col|
|
240
|
+
col.identity = true if col.sql_type =~ /identity/i
|
241
|
+
col.is_special = true if col.sql_type =~ /text|ntext|image/i
|
242
|
+
end
|
243
|
+
cc
|
244
|
+
end
|
245
|
+
|
246
|
+
def _execute(sql, name = nil)
|
247
|
+
if sql.lstrip =~ /^insert/i
|
248
|
+
if query_requires_identity_insert?(sql)
|
249
|
+
table_name = get_table_name(sql)
|
250
|
+
with_identity_insert_enabled(table_name) do
|
251
|
+
id = @connection.execute_insert(sql)
|
252
|
+
end
|
253
|
+
else
|
254
|
+
@connection.execute_insert(sql)
|
255
|
+
end
|
256
|
+
elsif sql.lstrip =~ /^\(?\s*(select|show)/i
|
257
|
+
repair_special_columns(sql)
|
258
|
+
@connection.execute_query(sql)
|
259
|
+
else
|
260
|
+
@connection.execute_update(sql)
|
261
|
+
end
|
262
|
+
end
|
263
|
+
|
264
|
+
#SELECT .. FOR UPDATE is not supported on Microsoft SQL Server
|
265
|
+
def add_lock!(sql, options)
|
266
|
+
sql
|
267
|
+
end
|
268
|
+
|
269
|
+
private
|
270
|
+
# Turns IDENTITY_INSERT ON for table during execution of the block
|
271
|
+
# N.B. This sets the state of IDENTITY_INSERT to OFF after the
|
272
|
+
# block has been executed without regard to its previous state
|
273
|
+
|
274
|
+
def with_identity_insert_enabled(table_name, &block)
|
275
|
+
set_identity_insert(table_name, true)
|
276
|
+
yield
|
277
|
+
ensure
|
278
|
+
set_identity_insert(table_name, false)
|
279
|
+
end
|
280
|
+
|
281
|
+
def set_identity_insert(table_name, enable = true)
|
282
|
+
execute "SET IDENTITY_INSERT #{table_name} #{enable ? 'ON' : 'OFF'}"
|
283
|
+
rescue Exception => e
|
284
|
+
raise ActiveRecord::ActiveRecordError, "IDENTITY_INSERT could not be turned #{enable ? 'ON' : 'OFF'} for table #{table_name}"
|
285
|
+
end
|
286
|
+
|
287
|
+
def get_table_name(sql)
|
288
|
+
if sql =~ /^\s*insert\s+into\s+([^\(\s,]+)\s*|^\s*update\s+([^\(\s,]+)\s*/i
|
289
|
+
$1
|
290
|
+
elsif sql =~ /from\s+([^\(\s,]+)\s*/i
|
291
|
+
$1
|
292
|
+
else
|
293
|
+
nil
|
294
|
+
end
|
295
|
+
end
|
296
|
+
|
297
|
+
def identity_column(table_name)
|
298
|
+
@table_columns = {} unless @table_columns
|
299
|
+
@table_columns[table_name] = columns(table_name) if @table_columns[table_name] == nil
|
300
|
+
@table_columns[table_name].each do |col|
|
301
|
+
return col.name if col.identity
|
302
|
+
end
|
303
|
+
|
304
|
+
return nil
|
305
|
+
end
|
306
|
+
|
307
|
+
def query_requires_identity_insert?(sql)
|
308
|
+
table_name = get_table_name(sql)
|
309
|
+
id_column = identity_column(table_name)
|
310
|
+
sql =~ /\[#{id_column}\]/ ? table_name : nil
|
311
|
+
end
|
312
|
+
|
313
|
+
def get_special_columns(table_name)
|
314
|
+
special = []
|
315
|
+
@table_columns ||= {}
|
316
|
+
@table_columns[table_name] ||= columns(table_name)
|
317
|
+
@table_columns[table_name].each do |col|
|
318
|
+
special << col.name if col.is_special
|
319
|
+
end
|
320
|
+
special
|
321
|
+
end
|
322
|
+
|
323
|
+
def repair_special_columns(sql)
|
324
|
+
special_cols = get_special_columns(get_table_name(sql))
|
325
|
+
for col in special_cols.to_a
|
326
|
+
sql.gsub!(Regexp.new(" #{col.to_s} = "), " #{col.to_s} LIKE ")
|
327
|
+
sql.gsub!(/ORDER BY #{col.to_s}/i, '')
|
328
|
+
end
|
329
|
+
sql
|
330
|
+
end
|
331
|
+
end
|
332
|
+
end
|
333
|
+
|