summaryrefslogtreecommitdiff
path: root/app/models/experiment.rb
blob: ac8b6516d02b1515bf9ade29bf8ff3c951f45445 (plain)
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
# frozen_string_literal: true

class Experiment < ApplicationRecord
  has_many :experiment_users
  has_many :experiment_subjects, inverse_of: :experiment

  validates :name, presence: true, uniqueness: true, length: { maximum: 255 }

  def self.add_user(name, group_type, user, context = {})
    find_or_create_by!(name: name).record_user_and_group(user, group_type, context)
  end

  def self.add_group(name, variant:, group:)
    find_or_create_by!(name: name).record_group_and_variant!(group, variant)
  end

  def self.record_conversion_event(name, user, context = {})
    find_or_create_by!(name: name).record_conversion_event_for_user(user, context)
  end

  # Create or update the recorded experiment_user row for the user in this experiment.
  def record_user_and_group(user, group_type, context = {})
    experiment_user = experiment_users.find_or_initialize_by(user: user)
    experiment_user.update!(group_type: group_type, context: merged_context(experiment_user, context))
  end

  def record_conversion_event_for_user(user, context = {})
    experiment_user = experiment_users.find_by(user: user)
    return unless experiment_user

    experiment_user.update!(converted_at: Time.current, context: merged_context(experiment_user, context))
  end

  def record_group_and_variant!(group, variant)
    experiment_subjects.find_or_initialize_by(group: group).update!(variant: variant)
  end

  private

  def merged_context(experiment_user, new_context)
    experiment_user.context.deep_merge(new_context.deep_stringify_keys)
  end
end