60 lines
1.7 KiB
Ruby
60 lines
1.7 KiB
Ruby
module Danbooru
|
|
module Extensions
|
|
module ActiveRecord
|
|
extend ActiveSupport::Concern
|
|
|
|
module ClassMethods
|
|
def without_timeout
|
|
connection.execute("SET STATEMENT_TIMEOUT = 0") unless Rails.env == "test"
|
|
yield
|
|
ensure
|
|
connection.execute("SET STATEMENT_TIMEOUT = #{CurrentUser.user.statement_timeout}") unless Rails.env == "test"
|
|
end
|
|
|
|
def with_timeout(n, default_value)
|
|
connection.execute("SET STATEMENT_TIMEOUT = #{n}") unless Rails.env == "test"
|
|
yield
|
|
rescue ::ActiveRecord::StatementInvalid
|
|
return default_value
|
|
ensure
|
|
connection.execute("SET STATEMENT_TIMEOUT = #{CurrentUser.user.statement_timeout}") unless Rails.env == "test"
|
|
end
|
|
end
|
|
|
|
%w(execute select_value select_values select_all).each do |method_name|
|
|
define_method("#{method_name}_sql") do |sql, *params|
|
|
connection.__send__(method_name, self.class.sanitize_sql_array([sql, *params]))
|
|
end
|
|
|
|
self.class.__send__(:define_method, "#{method_name}_sql") do |sql, *params|
|
|
connection.__send__(method_name, sanitize_sql_array([sql, *params]))
|
|
end
|
|
end
|
|
|
|
def arbitrary_sql_order_clause(ids, table_name = nil)
|
|
table_name = self.class.table_name if table_name.nil?
|
|
|
|
if ids.empty?
|
|
return "#{table_name}.id desc"
|
|
end
|
|
|
|
conditions = []
|
|
|
|
ids.each_with_index do |x, n|
|
|
conditions << "when #{x} then #{n}"
|
|
end
|
|
|
|
"case #{table_name}.id " + conditions.join(" ") + " end"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
class ActiveRecord::Base
|
|
class << self
|
|
public :sanitize_sql_array
|
|
end
|
|
|
|
include Danbooru::Extensions::ActiveRecord
|
|
end
|