Class: Optimizely::ProjectConfig

Inherits:
Object
  • Object
show all
Defined in:
lib/optimizely/project_config.rb

Constant Summary collapse

RUNNING_EXPERIMENT_STATUS =

Representation of the Optimizely project config.

['Running'].freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(datafile, logger, error_handler) ⇒ ProjectConfig

Returns a new instance of ProjectConfig.



68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
 
# File 'lib/optimizely/project_config.rb', line 68

def initialize(datafile, logger, error_handler)
  # ProjectConfig init method to fetch and set project config data
  #
  # datafile - JSON string representing the project

  config = JSON.parse(datafile)

  @parsing_succeeded = false
  @error_handler = error_handler
  @logger = logger
  @version = config['version']

  return if UNSUPPORTED_VERSIONS.include?(@version)

  @account_id = config['accountId']
  @attributes = config.fetch('attributes', [])
  @audiences = config.fetch('audiences', [])
  @events = config.fetch('events', [])
  @experiments = config['experiments']
  @feature_flags = config.fetch('featureFlags', [])
  @groups = config.fetch('groups', [])
  @project_id = config['projectId']
  @anonymize_ip = config.key? 'anonymizeIP' ? config['anonymizeIP'] : false
  @revision = config['revision']
  @rollouts = config.fetch('rollouts', [])

  # Utility maps for quick lookup
  @attribute_key_map = generate_key_map(@attributes, 'key')
  @event_key_map = generate_key_map(@events, 'key')
  @group_key_map = generate_key_map(@groups, 'id')
  @group_key_map.each do |key, group|
    exps = group.fetch('experiments')
    exps.each do |exp|
      @experiments.push(exp.merge('groupId' => key))
    end
  end
  @experiment_key_map = generate_key_map(@experiments, 'key')
  @experiment_id_map = generate_key_map(@experiments, 'id')
  @audience_id_map = generate_key_map(@audiences, 'id')
  @variation_id_map = {}
  @variation_key_map = {}
  @forced_variation_map = {}
  @variation_id_to_variable_usage_map = {}
  @variation_id_to_experiment_map = {}
  @experiment_key_map.each_value do |exp|
    # Excludes experiments from rollouts
    variations = exp.fetch('variations')
    variations.each do |variation|
      variation_id = variation['id']
      @variation_id_to_experiment_map[variation_id] = exp
    end
  end
  @rollout_id_map = generate_key_map(@rollouts, 'id')
  # split out the experiment id map for rollouts
  @rollout_experiment_id_map = {}
  @rollout_id_map.each_value do |rollout|
    exps = rollout.fetch('experiments')
    @rollout_experiment_id_map = @rollout_experiment_id_map.merge(generate_key_map(exps, 'id'))
  end
  @all_experiments = @experiment_key_map.merge(@rollout_experiment_id_map)
  @all_experiments.each do |key, exp|
    variations = exp.fetch('variations')
    variations.each do |variation|
      variation_id = variation['id']
      variation['featureEnabled'] = variation['featureEnabled'] == true
      variation_variables = variation['variables']
      unless variation_variables.nil?
        @variation_id_to_variable_usage_map[variation_id] = generate_key_map(variation_variables, 'id')
      end
    end
    @variation_id_map[key] = generate_key_map(variations, 'id')
    @variation_key_map[key] = generate_key_map(variations, 'key')
  end
  @feature_flag_key_map = generate_key_map(@feature_flags, 'key')
  @feature_variable_key_map = {}
  @feature_flag_key_map.each do |key, feature_flag|
    @feature_variable_key_map[key] = generate_key_map(feature_flag['variables'], 'key')
  end
  @parsing_succeeded = true
end

Instance Attribute Details

#account_id ⇒ Object (readonly)

Returns the value of attribute account_id.



33
34
35
 
# File 'lib/optimizely/project_config.rb', line 33

def 
  @account_id
end

#anonymize_ip ⇒ Object (readonly)

Boolean - denotes if Optimizely should remove the last block of visitors’ IP address before storing event data



43
44
45
 
# File 'lib/optimizely/project_config.rb', line 43

def anonymize_ip
  @anonymize_ip
end

#attribute_key_map ⇒ Object (readonly)

Returns the value of attribute attribute_key_map.



48
49
50
 
# File 'lib/optimizely/project_config.rb', line 48

def attribute_key_map
  @attribute_key_map
end

#attributes ⇒ Object (readonly)

Returns the value of attribute attributes.



34
35
36
 
# File 'lib/optimizely/project_config.rb', line 34

def attributes
  @attributes
end

#audience_id_map ⇒ Object (readonly)

Returns the value of attribute audience_id_map.



49
50
51
 
# File 'lib/optimizely/project_config.rb', line 49

def audience_id_map
  @audience_id_map
end

#audiences ⇒ Object (readonly)

Returns the value of attribute audiences.



35
36
37
 
# File 'lib/optimizely/project_config.rb', line 35

def audiences
  @audiences
end

#error_handler ⇒ Object (readonly)

Gets project config attributes.



30
31
32
 
# File 'lib/optimizely/project_config.rb', line 30

def error_handler
  @error_handler
end

#event_key_map ⇒ Object (readonly)

Returns the value of attribute event_key_map.



50
51
52
 
# File 'lib/optimizely/project_config.rb', line 50

def event_key_map
  @event_key_map
end

#events ⇒ Object (readonly)

Returns the value of attribute events.



36
37
38
 
# File 'lib/optimizely/project_config.rb', line 36

def events
  @events
end

#experiment_id_map ⇒ Object (readonly)

Returns the value of attribute experiment_id_map.



51
52
53
 
# File 'lib/optimizely/project_config.rb', line 51

def experiment_id_map
  @experiment_id_map
end

#experiment_key_map ⇒ Object (readonly)

Returns the value of attribute experiment_key_map.



52
53
54
 
# File 'lib/optimizely/project_config.rb', line 52

def experiment_key_map
  @experiment_key_map
end

#experiments ⇒ Object (readonly)

Returns the value of attribute experiments.



37
38
39
 
# File 'lib/optimizely/project_config.rb', line 37

def experiments
  @experiments
end

#feature_flag_key_map ⇒ Object (readonly)

Returns the value of attribute feature_flag_key_map.



53
54
55
 
# File 'lib/optimizely/project_config.rb', line 53

def feature_flag_key_map
  @feature_flag_key_map
end

#feature_flags ⇒ Object (readonly)

Returns the value of attribute feature_flags.



38
39
40
 
# File 'lib/optimizely/project_config.rb', line 38

def feature_flags
  @feature_flags
end

#feature_variable_key_map ⇒ Object (readonly)

Returns the value of attribute feature_variable_key_map.



54
55
56
 
# File 'lib/optimizely/project_config.rb', line 54

def feature_variable_key_map
  @feature_variable_key_map
end

#forced_variation_map ⇒ Object (readonly)

Hash of user IDs to a Hash of experiments to variations. This contains all the forced variations set by the user by calling setForcedVariation (it is not the same as the whitelisting forcedVariations data structure in the Experiments class).



66
67
68
 
# File 'lib/optimizely/project_config.rb', line 66

def forced_variation_map
  @forced_variation_map
end

#group_key_map ⇒ Object (readonly)

Returns the value of attribute group_key_map.



55
56
57
 
# File 'lib/optimizely/project_config.rb', line 55

def group_key_map
  @group_key_map
end

#groups ⇒ Object (readonly)

Returns the value of attribute groups.



39
40
41
 
# File 'lib/optimizely/project_config.rb', line 39

def groups
  @groups
end

#logger ⇒ Object (readonly)

Returns the value of attribute logger.



31
32
33
 
# File 'lib/optimizely/project_config.rb', line 31

def logger
  @logger
end

#parsing_succeeded ⇒ Object (readonly)

Returns the value of attribute parsing_succeeded.



40
41
42
 
# File 'lib/optimizely/project_config.rb', line 40

def parsing_succeeded
  @parsing_succeeded
end

#project_id ⇒ Object (readonly)

Returns the value of attribute project_id.



41
42
43
 
# File 'lib/optimizely/project_config.rb', line 41

def project_id
  @project_id
end

#revision ⇒ Object (readonly)

Returns the value of attribute revision.



44
45
46
 
# File 'lib/optimizely/project_config.rb', line 44

def revision
  @revision
end

#rollout_experiment_id_map ⇒ Object (readonly)

Returns the value of attribute rollout_experiment_id_map.



57
58
59
 
# File 'lib/optimizely/project_config.rb', line 57

def rollout_experiment_id_map
  @rollout_experiment_id_map
end

#rollout_id_map ⇒ Object (readonly)

Returns the value of attribute rollout_id_map.



56
57
58
 
# File 'lib/optimizely/project_config.rb', line 56

def rollout_id_map
  @rollout_id_map
end

#rollouts ⇒ Object (readonly)

Returns the value of attribute rollouts.



45
46
47
 
# File 'lib/optimizely/project_config.rb', line 45

def rollouts
  @rollouts
end

#variation_id_map ⇒ Object (readonly)

Returns the value of attribute variation_id_map.



58
59
60
 
# File 'lib/optimizely/project_config.rb', line 58

def variation_id_map
  @variation_id_map
end

#variation_id_to_variable_usage_map ⇒ Object (readonly)

Returns the value of attribute variation_id_to_variable_usage_map.



59
60
61
 
# File 'lib/optimizely/project_config.rb', line 59

def variation_id_to_variable_usage_map
  @variation_id_to_variable_usage_map
end

#variation_key_map ⇒ Object (readonly)

Returns the value of attribute variation_key_map.



60
61
62
 
# File 'lib/optimizely/project_config.rb', line 60

def variation_key_map
  @variation_key_map
end

#version ⇒ Object (readonly)

Returns the value of attribute version.



46
47
48
 
# File 'lib/optimizely/project_config.rb', line 46

def version
  @version
end

Instance Method Details

#experiment_running?(experiment) ⇒ Boolean

Returns:

  • (Boolean) 


149
150
151
152
153
154
155
156
 
# File 'lib/optimizely/project_config.rb', line 149

def experiment_running?(experiment)
  # Determine if experiment corresponding to given key is running
  #
  # experiment - Experiment
  #
  # Returns true if experiment is running
  RUNNING_EXPERIMENT_STATUS.include?(experiment['status'])
end

#get_attribute_id(attribute_key) ⇒ Object 



365
366
367
368
369
370
371
 
# File 'lib/optimizely/project_config.rb', line 365

def get_attribute_id(attribute_key)
  attribute = @attribute_key_map[attribute_key]
  return attribute['id'] if attribute
  @logger.log Logger::ERROR, "Attribute key '#{attribute_key}' is not in datafile."
  @error_handler.handle_error InvalidAttributeError
  nil
end

#get_audience_from_id(audience_id) ⇒ Object 



200
201
202
203
204
205
206
207
208
209
210
211
212
 
# File 'lib/optimizely/project_config.rb', line 200

def get_audience_from_id(audience_id)
  # Get audience for the provided audience ID
  #
  # audience_id - ID of the audience
  #
  # Returns the audience

  audience = @audience_id_map[audience_id]
  return audience if audience
  @logger.log Logger::ERROR, "Audience '#{audience_id}' is not in datafile."
  @error_handler.handle_error InvalidAudienceError
  nil
end

#get_experiment_from_key(experiment_key) ⇒ Object 



158
159
160
161
162
163
164
165
166
167
168
169
170
 
# File 'lib/optimizely/project_config.rb', line 158

def get_experiment_from_key(experiment_key)
  # Retrieves experiment ID for a given key
  #
  # experiment_key - String key representing the experiment
  #
  # Returns Experiment or nil if not found

  experiment = @experiment_key_map[experiment_key]
  return experiment if experiment
  @logger.log Logger::ERROR, "Experiment key '#{experiment_key}' is not in datafile."
  @error_handler.handle_error InvalidExperimentError
  nil
end

#get_experiment_ids_for_event(event_key) ⇒ Object 



186
187
188
189
190
191
192
193
194
195
196
197
198
 
# File 'lib/optimizely/project_config.rb', line 186

def get_experiment_ids_for_event(event_key)
  # Get experiment IDs for the provided event key.
  #
  # event_key - Event key for which experiment IDs are to be retrieved.
  #
  # Returns array of all experiment IDs for the event.

  event = @event_key_map[event_key]
  return event['experimentIds'] if event
  @logger.log Logger::ERROR, "Event '#{event_key}' is not in datafile."
  @error_handler.handle_error InvalidEventError
  []
end

#get_experiment_key(experiment_id) ⇒ Object 



172
173
174
175
176
177
178
179
180
181
182
183
184
 
# File 'lib/optimizely/project_config.rb', line 172

def get_experiment_key(experiment_id)
  # Retrieves experiment key for a given ID.
  #
  # experiment_id - String ID representing the experiment.
  #
  # Returns String key.

  experiment = @experiment_id_map[experiment_id]
  return experiment['key'] unless experiment.nil?
  @logger.log Logger::ERROR, "Experiment id '#{experiment_id}' is not in datafile."
  @error_handler.handle_error InvalidExperimentError
  nil
end

#get_feature_flag_from_key(feature_flag_key) ⇒ Object 



401
402
403
404
405
406
407
408
409
410
411
 
# File 'lib/optimizely/project_config.rb', line 401

def get_feature_flag_from_key(feature_flag_key)
  # Retrieves the feature flag with the given key
  #
  # feature_flag_key - String feature key
  #
  # Returns feature flag if found, otherwise nil
  feature_flag = @feature_flag_key_map[feature_flag_key]
  return feature_flag if feature_flag
  @logger.log Logger::ERROR, "Feature flag key '#{feature_flag_key}' is not in datafile."
  nil
end

#get_feature_variable(feature_flag, variable_key) ⇒ Object 



413
414
415
416
417
418
419
420
421
422
423
424
425
426
 
# File 'lib/optimizely/project_config.rb', line 413

def get_feature_variable(feature_flag, variable_key)
  # Retrieves the variable with the given key for the given feature
  #
  # feature_flag - The feature flag for which we are retrieving the variable
  # variable_key - String variable key
  #
  # Returns variable if found, otherwise nil
  feature_flag_key = feature_flag['key']
  variable = @feature_variable_key_map[feature_flag_key][variable_key]
  return variable if variable
  @logger.log Logger::ERROR, "No feature variable was found for key '#{variable_key}' in feature flag "\
              "'#{feature_flag_key}'."
  nil
end

#get_forced_variation(experiment_key, user_id) ⇒ Object 



271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
 
# File 'lib/optimizely/project_config.rb', line 271

def get_forced_variation(experiment_key, user_id)
  # Gets the forced variation for the given user and experiment.
  #
  # experiment_key - String Key for experiment.
  # user_id - String ID for user
  #
  # Returns Variation The variation which the given user and experiment should be forced into.

  return nil unless Optimizely::Helpers::Validator.inputs_valid?(
    {
      experiment_key: experiment_key,
      user_id: user_id
    }, @logger, Logger::DEBUG
  )

  unless @forced_variation_map.key? user_id
    @logger.log(Logger::DEBUG, "User '#{user_id}' is not in the forced variation map.")
    return nil
  end

  experiment_to_variation_map = @forced_variation_map[user_id]
  experiment = get_experiment_from_key(experiment_key)
  experiment_id = experiment['id'] if experiment
  # check for nil and empty string experiment ID
  # this case is logged in get_experiment_from_key
  return nil if experiment_id.nil? || experiment_id.empty?

  unless experiment_to_variation_map.key? experiment_id
    @logger.log(Logger::DEBUG, "No experiment '#{experiment_key}' mapped to user '#{user_id}' "\
                'in the forced variation map.')
    return nil
  end

  variation_id = experiment_to_variation_map[experiment_id]
  variation_key = ''
  variation = get_variation_from_id(experiment_key, variation_id)
  variation_key = variation['key'] if variation

  # check if the variation exists in the datafile
  # this case is logged in get_variation_from_id
  return nil if variation_key.empty?

  @logger.log(Logger::DEBUG, "Variation '#{variation_key}' is mapped to experiment '#{experiment_key}' "\
              "and user '#{user_id}' in the forced variation map")

  variation
end

#get_rollout_from_id(rollout_id) ⇒ Object 



428
429
430
431
432
433
434
435
436
437
438
 
# File 'lib/optimizely/project_config.rb', line 428

def get_rollout_from_id(rollout_id)
  # Retrieves the rollout with the given ID
  #
  # rollout_id - String rollout ID
  #
  # Returns the rollout if found, otherwise nil
  rollout = @rollout_id_map[rollout_id]
  return rollout if rollout
  @logger.log Logger::ERROR, "Rollout with ID '#{rollout_id}' is not in the datafile."
  nil
end

#get_variation_from_id(experiment_key, variation_id) ⇒ Object 



214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
 
# File 'lib/optimizely/project_config.rb', line 214

def get_variation_from_id(experiment_key, variation_id)
  # Get variation given experiment key and variation ID
  #
  # experiment_key - Key representing parent experiment of variation
  # variation_id - ID of the variation
  #
  # Returns the variation or nil if not found

  variation_id_map = @variation_id_map[experiment_key]
  if variation_id_map
    variation = variation_id_map[variation_id]
    return variation if variation
    @logger.log Logger::ERROR, "Variation id '#{variation_id}' is not in datafile."
    @error_handler.handle_error InvalidVariationError
    return nil
  end

  @logger.log Logger::ERROR, "Experiment key '#{experiment_key}' is not in datafile."
  @error_handler.handle_error InvalidExperimentError
  nil
end

#get_variation_id_from_key(experiment_key, variation_key) ⇒ Object 



236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
 
# File 'lib/optimizely/project_config.rb', line 236

def get_variation_id_from_key(experiment_key, variation_key)
  # Get variation ID given experiment key and variation key
  #
  # experiment_key - Key representing parent experiment of variation
  # variation_key - Key of the variation
  #
  # Returns ID of the variation

  variation_key_map = @variation_key_map[experiment_key]
  if variation_key_map
    variation = variation_key_map[variation_key]
    return variation['id'] if variation
    @logger.log Logger::ERROR, "Variation key '#{variation_key}' is not in datafile."
    @error_handler.handle_error InvalidVariationError
    return nil
  end

  @logger.log Logger::ERROR, "Experiment key '#{experiment_key}' is not in datafile."
  @error_handler.handle_error InvalidExperimentError
  nil
end

#get_whitelisted_variations(experiment_key) ⇒ Object 



258
259
260
261
262
263
264
265
266
267
268
269
 
# File 'lib/optimizely/project_config.rb', line 258

def get_whitelisted_variations(experiment_key)
  # Retrieves whitelisted variations for a given experiment Key
  #
  # experiment_key - String Key representing the experiment
  #
  # Returns whitelisted variations for the experiment or nil

  experiment = @experiment_key_map[experiment_key]
  return experiment['forcedVariations'] if experiment
  @logger.log Logger::ERROR, "Experiment key '#{experiment_key}' is not in datafile."
  @error_handler.handle_error InvalidExperimentError
end

#parsing_succeeded? ⇒ Boolean

Returns:

  • (Boolean) 


373
374
375
376
377
378
379
 
# File 'lib/optimizely/project_config.rb', line 373

def parsing_succeeded?
  # Helper method to determine if parsing the datafile was successful.
  #
  # Returns Boolean depending on whether parsing the datafile succeeded or not.

  @parsing_succeeded
end

#set_forced_variation(experiment_key, user_id, variation_key) ⇒ Object 



319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
 
# File 'lib/optimizely/project_config.rb', line 319

def set_forced_variation(experiment_key, user_id, variation_key)
  # Sets a Hash of user IDs to a Hash of experiments to forced variations.
  #
  # experiment_key - String Key for experiment.
  # user_id - String ID for user.
  # variation_key - String Key for variation. If null, then clear the existing experiment-to-variation mapping.
  #
  # Returns a boolean value that indicates if the set completed successfully.

  return false unless Optimizely::Helpers::Validator.inputs_valid?(
    {
      user_id: user_id,
      experiment_key: experiment_key
    }, @logger, Logger::DEBUG
  )

  experiment = get_experiment_from_key(experiment_key)
  experiment_id = experiment['id'] if experiment
  #  check if the experiment exists in the datafile
  return false if experiment_id.nil? || experiment_id.empty?

  #  clear the forced variation if the variation key is null
  if variation_key.nil? || variation_key.empty?
    @forced_variation_map[user_id].delete(experiment_id) if @forced_variation_map.key? user_id
    @logger.log(Logger::DEBUG, "Variation mapped to experiment '#{experiment_key}' has been removed for user "\
                "'#{user_id}'.")
    return true
  end

  variation_id = get_variation_id_from_key(experiment_key, variation_key)

  #  check if the variation exists in the datafile
  unless variation_id
    #  this case is logged in get_variation_id_from_key
    return false
  end

  unless @forced_variation_map.key? user_id
    @forced_variation_map[user_id] = {}
  end
  @forced_variation_map[user_id][experiment_id] = variation_id
  @logger.log(Logger::DEBUG, "Set variation '#{variation_id}' for experiment '#{experiment_id}' and "\
              "user '#{user_id}' in the forced variation map.")
  true
end

#variation_id_exists?(experiment_id, variation_id) ⇒ Boolean

Returns:

  • (Boolean) 


381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
 
# File 'lib/optimizely/project_config.rb', line 381

def variation_id_exists?(experiment_id, variation_id)
  # Determines if a given experiment ID / variation ID pair exists in the datafile
  #
  # experiment_id - String experiment ID
  # variation_id - String variation ID
  #
  # Returns true if variation is in datafile

  experiment_key = get_experiment_key(experiment_id)
  variation_id_map = @variation_id_map[experiment_key]
  if variation_id_map
    variation = variation_id_map[variation_id]
    return true if variation
    @logger.log Logger::ERROR, "Variation ID '#{variation_id}' is not in datafile."
    @error_handler.handle_error InvalidVariationError
  end

  false
end