Add escalations table for issues

config/initializers/postgres_partitioning.rb

@@ -8,6 +8,7 @@ Gitlab::Database::Partitioning::PartitionManager.register(WebHookLog)
 
 if Gitlab.ee?
   Gitlab::Database::Partitioning::PartitionManager.register(IncidentManagement::PendingEscalations::Alert)
+  Gitlab::Database::Partitioning::PartitionManager.register(IncidentManagement::PendingEscalations::Issue)
 end
 
 begin

db/migrate/20210730194555_create_incident_management_pending_issue_escalations.rb

+# frozen_string_literal: true
+
+class CreateIncidentManagementPendingIssueEscalations < ActiveRecord::Migration[6.1]
+  include Gitlab::Database::MigrationHelpers
+
+  def up
+    with_lock_retries do
+      execute(<<~SQL)
+        CREATE TABLE incident_management_pending_issue_escalations (
+          id bigserial NOT NULL,
+          rule_id bigint  NOT NULL,
+          issue_id bigint NOT NULL,
+          process_at timestamp with time zone NOT NULL,
+          created_at timestamp with time zone NOT NULL,
+          updated_at timestamp with time zone NOT NULL,
+          PRIMARY KEY (id, process_at)
+        ) PARTITION BY RANGE (process_at);
+
+        CREATE INDEX index_incident_management_pending_issue_escalations_on_issue_id
+          ON incident_management_pending_issue_escalations USING btree (issue_id);
+
+        CREATE INDEX index_incident_management_pending_issue_escalations_on_rule_id
+          ON incident_management_pending_issue_escalations USING btree (rule_id);
+      SQL
+    end
+  end
+
+  def down
+    with_lock_retries do
+      drop_table :incident_management_pending_issue_escalations
+    end
+  end
+end

db/migrate/20210820171834_add_foreign_keys_for_pending_issue_escalations.rb

+# frozen_string_literal: true
+
+class AddForeignKeysForPendingIssueEscalations < ActiveRecord::Migration[6.1]
+  include Gitlab::Database::PartitioningMigrationHelpers
+
+  disable_ddl_transaction!
+
+  def up
+    add_concurrent_partitioned_foreign_key :incident_management_pending_issue_escalations,
+      :incident_management_escalation_rules,
+      column: :rule_id
+
+    add_concurrent_partitioned_foreign_key :incident_management_pending_issue_escalations,
+      :issues,
+      column: :issue_id
+  end
+
+  def down
+    remove_foreign_key_if_exists :incident_management_pending_issue_escalations, :incident_management_escalation_rules, column: :rule_id
+    remove_foreign_key_if_exists :incident_management_pending_issue_escalations, :issues, column: :issue_id
+  end
+end

db/schema_migrations/20210730194555

+2d0399beca58815197487d310318ed1cb3d8e85671d55581a6256ceac7667b43
\ No newline at end of file

db/schema_migrations/20210820171834

+892a71a3f6fdeb20cb2837a426d6d0931c756f8bf3d647e520a72a0bb6f78309
\ No newline at end of file

db/structure.sql

@@ -267,6 +267,16 @@ CREATE TABLE incident_management_pending_alert_escalations (
 )
 PARTITION BY RANGE (process_at);
 
+CREATE TABLE incident_management_pending_issue_escalations (
+    id bigint NOT NULL,
+    rule_id bigint NOT NULL,
+    issue_id bigint NOT NULL,
+    process_at timestamp with time zone NOT NULL,
+    created_at timestamp with time zone NOT NULL,
+    updated_at timestamp with time zone NOT NULL
+)
+PARTITION BY RANGE (process_at);
+
 CREATE TABLE web_hook_logs (
     id bigint NOT NULL,
     web_hook_id integer NOT NULL,
@@ -14184,6 +14194,15 @@ CREATE SEQUENCE incident_management_pending_alert_escalations_id_seq
 
 ALTER SEQUENCE incident_management_pending_alert_escalations_id_seq OWNED BY incident_management_pending_alert_escalations.id;
 
+CREATE SEQUENCE incident_management_pending_issue_escalations_id_seq
+    START WITH 1
+    INCREMENT BY 1
+    NO MINVALUE
+    NO MAXVALUE
+    CACHE 1;
+
+ALTER SEQUENCE incident_management_pending_issue_escalations_id_seq OWNED BY incident_management_pending_issue_escalations.id;
+
 CREATE TABLE index_statuses (
     id integer NOT NULL,
     project_id integer NOT NULL,
@@ -20566,6 +20585,8 @@ ALTER TABLE ONLY incident_management_oncall_shifts ALTER COLUMN id SET DEFAULT n
 
 ALTER TABLE ONLY incident_management_pending_alert_escalations ALTER COLUMN id SET DEFAULT nextval('incident_management_pending_alert_escalations_id_seq'::regclass);
 
+ALTER TABLE ONLY incident_management_pending_issue_escalations ALTER COLUMN id SET DEFAULT nextval('incident_management_pending_issue_escalations_id_seq'::regclass);
+
 ALTER TABLE ONLY index_statuses ALTER COLUMN id SET DEFAULT nextval('index_statuses_id_seq'::regclass);
 
 ALTER TABLE ONLY insights ALTER COLUMN id SET DEFAULT nextval('insights_id_seq'::regclass);
@@ -21999,6 +22020,9 @@ ALTER TABLE ONLY incident_management_oncall_shifts
 ALTER TABLE ONLY incident_management_pending_alert_escalations
     ADD CONSTRAINT incident_management_pending_alert_escalations_pkey PRIMARY KEY (id, process_at);
 
+ALTER TABLE ONLY incident_management_pending_issue_escalations
+    ADD CONSTRAINT incident_management_pending_issue_escalations_pkey PRIMARY KEY (id, process_at);
+
 ALTER TABLE ONLY index_statuses
     ADD CONSTRAINT index_statuses_pkey PRIMARY KEY (id);
 
@@ -24298,6 +24322,10 @@ CREATE INDEX index_incident_management_pending_alert_escalations_on_rule_id ON O
 
 CREATE INDEX index_incident_management_pending_alert_escalations_on_schedule ON ONLY incident_management_pending_alert_escalations USING btree (schedule_id);
 
+CREATE INDEX index_incident_management_pending_issue_escalations_on_issue_id ON ONLY incident_management_pending_issue_escalations USING btree (issue_id);
+
+CREATE INDEX index_incident_management_pending_issue_escalations_on_rule_id ON ONLY incident_management_pending_issue_escalations USING btree (rule_id);
+
 CREATE UNIQUE INDEX index_index_statuses_on_project_id ON index_statuses USING btree (project_id);
 
 CREATE INDEX index_insights_on_namespace_id ON insights USING btree (namespace_id);
@@ -27106,6 +27134,9 @@ ALTER TABLE ONLY incident_management_oncall_participants
 ALTER TABLE ONLY events
     ADD CONSTRAINT fk_rails_0434b48643 FOREIGN KEY (project_id) REFERENCES projects(id) ON DELETE CASCADE;
 
+ALTER TABLE incident_management_pending_issue_escalations
+    ADD CONSTRAINT fk_rails_0470889ee5 FOREIGN KEY (rule_id) REFERENCES incident_management_escalation_rules(id) ON DELETE CASCADE;
+
 ALTER TABLE ONLY ip_restrictions
     ADD CONSTRAINT fk_rails_04a93778d5 FOREIGN KEY (group_id) REFERENCES namespaces(id) ON DELETE CASCADE;
 
@@ -27712,6 +27743,9 @@ ALTER TABLE ONLY status_page_published_incidents
 ALTER TABLE ONLY deployment_clusters
     ADD CONSTRAINT fk_rails_6359a164df FOREIGN KEY (deployment_id) REFERENCES deployments(id) ON DELETE CASCADE;
 
+ALTER TABLE incident_management_pending_issue_escalations
+    ADD CONSTRAINT fk_rails_636678b3bd FOREIGN KEY (issue_id) REFERENCES issues(id) ON DELETE CASCADE;
+
 ALTER TABLE ONLY evidences
     ADD CONSTRAINT fk_rails_6388b435a6 FOREIGN KEY (release_id) REFERENCES releases(id) ON DELETE CASCADE;
 

ee/app/models/concerns/incident_management/base_pending_escalation.rb

+# frozen_string_literal: true
+
+module IncidentManagement
+  # Functionality needed for models which represent escalations.
+  #
+  # Implemeting classes should alias `target` to the attribute
+  # of the relevant escalatable.
+  #
+  # EX) `alias_attribute :target, :alert`
+  module BasePendingEscalation
+    extend ActiveSupport::Concern
+
+    include PartitionedTable
+    include EachBatch
+
+    ESCALATION_BUFFER = 1.month.freeze
+
+    included do
+      # Required to find records by id on partitioned tables.
+      self.primary_key = :id
+
+      partitioned_by :process_at, strategy: :monthly
+
+      belongs_to :rule, class_name: 'EscalationRule', foreign_key: 'rule_id'
+
+      validates :process_at, :rule_id, presence: true
+
+      scope :processable, -> { where(process_at: ESCALATION_BUFFER.ago..Time.current) }
+
+      delegate :project, to: :target
+    end
+  end
+end

ee/app/models/ee/issue.rb

@@ -70,6 +70,8 @@ module EE
       has_many :feature_flag_issues
       has_many :feature_flags, through: :feature_flag_issues, class_name: '::Operations::FeatureFlag'
 
+      has_many :pending_escalations, class_name: 'IncidentManagement::PendingEscalations::Issue', foreign_key: :issue_id, inverse_of: :issue
+
       validates :weight, allow_nil: true, numericality: { greater_than_or_equal_to: 0 }
       validate :validate_confidential_epic
 

ee/app/models/incident_management/pending_escalations/alert.rb

@@ -3,30 +3,18 @@
 module IncidentManagement
   module PendingEscalations
     class Alert < ApplicationRecord
-      include PartitionedTable
-      include EachBatch
+      include ::IncidentManagement::BasePendingEscalation
       include IgnorableColumns
 
       ignore_columns :schedule_id, :status, remove_with: '14.4', remove_after: '2021-09-22'
 
-      alias_attribute :target, :alert
-
-      self.primary_key = :id
       self.table_name = 'incident_management_pending_alert_escalations'
 
-      ESCALATION_BUFFER = 1.month.freeze
-
-      partitioned_by :process_at, strategy: :monthly
+      alias_attribute :target, :alert
 
       belongs_to :alert, class_name: 'AlertManagement::Alert', foreign_key: 'alert_id', inverse_of: :pending_escalations
-      belongs_to :rule, class_name: 'EscalationRule', foreign_key: 'rule_id'
-
-      scope :processable, -> { where(process_at: ESCALATION_BUFFER.ago..Time.current) }
-
-      validates :process_at, presence: true
-      validates :rule_id, presence: true, uniqueness: { scope: [:alert_id] }
 
-      delegate :project, to: :alert
+      validates :rule_id, uniqueness: { scope: [:alert_id] }
     end
   end
 end

ee/app/models/incident_management/pending_escalations/issue.rb

+# frozen_string_literal: true
+
+module IncidentManagement
+  module PendingEscalations
+    class Issue < ApplicationRecord
+      include ::IncidentManagement::BasePendingEscalation
+
+      self.table_name = 'incident_management_pending_issue_escalations'
+
+      alias_attribute :target, :issue
+
+      belongs_to :issue, class_name: '::Issue', foreign_key: 'issue_id', inverse_of: :pending_escalations
+
+      validates :rule_id, uniqueness: { scope: [:issue_id] }
+    end
+  end
+end

ee/spec/factories/incident_management/pending_issue_escalations.rb

+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :incident_management_pending_issue_escalation, class: 'IncidentManagement::PendingEscalations::Issue' do
+    transient do
+      project { association :project }
+      policy { association :incident_management_escalation_policy, project: project }
+    end
+
+    rule { association :incident_management_escalation_rule, policy: policy }
+    issue { association :issue, project: rule.policy.project }
+    process_at { 5.minutes.from_now }
+  end
+end

ee/spec/models/incident_management/pending_escalations/alert_spec.rb

@@ -3,34 +3,5 @@
 require 'spec_helper'
 
 RSpec.describe IncidentManagement::PendingEscalations::Alert do
-  subject { create(:incident_management_pending_alert_escalation) }
-
-  it { is_expected.to be_valid }
-
-  describe 'validations' do
-    it { is_expected.to validate_presence_of(:process_at) }
-    it { is_expected.to delegate_method(:project).to(:alert) }
-    it { is_expected.to validate_uniqueness_of(:rule_id).scoped_to([:alert_id]) }
-  end
-
-  describe 'associations' do
-    it { is_expected.to belong_to(:alert) }
-    it { is_expected.to belong_to(:rule) }
-  end
-
-  describe 'scopes' do
-    describe '.processable' do
-      subject { described_class.processable }
-
-      let_it_be(:policy) { create(:incident_management_escalation_policy) }
-      let_it_be(:rule) { policy.rules.first }
-
-      let_it_be(:two_months_ago_escalation) { create(:incident_management_pending_alert_escalation, rule: rule, process_at: 2.months.ago) }
-      let_it_be(:three_weeks_ago_escalation) { create(:incident_management_pending_alert_escalation, rule: rule, process_at: 3.weeks.ago) }
-      let_it_be(:three_days_ago_escalation) { create(:incident_management_pending_alert_escalation, rule: rule, process_at: 3.days.ago) }
-      let_it_be(:future_escalation) { create(:incident_management_pending_alert_escalation, rule: rule, process_at: 5.minutes.from_now) }
-
-      it { is_expected.to eq [three_weeks_ago_escalation, three_days_ago_escalation] }
-    end
-  end
+  include_examples 'IncidentManagement::PendingEscalation model'
 end

ee/spec/models/incident_management/pending_escalations/issue_spec.rb

+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe IncidentManagement::PendingEscalations::Issue do
+  include_examples 'IncidentManagement::PendingEscalation model'
+end

ee/spec/models/issue_spec.rb

@@ -14,6 +14,7 @@ RSpec.describe Issue do
     it { is_expected.to have_many(:resource_iteration_events) }
     it { is_expected.to have_one(:issuable_sla) }
     it { is_expected.to have_many(:metric_images) }
+    it { is_expected.to have_many(:pending_escalations) }
 
     it { is_expected.to have_one(:requirement) }
     it { is_expected.to have_many(:test_reports) }

ee/spec/support/shared_examples/models/incident_management/pending_escalations_shared_examples.rb

+# frozen_string_literal: true
+
+RSpec.shared_examples 'IncidentManagement::PendingEscalation model' do
+  let_it_be(:escalatable_type) { described_class.name.demodulize.downcase.to_sym }
+  let_it_be(:pending_escalation) { create_escalation }
+
+  subject { pending_escalation }
+
+  it { is_expected.to be_valid }
+
+  describe 'validations' do
+    it { is_expected.to validate_presence_of(:process_at) }
+    it { is_expected.to validate_presence_of(:rule_id) }
+    it { is_expected.to delegate_method(:project).to(escalatable_type) }
+    it { is_expected.to validate_uniqueness_of(:rule_id).scoped_to([:"#{escalatable_type}_id"]) }
+  end
+
+  describe 'associations' do
+    it { is_expected.to belong_to(escalatable_type) }
+    it { is_expected.to belong_to(:rule) }
+  end
+
+  describe 'scopes' do
+    describe '.processable' do
+      subject { described_class.processable }
+
+      let_it_be(:policy) { create(:incident_management_escalation_policy) }
+      let_it_be(:rule) { policy.rules.first }
+
+      let_it_be(:two_months_ago_escalation) { create_escalation(rule: rule, process_at: 2.months.ago) }
+      let_it_be(:three_weeks_ago_escalation) { create_escalation(rule: rule, process_at: 3.weeks.ago) }
+      let_it_be(:three_days_ago_escalation) { create_escalation(rule: rule, process_at: 3.days.ago) }
+      let_it_be(:future_escalation) { create_escalation(rule: rule, process_at: 5.minutes.from_now) }
+
+      it { is_expected.to eq [three_weeks_ago_escalation, three_days_ago_escalation] }
+    end
+  end
+
+  private
+
+  def create_escalation(type: escalatable_type, **options)
+    create(
+      :"incident_management_pending_#{type}_escalation",
+      **options
+    )
+  end
+end

spec/lib/gitlab/import_export/all_models.yml

@@ -58,6 +58,7 @@ issues:
 - test_reports
 - requirement
 - incident_management_issuable_escalation_status
+- pending_escalations
 work_item_type:
 - issues
 events: