46 lines
1.1 KiB
Ruby
46 lines
1.1 KiB
Ruby
class ActiveRecord::Base
|
|
|
|
# Execute SQL manually
|
|
def self.exec_sql(*args)
|
|
conn = ActiveRecord::Base.connection
|
|
sql = ActiveRecord::Base.send(:sanitize_sql_array, args)
|
|
conn.raw_connection.exec(sql)
|
|
end
|
|
|
|
def self.exec_sql_row_count(*args)
|
|
exec_sql(*args).cmd_tuples
|
|
end
|
|
|
|
def self.sql_fragment(*sql_array)
|
|
ActiveRecord::Base.send(:sanitize_sql_array, sql_array)
|
|
end
|
|
|
|
def exec_sql(*args)
|
|
ActiveRecord::Base.exec_sql(*args)
|
|
end
|
|
|
|
# Executes the given block +retries+ times (or forever, if explicitly given nil),
|
|
# catching and retrying SQL Deadlock errors.
|
|
#
|
|
# Thanks to: http://stackoverflow.com/a/7427186/165668
|
|
#
|
|
def self.retry_lock_error(retries = 5, &block)
|
|
begin
|
|
yield
|
|
rescue ActiveRecord::StatementInvalid => e
|
|
if e.message =~ /deadlock detected/ && (retries.nil? || retries > 0)
|
|
retry_lock_error(retries ? retries - 1 : nil, &block)
|
|
else
|
|
raise e
|
|
end
|
|
end
|
|
end
|
|
|
|
# Support for psql. If we want to support multiple RDBMs in the future we can
|
|
# split this.
|
|
def exec_sql_row_count(*args)
|
|
exec_sql(*args).cmd_tuples
|
|
end
|
|
|
|
end
|