summaryrefslogtreecommitdiff
path: root/lib/gitlab/jira_import.rb
blob: 3f56094956aba048059a2aa805309af4b2a835c5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
# frozen_string_literal: true

module Gitlab
  module JiraImport
    JIRA_IMPORT_CACHE_TIMEOUT = 10.seconds.to_i

    FAILED_ISSUES_COUNTER_KEY = 'jira-import/failed/%{project_id}/%{collection_type}'
    NEXT_ITEMS_START_AT_KEY = 'jira-import/paginator/%{project_id}/%{collection_type}'
    JIRA_IMPORT_LABEL = 'jira-import/import-label/%{project_id}'
    ITEMS_MAPPER_CACHE_KEY = 'jira-import/items-mapper/%{project_id}/%{collection_type}/%{jira_isssue_id}'
    ALREADY_IMPORTED_ITEMS_CACHE_KEY = 'jira-importer/already-imported/%{project}/%{collection_type}'

    def self.jira_issue_cache_key(project_id, jira_issue_id)
      ITEMS_MAPPER_CACHE_KEY % { project_id: project_id, collection_type: :issues, jira_isssue_id: jira_issue_id }
    end

    def self.already_imported_cache_key(collection_type, project_id)
      ALREADY_IMPORTED_ITEMS_CACHE_KEY % { collection_type: collection_type, project: project_id }
    end

    def self.jira_issues_next_page_cache_key(project_id)
      NEXT_ITEMS_START_AT_KEY % { project_id: project_id, collection_type: :issues }
    end

    def self.failed_issues_counter_cache_key(project_id)
      FAILED_ISSUES_COUNTER_KEY % { project_id: project_id, collection_type: :issues }
    end

    def self.import_label_cache_key(project_id)
      JIRA_IMPORT_LABEL % { project_id: project_id }
    end

    def self.increment_issue_failures(project_id)
      cache_class.increment(self.failed_issues_counter_cache_key(project_id))
    end

    def self.issue_failures(project_id)
      cache_class.read(self.failed_issues_counter_cache_key(project_id)).to_i
    end

    def self.get_issues_next_start_at(project_id)
      cache_class.read(self.jira_issues_next_page_cache_key(project_id)).to_i
    end

    def self.store_issues_next_started_at(project_id, value)
      cache_key = self.jira_issues_next_page_cache_key(project_id)
      cache_class.write(cache_key, value)
    end

    def self.cache_issue_mapping(issue_id, jira_issue_id, project_id)
      cache_key = JiraImport.jira_issue_cache_key(project_id, jira_issue_id)
      cache_class.write(cache_key, issue_id)
    end

    def self.get_import_label_id(project_id)
      cache_class.read(JiraImport.import_label_cache_key(project_id))
    end

    def self.cache_import_label_id(project_id, label_id)
      cache_class.write(JiraImport.import_label_cache_key(project_id), label_id)
    end

    def self.cache_cleanup(project_id)
      cache_class.expire(self.import_label_cache_key(project_id), JIRA_IMPORT_CACHE_TIMEOUT)
      cache_class.expire(self.failed_issues_counter_cache_key(project_id), JIRA_IMPORT_CACHE_TIMEOUT)
      cache_class.expire(self.jira_issues_next_page_cache_key(project_id), JIRA_IMPORT_CACHE_TIMEOUT)
      cache_class.expire(self.already_imported_cache_key(:issues, project_id), JIRA_IMPORT_CACHE_TIMEOUT)
    end

    def self.cache_class
      Gitlab::Cache::Import::Caching
    end
  end
end