Sha256: 3c7ea8c416ed4c5541abf970a9fdecbba9a1a67d093f57f7f88f371fce32d291
Contents?: true
Size: 1.64 KB
Versions: 5
Compression:
Stored size: 1.64 KB
Contents
module ActiveRecord::Turntable
module ActiveRecordExt
module LockingOptimistic
::ActiveRecord::Locking::Optimistic.class_eval <<-EOD
private
# @note Override to add sharding condition on optimistic locking
def _update_record(attribute_names = self.attribute_names) #:nodoc:
return super unless locking_enabled?
return 0 if attribute_names.empty?
klass = self.class
lock_col = self.class.locking_column
previous_lock_value = send(lock_col).to_i
increment_lock
attribute_names += [lock_col]
attribute_names.uniq!
begin
relation = self.class.unscoped
condition_scope = relation.where(
self.class.primary_key => id,
lock_col => previous_lock_value,
)
if klass.turntable_enabled? and klass.primary_key != klass.turntable_shard_key.to_s
condition_scope = condition_scope.where(
klass.turntable_shard_key => self.send(klass.turntable_shard_key)
)
end
affected_rows = condition_scope.update_all(
attributes_for_update(attribute_names).map do |name|
[name, _read_attribute(name)]
end.to_h
)
unless affected_rows == 1
raise ActiveRecord::StaleObjectError.new(self, "update")
end
affected_rows
# If something went wrong, revert the version.
rescue Exception
send(lock_col + '=', previous_lock_value)
raise
end
end
EOD
end
end
end
Version data entries
5 entries across 5 versions & 1 rubygems