debian-mirror-gitlab/app/models/concerns/atomic_internal_id.rb

73 lines
2.5 KiB
Ruby
Raw Normal View History

2018-11-20 20:47:30 +05:30
# frozen_string_literal: true
2018-05-09 12:01:36 +05:30
# Include atomic internal id generation scheme for a model
#
# This allows us to atomically generate internal ids that are
# unique within a given scope.
#
# For example, let's generate internal ids for Issue per Project:
# ```
2019-07-07 11:18:12 +05:30
# class Issue < ApplicationRecord
2018-05-09 12:01:36 +05:30
# has_internal_id :iid, scope: :project, init: ->(s) { s.project.issues.maximum(:iid) }
# end
# ```
#
# This generates unique internal ids per project for newly created issues.
# The generated internal id is saved in the `iid` attribute of `Issue`.
#
# This concern uses InternalId records to facilitate atomicity.
# In the absence of a record for the given scope, one will be created automatically.
# In this situation, the `init` block is called to calculate the initial value.
# In the example above, we calculate the maximum `iid` of all issues
# within the given project.
#
# Note that a model may have more than one internal id associated with possibly
# different scopes.
module AtomicInternalId
extend ActiveSupport::Concern
2018-11-20 20:47:30 +05:30
class_methods do
2018-11-08 19:23:39 +05:30
def has_internal_id(column, scope:, init:, presence: true) # rubocop:disable Naming/PredicateName
2018-11-18 11:00:15 +05:30
# We require init here to retain the ability to recalculate in the absence of a
# InternaLId record (we may delete records in `internal_ids` for example).
raise "has_internal_id requires a init block, none given." unless init
2018-11-08 19:23:39 +05:30
before_validation :"ensure_#{scope}_#{column}!", on: :create
validates column, presence: presence
define_method("ensure_#{scope}_#{column}!") do
2018-10-15 14:42:47 +05:30
scope_value = association(scope).reader
2018-11-18 11:00:15 +05:30
value = read_attribute(column)
return value unless scope_value
2018-11-08 19:23:39 +05:30
2018-11-18 11:00:15 +05:30
scope_attrs = { scope_value.class.table_name.singularize.to_sym => scope_value }
usage = self.class.table_name.to_sym
2018-05-09 12:01:36 +05:30
2018-11-18 11:00:15 +05:30
if value.present?
InternalId.track_greatest(self, scope_attrs, usage, value, init)
else
value = InternalId.generate_next(self, scope_attrs, usage, init)
write_attribute(column, value)
2018-05-09 12:01:36 +05:30
end
2018-11-18 11:00:15 +05:30
value
2018-11-08 19:23:39 +05:30
end
2019-07-31 22:56:46 +05:30
define_method("reset_#{scope}_#{column}") do
if value = read_attribute(column)
scope_value = association(scope).reader
scope_attrs = { scope_value.class.table_name.singularize.to_sym => scope_value }
usage = self.class.table_name.to_sym
if InternalId.reset(self, scope_attrs, usage, value)
write_attribute(column, nil)
end
end
read_attribute(column)
end
2018-05-09 12:01:36 +05:30
end
end
end