lib/gitlab/experimentation/experiment.rb


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

# frozen_string_literal: true

module Gitlab
  module Experimentation
    class Experiment
      FEATURE_FLAG_SUFFIX = "_experiment_percentage"

      attr_reader :key, :tracking_category, :use_backwards_compatible_subject_index, :rollout_strategy

      def initialize(key, **params)
        @key = key
        @tracking_category = params[:tracking_category]
        @use_backwards_compatible_subject_index = params[:use_backwards_compatible_subject_index]
        @rollout_strategy = params[:rollout_strategy] || :cookie
      end

      def active?
        # TODO: just touch a feature flag
        # Temporary change, we will change `experiment_percentage` in future to `Feature.enabled?
        Feature.enabled?(feature_flag_name, type: :experiment, default_enabled: :yaml)

        ::Gitlab.dev_env_or_com? && experiment_percentage > 0
      end

      def enabled_for_index?(index)
        return false if index.blank?

        index <= experiment_percentage
      end

      private

      def experiment_percentage
        feature_flag.percentage_of_time_value
      end

      def feature_flag
        Feature.get(feature_flag_name) # rubocop:disable Gitlab/AvoidFeatureGet
      end

      def feature_flag_name
        :"#{key}#{FEATURE_FLAG_SUFFIX}"
      end
    end
  end
end