A Ruby library for carefully refactoring critical paths.
Let's pretend you're changing the way you handle permissions in a large web app. Tests can help guide your refactoring, but you really want to compare the current and refactored behaviors under load.
require "scientist"
class MyWidget
def allows?(user)
experiment = Scientist::Default.new "widget-permissions"
experiment.use { model.check_user?(user).valid? } # old way
experiment.try { user.can?(:read, model) } # new way
experiment.run
end
end
Wrap a use
block around the code's original behavior, and wrap try
around the new behavior. experiment.run
will always return whatever the use
block returns, but it does a bunch of stuff behind the scenes:
- It decides whether or not to run the
try
block, - Randomizes the order in which
use
andtry
blocks are run, - Measures the durations of all behaviors,
- Compares the result of
try
to the result ofuse
, - Swallows (but records) any exceptions raised in the
try
block, and - Publishes all this information.
The use
block is called the control. The try
block is called the candidate.
Creating an experiment is wordy, but when you include the Scientist
module, the science
helper will instantiate an experiment and call run
for you:
require "scientist"
class MyWidget
include Scientist
def allows?(user)
science "widget-permissions" do |e|
e.use { model.check_user(user).valid? } # old way
e.try { user.can?(:read, model) } # new way
end # returns the control value
end
end
If you don't declare any try
blocks, none of the Scientist machinery is invoked and the control value is always returned.
The examples above will run, but they're not really doing anything. The try
blocks run every time and none of the results get published. Replace the default experiment implementation to control execution and reporting:
require "scientist"
class MyExperiment
include ActiveModel::Model
include Scientist::Experiment
attr_accessor :name
def enabled?
# see "Ramping up experiments" below
super
end
def publish(result)
# see "Publishing results" below
super
end
end
# replace `Scientist::Default` as the default implementation
module Scientist::Experiment
def self.new(name)
MyExperiment.new(name: name)
end
end
Now calls to the science
helper will load instances of MyExperiment
.
Scientist compares control and candidate values using ==
. To override this behavior, use compare
to define how to compare observed values instead:
class MyWidget
include Scientist
def users
science "users" do |e|
e.use { User.all } # returns User instances
e.try { UserService.list } # returns UserService::User instances
e.compare do |control, candidate|
control.map(&:login) == candidate.map(&:login)
end
end
end
end
Results aren't very useful without some way to identify them. Use the context
method to add to or retrieve the context for an experiment:
science "widget-permissions" do |e|
e.context :user => user
e.use { model.check_user(user).valid? }
e.try { user.can?(:read, model) }
end
context
takes a Symbol-keyed Hash of extra data. The data is available in Experiment#publish
via the context
method. If you're using the science
helper a lot in a class, you can provide a default context:
class MyWidget
include Scientist
def allows?(user)
science "widget-permissions" do |e|
e.context :user => user
e.use { model.check_user(user).valid? }
e.try { user.can?(:read, model) }
end
end
def destroy
science "widget-destruction" do |e|
e.use { old_scary_destroy }
e.try { new_safe_destroy }
end
end
def default_scientist_context
{ :widget => self }
end
end
The widget-permissions
and widget-destruction
experiments will both have a :widget
key in their contexts.
If an experiment requires expensive setup that should only occur when the experiment is going to be run, define it with the before_run
method:
# Code under test modifies this in-place. We want to copy it for the
# candidate code, but only when needed:
value_for_original_code = big_object
value_for_new_code = nil
science "expensive-but-worthwhile" do |e|
e.before_run do
value_for_new_code = big_object.deep_copy
end
e.use { original_code(value_for_original_code) }
e.try { new_code(value_for_new_code) }
end
Sometimes you don't want to store the full value for later analysis. For example, an experiment may return User
instances, but when researching a mismatch, all you care about is the logins. You can define how to clean these values in an experiment:
class MyWidget
include Scientist
def users
science "users" do |e|
e.use { User.all }
e.try { UserService.list }
e.clean do |value|
value.map(&:login).sort
end
end
end
end
And this cleaned value is available in observations in the final published result:
class MyExperiment < ActiveRecord::Base
include Scientist::Experiment
def publish(result)
result.control.value # [<User alice>, <User bob>, <User carol>]
result.control.cleaned_value # ["alice", "bob", "carol"]
end
end
During the early stages of an experiment, it's possible that some of your code will always generate a mismatch for reasons you know and understand but haven't yet fixed. Instead of these known cases always showing up as mismatches in your metrics or analysis, you can tell an experiment whether or not to ignore a mismatch using the ignore
method. You may include more than one block if needed:
def admin?(user)
science "widget-permissions" do |e|
e.use { model.check_user(user).admin? }
e.try { user.can?(:admin, model) }
e.ignore { user.staff? } # user is staff, always an admin in the new system
e.ignore do |control, candidate|
# new system doesn't handle unconfirmed users yet:
control && !candidate && !user.confirmed_email?
end
end
end
The ignore blocks are only called if the values don't match. If one observation raises an exception and the other doesn't, it's always considered a mismatch. If both observations raise different exceptions, that is also considered a mismatch.
Sometimes you don't want an experiment to run. Say, disabling a new codepath for anyone who isn't staff. You can disable an experiment by setting a run_if
block. If this returns false
, the experiment will merely return the control value. Otherwise, it defers to the experiment's configured enabled?
method.
class DashboardController
include Scientist
def dashboard_items
science "dashboard-items" do |e|
# only run this experiment for staff members
e.run_if { current_user.staff? }
# ...
end
end
As a scientist, you know it's always important to be able to turn your experiment off, lest it run amok and result in villagers with pitchforks on your doorstep. In order to control whether or not an experiment is enabled, you must include the enabled?
method in your Scientist::Experiment
implementation.
class MyExperiment < ActiveRecord::Base
include Scientist::Experiment
def enabled?
percent_enabled > 0 && rand(100) < percent_enabled
end
end
This code will be invoked for every method with an experiment every time, so be sensitive about its performance. For example, you can store an experiment in the database but wrap it in various levels of caching such as memcache or per-request thread-locals.
What good is science if you can't publish your results?
You must implement the publish(result)
method, and can publish data however you like. For example, timing data can be sent to graphite, and mismatches can be placed in a capped collection in redis for debugging later.
The publish
method is given a Scientist::Result
instance with its associated Scientist::Observation
s:
class MyExperiment
include Scientist::Experiment
# ...
def publish(result)
# Store the timing for the control value,
$statsd.timing "science.#{name}.control", result.control.duration
# for the candidate (only the first, see "Breaking the rules" below,
$statsd.timing "science.#{name}.candidate", result.candidates.first.duration
# and counts for match/ignore/mismatch:
if result.matched?
$statsd.increment "science.#{name}.matched"
elsif result.ignored?
$statsd.increment "science.#{name}.ignored"
else
$statsd.increment "science.#{name}.mismatched"
# Finally, store mismatches in redis so they can be retrieved and examined
# later on, for debugging and research.
store_mismatch_data(result)
end
end
def store_mismatch_data(result)
payload = {
:name => name,
:context => context,
:control => observation_payload(result.control),
:candidate => observation_payload(result.candidates.first),
:execution_order => result.observations.map(&:name)
}
key = "science.#{name}.mismatch"
$redis.lpush key, payload
$redis.ltrim key, 0, 1000
end
def observation_payload(observation)
if observation.raised?
{
:exception => observation.exception.class,
:message => observation.exception.message,
:backtrace => observation.exception.backtrace
}
else
{
# see "Keeping it clean" below
:value => observation.cleaned_value
}
end
end
end
When running your test suite, it's helpful to know that the experimental results always match. To help with testing, Scientist defines a raise_on_mismatches
class attribute when you include Scientist::Experiment
. Only do this in your test suite!
To raise on mismatches:
class MyExperiment
include Scientist::Experiment
# ... implementation
end
MyExperiment.raise_on_mismatches = true
Scientist will raise a Scientist::Experiment::MismatchError
exception if any observations don't match.
If an exception is raised within any of scientist's internal helpers, like publish
, compare
, or clean
, the raised
method is called with the symbol name of the internal operation that failed and the exception that was raised. The default behavior of Scientist::Default
is to simply re-raise the exception. Since this halts the experiment entirely, it's often a better idea to handle this error and continue so the experiment as a whole isn't canceled entirely:
class MyExperiment
include Scientist::Experiment
# ...
def raised(operation, error)
InternalErrorTracker.track! "science failure in #{name}: #{operation}", error
end
end
The operations that may be handled here are:
:clean
- an exception is raised in aclean
block:compare
- an exception is raised in acompare
block:enabled
- an exception is raised in theenabled?
method:ignore
- an exception is raised in anignore
block:publish
- an exception is raised in thepublish
method:run_if
- an exception is raised in arun_if
block
Because enabled?
and run_if
determine when a candidate runs, it's impossible to guarantee that it will run every time. For this reason, Scientist is only safe for wrapping methods that aren't changing data.
When using Scientist, we've found it most useful to modify both the existing and new systems simultaneously anywhere writes happen, and verify the results at read time with science
. raise_on_mismatches
has also been useful to ensure that the correct data was written during tests, and reviewing published mismatches has helped us find any situations we overlooked with our production data at runtime. When writing to and reading from two systems, it's also useful to write some data reconciliation scripts to verify and clean up production data alongside any running experiments.
As your candidate behavior converges on the controls, you'll start thinking about removing an experiment and using the new behavior.
- If there are any ignore blocks, the candidate behavior is guaranteed to be different. If this is unacceptable, you'll need to remove the ignore blocks and resolve any ongoing mismatches in behavior until the observations match perfectly every time.
- When removing a read-behavior experiment, it's a good idea to keep any write-side duplication between an old and new system in place until well after the new behavior has been in production, in case you need to roll back.
Sometimes scientists just gotta do weird stuff. We understand.
Science is useful even when all you care about is the timing data or even whether or not a new code path blew up. If you have the ability to incrementally control how often an experiment runs via your enabled?
method, you can use it to silently and carefully test new code paths and ignore the results altogether. You can do this by setting ignore { true }
, or for greater efficiency, compare { true }
.
This will still log mismatches if any exceptions are raised, but will disregard the values entirely.
It's not usually a good idea to try more than one alternative simultaneously. Behavior isn't guaranteed to be isolated and reporting + visualization get quite a bit harder. Still, it's sometimes useful.
To try more than one alternative at once, add names to some try
blocks:
require "scientist"
class MyWidget
include Scientist
def allows?(user)
science "widget-permissions" do |e|
e.use { model.check_user(user).valid? } # old way
e.try("api") { user.can?(:read, model) } # new service API
e.try("raw-sql") { user.can_sql?(:read, model) } # raw query
end
end
end
When the experiment runs, all candidate behaviors are tested and each candidate observation is compared with the control in turn.
Define the candidates with named try
blocks, omit a use
, and pass a candidate name to run
:
experiment = MyExperiment.new("various-ways") do |e|
e.try("first-way") { ... }
e.try("second-way") { ... }
end
experiment.run("second-way")
The science
helper also knows this trick:
science "various-ways", run: "first-way" do |e|
e.try("first-way") { ... }
e.try("second-way") { ... }
end
Be on a Unixy box. Make sure a modern Bundler is available. script/test
runs the unit tests. All development dependencies are installed automatically. Science requires Ruby 1.9 or newer.
@jbarnette, @jesseplusplus, @rick, and @zerowidth