Rails ActiveRecord 使用 SQL 函数作为创建属性的值

Question

我在 6 号轨道上，

我想在创建记录时使用 sql 函数作为某些属性的值。

该 sql 函数将是动态的，如果它是静态的，则使用它作为默认值是有意义的。

我能做的是写一个before_create 回调，类似于：

def populate_value_from_sql
  self.attr_name = self.connection.select_value('SELECT dynamic_function()')
end 

但这会导致额外的 sql 查询，我想避免额外的查询，只需编译 inset 的 sql 以使用 sql 函数而不是硬编码值。 而且我想继续使用 Rails 方式，这样我的验证和回调等仍然可以正常工作，我不想采用完整的原始 sql 路由。

谢谢，

Answer 1

我还没有用触发器做过，但我发现这个answer 我也打算试试。但我不明白为什么它不起作用。我更改了答案以尝试使用%1$s 语法进行插值，因此您不必一遍又一遍地列出相同的变量。

CREATE OR REPLACE FUNCTION trigen(tbl text) RETURNS void AS $T1$
BEGIN
    EXECUTE format(
    'CREATE FUNCTION %1$s_function_name() RETURNS TRIGGER AS $T2$
    BEGIN
        UPDATE insertions SET n = n + 1 WHERE tablename = %1$s;
        RETURN NEW;
    END
    $T2$ LANGUAGE plpgsql', $1, quote_nullable($1));

    EXECUTE format('CREATE TRIGGER %1$s_inCnt BEFORE INSERT ON %s
    FOR EACH ROW EXECUTE PROCEDURE %1$s_insertCnt();', $1);
    END
$T1$ LANGUAGE plpgsql;

我不知道如何将动态名称传递给方法，但 Rails 中的示例方法可以是：

def generate_trigger(table)
  ActiveRecord::Base.connection.execute("SELECT trigen(#{table});")
end

要在你的数据库中创建触发函数，你可以创建一个迁移

def change
  reversible do |dir|
    dir.up
      execute <<-SQL
        #trigen function snippet
      SQL
    end

    dir.down do
     execute <<-SQL
       DROP FUNCTION IF EXISTS trigen() CASCADE
      SQL
    end
  end
end

为了删除触发器和触发函数本身，您可以在 Rails 代码中创建另一个方法，

def drop_trigger(trigger)
  ActiveRecord::Base.connection.base.execute("DROP TRIGGER IF EXISTS #{trigger} CASCADE;")
end

def drop_function(function)
  ActiveRecord::Base.connection.base.execute("DROP FUNCTION IF EXISTS #{function}() CASCADE;")
end