Sha256: 21ddf2f961b9d6834ccfc3081905e662ea173a2bce332aab6ab847fe0d874d7c
Contents?: true
Size: 1.32 KB
Versions: 2
Compression:
Stored size: 1.32 KB
Contents
require "dino_utils/version" require 'active_record' module Dino # Provides a simple way to do an create or update of an ActiveRecord object. module Upsert extend ActiveSupport::Concern # User.upsert module ClassMethods def upsert(*args) Dino::Upsert.upsert(self, *args) end end # klass: The ActiveRecord class we are upserting against. # conditions: what should match the upsert # options: what we are updating/inserting # If provided, the block gets the object before it's saved, in case # there's special init options necessary for it. # rubocop:disable MethodLength def self.upsert(klass, conditions, options = {}, &block) retry_count = 0 begin klass.transaction(requires_new: true) do klass.where(conditions).first_or_initialize(&block).tap { |t| t.update!(options) } end rescue PG::UniqueViolation, ActiveRecord::RecordNotUnique # If there's a unique violation, retry this. But only a certain amount # of times or we'll get into an infinite loop if something's messed up. # (like an incorrect unique index or something) if retry_count < 10 retry_count += 1 && retry else raise end end end end end ActiveRecord::Base.send(:include, Dino::Upsert)
Version data entries
2 entries across 2 versions & 1 rubygems
Version | Path |
---|---|
dino_utils-0.1.2 | lib/dino_utils.rb |
dino_utils-0.1.1 | lib/dino_upsert.rb |