Add latest changes from gitlab-org/gitlab@master

1 files changed, 53 insertions, 0 deletions

diff --git a/lib/gitlab/alert_management/alert_status_counts.rb b/lib/gitlab/alert_management/alert_status_counts.rb
new file mode 100644
index 00000000000..382026236e0
--- /dev/null
+++ b/lib/gitlab/alert_management/alert_status_counts.rb
@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module AlertManagement
+    # Represents counts of each status or category of statuses
+    class AlertStatusCounts
+      include Gitlab::Utils::StrongMemoize
+
+      STATUSES = ::AlertManagement::Alert::STATUSES
+
+      attr_reader :project
+
+      def self.declarative_policy_class
+        'AlertManagement::AlertPolicy'
+      end
+
+      def initialize(current_user, project, params)
+        @project = project
+        @current_user = current_user
+        @params = params
+      end
+
+      # Define method for each status
+      STATUSES.each_key do |status|
+        define_method(status) { counts[status] }
+      end
+
+      def open
+        counts[:triggered] + counts[:acknowledged]
+      end
+
+      def all
+        counts.values.sum # rubocop:disable CodeReuse/ActiveRecord
+      end
+
+      private
+
+      attr_reader :current_user, :params
+
+      def counts
+        strong_memoize(:counts) do
+          Hash.new(0).merge(counts_by_status)
+        end
+      end
+
+      def counts_by_status
+        ::AlertManagement::AlertsFinder
+          .counts_by_status(current_user, project, params)
+          .transform_keys { |status_id| STATUSES.key(status_id) }
+      end
+    end
+  end
+end