debian-mirror-gitlab/app/models/jira_import_state.rb
2020-05-24 23:13:21 +05:30

105 lines
2.8 KiB
Ruby

# frozen_string_literal: true
class JiraImportState < ApplicationRecord
include AfterCommitQueue
include ImportState::SidekiqJobTracker
include UsageStatistics
self.table_name = 'jira_imports'
STATUSES = { initial: 0, scheduled: 1, started: 2, failed: 3, finished: 4 }.freeze
belongs_to :project
belongs_to :user
belongs_to :label
scope :by_jira_project_key, -> (jira_project_key) { where(jira_project_key: jira_project_key) }
validates :project, presence: true
validates :jira_project_key, presence: true
validates :jira_project_name, presence: true
validates :jira_project_xid, presence: true
validates :project, uniqueness: {
conditions: -> { where.not(status: STATUSES.values_at(:failed, :finished)) },
message: _('Cannot have multiple Jira imports running at the same time')
}
alias_method :scheduled_by, :user
state_machine :status, initial: :initial do
event :schedule do
transition initial: :scheduled
end
event :start do
transition scheduled: :started
end
event :finish do
transition started: :finished
end
event :do_fail do
transition [:initial, :scheduled, :started] => :failed
end
after_transition initial: :scheduled do |state, _|
state.run_after_commit do
job_id = Gitlab::JiraImport::Stage::StartImportWorker.perform_async(project.id)
state.update(jid: job_id, scheduled_at: Time.now) if job_id
end
end
before_transition any => :finished do |state, _|
InternalId.flush_records!(project: state.project)
state.project.update_project_counter_caches
state.store_issue_counts
end
after_transition any => :finished do |state, _|
if state.jid.present?
Gitlab::SidekiqStatus.unset(state.jid)
state.update_column(:jid, nil)
end
end
# Supress warning:
# both JiraImportState and its :status machine have defined a different default for "status".
# although both have same value but represented in 2 ways: integer(0) and symbol(:initial)
def owner_class_attribute_default
'initial'
end
end
enum status: STATUSES
def in_progress?
scheduled? || started?
end
def non_initial?
!initial?
end
def store_issue_counts
import_label_id = Gitlab::JiraImport.get_import_label_id(project.id)
failed_to_import_count = Gitlab::JiraImport.issue_failures(project.id)
successfully_imported_count = project.issues.with_label_ids(import_label_id).count
total_issue_count = successfully_imported_count + failed_to_import_count
update(
{
failed_to_import_count: failed_to_import_count,
imported_issues_count: successfully_imported_count,
total_issue_count: total_issue_count
}
)
end
def self.finished_imports_count
finished.sum(:imported_issues_count)
end
end