在 Rails ActiveRecord 中更新插入

Question

ActiveRecord 是否有内置的更新插入功能？我知道我可以自己写它，但如果这样的东西已经存在，我显然不想这样做。

Answer 1

Rails 6 中有一个很棒的新功能：他们向 ActiveRecord 添加了 upsert 和 upsert_all

可以在此处找到更多信息 https://edgeapi.rubyonrails.org/classes/ActiveRecord/Persistence/ClassMethods.html #method-i-upsert_all

Answer 2

Model.find_or_initialize 可能会满足您的需求。如果有意义的话，您可以将其与 save 或 update_attributes 链接起来。

更多信息请参见 Rails 指南。

Answer 3

我刚刚跑过这个图书馆：
https://github.com/seamusabshere/upsert

我还没有测试过，但看起来很有希望

Answer 4

Rails 6 针对这种情况引入了 create_or_find_by https://github.com/rails/rails/pull/31989

此外，对于大量记录，可以使用 https://github.com/zdennis/ activerecord-import

示例：

Book.import [book], on_duplicate_key_update: [:title]

Answer 5

IMO Upsert 机制需要为每个模型进行自定义配置。

因此，最好的解决方案是为模型实现自定义 SQL 查询，例如

insert into <table> (<field_1>, ..., <field_n>) 
  values (<field_1_value>, ..., <field_n_value>)
on duplicate key update
  field_x = field_x_value,
  ...
  field_z = field_z_value;

Answer 6

还有Model.find_or_create

Answer 7

从 Rails 6 开始，它有 upsert 方法，文档：

1. https://apidock.com/rails/v6.0.0/ActiveRecord/Persistence/ClassMethods/upsert
2. Rails 中的 upsert 函数如何工作？

用法：

Book table
---
id: integer
author_name: string
title: string

Usage : ( into rails c )

> Book.all   #=> []
> Book.upsert({ id: 1, author_name: 'Liu', title: 'Tripple body 1'})
> Book.upsert({ id: 1, author_name: 'Liu', title: 'Tripple body 1'})
> Book.upsert({ id: 1, author_name: 'Liu', title: 'Tripple body 1'})
> Book.all  # => only 1 book, with tile: 'Tripple body 1'

，您可能会看到原始 SQL 如下所示： （在 Postgres 14.2 中）

INSERT INTO "books" ("author_name","title","created_at","updated_at") 
  VALUES ('Liu', 'Tripple body 1', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP) 
  ON CONFLICT ("id") 
  DO UPDATE SET updated_at=(
      CASE WHEN (
        "books"."author_name" IS NOT DISTINCT 
          FROM excluded."author_name" 
        AND "books"."title" IS NOT DISTINCT 
          FROM excluded."title"
      ) 
      THEN "books".updated_at ELSE CURRENT_TIMESTAMP END
    ),
    "author_name"=excluded."author_name",
    "title"=excluded."title" 
  RETURNING "id"

因此请注意：

1. upsert 将从哈希参数确定唯一性，因此请确保其中包含包含在表的唯一列中的键列（例如 id 列）
2. 如果您没有在参数中传递 id 或类似的唯一列，它总是将新记录插入表
3. upsert 将跳过模型回调并且验证。

顺便说一句，我讨厌这种方法，我更喜欢 find_or_create_by 或只是：

unless Book.exists?("...")
  Book.create ...
end

这更清晰。

Answer 8

我写了一篇关于我们如何实现这一目标的博客文章。请查看此处。

您必须编写一个活动记录扩展。它看起来像这样。

module ActiveRecordExtension
  extend ActiveSupport::Concern

  def self.upsert(attributes)
    begin
        create(attributes)
    rescue ActiveRecord::RecordNotUnique, PG::UniqueViolation => e
        find_by_primary_key(attributes['primary_key']).
        update(attributes)
    end
  end
end

ActiveRecord::Base.send(:include, ActiveRecordExtension)