Module: ActiveRecord::ConnectionAdapters::SQLServer::DatabaseStatements
- Includes:
- Savepoints
- Included in:
- ActiveRecord::ConnectionAdapters::SQLServerAdapter
- Defined in:
- lib/active_record/connection_adapters/sqlserver/database_statements.rb
Instance Method Summary collapse
- #begin_db_transaction ⇒ Object
- #begin_isolated_db_transaction(isolation) ⇒ Object
- #case_sensitive_comparison(table, attribute, column, value) ⇒ Object
- #commit_db_transaction ⇒ Object
- #create_savepoint(name = current_savepoint_name) ⇒ Object
- #exec_delete(sql, name, binds) ⇒ Object
- #exec_insert(sql, name = nil, binds = [], pk = nil, _sequence_name = nil) ⇒ Object
- #exec_query(sql, name = 'SQL', binds = [], prepare: false) ⇒ Object
- #exec_rollback_db_transaction ⇒ Object
- #exec_rollback_to_savepoint(name = current_savepoint_name) ⇒ Object
- #exec_update(sql, name, binds) ⇒ Object
- #execute(sql, name = nil) ⇒ Object
-
#execute_procedure(proc_name, *variables) ⇒ Object
SQLServer Specific ======================================== #.
-
#insert_fixtures_set(fixture_set, tables_to_delete = []) ⇒ Object
We should propose this change to Rails team.
- #newid_function ⇒ Object
- #newsequentialid_function ⇒ Object
- #release_savepoint(name = current_savepoint_name) ⇒ Object
- #set_transaction_isolation_level(isolation_level) ⇒ Object
- #transaction_isolation_levels ⇒ Object
- #use_database(database = nil) ⇒ Object
- #user_options ⇒ Object
- #user_options_dateformat ⇒ Object
- #user_options_isolation_level ⇒ Object
- #user_options_language ⇒ Object
- #with_identity_insert_enabled(table_name) ⇒ Object
Instance Method Details
#begin_db_transaction ⇒ Object
36 37 38 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 36 def begin_db_transaction do_execute 'BEGIN TRANSACTION' end |
#begin_isolated_db_transaction(isolation) ⇒ Object
44 45 46 47 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 44 def begin_isolated_db_transaction(isolation) set_transaction_isolation_level transaction_isolation_levels.fetch(isolation) begin_db_transaction end |
#case_sensitive_comparison(table, attribute, column, value) ⇒ Object
74 75 76 77 78 79 80 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 74 def case_sensitive_comparison(table, attribute, column, value) if column.collation && !column.case_sensitive? table[attribute].eq(Arel::Nodes::Bin.new(value)) else super end end |
#commit_db_transaction ⇒ Object
53 54 55 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 53 def commit_db_transaction do_execute 'COMMIT TRANSACTION' end |
#create_savepoint(name = current_savepoint_name) ⇒ Object
63 64 65 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 63 def create_savepoint(name = current_savepoint_name) do_execute "SAVE TRANSACTION #{name}" end |
#exec_delete(sql, name, binds) ⇒ Object
26 27 28 29 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 26 def exec_delete(sql, name, binds) sql = sql.dup << '; SELECT @@ROWCOUNT AS AffectedRows' super(sql, name, binds).rows.first.first end |
#exec_insert(sql, name = nil, binds = [], pk = nil, _sequence_name = nil) ⇒ Object
18 19 20 21 22 23 24 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 18 def exec_insert(sql, name = nil, binds = [], pk = nil, _sequence_name = nil) if id_insert_table_name = exec_insert_requires_identity?(sql, pk, binds) with_identity_insert_enabled(id_insert_table_name) { super(sql, name, binds, pk) } else super(sql, name, binds, pk) end end |
#exec_query(sql, name = 'SQL', binds = [], prepare: false) ⇒ Object
14 15 16 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 14 def exec_query(sql, name = 'SQL', binds = [], prepare: false) sp_executesql(sql, name, binds, prepare: prepare) end |
#exec_rollback_db_transaction ⇒ Object
57 58 59 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 57 def exec_rollback_db_transaction do_execute 'IF @@TRANCOUNT > 0 ROLLBACK TRANSACTION' end |
#exec_rollback_to_savepoint(name = current_savepoint_name) ⇒ Object
67 68 69 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 67 def exec_rollback_to_savepoint(name = current_savepoint_name) do_execute "ROLLBACK TRANSACTION #{name}" end |
#exec_update(sql, name, binds) ⇒ Object
31 32 33 34 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 31 def exec_update(sql, name, binds) sql = sql.dup << '; SELECT @@ROWCOUNT AS AffectedRows' super(sql, name, binds).rows.first.first end |
#execute(sql, name = nil) ⇒ Object
6 7 8 9 10 11 12 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 6 def execute(sql, name = nil) if id_insert_table_name = query_requires_identity_insert?(sql) with_identity_insert_enabled(id_insert_table_name) { do_execute(sql, name) } else do_execute(sql, name) end end |
#execute_procedure(proc_name, *variables) ⇒ Object
SQLServer Specific ======================================== #
127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 127 def execute_procedure(proc_name, *variables) vars = if variables.any? && variables.first.is_a?(Hash) variables.first.map { |k, v| "@#{k} = #{quote(v)}" } else variables.map { |v| quote(v) } end.join(', ') sql = "EXEC #{proc_name} #{vars}".strip name = 'Execute Procedure' log(sql, name) do case @connection_options[:mode] when :dblib result = @connection.execute(sql) = { as: :hash, cache_rows: true, timezone: ActiveRecord::Base.default_timezone || :utc } result.each() do |row| r = row.with_indifferent_access yield(r) if block_given? end result.each.map { |row| row.is_a?(Hash) ? row.with_indifferent_access : row } end end end |
#insert_fixtures_set(fixture_set, tables_to_delete = []) ⇒ Object
We should propose this change to Rails team
83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 83 def insert_fixtures_set(fixture_set, tables_to_delete = []) fixture_inserts = [] fixture_set.each do |table_name, fixtures| fixtures.each_slice(insert_rows_length) do |batch| fixture_inserts << build_fixture_sql(batch, table_name) end end table_deletes = tables_to_delete.map { |table| "DELETE FROM #{quote_table_name table}".dup } total_sql = Array.wrap(combine_multi_statements(table_deletes + fixture_inserts)) disable_referential_integrity do transaction(requires_new: true) do total_sql.each do |sql| execute sql, "Fixtures Load" yield if block_given? end end end end |
#newid_function ⇒ Object
213 214 215 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 213 def newid_function select_value 'SELECT NEWID()' end |
#newsequentialid_function ⇒ Object
217 218 219 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 217 def newsequentialid_function select_value 'SELECT NEWSEQUENTIALID()' end |
#release_savepoint(name = current_savepoint_name) ⇒ Object
71 72 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 71 def release_savepoint(name = current_savepoint_name) end |
#set_transaction_isolation_level(isolation_level) ⇒ Object
49 50 51 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 49 def set_transaction_isolation_level(isolation_level) do_execute "SET TRANSACTION ISOLATION LEVEL #{isolation_level}" end |
#transaction_isolation_levels ⇒ Object
40 41 42 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 40 def transaction_isolation_levels super.merge snapshot: "SNAPSHOT" end |
#use_database(database = nil) ⇒ Object
157 158 159 160 161 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 157 def use_database(database = nil) return if sqlserver_azure? name = SQLServer::Utils.extract_identifiers(database || @connection_options[:database]).quoted do_execute "USE #{name}" unless name.blank? end |
#user_options ⇒ Object
163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 163 def return {} if sqlserver_azure? rows = select_rows('DBCC USEROPTIONS WITH NO_INFOMSGS', 'SCHEMA') rows = rows.first if rows.size == 2 && rows.last.empty? rows.reduce(HashWithIndifferentAccess.new) do |values, row| if row.instance_of? Hash set_option = row.values[0].gsub(/\s+/, '_') user_value = row.values[1] elsif row.instance_of? Array set_option = row[0].gsub(/\s+/, '_') user_value = row[1] end values[set_option] = user_value values end end |
#user_options_dateformat ⇒ Object
180 181 182 183 184 185 186 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 180 def if sqlserver_azure? select_value 'SELECT [dateformat] FROM [sys].[syslanguages] WHERE [langid] = @@LANGID', 'SCHEMA' else ['dateformat'] end end |
#user_options_isolation_level ⇒ Object
188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 188 def if sqlserver_azure? sql = %(SELECT CASE [transaction_isolation_level] WHEN 0 THEN NULL WHEN 1 THEN 'READ UNCOMMITTED' WHEN 2 THEN 'READ COMMITTED' WHEN 3 THEN 'REPEATABLE READ' WHEN 4 THEN 'SERIALIZABLE' WHEN 5 THEN 'SNAPSHOT' END AS [isolation_level] FROM [sys].[dm_exec_sessions] WHERE [session_id] = @@SPID).squish select_value sql, 'SCHEMA' else ['isolation_level'] end end |
#user_options_language ⇒ Object
205 206 207 208 209 210 211 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 205 def if sqlserver_azure? select_value 'SELECT @@LANGUAGE AS [language]', 'SCHEMA' else ['language'] end end |
#with_identity_insert_enabled(table_name) ⇒ Object
149 150 151 152 153 154 155 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 149 def with_identity_insert_enabled(table_name) table_name = quote_table_name(table_name) set_identity_insert(table_name, true) yield ensure set_identity_insert(table_name, false) end |