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, 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 ======================================== #.
- #newid_function ⇒ Object
- #newsequentialid_function ⇒ Object
- #release_savepoint(name = current_savepoint_name) ⇒ Object
- #select_rows(sql, name = nil, binds = []) ⇒ Object
- #set_transaction_isolation_level(isolation_level) ⇒ Object
- #supports_statement_cache? ⇒ Boolean
- #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
44 45 46 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 44 def begin_db_transaction do_execute 'BEGIN TRANSACTION' end |
#begin_isolated_db_transaction(isolation) ⇒ Object
52 53 54 55 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 52 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
82 83 84 85 86 87 88 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 82 def case_sensitive_comparison(table, attribute, column, value) if value && value.acts_like?(:string) table[attribute].eq(Arel::Nodes::Bin.new(Arel::Nodes::BindParam.new)) else super end end |
#commit_db_transaction ⇒ Object
61 62 63 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 61 def commit_db_transaction do_execute 'COMMIT TRANSACTION' end |
#create_savepoint(name = current_savepoint_name) ⇒ Object
71 72 73 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 71 def create_savepoint(name = current_savepoint_name) do_execute "SAVE TRANSACTION #{name}" end |
#exec_delete(sql, name, binds) ⇒ Object
30 31 32 33 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 30 def exec_delete(sql, name, binds) sql << '; SELECT @@ROWCOUNT AS AffectedRows' super.rows.first.first end |
#exec_insert(sql, name, binds, pk = nil, _sequence_name = nil) ⇒ Object
22 23 24 25 26 27 28 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 22 def exec_insert(sql, name, binds, pk = nil, _sequence_name = nil) if pk && id_insert_table_name = query_requires_identity_insert?(sql) with_identity_insert_enabled(id_insert_table_name) { exec_query(sql, name, binds) } else exec_query(sql, name, binds) end end |
#exec_query(sql, name = 'SQL', binds = [], prepare: false) ⇒ Object
18 19 20 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 18 def exec_query(sql, name = 'SQL', binds = [], prepare: false) sp_executesql(sql, name, binds, prepare: prepare) end |
#exec_rollback_db_transaction ⇒ Object
65 66 67 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 65 def exec_rollback_db_transaction do_execute 'IF @@TRANCOUNT > 0 ROLLBACK TRANSACTION' end |
#exec_rollback_to_savepoint(name = current_savepoint_name) ⇒ Object
75 76 77 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 75 def exec_rollback_to_savepoint(name = current_savepoint_name) do_execute "ROLLBACK TRANSACTION #{name}" end |
#exec_update(sql, name, binds) ⇒ Object
35 36 37 38 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 35 def exec_update(sql, name, binds) sql << '; SELECT @@ROWCOUNT AS AffectedRows' super.rows.first.first end |
#execute(sql, name = nil) ⇒ Object
10 11 12 13 14 15 16 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 10 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 ======================================== #
97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 97 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 |
#newid_function ⇒ Object
183 184 185 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 183 def newid_function select_value 'SELECT NEWID()' end |
#newsequentialid_function ⇒ Object
187 188 189 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 187 def newsequentialid_function select_value 'SELECT NEWSEQUENTIALID()' end |
#release_savepoint(name = current_savepoint_name) ⇒ Object
79 80 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 79 def release_savepoint(name = current_savepoint_name) end |
#select_rows(sql, name = nil, binds = []) ⇒ Object
6 7 8 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 6 def select_rows(sql, name = nil, binds = []) sp_executesql sql, name, binds, fetch: :rows end |
#set_transaction_isolation_level(isolation_level) ⇒ Object
57 58 59 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 57 def set_transaction_isolation_level(isolation_level) do_execute "SET TRANSACTION ISOLATION LEVEL #{isolation_level}" end |
#supports_statement_cache? ⇒ Boolean
40 41 42 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 40 def supports_statement_cache? true end |
#transaction_isolation_levels ⇒ Object
48 49 50 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 48 def transaction_isolation_levels super.merge snapshot: "SNAPSHOT" end |
#use_database(database = nil) ⇒ Object
127 128 129 130 131 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 127 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
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 133 def return {} if sqlserver_azure? rows = select_rows('dbcc useroptions', '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
150 151 152 153 154 155 156 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 150 def if sqlserver_azure? select_value 'SELECT [dateformat] FROM [sys].[syslanguages] WHERE [langid] = @@LANGID', 'SCHEMA' else ['dateformat'] end end |
#user_options_isolation_level ⇒ Object
158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 158 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
175 176 177 178 179 180 181 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 175 def if sqlserver_azure? select_value 'SELECT @@LANGUAGE AS [language]', 'SCHEMA' else ['language'] end end |
#with_identity_insert_enabled(table_name) ⇒ Object
119 120 121 122 123 124 125 |
# File 'lib/active_record/connection_adapters/sqlserver/database_statements.rb', line 119 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 |