Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(15)

Unified Diff: infra_libs/experiments.py

Issue 1365583002: Have service_manager start a cloudtail attached to each service it starts. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master
Patch Set: Just use the default project ID Created 5 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: infra_libs/experiments.py
diff --git a/infra_libs/experiments.py b/infra_libs/experiments.py
new file mode 100644
index 0000000000000000000000000000000000000000..018d612de62059166c5d7f3cdcc6fb3107381b49
--- /dev/null
+++ b/infra_libs/experiments.py
@@ -0,0 +1,44 @@
+# Copyright 2015 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""Tools for gradually enabling a feature on a deterministic set of machines.
+
+Add a flag to your program to control the percentage of machines that a new
+feature should be enabled on::
+
+ def add_argparse_options(self, parser):
+ parser.add_argument('--myfeature-percent', type=int, default=0)
+
+ def main(self, opts):
+ if experiments.is_active_for_host('myfeature', opts.myfeature_percent):
+ # do myfeature
+"""
+
+import hashlib
+import logging
+import socket
+
+
+def _is_active(labels, percent):
+ h = hashlib.md5()
+ for label, value in labels.iteritems():
Vadim Sh. 2015/09/23 17:40:22 sorted(labels.iteritems())
dsansome 2015/09/24 04:34:13 Done.
+ h.update(label)
+ h.update(value)
+
+ digest = h.digest()
+
+ print hash(digest) % 100
Vadim Sh. 2015/09/23 17:40:22 remove :)
dsansome 2015/09/24 04:34:13 Oops!
+ return (hash(digest) % 100) < percent
+
+
+def is_active_for_host(experiment_name, percent):
+ ret = _is_active({
+ 'name': experiment_name,
+ 'host': socket.getfqdn(),
Vadim Sh. 2015/09/23 17:40:21 This thing can return nonsense (e.g. 1.1.1.1.1.1.a
dsansome 2015/09/24 04:34:13 Nonsense is fine - it's just being hashed anyway.
+ }, percent)
+
+ if ret:
+ logging.info('Experiment "%s" is active', experiment_name)
+
+ return ret

Powered by Google App Engine
This is Rietveld 408576698