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

93 lines
3 KiB
Ruby
Raw Normal View History

2019-07-31 22:56:46 +05:30
# frozen_string_literal: true
# This module is providing helpers for updating `ProjectStatistics` with `after_save` and `before_destroy` hooks.
#
# It deals with `ProjectStatistics.increment_statistic` making sure not to update statistics on a cascade delete from the
# project, and keeping track of value deltas on each save. It updates the DB only when a change is needed.
#
2019-09-04 21:01:54 +05:30
# Example:
2019-07-31 22:56:46 +05:30
#
2019-09-04 21:01:54 +05:30
# module Ci
# class JobArtifact < ApplicationRecord
# include UpdateProjectStatistics
#
# update_project_statistics project_statistics_name: :build_artifacts_size
# end
# end
#
# Expectation:
#
# - `statistic_attribute` must be an ActiveRecord attribute
2019-07-31 22:56:46 +05:30
# - The model must implement `project` and `project_id`. i.e. direct Project relationship or delegation
module UpdateProjectStatistics
extend ActiveSupport::Concern
2019-09-30 21:07:59 +05:30
include AfterCommitQueue
2019-07-31 22:56:46 +05:30
class_methods do
2019-09-04 21:01:54 +05:30
attr_reader :project_statistics_name, :statistic_attribute
2019-07-31 22:56:46 +05:30
2019-09-04 21:01:54 +05:30
# Configure the model to update `project_statistics_name` on ProjectStatistics,
# when `statistic_attribute` changes
#
# - project_statistics_name: A column of `ProjectStatistics` to update
# - statistic_attribute: An attribute of the current model, default to `size`
def update_project_statistics(project_statistics_name:, statistic_attribute: :size)
@project_statistics_name = project_statistics_name
@statistic_attribute = statistic_attribute
2019-07-31 22:56:46 +05:30
after_save(:update_project_statistics_after_save, if: :update_project_statistics_attribute_changed?)
after_destroy(:update_project_statistics_after_destroy, unless: :project_destroyed?)
end
2019-09-04 21:01:54 +05:30
2019-07-31 22:56:46 +05:30
private :update_project_statistics
end
included do
private
def update_project_statistics_after_save
attr = self.class.statistic_attribute
delta = read_attribute(attr).to_i - attribute_before_last_save(attr).to_i
2020-01-01 13:55:28 +05:30
schedule_update_project_statistic(delta)
2019-07-31 22:56:46 +05:30
end
2019-09-04 21:01:54 +05:30
def update_project_statistics_attribute_changed?
saved_change_to_attribute?(self.class.statistic_attribute)
end
2019-07-31 22:56:46 +05:30
def update_project_statistics_after_destroy
2020-01-01 13:55:28 +05:30
delta = -read_attribute(self.class.statistic_attribute).to_i
2019-09-30 21:07:59 +05:30
2020-01-01 13:55:28 +05:30
schedule_update_project_statistic(delta)
2019-07-31 22:56:46 +05:30
end
2019-09-04 21:01:54 +05:30
def project_destroyed?
project.pending_delete?
end
2020-01-01 13:55:28 +05:30
def schedule_update_project_statistic(delta)
return if delta.zero?
if Feature.enabled?(:update_project_statistics_after_commit, default_enabled: true)
# Update ProjectStatistics after the transaction
run_after_commit do
ProjectStatistics.increment_statistic(
project_id, self.class.project_statistics_name, delta)
end
else
# Use legacy-way to update within transaction
ProjectStatistics.increment_statistic(
project_id, self.class.project_statistics_name, delta)
end
2019-09-30 21:07:59 +05:30
run_after_commit do
2019-10-12 21:52:04 +05:30
next if project.nil?
2019-09-30 21:07:59 +05:30
2020-01-01 13:55:28 +05:30
Namespaces::ScheduleAggregationWorker.perform_async(
project.namespace_id)
2019-09-30 21:07:59 +05:30
end
end
2019-07-31 22:56:46 +05:30
end
end