activerecord-jdbc-adapter 0.9.2 → 1.1.0
Sign up to get free protection for your applications and to get access to all the features.
- data/History.txt +163 -0
- data/Manifest.txt +130 -65
- data/README.txt +64 -72
- data/lib/active_record/connection_adapters/derby_adapter.rb +1 -13
- data/lib/active_record/connection_adapters/h2_adapter.rb +1 -13
- data/lib/active_record/connection_adapters/hsqldb_adapter.rb +1 -13
- data/lib/active_record/connection_adapters/informix_adapter.rb +1 -1
- data/lib/active_record/connection_adapters/jdbc_adapter.rb +1 -633
- data/lib/active_record/connection_adapters/jndi_adapter.rb +1 -1
- 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 -13
- data/lib/active_record/connection_adapters/oracle_adapter.rb +1 -1
- data/lib/active_record/connection_adapters/postgresql_adapter.rb +1 -13
- data/lib/active_record/connection_adapters/sqlite3_adapter.rb +1 -13
- data/lib/activerecord-jdbc-adapter.rb +6 -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/db2.rb +15 -0
- data/lib/arel/visitors/derby.rb +19 -0
- data/lib/arel/visitors/hsqldb.rb +24 -0
- data/lib/arel/visitors/mssql.rb +44 -0
- data/lib/arjdbc.rb +29 -0
- data/lib/arjdbc/db2.rb +2 -0
- data/lib/arjdbc/db2/adapter.rb +413 -0
- data/lib/arjdbc/derby.rb +7 -0
- data/lib/{jdbc_adapter/jdbc_derby.rb → arjdbc/derby/adapter.rb} +58 -132
- data/lib/arjdbc/derby/connection_methods.rb +18 -0
- data/lib/arjdbc/discover.rb +92 -0
- data/lib/arjdbc/firebird.rb +2 -0
- data/lib/{jdbc_adapter/jdbc_firebird.rb → arjdbc/firebird/adapter.rb} +36 -14
- data/lib/arjdbc/h2.rb +4 -0
- data/lib/arjdbc/h2/adapter.rb +36 -0
- data/lib/arjdbc/h2/connection_methods.rb +12 -0
- data/lib/arjdbc/hsqldb.rb +4 -0
- data/lib/{jdbc_adapter/jdbc_hsqldb.rb → arjdbc/hsqldb/adapter.rb} +34 -62
- data/lib/arjdbc/hsqldb/connection_methods.rb +14 -0
- data/lib/arjdbc/informix.rb +3 -0
- data/lib/{jdbc_adapter/jdbc_informix.rb → arjdbc/informix/adapter.rb} +10 -19
- data/lib/arjdbc/informix/connection_methods.rb +10 -0
- data/lib/arjdbc/jdbc.rb +2 -0
- data/lib/arjdbc/jdbc/adapter.rb +285 -0
- data/lib/arjdbc/jdbc/adapter_java.jar +0 -0
- data/lib/arjdbc/jdbc/callbacks.rb +44 -0
- data/lib/arjdbc/jdbc/column.rb +38 -0
- data/lib/arjdbc/jdbc/compatibility.rb +51 -0
- data/lib/arjdbc/jdbc/connection.rb +128 -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 +44 -0
- data/lib/arjdbc/jdbc/extension.rb +47 -0
- data/lib/arjdbc/jdbc/java.rb +14 -0
- data/lib/{jdbc_adapter → arjdbc/jdbc}/jdbc.rake +36 -26
- data/lib/{jdbc_adapter → arjdbc/jdbc}/missing_functionality_helper.rb +5 -5
- 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/{jdbc_adapter/jdbc_mimer.rb → arjdbc/mimer/adapter.rb} +17 -16
- data/lib/arjdbc/mssql.rb +4 -0
- data/lib/arjdbc/mssql/adapter.rb +472 -0
- data/lib/arjdbc/mssql/connection_methods.rb +30 -0
- data/lib/arjdbc/mssql/limit_helpers.rb +92 -0
- data/lib/{jdbc_adapter → arjdbc/mssql}/tsql_helper.rb +2 -1
- data/lib/arjdbc/mysql.rb +4 -0
- data/lib/arjdbc/mysql/adapter.rb +416 -0
- data/lib/arjdbc/mysql/connection_methods.rb +27 -0
- data/lib/arjdbc/oracle.rb +3 -0
- data/lib/{jdbc_adapter/jdbc_oracle.rb → arjdbc/oracle/adapter.rb} +68 -33
- data/lib/arjdbc/oracle/connection_methods.rb +11 -0
- data/lib/arjdbc/postgresql.rb +4 -0
- data/lib/{jdbc_adapter/jdbc_postgre.rb → arjdbc/postgresql/adapter.rb} +147 -72
- data/lib/arjdbc/postgresql/connection_methods.rb +21 -0
- data/lib/arjdbc/sqlite3.rb +4 -0
- data/lib/arjdbc/sqlite3/adapter.rb +381 -0
- data/lib/arjdbc/sqlite3/connection_methods.rb +34 -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/jdbc_generator.rb +9 -0
- data/lib/jdbc_adapter.rb +2 -27
- data/lib/jdbc_adapter/rake_tasks.rb +3 -10
- data/lib/jdbc_adapter/version.rb +3 -5
- data/lib/pg.rb +23 -1
- data/rails_generators/jdbc_generator.rb +2 -2
- data/rails_generators/templates/{jdbc.rb → config/initializers/jdbc.rb} +1 -1
- data/rails_generators/templates/{jdbc.rake → lib/tasks/jdbc.rake} +2 -2
- data/rakelib/compile.rake +5 -3
- data/rakelib/db.rake +19 -0
- data/rakelib/package.rake +12 -6
- data/rakelib/test.rake +24 -14
- data/src/java/{jdbc_adapter/JdbcDerbySpec.java → arjdbc/derby/DerbyModule.java} +63 -34
- data/src/java/arjdbc/h2/H2RubyJdbcConnection.java +70 -0
- data/src/java/{jdbc_adapter/MssqlRubyJdbcConnection.java → arjdbc/informix/InformixRubyJdbcConnection.java} +15 -12
- data/src/java/{jdbc_adapter/JdbcAdapterInternalService.java → arjdbc/jdbc/AdapterJavaService.java} +21 -8
- data/src/java/{jdbc_adapter → arjdbc/jdbc}/JdbcConnectionFactory.java +6 -6
- data/src/java/{jdbc_adapter → arjdbc/jdbc}/RubyJdbcConnection.java +210 -67
- data/src/java/arjdbc/jdbc/SQLBlock.java +48 -0
- data/src/java/arjdbc/mssql/MssqlRubyJdbcConnection.java +127 -0
- data/src/java/{jdbc_adapter/JdbcMySQLSpec.java → arjdbc/mysql/MySQLModule.java} +12 -12
- data/src/java/arjdbc/mysql/MySQLRubyJdbcConnection.java +87 -0
- data/src/java/arjdbc/oracle/OracleRubyJdbcConnection.java +85 -0
- data/src/java/{jdbc_adapter/PostgresRubyJdbcConnection.java → arjdbc/postgresql/PostgresqlRubyJdbcConnection.java} +11 -9
- data/src/java/arjdbc/sqlite3/Sqlite3RubyJdbcConnection.java +64 -0
- data/test/abstract_db_create.rb +98 -19
- data/test/activerecord/connection_adapters/type_conversion_test.rb +1 -1
- data/test/db/db2.rb +4 -2
- data/test/db/derby.rb +12 -14
- data/test/db/hsqldb.rb +3 -2
- data/test/db/jndi_config.rb +22 -12
- data/test/db/mssql.rb +5 -5
- data/test/db/sqlite3.rb +2 -6
- data/test/db2_simple_test.rb +56 -0
- data/test/derby_migration_test.rb +50 -3
- data/test/derby_simple_test.rb +87 -0
- data/test/generic_jdbc_connection_test.rb +21 -1
- data/test/h2_simple_test.rb +35 -0
- data/test/jdbc_common.rb +4 -5
- data/test/jndi_callbacks_test.rb +3 -1
- data/test/jndi_test.rb +1 -11
- data/test/models/entry.rb +20 -0
- data/test/models/mixed_case.rb +7 -2
- data/test/models/string_id.rb +18 -0
- data/test/mssql_db_create_test.rb +26 -0
- data/test/mssql_identity_insert_test.rb +19 -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_simple_test.rb +49 -0
- data/test/mysql_db_create_test.rb +8 -6
- data/test/mysql_info_test.rb +123 -0
- data/test/mysql_simple_test.rb +19 -1
- data/test/oracle_simple_test.rb +9 -20
- data/test/oracle_specific_test.rb +83 -0
- data/test/pick_rails_version.rb +3 -0
- data/test/postgres_db_create_test.rb +11 -0
- data/test/postgres_drop_db_test.rb +16 -0
- data/test/postgres_mixed_case_test.rb +11 -1
- data/test/postgres_nonseq_pkey_test.rb +0 -2
- data/test/postgres_schema_search_path_test.rb +44 -0
- data/test/postgres_simple_test.rb +53 -0
- data/test/postgres_table_alias_length_test.rb +15 -0
- data/test/simple.rb +103 -9
- data/test/sqlite3_simple_test.rb +1 -0
- data/test/sybase_jtds_simple_test.rb +22 -0
- metadata +239 -137
- data/lib/active_record/connection_adapters/cachedb_adapter.rb +0 -1
- data/lib/active_record/connection_adapters/jdbc_adapter_spec.rb +0 -26
- data/lib/jdbc_adapter/jdbc_adapter_internal.jar +0 -0
- data/lib/jdbc_adapter/jdbc_cachedb.rb +0 -33
- data/lib/jdbc_adapter/jdbc_db2.rb +0 -191
- data/lib/jdbc_adapter/jdbc_mssql.rb +0 -333
- data/lib/jdbc_adapter/jdbc_mysql.rb +0 -236
- data/lib/jdbc_adapter/jdbc_sqlite3.rb +0 -379
- data/lib/jdbc_adapter/jdbc_sybase.rb +0 -50
- data/src/java/jdbc_adapter/SQLBlock.java +0 -27
- data/src/java/jdbc_adapter/Sqlite3RubyJdbcConnection.java +0 -41
- data/test/cachedb_simple_test.rb +0 -6
- data/test/db/cachedb.rb +0 -9
- data/test/jdbc_adapter/jdbc_db2_test.rb +0 -26
- data/test/jdbc_adapter/jdbc_sybase_test.rb +0 -33
- data/test/minirunit.rb +0 -109
- data/test/minirunit/testConnect.rb +0 -14
- data/test/minirunit/testH2.rb +0 -73
- data/test/minirunit/testHsqldb.rb +0 -73
- data/test/minirunit/testLoadActiveRecord.rb +0 -3
- data/test/minirunit/testMysql.rb +0 -83
- data/test/minirunit/testRawSelect.rb +0 -24
@@ -1,4 +1,4 @@
|
|
1
|
-
module
|
1
|
+
module ArJdbc
|
2
2
|
module MissingFunctionalityHelper
|
3
3
|
#Taken from SQLite adapter
|
4
4
|
|
@@ -15,7 +15,7 @@ module JdbcSpec
|
|
15
15
|
move_table(altered_table_name, table_name, &caller)
|
16
16
|
end
|
17
17
|
end
|
18
|
-
|
18
|
+
|
19
19
|
def move_table(from, to, options = {}, &block) #:nodoc:
|
20
20
|
copy_table(from, to, options, &block)
|
21
21
|
drop_table(from)
|
@@ -30,7 +30,7 @@ module JdbcSpec
|
|
30
30
|
(options[:rename][column.name] ||
|
31
31
|
options[:rename][column.name.to_sym] ||
|
32
32
|
column.name) : column.name
|
33
|
-
|
33
|
+
|
34
34
|
@definition.column(column_name, column.type,
|
35
35
|
:limit => column.limit, :default => column.default,
|
36
36
|
:null => column.null)
|
@@ -44,7 +44,7 @@ module JdbcSpec
|
|
44
44
|
@definition.columns.map {|column| column.name},
|
45
45
|
options[:rename] || {})
|
46
46
|
end
|
47
|
-
|
47
|
+
|
48
48
|
def copy_table_indexes(from, to, rename = {}) #:nodoc:
|
49
49
|
indexes(from).each do |index|
|
50
50
|
name = index.name.downcase
|
@@ -67,7 +67,7 @@ module JdbcSpec
|
|
67
67
|
end
|
68
68
|
end
|
69
69
|
end
|
70
|
-
|
70
|
+
|
71
71
|
def copy_table_contents(from, to, columns, rename = {}) #:nodoc:
|
72
72
|
column_mappings = Hash[*columns.map {|name| [name, name]}.flatten]
|
73
73
|
rename.inject(column_mappings) {|map, a| map[a.last] = a.first; map}
|
@@ -0,0 +1,28 @@
|
|
1
|
+
module ArJdbc
|
2
|
+
module QuotedPrimaryKeyExtension
|
3
|
+
def self.extended(base)
|
4
|
+
# Rails 3 method Rails 2 method
|
5
|
+
meth = [:arel_attributes_values, :attributes_with_quotes].detect do |m|
|
6
|
+
base.private_instance_methods.include?(m.to_s)
|
7
|
+
end
|
8
|
+
pk_hash_key = "self.class.primary_key"
|
9
|
+
pk_hash_value = '"?"'
|
10
|
+
if meth == :arel_attributes_values
|
11
|
+
pk_hash_key = "self.class.arel_table[#{pk_hash_key}]"
|
12
|
+
pk_hash_value = "Arel::SqlLiteral.new(#{pk_hash_value})"
|
13
|
+
end
|
14
|
+
if meth
|
15
|
+
base.module_eval <<-PK, __FILE__, __LINE__
|
16
|
+
alias :#{meth}_pre_pk :#{meth}
|
17
|
+
def #{meth}(include_primary_key = true, *args) #:nodoc:
|
18
|
+
aq = #{meth}_pre_pk(include_primary_key, *args)
|
19
|
+
if connection.is_a?(ArJdbc::Oracle) || connection.is_a?(ArJdbc::Mimer)
|
20
|
+
aq[#{pk_hash_key}] = #{pk_hash_value} if include_primary_key && aq[#{pk_hash_key}].nil?
|
21
|
+
end
|
22
|
+
aq
|
23
|
+
end
|
24
|
+
PK
|
25
|
+
end
|
26
|
+
end
|
27
|
+
end
|
28
|
+
end
|
@@ -0,0 +1,10 @@
|
|
1
|
+
if defined?(Rake.application) && Rake.application && ENV["SKIP_AR_JDBC_RAKE_REDEFINES"].nil?
|
2
|
+
jdbc_rakefile = File.dirname(__FILE__) + "/jdbc.rake"
|
3
|
+
if Rake.application.lookup("db:create")
|
4
|
+
# rails tasks already defined; load the override tasks now
|
5
|
+
load jdbc_rakefile
|
6
|
+
else
|
7
|
+
# rails tasks not loaded yet; load as an import
|
8
|
+
Rake.application.add_import(jdbc_rakefile)
|
9
|
+
end
|
10
|
+
end
|
@@ -0,0 +1,16 @@
|
|
1
|
+
module Kernel
|
2
|
+
# load a JDBC driver library/gem, failing silently. If failed, trust
|
3
|
+
# that the driver jar is already present through some other means
|
4
|
+
def jdbc_require_driver(path, gem_name = nil)
|
5
|
+
gem_name ||= path.sub('/', '-')
|
6
|
+
2.times do
|
7
|
+
begin
|
8
|
+
require path
|
9
|
+
break
|
10
|
+
rescue LoadError
|
11
|
+
require 'rubygems'
|
12
|
+
begin; gem gem_name; rescue LoadError; end
|
13
|
+
end
|
14
|
+
end
|
15
|
+
end
|
16
|
+
end
|
@@ -0,0 +1,126 @@
|
|
1
|
+
module ActiveRecord
|
2
|
+
module ConnectionAdapters
|
3
|
+
# I want to use JDBC's DatabaseMetaData#getTypeInfo to choose the best native types to
|
4
|
+
# use for ActiveRecord's Adapter#native_database_types in a database-independent way,
|
5
|
+
# but apparently a database driver can return multiple types for a given
|
6
|
+
# java.sql.Types constant. So this type converter uses some heuristics to try to pick
|
7
|
+
# the best (most common) type to use. It's not great, it would be better to just
|
8
|
+
# delegate to each database's existin AR adapter's native_database_types method, but I
|
9
|
+
# wanted to try to do this in a way that didn't pull in all the other adapters as
|
10
|
+
# dependencies. Suggestions appreciated.
|
11
|
+
class JdbcTypeConverter
|
12
|
+
# The basic ActiveRecord types, mapped to an array of procs that are used to #select
|
13
|
+
# the best type. The procs are used as selectors in order until there is only one
|
14
|
+
# type left. If all the selectors are applied and there is still more than one
|
15
|
+
# type, an exception will be raised.
|
16
|
+
AR_TO_JDBC_TYPES = {
|
17
|
+
:string => [ lambda {|r| Jdbc::Types::VARCHAR == r['data_type'].to_i},
|
18
|
+
lambda {|r| r['type_name'] =~ /^varchar/i},
|
19
|
+
lambda {|r| r['type_name'] =~ /^varchar$/i},
|
20
|
+
lambda {|r| r['type_name'] =~ /varying/i}],
|
21
|
+
:text => [ lambda {|r| [Jdbc::Types::LONGVARCHAR, Jdbc::Types::CLOB].include?(r['data_type'].to_i)},
|
22
|
+
lambda {|r| r['type_name'] =~ /^text$/i}, # For Informix
|
23
|
+
lambda {|r| r['type_name'] =~ /sub_type 1$/i}, # For FireBird
|
24
|
+
lambda {|r| r['type_name'] =~ /^(text|clob)$/i},
|
25
|
+
lambda {|r| r['type_name'] =~ /^character large object$/i},
|
26
|
+
lambda {|r| r['sql_data_type'] == 2005}],
|
27
|
+
:integer => [ lambda {|r| Jdbc::Types::INTEGER == r['data_type'].to_i},
|
28
|
+
lambda {|r| r['type_name'] =~ /^integer$/i},
|
29
|
+
lambda {|r| r['type_name'] =~ /^int4$/i},
|
30
|
+
lambda {|r| r['type_name'] =~ /^int$/i}],
|
31
|
+
:decimal => [ lambda {|r| Jdbc::Types::DECIMAL == r['data_type'].to_i},
|
32
|
+
lambda {|r| r['type_name'] =~ /^decimal$/i},
|
33
|
+
lambda {|r| r['type_name'] =~ /^numeric$/i},
|
34
|
+
lambda {|r| r['type_name'] =~ /^number$/i},
|
35
|
+
lambda {|r| r['type_name'] =~ /^real$/i},
|
36
|
+
lambda {|r| r['precision'] == '38'},
|
37
|
+
lambda {|r| r['data_type'] == '2'}],
|
38
|
+
:float => [ lambda {|r| [Jdbc::Types::FLOAT,Jdbc::Types::DOUBLE, Jdbc::Types::REAL].include?(r['data_type'].to_i)},
|
39
|
+
lambda {|r| r['data_type'].to_i == Jdbc::Types::REAL}, #Prefer REAL to DOUBLE for Postgresql
|
40
|
+
lambda {|r| r['type_name'] =~ /^float/i},
|
41
|
+
lambda {|r| r['type_name'] =~ /^double$/i},
|
42
|
+
lambda {|r| r['type_name'] =~ /^real$/i},
|
43
|
+
lambda {|r| r['precision'] == '15'}],
|
44
|
+
:datetime => [ lambda {|r| Jdbc::Types::TIMESTAMP == r['data_type'].to_i},
|
45
|
+
lambda {|r| r['type_name'] =~ /^datetime$/i},
|
46
|
+
lambda {|r| r['type_name'] =~ /^timestamp$/i},
|
47
|
+
lambda {|r| r['type_name'] =~ /^date/i},
|
48
|
+
lambda {|r| r['type_name'] =~ /^integer/i}], #Num of milliseconds for SQLite3 JDBC Driver
|
49
|
+
:timestamp => [ lambda {|r| Jdbc::Types::TIMESTAMP == r['data_type'].to_i},
|
50
|
+
lambda {|r| r['type_name'] =~ /^timestamp$/i},
|
51
|
+
lambda {|r| r['type_name'] =~ /^datetime/i},
|
52
|
+
lambda {|r| r['type_name'] =~ /^date/i},
|
53
|
+
lambda {|r| r['type_name'] =~ /^integer/i}], #Num of milliseconds for SQLite3 JDBC Driver
|
54
|
+
:time => [ lambda {|r| Jdbc::Types::TIME == r['data_type'].to_i},
|
55
|
+
lambda {|r| r['type_name'] =~ /^time$/i},
|
56
|
+
lambda {|r| r['type_name'] =~ /^datetime/i}, # For Informix
|
57
|
+
lambda {|r| r['type_name'] =~ /^date/i},
|
58
|
+
lambda {|r| r['type_name'] =~ /^integer/i}], #Num of milliseconds for SQLite3 JDBC Driver
|
59
|
+
:date => [ lambda {|r| Jdbc::Types::DATE == r['data_type'].to_i},
|
60
|
+
lambda {|r| r['type_name'] =~ /^date$/i},
|
61
|
+
lambda {|r| r['type_name'] =~ /^date/i},
|
62
|
+
lambda {|r| r['type_name'] =~ /^integer/i}], #Num of milliseconds for SQLite3 JDBC Driver3
|
63
|
+
:binary => [ lambda {|r| [Jdbc::Types::LONGVARBINARY,Jdbc::Types::BINARY,Jdbc::Types::BLOB].include?(r['data_type'].to_i)},
|
64
|
+
lambda {|r| r['type_name'] =~ /^blob/i},
|
65
|
+
lambda {|r| r['type_name'] =~ /sub_type 0$/i}, # For FireBird
|
66
|
+
lambda {|r| r['type_name'] =~ /^varbinary$/i}, # We want this sucker for Mimer
|
67
|
+
lambda {|r| r['type_name'] =~ /^binary$/i}, ],
|
68
|
+
:boolean => [ lambda {|r| [Jdbc::Types::TINYINT].include?(r['data_type'].to_i)},
|
69
|
+
lambda {|r| r['type_name'] =~ /^bool/i},
|
70
|
+
lambda {|r| r['data_type'] == '-7'},
|
71
|
+
lambda {|r| r['type_name'] =~ /^tinyint$/i},
|
72
|
+
lambda {|r| r['type_name'] =~ /^decimal$/i},
|
73
|
+
lambda {|r| r['type_name'] =~ /^integer$/i}]
|
74
|
+
}
|
75
|
+
|
76
|
+
def initialize(types)
|
77
|
+
@types = types
|
78
|
+
@types.each {|t| t['type_name'] ||= t['local_type_name']} # Sybase driver seems to want 'local_type_name'
|
79
|
+
end
|
80
|
+
|
81
|
+
def choose_best_types
|
82
|
+
type_map = {}
|
83
|
+
@types.each do |row|
|
84
|
+
name = row['type_name'].downcase
|
85
|
+
k = name.to_sym
|
86
|
+
type_map[k] = { :name => name }
|
87
|
+
set_limit_to_nonzero_precision(type_map[k], row)
|
88
|
+
end
|
89
|
+
|
90
|
+
AR_TO_JDBC_TYPES.keys.each do |k|
|
91
|
+
typerow = choose_type(k)
|
92
|
+
type_map[k] = { :name => typerow['type_name'].downcase }
|
93
|
+
case k
|
94
|
+
when :integer, :string, :decimal
|
95
|
+
set_limit_to_nonzero_precision(type_map[k], typerow)
|
96
|
+
when :boolean
|
97
|
+
type_map[k][:limit] = 1
|
98
|
+
end
|
99
|
+
end
|
100
|
+
type_map
|
101
|
+
end
|
102
|
+
|
103
|
+
def choose_type(ar_type)
|
104
|
+
procs = AR_TO_JDBC_TYPES[ar_type]
|
105
|
+
types = @types
|
106
|
+
procs.each do |p|
|
107
|
+
new_types = types.reject {|r| r["data_type"].to_i == Jdbc::Types::OTHER}
|
108
|
+
new_types = new_types.select(&p)
|
109
|
+
new_types = new_types.inject([]) do |typs,t|
|
110
|
+
typs << t unless typs.detect {|el| el['type_name'] == t['type_name']}
|
111
|
+
typs
|
112
|
+
end
|
113
|
+
return new_types.first if new_types.length == 1
|
114
|
+
types = new_types if new_types.length > 0
|
115
|
+
end
|
116
|
+
raise "unable to choose type for #{ar_type} from:\n#{types.collect{|t| t['type_name']}.inspect}"
|
117
|
+
end
|
118
|
+
|
119
|
+
def set_limit_to_nonzero_precision(map, row)
|
120
|
+
if row['precision'] && row['precision'].to_i > 0
|
121
|
+
map[:limit] = row['precision'].to_i
|
122
|
+
end
|
123
|
+
end
|
124
|
+
end
|
125
|
+
end
|
126
|
+
end
|
data/lib/arjdbc/mimer.rb
ADDED
@@ -1,7 +1,8 @@
|
|
1
|
-
module
|
1
|
+
module ArJdbc
|
2
2
|
module Mimer
|
3
|
-
def self.
|
4
|
-
|
3
|
+
def self.extended(mod)
|
4
|
+
require 'arjdbc/jdbc/quoted_primary_key'
|
5
|
+
ActiveRecord::Base.extend ArJdbc::QuotedPrimaryKeyExtension
|
5
6
|
end
|
6
7
|
|
7
8
|
def modify_types(tp)
|
@@ -16,7 +17,7 @@ module JdbcSpec
|
|
16
17
|
tp[:date] = { :name => "TIMESTAMP" }
|
17
18
|
tp
|
18
19
|
end
|
19
|
-
|
20
|
+
|
20
21
|
def default_sequence_name(table, column) #:nodoc:
|
21
22
|
"#{table}_seq"
|
22
23
|
end
|
@@ -50,13 +51,13 @@ module JdbcSpec
|
|
50
51
|
log(sql, name) { @connection.execute_insert sql,pk }
|
51
52
|
else # Assume the sql contains a bind-variable for the id
|
52
53
|
id_value = select_one("SELECT NEXT_VALUE OF #{sequence_name} AS val FROM MIMER.ONEROW")['val']
|
53
|
-
log(sql, name) {
|
54
|
+
log(sql, name) {
|
54
55
|
execute_prepared_insert(sql,id_value)
|
55
56
|
}
|
56
57
|
end
|
57
58
|
id_value
|
58
59
|
end
|
59
|
-
|
60
|
+
|
60
61
|
def execute_prepared_insert(sql, id)
|
61
62
|
@stmts ||= {}
|
62
63
|
@stmts[sql] ||= @connection.ps(sql)
|
@@ -68,24 +69,24 @@ module JdbcSpec
|
|
68
69
|
|
69
70
|
def quote(value, column = nil) #:nodoc:
|
70
71
|
return value.quoted_id if value.respond_to?(:quoted_id)
|
71
|
-
|
72
|
+
|
72
73
|
if String === value && column && column.type == :binary
|
73
74
|
return "X'#{quote_string(value.unpack("C*").collect {|v| v.to_s(16)}.join)}'"
|
74
75
|
end
|
75
76
|
case value
|
76
|
-
when String
|
77
|
+
when String
|
77
78
|
%Q{'#{quote_string(value)}'}
|
78
|
-
when NilClass
|
79
|
+
when NilClass
|
79
80
|
'NULL'
|
80
|
-
when TrueClass
|
81
|
+
when TrueClass
|
81
82
|
'1'
|
82
|
-
when FalseClass
|
83
|
+
when FalseClass
|
83
84
|
'0'
|
84
|
-
when Numeric
|
85
|
+
when Numeric
|
85
86
|
value.to_s
|
86
|
-
when Date, Time
|
87
|
+
when Date, Time
|
87
88
|
%Q{TIMESTAMP '#{value.strftime("%Y-%m-%d %H:%M:%S")}'}
|
88
|
-
else
|
89
|
+
else
|
89
90
|
%Q{'#{quote_string(value.to_yaml)}'}
|
90
91
|
end
|
91
92
|
end
|
@@ -102,7 +103,7 @@ module JdbcSpec
|
|
102
103
|
@limit = options[:limit]
|
103
104
|
@offset = options[:offset]
|
104
105
|
end
|
105
|
-
|
106
|
+
|
106
107
|
def select_all(sql, name = nil)
|
107
108
|
@offset ||= 0
|
108
109
|
if !@limit || @limit == -1
|
@@ -114,7 +115,7 @@ module JdbcSpec
|
|
114
115
|
ensure
|
115
116
|
@limit = @offset = nil
|
116
117
|
end
|
117
|
-
|
118
|
+
|
118
119
|
def select_one(sql, name = nil)
|
119
120
|
@offset ||= 0
|
120
121
|
select(sql, name)[@offset]
|
data/lib/arjdbc/mssql.rb
ADDED
@@ -0,0 +1,472 @@
|
|
1
|
+
require 'arjdbc/mssql/tsql_helper'
|
2
|
+
require 'arjdbc/mssql/limit_helpers'
|
3
|
+
|
4
|
+
module ::ArJdbc
|
5
|
+
module MsSQL
|
6
|
+
include TSqlMethods
|
7
|
+
include LimitHelpers
|
8
|
+
|
9
|
+
def self.extended(mod)
|
10
|
+
unless @lob_callback_added
|
11
|
+
ActiveRecord::Base.class_eval do
|
12
|
+
def after_save_with_mssql_lob
|
13
|
+
self.class.columns.select { |c| c.sql_type =~ /image/i }.each do |c|
|
14
|
+
value = self[c.name]
|
15
|
+
value = value.to_yaml if unserializable_attribute?(c.name, c)
|
16
|
+
next if value.nil? || (value == '')
|
17
|
+
|
18
|
+
connection.write_large_object(c.type == :binary, c.name, self.class.table_name, self.class.primary_key, quote_value(id), value)
|
19
|
+
end
|
20
|
+
end
|
21
|
+
end
|
22
|
+
|
23
|
+
ActiveRecord::Base.after_save :after_save_with_mssql_lob
|
24
|
+
@lob_callback_added = true
|
25
|
+
end
|
26
|
+
mod.add_version_specific_add_limit_offset
|
27
|
+
end
|
28
|
+
|
29
|
+
def self.column_selector
|
30
|
+
[/sqlserver|tds|Microsoft SQL/i, lambda {|cfg,col| col.extend(::ArJdbc::MsSQL::Column)}]
|
31
|
+
end
|
32
|
+
|
33
|
+
def self.jdbc_connection_class
|
34
|
+
::ActiveRecord::ConnectionAdapters::MssqlJdbcConnection
|
35
|
+
end
|
36
|
+
|
37
|
+
def arel2_visitors
|
38
|
+
require 'arel/visitors/mssql'
|
39
|
+
visitor_class = sqlserver_version == "2000" ? ::Arel::Visitors::SQLServer2000 : ::Arel::Visitors::SQLServer
|
40
|
+
{ 'mssql' => visitor_class, 'jdbcmssql' => visitor_class}
|
41
|
+
end
|
42
|
+
|
43
|
+
def sqlserver_version
|
44
|
+
@sqlserver_version ||= select_value("select @@version")[/Microsoft SQL Server\s+(\d{4})/, 1]
|
45
|
+
end
|
46
|
+
|
47
|
+
def add_version_specific_add_limit_offset
|
48
|
+
if sqlserver_version == "2000"
|
49
|
+
extend LimitHelpers::SqlServer2000AddLimitOffset
|
50
|
+
else
|
51
|
+
extend LimitHelpers::SqlServerAddLimitOffset
|
52
|
+
end
|
53
|
+
end
|
54
|
+
|
55
|
+
def modify_types(tp) #:nodoc:
|
56
|
+
super(tp)
|
57
|
+
tp[:string] = {:name => "NVARCHAR", :limit => 255}
|
58
|
+
if sqlserver_version == "2000"
|
59
|
+
tp[:text] = {:name => "NTEXT"}
|
60
|
+
else
|
61
|
+
tp[:text] = {:name => "NVARCHAR(MAX)"}
|
62
|
+
end
|
63
|
+
tp
|
64
|
+
end
|
65
|
+
|
66
|
+
def type_to_sql(type, limit = nil, precision = nil, scale = nil) #:nodoc:
|
67
|
+
# MSSQL's NVARCHAR(n | max) column supports either a number between 1 and
|
68
|
+
# 4000, or the word "MAX", which corresponds to 2**30-1 UCS-2 characters.
|
69
|
+
#
|
70
|
+
# It does not accept NVARCHAR(1073741823) here, so we have to change it
|
71
|
+
# to NVARCHAR(MAX), even though they are logically equivalent.
|
72
|
+
#
|
73
|
+
# MSSQL Server 2000 is skipped here because I don't know how it will behave.
|
74
|
+
#
|
75
|
+
# See: http://msdn.microsoft.com/en-us/library/ms186939.aspx
|
76
|
+
if type.to_s == 'string' and limit == 1073741823 and sqlserver_version != "2000"
|
77
|
+
'NVARCHAR(MAX)'
|
78
|
+
elsif %w( boolean date datetime ).include?(type.to_s)
|
79
|
+
super(type) # cannot specify limit/precision/scale with these types
|
80
|
+
else
|
81
|
+
super
|
82
|
+
end
|
83
|
+
end
|
84
|
+
|
85
|
+
module Column
|
86
|
+
attr_accessor :identity, :is_special
|
87
|
+
|
88
|
+
def simplified_type(field_type)
|
89
|
+
case field_type
|
90
|
+
when /int|bigint|smallint|tinyint/i then :integer
|
91
|
+
when /numeric/i then (@scale.nil? || @scale == 0) ? :integer : :decimal
|
92
|
+
when /float|double|decimal|money|real|smallmoney/i then :decimal
|
93
|
+
when /datetime|smalldatetime/i then :datetime
|
94
|
+
when /timestamp/i then :timestamp
|
95
|
+
when /time/i then :time
|
96
|
+
when /date/i then :date
|
97
|
+
when /text|ntext|xml/i then :text
|
98
|
+
when /binary|image|varbinary/i then :binary
|
99
|
+
when /char|nchar|nvarchar|string|varchar/i then (@limit == 1073741823 ? (@limit = nil; :text) : :string)
|
100
|
+
when /bit/i then :boolean
|
101
|
+
when /uniqueidentifier/i then :string
|
102
|
+
end
|
103
|
+
end
|
104
|
+
|
105
|
+
def default_value(value)
|
106
|
+
return $1 if value =~ /^\(N?'(.*)'\)$/
|
107
|
+
value
|
108
|
+
end
|
109
|
+
|
110
|
+
def type_cast(value)
|
111
|
+
return nil if value.nil? || value == "(null)" || value == "(NULL)"
|
112
|
+
case type
|
113
|
+
when :integer then value.to_i rescue unquote(value).to_i rescue value ? 1 : 0
|
114
|
+
when :primary_key then value == true || value == false ? value == true ? 1 : 0 : value.to_i
|
115
|
+
when :decimal then self.class.value_to_decimal(unquote(value))
|
116
|
+
when :datetime then cast_to_datetime(value)
|
117
|
+
when :timestamp then cast_to_time(value)
|
118
|
+
when :time then cast_to_time(value)
|
119
|
+
when :date then cast_to_date(value)
|
120
|
+
when :boolean then value == true or (value =~ /^t(rue)?$/i) == 0 or unquote(value)=="1"
|
121
|
+
when :binary then unquote value
|
122
|
+
else value
|
123
|
+
end
|
124
|
+
end
|
125
|
+
|
126
|
+
def extract_limit(sql_type)
|
127
|
+
case sql_type
|
128
|
+
when /text|ntext|xml|binary|image|varbinary|bit/
|
129
|
+
nil
|
130
|
+
else
|
131
|
+
super
|
132
|
+
end
|
133
|
+
end
|
134
|
+
|
135
|
+
def is_utf8?
|
136
|
+
sql_type =~ /nvarchar|ntext|nchar/i
|
137
|
+
end
|
138
|
+
|
139
|
+
def unquote(value)
|
140
|
+
value.to_s.sub(/\A\([\(\']?/, "").sub(/[\'\)]?\)\Z/, "")
|
141
|
+
end
|
142
|
+
|
143
|
+
def cast_to_time(value)
|
144
|
+
return value if value.is_a?(Time)
|
145
|
+
time_array = ParseDate.parsedate(value)
|
146
|
+
return nil if !time_array.any?
|
147
|
+
time_array[0] ||= 2000
|
148
|
+
time_array[1] ||= 1
|
149
|
+
time_array[2] ||= 1
|
150
|
+
return Time.send(ActiveRecord::Base.default_timezone, *time_array) rescue nil
|
151
|
+
|
152
|
+
# Try DateTime instead - the date may be outside the time period support by Time.
|
153
|
+
DateTime.new(*time_array[0..5]) rescue nil
|
154
|
+
end
|
155
|
+
|
156
|
+
def cast_to_date(value)
|
157
|
+
return value if value.is_a?(Date)
|
158
|
+
return Date.parse(value) rescue nil
|
159
|
+
end
|
160
|
+
|
161
|
+
def cast_to_datetime(value)
|
162
|
+
if value.is_a?(Time)
|
163
|
+
if value.year != 0 and value.month != 0 and value.day != 0
|
164
|
+
return value
|
165
|
+
else
|
166
|
+
return Time.mktime(2000, 1, 1, value.hour, value.min, value.sec) rescue nil
|
167
|
+
end
|
168
|
+
end
|
169
|
+
if value.is_a?(DateTime)
|
170
|
+
begin
|
171
|
+
# Attempt to convert back to a Time, but it could fail for dates significantly in the past/future.
|
172
|
+
return Time.mktime(value.year, value.mon, value.day, value.hour, value.min, value.sec)
|
173
|
+
rescue ArgumentError
|
174
|
+
return value
|
175
|
+
end
|
176
|
+
end
|
177
|
+
|
178
|
+
return cast_to_time(value) if value.is_a?(Date) or value.is_a?(String) rescue nil
|
179
|
+
|
180
|
+
return value.is_a?(Date) ? value : nil
|
181
|
+
end
|
182
|
+
|
183
|
+
# These methods will only allow the adapter to insert binary data with a length of 7K or less
|
184
|
+
# because of a SQL Server statement length policy.
|
185
|
+
def self.string_to_binary(value)
|
186
|
+
''
|
187
|
+
end
|
188
|
+
|
189
|
+
end
|
190
|
+
|
191
|
+
def quote(value, column = nil)
|
192
|
+
return value.quoted_id if value.respond_to?(:quoted_id)
|
193
|
+
|
194
|
+
case value
|
195
|
+
# SQL Server 2000 doesn't let you insert an integer into a NVARCHAR
|
196
|
+
# column, so we include Integer here.
|
197
|
+
when String, ActiveSupport::Multibyte::Chars, Integer
|
198
|
+
value = value.to_s
|
199
|
+
if column && column.type == :binary
|
200
|
+
"'#{quote_string(ArJdbc::MsSQL::Column.string_to_binary(value))}'" # ' (for ruby-mode)
|
201
|
+
elsif column && [:integer, :float].include?(column.type)
|
202
|
+
value = column.type == :integer ? value.to_i : value.to_f
|
203
|
+
value.to_s
|
204
|
+
elsif !column.respond_to?(:is_utf8?) || column.is_utf8?
|
205
|
+
"N'#{quote_string(value)}'" # ' (for ruby-mode)
|
206
|
+
else
|
207
|
+
super
|
208
|
+
end
|
209
|
+
when TrueClass then '1'
|
210
|
+
when FalseClass then '0'
|
211
|
+
else super
|
212
|
+
end
|
213
|
+
end
|
214
|
+
|
215
|
+
def quote_string(string)
|
216
|
+
string.gsub(/\'/, "''")
|
217
|
+
end
|
218
|
+
|
219
|
+
def quote_table_name(name)
|
220
|
+
name
|
221
|
+
end
|
222
|
+
|
223
|
+
def quote_column_name(name)
|
224
|
+
"[#{name}]"
|
225
|
+
end
|
226
|
+
|
227
|
+
def quoted_true
|
228
|
+
quote true
|
229
|
+
end
|
230
|
+
|
231
|
+
def quoted_false
|
232
|
+
quote false
|
233
|
+
end
|
234
|
+
|
235
|
+
def adapter_name #:nodoc:
|
236
|
+
'MsSQL'
|
237
|
+
end
|
238
|
+
|
239
|
+
def change_order_direction(order)
|
240
|
+
order.split(",").collect do |fragment|
|
241
|
+
case fragment
|
242
|
+
when /\bDESC\b/i then fragment.gsub(/\bDESC\b/i, "ASC")
|
243
|
+
when /\bASC\b/i then fragment.gsub(/\bASC\b/i, "DESC")
|
244
|
+
else String.new(fragment).split(',').join(' DESC,') + ' DESC'
|
245
|
+
end
|
246
|
+
end.join(",")
|
247
|
+
end
|
248
|
+
|
249
|
+
def supports_ddl_transactions?
|
250
|
+
true
|
251
|
+
end
|
252
|
+
|
253
|
+
def recreate_database(name)
|
254
|
+
drop_database(name)
|
255
|
+
create_database(name)
|
256
|
+
end
|
257
|
+
|
258
|
+
def drop_database(name)
|
259
|
+
execute "USE master"
|
260
|
+
execute "DROP DATABASE #{name}"
|
261
|
+
end
|
262
|
+
|
263
|
+
def create_database(name)
|
264
|
+
execute "CREATE DATABASE #{name}"
|
265
|
+
execute "USE #{name}"
|
266
|
+
end
|
267
|
+
|
268
|
+
def rename_table(name, new_name)
|
269
|
+
clear_cached_table(name)
|
270
|
+
execute "EXEC sp_rename '#{name}', '#{new_name}'"
|
271
|
+
end
|
272
|
+
|
273
|
+
# Adds a new column to the named table.
|
274
|
+
# See TableDefinition#column for details of the options you can use.
|
275
|
+
def add_column(table_name, column_name, type, options = {})
|
276
|
+
clear_cached_table(table_name)
|
277
|
+
add_column_sql = "ALTER TABLE #{table_name} ADD #{quote_column_name(column_name)} #{type_to_sql(type, options[:limit], options[:precision], options[:scale])}"
|
278
|
+
add_column_options!(add_column_sql, options)
|
279
|
+
# TODO: Add support to mimic date columns, using constraints to mark them as such in the database
|
280
|
+
# 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
|
281
|
+
execute(add_column_sql)
|
282
|
+
end
|
283
|
+
|
284
|
+
def rename_column(table, column, new_column_name)
|
285
|
+
clear_cached_table(table)
|
286
|
+
execute "EXEC sp_rename '#{table}.#{column}', '#{new_column_name}'"
|
287
|
+
end
|
288
|
+
|
289
|
+
def change_column(table_name, column_name, type, options = {}) #:nodoc:
|
290
|
+
clear_cached_table(table_name)
|
291
|
+
change_column_type(table_name, column_name, type, options)
|
292
|
+
change_column_default(table_name, column_name, options[:default]) if options_include_default?(options)
|
293
|
+
end
|
294
|
+
|
295
|
+
def change_column_type(table_name, column_name, type, options = {}) #:nodoc:
|
296
|
+
clear_cached_table(table_name)
|
297
|
+
sql = "ALTER TABLE #{table_name} ALTER COLUMN #{quote_column_name(column_name)} #{type_to_sql(type, options[:limit], options[:precision], options[:scale])}"
|
298
|
+
if options.has_key?(:null)
|
299
|
+
sql += (options[:null] ? " NULL" : " NOT NULL")
|
300
|
+
end
|
301
|
+
execute(sql)
|
302
|
+
end
|
303
|
+
|
304
|
+
def change_column_default(table_name, column_name, default) #:nodoc:
|
305
|
+
clear_cached_table(table_name)
|
306
|
+
remove_default_constraint(table_name, column_name)
|
307
|
+
unless default.nil?
|
308
|
+
execute "ALTER TABLE #{table_name} ADD CONSTRAINT DF_#{table_name}_#{column_name} DEFAULT #{quote(default)} FOR #{quote_column_name(column_name)}"
|
309
|
+
end
|
310
|
+
end
|
311
|
+
|
312
|
+
def remove_column(table_name, column_name)
|
313
|
+
clear_cached_table(table_name)
|
314
|
+
remove_check_constraints(table_name, column_name)
|
315
|
+
remove_default_constraint(table_name, column_name)
|
316
|
+
execute "ALTER TABLE #{table_name} DROP COLUMN [#{column_name}]"
|
317
|
+
end
|
318
|
+
|
319
|
+
def remove_default_constraint(table_name, column_name)
|
320
|
+
clear_cached_table(table_name)
|
321
|
+
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"
|
322
|
+
defaults.each {|constraint|
|
323
|
+
execute "ALTER TABLE #{table_name} DROP CONSTRAINT #{constraint["name"]}"
|
324
|
+
}
|
325
|
+
end
|
326
|
+
|
327
|
+
def remove_check_constraints(table_name, column_name)
|
328
|
+
clear_cached_table(table_name)
|
329
|
+
# TODO remove all constraints in single method
|
330
|
+
constraints = select "SELECT CONSTRAINT_NAME FROM INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE where TABLE_NAME = '#{table_name}' and COLUMN_NAME = '#{column_name}'"
|
331
|
+
constraints.each do |constraint|
|
332
|
+
execute "ALTER TABLE #{table_name} DROP CONSTRAINT #{constraint["CONSTRAINT_NAME"]}"
|
333
|
+
end
|
334
|
+
end
|
335
|
+
|
336
|
+
def remove_index(table_name, options = {})
|
337
|
+
execute "DROP INDEX #{table_name}.#{index_name(table_name, options)}"
|
338
|
+
end
|
339
|
+
|
340
|
+
def columns(table_name, name = nil)
|
341
|
+
# It's possible for table_name to be an empty string, or nil, if something attempts to issue SQL
|
342
|
+
# which doesn't involve a table. IE. "SELECT 1" or "SELECT * from someFunction()".
|
343
|
+
return [] if table_name.blank?
|
344
|
+
table_name = table_name.to_s if table_name.is_a?(Symbol)
|
345
|
+
|
346
|
+
# Remove []'s from around the table name, valid in a select statement, but not when matching metadata.
|
347
|
+
table_name = table_name.gsub(/[\[\]]/, '')
|
348
|
+
|
349
|
+
return [] if table_name =~ /^information_schema\./i
|
350
|
+
@table_columns = {} unless @table_columns
|
351
|
+
unless @table_columns[table_name]
|
352
|
+
@table_columns[table_name] = super
|
353
|
+
@table_columns[table_name].each do |col|
|
354
|
+
col.identity = true if col.sql_type =~ /identity/i
|
355
|
+
col.is_special = true if col.sql_type =~ /text|ntext|image|xml/i
|
356
|
+
end
|
357
|
+
end
|
358
|
+
@table_columns[table_name]
|
359
|
+
end
|
360
|
+
|
361
|
+
def _execute(sql, name = nil)
|
362
|
+
# Match the start of the sql to determine appropriate behaviour. Be aware of
|
363
|
+
# multi-line sql which might begin with 'create stored_proc' and contain 'insert into ...' lines.
|
364
|
+
# Possible improvements include ignoring comment blocks prior to the first statement.
|
365
|
+
if sql.lstrip =~ /\Ainsert/i
|
366
|
+
if query_requires_identity_insert?(sql)
|
367
|
+
table_name = get_table_name(sql)
|
368
|
+
with_identity_insert_enabled(table_name) do
|
369
|
+
id = @connection.execute_insert(sql)
|
370
|
+
end
|
371
|
+
else
|
372
|
+
@connection.execute_insert(sql)
|
373
|
+
end
|
374
|
+
elsif sql.lstrip =~ /\A(create|exec)/i
|
375
|
+
@connection.execute_update(sql)
|
376
|
+
elsif sql.lstrip =~ /\A\(?\s*(select|show)/i
|
377
|
+
repair_special_columns(sql)
|
378
|
+
@connection.execute_query(sql)
|
379
|
+
else
|
380
|
+
@connection.execute_update(sql)
|
381
|
+
end
|
382
|
+
end
|
383
|
+
|
384
|
+
def select(sql, name = nil)
|
385
|
+
log(sql, name) do
|
386
|
+
@connection.execute_query(sql)
|
387
|
+
end
|
388
|
+
end
|
389
|
+
|
390
|
+
#SELECT .. FOR UPDATE is not supported on Microsoft SQL Server
|
391
|
+
def add_lock!(sql, options)
|
392
|
+
sql
|
393
|
+
end
|
394
|
+
|
395
|
+
# Turns IDENTITY_INSERT ON for table during execution of the block
|
396
|
+
# N.B. This sets the state of IDENTITY_INSERT to OFF after the
|
397
|
+
# block has been executed without regard to its previous state
|
398
|
+
def with_identity_insert_enabled(table_name, &block)
|
399
|
+
set_identity_insert(table_name, true)
|
400
|
+
yield
|
401
|
+
ensure
|
402
|
+
set_identity_insert(table_name, false)
|
403
|
+
end
|
404
|
+
|
405
|
+
def set_identity_insert(table_name, enable = true)
|
406
|
+
execute "SET IDENTITY_INSERT #{table_name} #{enable ? 'ON' : 'OFF'}"
|
407
|
+
rescue Exception => e
|
408
|
+
raise ActiveRecord::ActiveRecordError, "IDENTITY_INSERT could not be turned #{enable ? 'ON' : 'OFF'} for table #{table_name}"
|
409
|
+
end
|
410
|
+
|
411
|
+
def identity_column(table_name)
|
412
|
+
columns(table_name).each do |col|
|
413
|
+
return col.name if col.identity
|
414
|
+
end
|
415
|
+
return nil
|
416
|
+
end
|
417
|
+
|
418
|
+
def query_requires_identity_insert?(sql)
|
419
|
+
table_name = get_table_name(sql)
|
420
|
+
id_column = identity_column(table_name)
|
421
|
+
if sql.strip =~ /insert into [^ ]+ ?\((.+?)\)/i
|
422
|
+
insert_columns = $1.split(/, */).map(&method(:unquote_column_name))
|
423
|
+
return table_name if insert_columns.include?(id_column)
|
424
|
+
end
|
425
|
+
end
|
426
|
+
|
427
|
+
def unquote_column_name(name)
|
428
|
+
if name =~ /^\[.*\]$/
|
429
|
+
name[1..-2]
|
430
|
+
else
|
431
|
+
name
|
432
|
+
end
|
433
|
+
end
|
434
|
+
|
435
|
+
def get_special_columns(table_name)
|
436
|
+
special = []
|
437
|
+
columns(table_name).each do |col|
|
438
|
+
special << col.name if col.is_special
|
439
|
+
end
|
440
|
+
special
|
441
|
+
end
|
442
|
+
|
443
|
+
def repair_special_columns(sql)
|
444
|
+
special_cols = get_special_columns(get_table_name(sql))
|
445
|
+
for col in special_cols.to_a
|
446
|
+
sql.gsub!(Regexp.new(" #{col.to_s} = "), " #{col.to_s} LIKE ")
|
447
|
+
sql.gsub!(/ORDER BY #{col.to_s}/i, '')
|
448
|
+
end
|
449
|
+
sql
|
450
|
+
end
|
451
|
+
|
452
|
+
def determine_order_clause(sql)
|
453
|
+
return $1 if sql =~ /ORDER BY (.*)$/
|
454
|
+
table_name = get_table_name(sql)
|
455
|
+
"#{table_name}.#{determine_primary_key(table_name)}"
|
456
|
+
end
|
457
|
+
|
458
|
+
def determine_primary_key(table_name)
|
459
|
+
primary_key = columns(table_name).detect { |column| column.primary || column.identity }
|
460
|
+
return primary_key.name if primary_key
|
461
|
+
# Look for an id column. Return it, without changing case, to cover dbs with a case-sensitive collation.
|
462
|
+
columns(table_name).each { |column| return column.name if column.name =~ /^id$/i }
|
463
|
+
# Give up and provide something which is going to crash almost certainly
|
464
|
+
columns(table_name)[0].name
|
465
|
+
end
|
466
|
+
|
467
|
+
def clear_cached_table(name)
|
468
|
+
(@table_columns ||= {}).delete(name.to_s)
|
469
|
+
end
|
470
|
+
end
|
471
|
+
end
|
472
|
+
|