debian-mirror-gitlab/lib/gitlab/experimentation/controller_concern.rb

157 lines
5.7 KiB
Ruby
Raw Normal View History

2021-01-29 00:20:46 +05:30
# frozen_string_literal: true
require 'zlib'
# Controller concern that checks if an `experimentation_subject_id cookie` is present and sets it if absent.
2021-02-22 17:27:13 +05:30
# Used for A/B testing of experimental features. Exposes the `experiment_enabled?(experiment_name, subject: nil)` method
2021-01-29 00:20:46 +05:30
# to controllers and views. It returns true when the experiment is enabled and the user is selected as part
# of the experimental group.
#
module Gitlab
module Experimentation
module ControllerConcern
include ::Gitlab::Experimentation::GroupTypes
2021-09-30 23:02:18 +05:30
include Gitlab::Tracking::Helpers
2021-01-29 00:20:46 +05:30
extend ActiveSupport::Concern
included do
before_action :set_experimentation_subject_id_cookie, unless: :dnt_enabled?
2022-01-26 12:08:38 +05:30
helper_method :experiment_enabled?, :experiment_tracking_category_and_group, :record_experiment_group
2021-01-29 00:20:46 +05:30
end
def set_experimentation_subject_id_cookie
2021-06-08 01:23:25 +05:30
if Gitlab.dev_env_or_com?
return if cookies[:experimentation_subject_id].present?
cookies.permanent.signed[:experimentation_subject_id] = {
value: SecureRandom.uuid,
secure: ::Gitlab.config.gitlab.https,
httponly: true
}
else
# We set the cookie before, although experiments are not conducted on self managed instances.
cookies.delete(:experimentation_subject_id)
end
2021-01-29 00:20:46 +05:30
end
2021-02-22 17:27:13 +05:30
def push_frontend_experiment(experiment_key, subject: nil)
2021-01-29 00:20:46 +05:30
var_name = experiment_key.to_s.camelize(:lower)
2021-02-22 17:27:13 +05:30
enabled = experiment_enabled?(experiment_key, subject: subject)
2021-01-29 00:20:46 +05:30
gon.push({ experiments: { var_name => enabled } }, true)
end
2021-02-22 17:27:13 +05:30
def experiment_enabled?(experiment_key, subject: nil)
return true if forced_enabled?(experiment_key)
2021-01-29 00:20:46 +05:30
return false if dnt_enabled?
2021-03-11 19:13:27 +05:30
Experimentation.log_invalid_rollout(experiment_key, subject)
2021-11-11 11:23:49 +05:30
subject ||= experimentation_subject_id
2021-01-29 00:20:46 +05:30
2021-02-22 17:27:13 +05:30
Experimentation.in_experiment_group?(experiment_key, subject: subject)
2021-01-29 00:20:46 +05:30
end
2021-02-22 17:27:13 +05:30
def track_experiment_event(experiment_key, action, value = nil, subject: nil)
2021-01-29 00:20:46 +05:30
return if dnt_enabled?
2021-02-22 17:27:13 +05:30
track_experiment_event_for(experiment_key, action, value, subject: subject) do |tracking_data|
2021-09-04 01:27:46 +05:30
::Gitlab::Tracking.event(tracking_data.delete(:category), tracking_data.delete(:action), **tracking_data.merge!(user: current_user))
2021-01-29 00:20:46 +05:30
end
end
2021-02-22 17:27:13 +05:30
def frontend_experimentation_tracking_data(experiment_key, action, value = nil, subject: nil)
2021-01-29 00:20:46 +05:30
return if dnt_enabled?
2021-02-22 17:27:13 +05:30
track_experiment_event_for(experiment_key, action, value, subject: subject) do |tracking_data|
2021-01-29 00:20:46 +05:30
gon.push(tracking_data: tracking_data)
end
end
2021-02-22 17:27:13 +05:30
def record_experiment_user(experiment_key, context = {})
return if dnt_enabled?
return unless Experimentation.active?(experiment_key) && current_user
2021-03-11 19:13:27 +05:30
subject = Experimentation.rollout_strategy(experiment_key) == :cookie ? nil : current_user
::Experiment.add_user(experiment_key, tracking_group(experiment_key, nil, subject: subject), current_user, context)
2021-02-22 17:27:13 +05:30
end
2021-04-17 20:07:23 +05:30
def record_experiment_group(experiment_key, group)
return if dnt_enabled?
return unless Experimentation.active?(experiment_key) && group
variant_subject = Experimentation.rollout_strategy(experiment_key) == :cookie ? nil : group
variant = tracking_group(experiment_key, nil, subject: variant_subject)
::Experiment.add_group(experiment_key, group: group, variant: variant)
end
def record_experiment_conversion_event(experiment_key, context = {})
2021-01-29 00:20:46 +05:30
return if dnt_enabled?
2021-02-22 17:27:13 +05:30
return unless current_user
return unless Experimentation.active?(experiment_key)
2021-01-29 00:20:46 +05:30
2021-04-17 20:07:23 +05:30
::Experiment.record_conversion_event(experiment_key, current_user, context)
2021-01-29 00:20:46 +05:30
end
2021-02-22 17:27:13 +05:30
def experiment_tracking_category_and_group(experiment_key, subject: nil)
"#{tracking_category(experiment_key)}:#{tracking_group(experiment_key, '_group', subject: subject)}"
2021-01-29 00:20:46 +05:30
end
private
def experimentation_subject_id
cookies.signed[:experimentation_subject_id]
end
2021-02-22 17:27:13 +05:30
def track_experiment_event_for(experiment_key, action, value, subject: nil)
return unless Experimentation.active?(experiment_key)
2021-01-29 00:20:46 +05:30
2021-02-22 17:27:13 +05:30
yield experimentation_tracking_data(experiment_key, action, value, subject: subject)
2021-01-29 00:20:46 +05:30
end
2021-02-22 17:27:13 +05:30
def experimentation_tracking_data(experiment_key, action, value, subject: nil)
2021-01-29 00:20:46 +05:30
{
category: tracking_category(experiment_key),
action: action,
2021-02-22 17:27:13 +05:30
property: tracking_group(experiment_key, "_group", subject: subject),
label: tracking_label(subject),
2021-01-29 00:20:46 +05:30
value: value
}.compact
end
def tracking_category(experiment_key)
2021-02-22 17:27:13 +05:30
Experimentation.get_experiment(experiment_key).tracking_category
2021-01-29 00:20:46 +05:30
end
2021-02-22 17:27:13 +05:30
def tracking_group(experiment_key, suffix = nil, subject: nil)
return unless Experimentation.active?(experiment_key)
2021-01-29 00:20:46 +05:30
2021-11-11 11:23:49 +05:30
subject ||= experimentation_subject_id
2021-02-22 17:27:13 +05:30
group = experiment_enabled?(experiment_key, subject: subject) ? GROUP_EXPERIMENTAL : GROUP_CONTROL
2021-01-29 00:20:46 +05:30
suffix ? "#{group}#{suffix}" : group
end
def forced_enabled?(experiment_key)
2021-03-08 18:12:59 +05:30
return true if params.has_key?(:force_experiment) && params[:force_experiment] == experiment_key.to_s
return false if cookies[:force_experiment].blank?
cookies[:force_experiment].to_s.split(',').any? { |experiment| experiment.strip == experiment_key.to_s }
2021-01-29 00:20:46 +05:30
end
2021-02-22 17:27:13 +05:30
2021-03-11 19:13:27 +05:30
def tracking_label(subject = nil)
2021-02-22 17:27:13 +05:30
return experimentation_subject_id if subject.blank?
if subject.respond_to?(:to_global_id)
Digest::MD5.hexdigest(subject.to_global_id.to_s)
else
Digest::MD5.hexdigest(subject.to_s)
end
end
2021-01-29 00:20:46 +05:30
end
end
end