Rails 事务启动挂钩

Question

我想将我公司的数据库从应用程序级审计转移到数据库级审计。我将使用此处详述的 91audit 触发器：https://github.com/2ndQuadrant/audit-trigger/blob/master/audit.sql

数据库级审计的问题在于它实际上并没有记录当前用户，它记录的是postgresql用户。

应用程序级别审计的缺点是我必须运行通过网络应用程序进行的每笔交易才能审计更改（如果我在同一个数据库上有多个应用程序则效果不佳）

我在这里找到了一个非常有趣的解决方案：http://schinckel.net/2014/06/01/in-database-audit-trail/

该解决方案依赖于 updating/inserting 在每个事务开始时临时 table 上的一行。

现在

为了让它对我有用，我想添加一个挂钩到 rails 以在任何保存、更新、销毁或删除调用时更新此临时 table。

我的问题是：

Rails 5.2+ 是否存在顶级 transaction-start 挂钩；最好是我可以在内部和初始化程序或 ApplicationRecord?

中设置的东西

我的第一个想法是添加一个初始化程序，但我不知道我要挂钩什么

我的备用计划是覆盖 ApplicationRecord.transaction

class ApplicationRecord < ActiveRecord::Base
  def self.transaction(*args)
    super(*args) do
      if Current.user
        ActiveRecord::Base.connection.execute(%Q(
          CREATE TEMP TABLE IF NOT EXISTS
          "_app_user" (user_id integer, ip_address inet);

          UPDATE
            _app_user
          SET
            user_id=#{Current.user.id},
            ip_address='#{Current.ip_address}';

          INSERT INTO
            _app_user (user_id, ip_address)
          SELECT
            #{Current.user.id}, '#{Current.ip_address}'
          WHERE NOT EXISTS (SELECT * FROM _app_user);
        ))
      end

      yield
    end
  end
end

Answer 1

覆盖 ActiveRecord.transaction 似乎工作得很好

class ApplicationRecord < ActiveRecord::Base
  def self.transaction(*args)
    super(*args) do
      if Current.user
        ip = Current.ip_address ? "'#{Current.ip_address}'" : 'NULL'
        ActiveRecord::Base.connection.execute <<-SQL
          CREATE TEMP TABLE IF NOT EXISTS
            "_app_user" (user_id integer, user_type text, ip_address inet);

          UPDATE
            _app_user
          SET
            user_id=#{Current.user.id},
            user_type='#{Current.user.class.to_s}',
            ip_address=#{ip};

          INSERT INTO
            _app_user (user_id, user_type, ip_address)
          SELECT
            #{Current.user.id},
            '#{Current.user.class.to_s}',
            #{ip}
          WHERE NOT EXISTS (SELECT * FROM _app_user);
        SQL
      end

      yield
    end
  end
end

Rails 事务启动挂钩

Rails transaction start hook

postgresql

audit

ruby-on-rails

ruby-on-rails-5.2