如何在 Rails 中的数据库连接上执行查询？

Question

我有一些初始化函数，用于在 PostgreSQL 的数据库服务器端（即不是 Rails）设置审核日志记录。在将数据插入或更新任何审核表之前，至少必须发出一个命令（设置当前用户），否则整个查询将彻底失败。

每次在代码中运行任何保存操作之前我都可以轻松地调用这些函数，但 DRY 让我觉得我应该在尽可能少的地方重复代码，特别是因为这与数据库不可知论的理想有很大的不同。目前，我正在尝试在初始值设定项中覆盖 ActiveRecord::Base.assessment_connection 来设置它，以便在自动连接后立即运行查询，但它的行为并不符合我的预期。这是初始化程序中的代码：

class ActiveRecord::Base
  # extend the class methods, not the instance methods
  class << self
    alias :old_establish_connection :establish_connection # hide the default

    def establish_connection(*args)
      ret = old_establish_connection(*args) # call the default

      # set up necessary session variables for audit logging
      # call these after calling default, to make sure conn is established 1st
      db = self.class.connection
      db.execute("SELECT SV.set('current_user', 'test@localhost')")
      db.execute("SELECT SV.set('audit_notes', NULL)") # end "empty variable" err

      ret # return the default's original value
    end
  end
end

puts "Loaded custom establish_connection into ActiveRecord::Base"

sycobuny:~/rails$ ruby script/server 
=> Booting WEBrick
=> Rails 2.3.5 application starting on http://0.0.0.0:3000
Loaded custom establish_connection into ActiveRecord::Base

这不会给我任何错误，不幸的是我无法检查该方法在内部是什么样子（我使用的是 ActiveRecord::Base.method(:assessment_connection)，但显然这创建了一个每次调用时都会创建 new Method 对象，这似乎毫无价值，因为我无法检查 object_id 是否有任何有价值的信息，而且我也无法反转编译）。

但是，该代码似乎从未被调用，因为正如我之前预测的那样，对数据库对象运行保存或更新的任何尝试都会失败。如果这不是在连接到数据库时立即执行代码的正确方法，那么什么是呢？

Answer 1

Rails 使用连接池，因此最好的选择是对 ActiveRecord::ConnectionAdapters::ConnectionPool#new_connection 使用 alias_method_chain

module ActiveRecord
  Module ConnectionAdapters
    class ConnectionPool
      alias_method_chain :new_connection, :my_stuff
      private
      def new_connection_with_my_stuff
        c = new_connection_without_my_stuff
        # Do your stuff with 
        # c.exec(<your sql command>)
        c
      end
    end
  end
end

Answer 2

您是否知道在建立数据库连接之前正在调用您的初始化程序？

如果没有关于可能出现问题的权威信息，我会利用 ActiveRecord 源位于我的硬盘上这一事实。

深入研究基类并进行一些调试，以查看相对于调用初始化程序的时间，ActiveRecord建立连接的调用时间。要么打开额外的详细调试。

Answer 3

在编写代码来挂钩 Rails 数据库调用和初始化方面，您可能需要使用 db-charmer 中的代码 作为模板。它演示了许多与 ActiveRecord 交互的有趣方法，其中一种可能可以满足您的需求。