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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
# frozen_string_literal: true
# PagesDeployment stores a zip archive containing GitLab Pages web-site
class PagesDeployment < ApplicationRecord
include GlobalID::Identification
include EachBatch
include FromUnion
include Sortable
include FileStoreMounter
include Gitlab::Utils::StrongMemoize
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) }
scope :expired, -> { where('expires_at < ?', Time.now.utc).order(:expires_at, :id) }
scope :deactivated, -> { where('deleted_at < ?', Time.now.utc) }
scope :versioned, -> { where.not(path_prefix: [nil, '']) }
scope :unversioned, -> { where(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.declarative_policy_class
'Pages::DeploymentPolicy'
end
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 self.count_versioned_deployments_for(project_or_namespace, limit)
namespace = project_or_namespace.try(:root_ancestor) || project_or_namespace
project_id_in(namespace.all_projects)
.active
.versioned
.limit(limit)
.count
end
def active?
deleted_at.nil?
end
def url
base_url = ::Gitlab::Pages::UrlBuilder
.new(project)
.pages_url
File.join(base_url.to_s, path_prefix.to_s)
end
def deactivate
update_attribute(:deleted_at, Time.now.utc)
end
def restore
update_attribute(:deleted_at, nil)
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
|
