Commit 94680e14 authored by Andrew Newdigate's avatar Andrew Newdigate Committed by Rémy Coutable

Gitaly feature toggles are on by default in development environments

parent 6cd0ec11
---
title: Gitaly feature toggles are on by default in development
merge_request: 13802
author:
type: other
...@@ -70,21 +70,41 @@ module Gitlab ...@@ -70,21 +70,41 @@ module Gitlab
params['gitaly_token'].presence || Gitlab.config.gitaly['token'] params['gitaly_token'].presence || Gitlab.config.gitaly['token']
end end
def self.feature_enabled?(feature, status: MigrationStatus::OPT_IN) # Evaluates whether a feature toggle is on or off
def self.feature_enabled?(feature_name, status: MigrationStatus::OPT_IN)
# Disabled features are always off!
return false if status == MigrationStatus::DISABLED return false if status == MigrationStatus::DISABLED
feature = Feature.get("gitaly_#{feature}") feature = Feature.get("gitaly_#{feature_name}")
# If the feature hasn't been set, turn it on if it's opt-out
return status == MigrationStatus::OPT_OUT unless Feature.persisted?(feature)
# If the feature has been set, always evaluate
if Feature.persisted?(feature)
if feature.percentage_of_time_value > 0 if feature.percentage_of_time_value > 0
# Probabilistically enable this feature # Probabilistically enable this feature
return Random.rand() * 100 < feature.percentage_of_time_value return Random.rand() * 100 < feature.percentage_of_time_value
end end
feature.enabled? return feature.enabled?
end
# If the feature has not been set, the default depends
# on it's status
case status
when MigrationStatus::OPT_OUT
true
when MigrationStatus::OPT_IN
opt_into_all_features?
else
false
end
end
# opt_into_all_features? returns true when the current environment
# is one in which we opt into features automatically
def self.opt_into_all_features?
Rails.env.development? || ENV["GITALY_FEATURE_DEFAULT_ON"] == "1"
end end
private_class_method :opt_into_all_features?
def self.migrate(feature, status: MigrationStatus::OPT_IN) def self.migrate(feature, status: MigrationStatus::OPT_IN)
is_enabled = feature_enabled?(feature, status: status) is_enabled = feature_enabled?(feature, status: status)
......
...@@ -102,6 +102,22 @@ describe Gitlab::GitalyClient, skip_gitaly_mock: true do ...@@ -102,6 +102,22 @@ describe Gitlab::GitalyClient, skip_gitaly_mock: true do
expect(described_class.feature_enabled?(feature_name, status: feature_status)).to be(false) expect(described_class.feature_enabled?(feature_name, status: feature_status)).to be(false)
end end
end end
context "when a feature is not persisted" do
it 'returns false when opt_into_all_features is off' do
allow(Feature).to receive(:persisted?).and_return(false)
allow(described_class).to receive(:opt_into_all_features?).and_return(false)
expect(described_class.feature_enabled?(feature_name, status: feature_status)).to be(false)
end
it 'returns true when the override is on' do
allow(Feature).to receive(:persisted?).and_return(false)
allow(described_class).to receive(:opt_into_all_features?).and_return(true)
expect(described_class.feature_enabled?(feature_name, status: feature_status)).to be(true)
end
end
end end
context 'when the feature_status is OPT_OUT' do context 'when the feature_status is OPT_OUT' do
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment