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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
# frozen_string_literal: true
# PagesDeployment stores a zip archive containing GitLab Pages web-site
class PagesDeployment < ApplicationRecord
include EachBatch
include FileStoreMounter
include Gitlab::Utils::StrongMemoize
include SafelyChangeColumnDefault
columns_changing_default :upload_ready
attribute :file_store, :integer, default: -> { ::Pages::DeploymentUploader.default_store }
belongs_to :project, optional: false
# ci_build is optional, because PagesDeployment must live even if its build/pipeline is removed.
belongs_to :ci_build, class_name: 'Ci::Build', optional: true
scope :older_than, ->(id) { where('id < ?', id) }
scope :with_files_stored_locally, -> { where(file_store: ::ObjectStorage::Store::LOCAL) }
scope :with_files_stored_remotely, -> { where(file_store: ::ObjectStorage::Store::REMOTE) }
scope :project_id_in, ->(ids) { where(project_id: ids) }
scope :ci_build_id_in, ->(ids) { where(ci_build_id: ids) }
scope :with_path_prefix, ->(prefix) { where("COALESCE(path_prefix, '') = ?", prefix.to_s) }
# We have to mark the PagesDeployment upload as ready to ensure we only
# serve PagesDeployment which files are already uploaded.
scope :upload_ready, -> { where(upload_ready: true) }
scope :active, -> { upload_ready.where(deleted_at: nil).order(created_at: :desc) }
scope :deactivated, -> { where('deleted_at < ?', Time.now.utc) }
scope :versioned, -> { where.not(path_prefix: [nil, '']) }
validates :file, presence: true
validates :file_store, presence: true, inclusion: { in: ObjectStorage::SUPPORTED_STORES }
validates :size, presence: true, numericality: { greater_than: 0, only_integer: true }
validates :file_count, presence: true, numericality: { greater_than_or_equal_to: 0, only_integer: true }
validates :file_sha256, presence: true
before_validation :set_size, if: :file_changed?
mount_file_store_uploader ::Pages::DeploymentUploader
skip_callback :save, :after, :store_file!
after_commit :store_file_after_commit!, on: [:create, :update]
def self.latest_pipeline_id
Ci::Build.id_in(pluck(:ci_build_id)).maximum(:commit_id)
end
def self.deactivate_all(project)
now = Time.now.utc
active
.project_id_in(project.id)
.update_all(updated_at: now, deleted_at: now)
end
def self.deactivate_deployments_older_than(deployment, time: nil)
now = Time.now.utc
active
.older_than(deployment.id)
.project_id_in(deployment.project_id)
.with_path_prefix(deployment.path_prefix)
.update_all(updated_at: now, deleted_at: time || now)
end
def self.deactivate
update(deleted_at: Time.now.utc)
end
def url
base_url = ::Gitlab::Pages::UrlBuilder
.new(project)
.pages_url(with_unique_domain: true)
File.join(base_url.to_s, path_prefix.to_s)
end
private
def set_size
self.size = file.size
end
def store_file_after_commit!
return unless previous_changes.key?(:file)
store_file_now!
mark_upload_as_finished!
end
# We have to mark the PagesDeployment upload as ready to ensure we only
# serve PagesDeployment which files are already uploaded.
#
# This is required because we're uploading the file outside of the db transaction
# (https://gitlab.com/gitlab-org/gitlab/-/merge_requests/114774)
def mark_upload_as_finished!
return unless file && file.exists?
update_column(:upload_ready, true)
end
end
PagesDeployment.prepend_mod
|
