如何在ActiveRecord中为PostgreSQL实现'upsert`？

Question

我有一个数据流，其中包含类别。我想维护一个类别表，它应该包含我遇到的每个类别一次。

我想实现一个id = Category.upsert(name)，它应该是原子的（当然），并且 - 如果可能的话 - 不要在DB端使用存储过程。

Answer 1

The upsert gem seems to do just that - 我发现它时，谷歌搜索是否“upsert”是一件事。

Answer 2

如何：

class Category < ActiveRecord::Base 
    ... 

    class << self 
    def upsert(name) 
     transaction { self.find_or_create_by(name: name).id } 
    end 
    end 
end 

Answer 3

你必须写一个ActiveRecordExtension。检查this了。

该代码将是这个样子

module ActiveRecordExtension 
    extend ActiveSupport::Concern 

    def 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)