autotest: add metrics for swarming bot manager's tick.
BUG=chromium:727926
TEST=None
Change-Id: Ie8b8f1f0cef5241f869a654f2b8f540a5c10eb64
Reviewed-on: https://chromium-review.googlesource.com/537059
Commit-Ready: Xixuan Wu <xixuan@chromium.org>
Tested-by: Xixuan Wu <xixuan@chromium.org>
Reviewed-by: Prathmesh Prabhu <pprabhu@chromium.org>
diff --git a/site_utils/chromeos_proxy/swarming_bot_manager.py b/site_utils/chromeos_proxy/swarming_bot_manager.py
index d317149..a45177b 100755
--- a/site_utils/chromeos_proxy/swarming_bot_manager.py
+++ b/site_utils/chromeos_proxy/swarming_bot_manager.py
@@ -99,6 +99,18 @@
return is_prod_proxy_server
+@metrics.SecondsTimerDecorator(
+ 'chromeos/autotest/swarming/bot_manager/tick')
+def tick(afe, bot_manager):
+ """One tick for swarming bot manager.
+
+ @param afe: the afe to check server role.
+ @param bot_manager: a swarming_bots.BotManager instance.
+ """
+ if is_server_in_prod(socket.getfqdn(), afe):
+ bot_manager.check()
+
+
def main(args):
"""Main func.
@@ -131,9 +143,7 @@
is_prod = False
retryable = True
while not _shut_down:
- if is_server_in_prod(socket.getfqdn(), args.afe):
- bot_manager.check()
-
+ tick(args.afe, bot_manager)
time.sleep(CHECK_INTERVAL)
