job_artifact.rb 2.75 KB
Newer Older
1
2
# frozen_string_literal: true

3
4
module Ci
  class JobArtifact < ActiveRecord::Base
5
    include AfterCommitQueue
6
    include ObjectStorage::BackgroundMove
7
8
    extend Gitlab::Ci::Model

9
10
11
12
    TEST_REPORT_FILE_TYPES = %w[junit].freeze
    DEFAULT_FILE_NAMES = { junit: 'junit.xml' }.freeze
    TYPE_AND_FORMAT_PAIRS = { archive: :zip, metadata: :gzip, trace: :raw, junit: :gzip }.freeze

13
    belongs_to :project
Zeger-Jan van de Weg's avatar
Zeger-Jan van de Weg committed
14
    belongs_to :job, class_name: "Ci::Build", foreign_key: :job_id
15

16
17
    mount_uploader :file, JobArtifactUploader

18
19
    validates :file_format, presence: true, unless: :trace?, on: :create
    validate :valid_file_format?, unless: :trace?, on: :create
20
    before_save :set_size, if: :file_changed?
21
22
    after_save :update_project_statistics_after_save, if: :size_changed?
    after_destroy :update_project_statistics_after_destroy, unless: :project_destroyed?
23

24
    after_save :update_file_store, if: :file_changed?
25

26
    scope :with_files_stored_locally, -> { where(file_store: [nil, ::JobArtifactUploader::Store::LOCAL]) }
27

28
29
30
31
32
33
    scope :test_reports, -> do
      types = self.file_types.select { |file_type| TEST_REPORT_FILE_TYPES.include?(file_type) }.values

      where(file_type: types)
    end

34
    delegate :exists?, :open, to: :file
35

36
37
    enum file_type: {
      archive: 1,
38
      metadata: 2,
39
40
      trace: 3,
      junit: 4
41
42
    }

43
44
45
46
47
48
49
50
51
52
53
54
    enum file_format: {
      raw: 1,
      zip: 2,
      gzip: 3
    }

    def valid_file_format?
      unless TYPE_AND_FORMAT_PAIRS[self.file_type&.to_sym] == self.file_format&.to_sym
        errors.add(:file_format, 'Invalid file format with specified file type')
      end
    end

55
    def update_file_store
56
57
58
      # The file.object_store is set during `uploader.store!`
      # which happens after object is inserted/updated
      self.update_column(:file_store, file.object_store)
59
60
    end

61
62
63
64
    def self.artifacts_size_for(project)
      self.where(project: project).sum(:size)
    end

65
66
67
68
    def local_store?
      [nil, ::JobArtifactUploader::Store::LOCAL].include?(self.file_store)
    end

69
70
71
72
73
74
75
76
77
78
    def expire_in
      expire_at - Time.now if expire_at
    end

    def expire_in=(value)
      self.expire_at =
        if value
          ChronicDuration.parse(value)&.seconds&.from_now
        end
    end
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

    private

    def set_size
      self.size = file.size
    end

    def update_project_statistics_after_save
      update_project_statistics(size.to_i - size_was.to_i)
    end

    def update_project_statistics_after_destroy
      update_project_statistics(-self.size)
    end

    def update_project_statistics(difference)
      ProjectStatistics.increment_statistic(project_id, :build_artifacts_size, difference)
    end

    def project_destroyed?
      # Use job.project to avoid extra DB query for project
      job.project.pending_delete?
    end
102
103
  end
end