debian-mirror-gitlab/app/models/environment.rb

262 lines
7.1 KiB
Ruby
Raw Normal View History

2018-11-18 11:00:15 +05:30
# frozen_string_literal: true
class Environment < ActiveRecord::Base
2019-02-15 15:39:39 +05:30
include Gitlab::Utils::StrongMemoize
2017-08-17 22:00:37 +05:30
# Used to generate random suffixes for the slug
LETTERS = 'a'..'z'
NUMBERS = '0'..'9'
SUFFIX_CHARS = LETTERS.to_a + NUMBERS.to_a
2018-03-27 19:54:05 +05:30
belongs_to :project, required: true
2018-12-13 13:39:08 +05:30
has_many :deployments, -> { success }, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent
2018-03-17 18:26:18 +05:30
2018-12-13 13:39:08 +05:30
has_one :last_deployment, -> { success.order('deployments.id DESC') }, class_name: 'Deployment'
2016-09-13 17:45:13 +05:30
before_validation :nullify_external_url
2017-08-17 22:00:37 +05:30
before_validation :generate_slug, if: ->(env) { env.slug.blank? }
2016-09-29 09:46:39 +05:30
before_save :set_environment_type
2016-09-13 17:45:13 +05:30
validates :name,
presence: true,
uniqueness: { scope: :project_id },
2017-08-17 22:00:37 +05:30
length: { maximum: 255 },
format: { with: Gitlab::Regex.environment_name_regex,
message: Gitlab::Regex.environment_name_regex_message }
2017-08-17 22:00:37 +05:30
validates :slug,
presence: true,
uniqueness: { scope: :project_id },
length: { maximum: 24 },
format: { with: Gitlab::Regex.environment_slug_regex,
message: Gitlab::Regex.environment_slug_regex_message }
2016-09-13 17:45:13 +05:30
validates :external_url,
length: { maximum: 255 },
allow_nil: true,
2018-11-08 19:23:39 +05:30
url: true
2016-09-13 17:45:13 +05:30
2017-08-17 22:00:37 +05:30
delegate :stop_action, :manual_actions, to: :last_deployment, allow_nil: true
2016-11-03 12:29:30 +05:30
scope :available, -> { with_state(:available) }
scope :stopped, -> { with_state(:stopped) }
2017-08-17 22:00:37 +05:30
scope :order_by_last_deployed_at, -> do
max_deployment_id_sql =
2017-09-10 17:25:29 +05:30
Deployment.select(Deployment.arel_table[:id].maximum)
.where(Deployment.arel_table[:environment_id].eq(arel_table[:id]))
.to_sql
2017-08-17 22:00:37 +05:30
order(Gitlab::Database.nulls_first_order("(#{max_deployment_id_sql})", 'ASC'))
end
2017-09-10 17:25:29 +05:30
scope :in_review_folder, -> { where(environment_type: "review") }
2018-12-13 13:39:08 +05:30
scope :for_name, -> (name) { where(name: name) }
2019-03-02 22:35:43 +05:30
##
# Search environments which have names like the given query.
# Do not set a large limit unless you've confirmed that it works on gitlab.com scale.
scope :for_name_like, -> (query, limit: 5) do
where('name LIKE ?', "#{sanitize_sql_like(query)}%").limit(limit)
end
2018-12-13 13:39:08 +05:30
scope :for_project, -> (project) { where(project_id: project) }
scope :with_deployment, -> (sha) { where('EXISTS (?)', Deployment.select(1).where('deployments.environment_id = environments.id').where(sha: sha)) }
2016-11-03 12:29:30 +05:30
state_machine :state, initial: :available do
event :start do
transition stopped: :available
end
event :stop do
transition available: :stopped
end
state :available
state :stopped
2017-09-10 17:25:29 +05:30
after_transition do |environment|
environment.expire_etag_cache
end
2016-11-03 12:29:30 +05:30
end
2019-03-02 22:35:43 +05:30
def self.pluck_names
pluck(:name)
end
2017-08-17 22:00:37 +05:30
def predefined_variables
2018-05-09 12:01:36 +05:30
Gitlab::Ci::Variables::Collection.new
.append(key: 'CI_ENVIRONMENT_NAME', value: name)
.append(key: 'CI_ENVIRONMENT_SLUG', value: slug)
2017-08-17 22:00:37 +05:30
end
def recently_updated_on_branch?(ref)
ref.to_s == last_deployment.try(:ref)
end
2016-09-13 17:45:13 +05:30
def nullify_external_url
self.external_url = nil if self.external_url.blank?
end
2016-09-29 09:46:39 +05:30
def set_environment_type
names = name.split('/')
2017-09-10 17:25:29 +05:30
self.environment_type = names.many? ? names.first : nil
2016-09-29 09:46:39 +05:30
end
2016-09-13 17:45:13 +05:30
def includes_commit?(commit)
return false unless last_deployment
last_deployment.includes_commit?(commit)
end
2016-09-29 09:46:39 +05:30
2017-08-17 22:00:37 +05:30
def last_deployed_at
last_deployment.try(:created_at)
end
2016-09-29 09:46:39 +05:30
def update_merge_request_metrics?
2017-09-10 17:25:29 +05:30
folder_name == "production"
2016-09-29 09:46:39 +05:30
end
2016-11-03 12:29:30 +05:30
2018-03-27 19:54:05 +05:30
def first_deployment_for(commit_sha)
ref = project.repository.ref_name_for_sha(ref_path, commit_sha)
2016-11-03 12:29:30 +05:30
return nil unless ref
deployment_iid = ref.split('/').last
deployments.find_by(iid: deployment_iid)
end
def ref_path
2018-03-17 18:26:18 +05:30
"refs/#{Repository::REF_ENVIRONMENTS}/#{slug}"
2016-11-03 12:29:30 +05:30
end
def formatted_external_url
return nil unless external_url
2018-03-17 18:26:18 +05:30
external_url.gsub(%r{\A.*?://}, '')
2016-11-03 12:29:30 +05:30
end
2018-11-18 11:00:15 +05:30
def stop_action_available?
2016-11-03 12:29:30 +05:30
available? && stop_action.present?
end
2017-08-17 22:00:37 +05:30
def stop_with_action!(current_user)
return unless available?
stop!
stop_action&.play(current_user)
end
def actions_for(environment)
return [] unless manual_actions
manual_actions.select do |action|
action.expanded_environment_name == environment
end
end
def has_terminals?
2018-03-17 18:26:18 +05:30
project.deployment_platform.present? && available? && last_deployment.present?
2017-08-17 22:00:37 +05:30
end
def terminals
2018-03-17 18:26:18 +05:30
project.deployment_platform.terminals(self) if has_terminals?
2017-08-17 22:00:37 +05:30
end
def has_metrics?
2018-12-13 13:39:08 +05:30
prometheus_adapter&.can_query? && available?
2017-08-17 22:00:37 +05:30
end
def metrics
2018-03-27 19:54:05 +05:30
prometheus_adapter.query(:environment, self) if has_metrics?
2017-09-10 17:25:29 +05:30
end
2018-03-27 19:54:05 +05:30
def additional_metrics
prometheus_adapter.query(:additional_metrics_environment, self) if has_metrics?
2017-09-10 17:25:29 +05:30
end
2018-12-05 23:21:45 +05:30
# rubocop: disable CodeReuse/ServiceClass
2018-03-27 19:54:05 +05:30
def prometheus_adapter
@prometheus_adapter ||= Prometheus::AdapterService.new(project, deployment_platform).prometheus_adapter
2017-08-17 22:00:37 +05:30
end
2018-12-05 23:21:45 +05:30
# rubocop: enable CodeReuse/ServiceClass
2017-08-17 22:00:37 +05:30
2018-03-17 18:26:18 +05:30
def slug
super.presence || generate_slug
end
2017-08-17 22:00:37 +05:30
# An environment name is not necessarily suitable for use in URLs, DNS
# or other third-party contexts, so provide a slugified version. A slug has
# the following properties:
# * contains only lowercase letters (a-z), numbers (0-9), and '-'
# * begins with a letter
# * has a maximum length of 24 bytes (OpenShift limitation)
# * cannot end with `-`
def generate_slug
# Lowercase letters and numbers only
2018-11-18 11:00:15 +05:30
slugified = +name.to_s.downcase.gsub(/[^a-z0-9]/, '-')
2017-08-17 22:00:37 +05:30
# Must start with a letter
slugified = 'env-' + slugified unless LETTERS.cover?(slugified[0])
# Repeated dashes are invalid (OpenShift limitation)
slugified.gsub!(/\-+/, '-')
# Maximum length: 24 characters (OpenShift limitation)
slugified = slugified[0..23]
# Cannot end with a dash (Kubernetes label limitation)
slugified.chop! if slugified.end_with?('-')
# Add a random suffix, shortening the current string if necessary, if it
# has been slugified. This ensures uniqueness.
if slugified != name
slugified = slugified[0..16]
slugified << '-' unless slugified.end_with?('-')
slugified << random_suffix
end
self.slug = slugified
end
def external_url_for(path, commit_sha)
return unless self.external_url
public_path = project.public_path_for_source_path(path, commit_sha)
return unless public_path
[external_url, public_path].join('/')
end
2017-09-10 17:25:29 +05:30
def expire_etag_cache
Gitlab::EtagCaching::Store.new.tap do |store|
store.touch(etag_cache_key)
end
end
def etag_cache_key
Gitlab::Routing.url_helpers.project_environments_path(
project,
format: :json)
end
def folder_name
self.environment_type || self.name
end
2018-03-27 19:54:05 +05:30
def deployment_platform
2019-02-15 15:39:39 +05:30
strong_memoize(:deployment_platform) do
project.deployment_platform(environment: self.name)
end
2018-03-27 19:54:05 +05:30
end
2017-08-17 22:00:37 +05:30
private
2016-11-03 12:29:30 +05:30
2017-08-17 22:00:37 +05:30
# Slugifying a name may remove the uniqueness guarantee afforded by it being
# based on name (which must be unique). To compensate, we add a random
# 6-byte suffix in those circumstances. This is not *guaranteed* uniqueness,
# but the chance of collisions is vanishingly small
def random_suffix
(0..5).map { SUFFIX_CHARS.sample }.join
2016-11-03 12:29:30 +05:30
end
end