1 files changed, 27 insertions, 0 deletions
diff --git a/app/models/experiment.rb b/app/models/experiment.rb
new file mode 100644
index 00000000000..25640385536
--- /dev/null
+++ b/app/models/experiment.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+class Experiment < ApplicationRecord
+  has_many :experiment_users
+  has_many :users, through: :experiment_users
+  has_many :control_group_users, -> { merge(ExperimentUser.control) }, through: :experiment_users, source: :user
+  has_many :experimental_group_users, -> { merge(ExperimentUser.experimental) }, through: :experiment_users, source: :user
+
+  validates :name, presence: true, uniqueness: true, length: { maximum: 255 }
+
+  def self.add_user(name, group_type, user)
+    experiment = find_or_create_by(name: name)
+
+    return unless experiment
+    return if experiment.experiment_users.where(user: user).exists?
+
+    group_type == ::Gitlab::Experimentation::GROUP_CONTROL ? experiment.add_control_user(user) : experiment.add_experimental_user(user)
+  end
+
+  def add_control_user(user)
+    control_group_users << user
+  end
+
+  def add_experimental_user(user)
+    experimental_group_users << user
+  end
+end