Module: ActiveRecord::ConnectionAdapters::DatabaseStatements

Included in:
AbstractAdapter
Defined in:
lib/active_record/connection_adapters/abstract/database_statements.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#transaction_manager ⇒ Object (readonly)

:nodoc:



238
239
240
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 238

def transaction_manager
  @transaction_manager
end

Instance Method Details

#add_transaction_record(record) ⇒ Object

Register a record with the current transaction so that its after_commit and after_rollback callbacks can be called.



252
253
254
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 252

def add_transaction_record(record)
  current_transaction.add_record(record)
end

#begin_db_transaction ⇒ Object

Begins the transaction (and turns off auto-committing).



261
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 261

def begin_db_transaction()    end

#begin_isolated_db_transaction(isolation) ⇒ Object

Begins the transaction with the isolation level set. Raises an error by default; adapters that support setting the isolation level should implement this method.

Raises:



275
276
277
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 275

def begin_isolated_db_transaction(isolation)
  raise ActiveRecord::TransactionIsolationError, "adapter does not support setting transaction isolation"
end

#cacheable_query(arel) ⇒ Object

This is used in the StatementCache object. It returns an object that can be used to query the database repeatedly.



21
22
23
24
25
26
27
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 21

def cacheable_query(arel) # :nodoc:
  if prepared_statements
    ActiveRecord::StatementCache.query visitor, arel.ast
  else
    ActiveRecord::StatementCache.partial_query visitor, arel.ast, collector
  end
end

#commit_db_transaction ⇒ Object

Commits the transaction (and turns on auto-committing).



280
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 280

def commit_db_transaction()   end

#default_sequence_name(table, column) ⇒ Object 



294
295
296
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 294

def default_sequence_name(table, column)
  nil
end

#delete(arel, name = nil, binds = []) ⇒ Object Also known as: delete_sql

Executes the delete statement and returns the number of rows affected.



139
140
141
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 139

def delete(arel, name = nil, binds = [])
  exec_delete(to_sql(arel, binds), name, binds)
end

#empty_insert_statement_value ⇒ Object 



329
330
331
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 329

def empty_insert_statement_value
  "DEFAULT VALUES"
end

#exec_delete(sql, name, binds) ⇒ Object

Executes delete sql statement in the context of this connection using binds as the bind substitutes. name is logged along with the executed sql statement.



98
99
100
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 98

def exec_delete(sql, name, binds)
  exec_query(sql, name, binds)
end

#exec_insert(sql, name, binds, pk = nil, sequence_name = nil) ⇒ Object

Executes insert sql statement in the context of this connection using binds as the bind substitutes. name is logged along with the executed sql statement.



91
92
93
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 91

def exec_insert(sql, name, binds, pk = nil, sequence_name = nil)
  exec_query(sql, name, binds)
end

#exec_query(sql, name = 'SQL', binds = [], prepare: false) ⇒ Object

Executes sql statement in the context of this connection using binds as the bind substitutes. name is logged along with the executed sql statement.

Raises:

  • (NotImplementedError) 


84
85
86
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 84

def exec_query(sql, name = 'SQL', binds = [], prepare: false)
  raise NotImplementedError
end

#exec_rollback_db_transaction ⇒ Object

:nodoc:



288
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 288

def exec_rollback_db_transaction() end

#exec_update(sql, name, binds) ⇒ Object

Executes update sql statement in the context of this connection using binds as the bind substitutes. name is logged along with the executed sql statement.



110
111
112
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 110

def exec_update(sql, name, binds)
  exec_query(sql, name, binds)
end

#execute(sql, name = nil) ⇒ Object

Executes the SQL statement in the context of this connection and returns the raw result from the connection adapter. Note: depending on your database connector, the result returned by this method may be manually memory managed. Consider using the exec_query wrapper instead.

Raises:

  • (NotImplementedError) 


77
78
79
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 77

def execute(sql, name = nil)
  raise NotImplementedError
end

#initialize ⇒ Object 



4
5
6
7
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 4

def initialize
  super
  reset_transaction
end

#insert(arel, name = nil, pk = nil, id_value = nil, sequence_name = nil, binds = []) ⇒ Object Also known as: create, insert_sql

Executes an INSERT query and returns the new record’s ID

id_value will be returned unless the value is nil, in which case the database will attempt to calculate the last inserted id and return that value.

If the next id was calculated in advance (as in Oracle), it should be passed in as id_value.



122
123
124
125
126
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 122

def insert(arel, name = nil, pk = nil, id_value = nil, sequence_name = nil, binds = [])
  sql, binds, pk, sequence_name = sql_for_insert(to_sql(arel, binds), pk, id_value, sequence_name, binds)
  value = exec_insert(sql, name, binds, pk, sequence_name)
  id_value || last_inserted_id(value)
end

#insert_fixture(fixture, table_name) ⇒ Object

Inserts the given fixture into the table. Overridden in adapters that require something beyond a simple insert (eg. Oracle).



305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 305

def insert_fixture(fixture, table_name)
  fixture = fixture.stringify_keys

  columns = schema_cache.columns_hash(table_name)
  binds = fixture.map do |name, value|
    if column = columns[name]
      type = lookup_cast_type_from_column(column)
      Relation::QueryAttribute.new(name, value, type)
    else
      raise Fixture::FixtureError, %(table "#{table_name}" has no column named #{name.inspect}.)
    end
  end
  key_list = fixture.keys.map { |name| quote_column_name(name) }
  value_list = prepare_binds_for_database(binds).map do |value|
    begin
      quote(value)
    rescue TypeError
      quote(YAML.dump(value))
    end
  end

  execute "INSERT INTO #{quote_table_name(table_name)} (#{key_list.join(', ')}) VALUES (#{value_list.join(', ')})", 'Fixture Insert'
end

#join_to_update(update, select, key) ⇒ Object Also known as: join_to_delete

The default strategy for an UPDATE with joins is to use a subquery. This doesn’t work on MySQL (even when aliasing the tables), but MySQL allows using JOIN directly in an UPDATE statement, so in the MySQL adapters we redefine this to do that.



355
356
357
358
359
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 355

def join_to_update(update, select, key) # :nodoc:
  subselect = subquery_for(key, select)

  update.where key.in(subselect)
end

#reset_sequence!(table, column, sequence = nil) ⇒ Object

Set the sequence to the max value of the table’s column.



299
300
301
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 299

def reset_sequence!(table, column, sequence = nil)
  # Do nothing by default. Implement for PostgreSQL, Oracle, ...
end

#reset_transaction ⇒ Object

:nodoc:



246
247
248
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 246

def reset_transaction #:nodoc:
  @transaction_manager = ConnectionAdapters::TransactionManager.new(self)
end

#rollback_db_transaction ⇒ Object

Rolls back the transaction (and turns on auto-committing). Must be done if the transaction block raises an exception or returns false.



284
285
286
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 284

def rollback_db_transaction
  exec_rollback_db_transaction
end

#rollback_to_savepoint(name = nil) ⇒ Object 



290
291
292
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 290

def rollback_to_savepoint(name = nil)
  exec_rollback_to_savepoint(name)
end

#sanitize_limit(limit) ⇒ Object

Sanitizes the given LIMIT parameter in order to prevent SQL injection.

The limit may be anything that can evaluate to a string via #to_s. It should look like an integer, or a comma-delimited list of integers, or an Arel SQL literal.

Returns Integer and Arel::Nodes::SqlLiteral limits as is. Returns the sanitized limit parameter, either as an integer, or as a string which contains a comma-delimited list of integers.



342
343
344
345
346
347
348
349
350
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 342

def sanitize_limit(limit)
  if limit.is_a?(Integer) || limit.is_a?(Arel::Nodes::SqlLiteral)
    limit
  elsif limit.to_s.include?(',')
    Arel.sql limit.to_s.split(',').map{ |i| Integer(i) }.join(',')
  else
    Integer(limit)
  end
end

#select_all(arel, name = nil, binds = [], preparable: nil) ⇒ Object

Returns an ActiveRecord::Result instance.



30
31
32
33
34
35
36
37
38
39
40
41
42
43
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 30

def select_all(arel, name = nil, binds = [], preparable: nil)
  arel, binds = binds_from_relation arel, binds
  sql = to_sql(arel, binds)
  if !prepared_statements || (arel.is_a?(String) && preparable.nil?)
    preparable = false
  else
    preparable = visitor.preparable
  end
  if prepared_statements && preparable
    select_prepared(sql, name, binds)
  else
    select(sql, name, binds)
  end
end

#select_one(arel, name = nil, binds = []) ⇒ Object

Returns a record hash with the column names as keys and column values as values.



47
48
49
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 47

def select_one(arel, name = nil, binds = [])
  select_all(arel, name, binds).first
end

#select_rows(sql, name = nil, binds = []) ⇒ Object

Returns an array of arrays containing the field values. Order is the same as that returned by columns.



68
69
70
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 68

def select_rows(sql, name = nil, binds = [])
  exec_query(sql, name, binds).rows
end

#select_value(arel, name = nil, binds = []) ⇒ Object

Returns a single value from a record



52
53
54
55
56
57
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 52

def select_value(arel, name = nil, binds = [])
  arel, binds = binds_from_relation arel, binds
  if result = select_rows(to_sql(arel, binds), name, binds).first
    result.first
  end
end

#select_values(arel, name = nil, binds = []) ⇒ Object

Returns an array of the values of the first column in a select:

select_values("SELECT id FROM companies LIMIT 3") => [1,2,3]



61
62
63
64
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 61

def select_values(arel, name = nil, binds = [])
  arel, binds = binds_from_relation arel, binds
  select_rows(to_sql(arel, binds), name, binds).map(&:first)
end

#supports_statement_cache? ⇒ Boolean

Returns true when the connection adapter supports prepared statement caching, otherwise returns false

Returns:

  • (Boolean) 


147
148
149
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 147

def supports_statement_cache?
  false
end

#to_sql(arel, binds = []) ⇒ Object

Converts an arel AST to SQL



10
11
12
13
14
15
16
17
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 10

def to_sql(arel, binds = [])
  if arel.respond_to?(:ast)
    collected = visitor.accept(arel.ast, collector)
    collected.compile(binds.dup, self)
  else
    arel
  end
end

#transaction(requires_new: nil, isolation: nil, joinable: true) ⇒ Object

Runs the given block in a database transaction, and returns the result of the block.

Nested transactions support

Most databases don’t support true nested transactions. At the time of writing, the only database that supports true nested transactions that we’re aware of, is MS-SQL.

In order to get around this problem, #transaction will emulate the effect of nested transactions, by using savepoints: dev.mysql.com/doc/refman/5.7/en/savepoint.html Savepoints are supported by MySQL and PostgreSQL. SQLite3 version >= ‘3.6.8’ supports savepoints.

It is safe to call this method if a database transaction is already open, i.e. if #transaction is called within another #transaction block. In case of a nested call, #transaction will behave as follows:

  • The block will be run without doing anything. All database statements that happen within the block are effectively appended to the already open database transaction.

  • However, if :requires_new is set, the block will be wrapped in a database savepoint acting as a sub-transaction.

Caveats

MySQL doesn’t support DDL transactions. If you perform a DDL operation, then any created savepoints will be automatically released. For example, if you’ve created a savepoint, then you execute a CREATE TABLE statement, then the savepoint that was created will be automatically released.

This means that, on MySQL, you shouldn’t execute DDL operations inside a #transaction call that you know might create a savepoint. Otherwise, #transaction will raise exceptions when it tries to release the already-automatically-released savepoints:

Model.connection.transaction do  # BEGIN
  Model.connection.transaction(requires_new: true) do  # CREATE SAVEPOINT active_record_1
    Model.connection.create_table(...)
    # active_record_1 now automatically released
  end  # RELEASE SAVEPOINT active_record_1  <--- BOOM! database error!
end

Transaction isolation

If your database supports setting the isolation level for a transaction, you can set it like so:

Post.transaction(isolation: :serializable) do
  # ...
end

Valid isolation levels are:

  • :read_uncommitted

  • :read_committed

  • :repeatable_read

  • :serializable

You should consult the documentation for your database to understand the semantics of these different levels:

An ActiveRecord::TransactionIsolationError will be raised if:

  • The adapter does not support setting the isolation level

  • You are joining an existing open transaction

  • You are creating a nested (savepoint) transaction

The mysql2 and postgresql adapters support setting the transaction isolation level.



225
226
227
228
229
230
231
232
233
234
235
236
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 225

def transaction(requires_new: nil, isolation: nil, joinable: true)
  if !requires_new && current_transaction.joinable?
    if isolation
      raise ActiveRecord::TransactionIsolationError, "cannot set isolation when joining a transaction"
    end
    yield
  else
    transaction_manager.within_new_transaction(isolation: isolation, joinable: joinable) { yield }
  end
rescue ActiveRecord::Rollback
  # rollbacks are silently swallowed
end

#transaction_isolation_levels ⇒ Object 



263
264
265
266
267
268
269
270
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 263

def transaction_isolation_levels
  {
    read_uncommitted: "READ UNCOMMITTED",
    read_committed:   "READ COMMITTED",
    repeatable_read:  "REPEATABLE READ",
    serializable:     "SERIALIZABLE"
  }
end

#transaction_open? ⇒ Boolean

Returns:

  • (Boolean) 


242
243
244
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 242

def transaction_open?
  current_transaction.open?
end

#transaction_state ⇒ Object 



256
257
258
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 256

def transaction_state
  current_transaction.state
end

#truncate(table_name, name = nil) ⇒ Object

Executes the truncate statement.

Raises:

  • (NotImplementedError) 


103
104
105
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 103

def truncate(table_name, name = nil)
  raise NotImplementedError
end

#update(arel, name = nil, binds = []) ⇒ Object Also known as: update_sql

Executes the update statement and returns the number of rows affected.



132
133
134
 
# File 'lib/active_record/connection_adapters/abstract/database_statements.rb', line 132

def update(arel, name = nil, binds = [])
  exec_update(to_sql(arel, binds), name, binds)
end