Count flakiness pipeline issue tracker requests

infra/services/builder_alerts/__main__.py

 #!/usr/bin/env python
 # Copyright 2014 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.
 
 import argparse
 import contextlib
 import cStringIO
 import datetime
 import gzip
 import json
 import logging
 import multiprocessing
 import os
 import sys
 import traceback
 
 import requests
 import requests_cache
 
 from infra_libs import logs
+from infra_libs import ts_mon
 from infra.libs.service_utils import outer_loop
 
 from infra.services.builder_alerts import alert_builder
 from infra.services.builder_alerts import analysis
 from infra.services.builder_alerts import buildbot
 from infra.services.builder_alerts import crbug_issues
 from infra.services.builder_alerts import gatekeeper_extras
 from infra.services.builder_alerts import string_helpers
 
 
@@ skipping 319 matching lines @@
   parser.add_argument('data_url', action='store', nargs='*')  # Deprecated
   parser.add_argument('--use-cache', action='store_true')
   parser.add_argument('--master-filter', action='store')
   parser.add_argument('--builder-filter', action='store')
   parser.add_argument('--processes', default=PARALLEL_TASKS, action='store',
                       type=int)
   parser.add_argument('--jobs', default=CONCURRENT_TASKS, action='store',
                       type=int)
   logs.add_argparse_options(parser)
   outer_loop.add_argparse_options(parser)
+  ts_mon.add_argparse_options(parser)
 
   gatekeeper_json = os.path.join(build_scripts_dir, 'slave', 'gatekeeper.json')
   parser.add_argument('--gatekeeper', action='store', default=gatekeeper_json)
   gatekeeper_trees_json = os.path.join(build_scripts_dir, 'slave',
                                        'gatekeeper_trees.json')
   parser.add_argument('--gatekeeper-trees', action='store',
                       default=gatekeeper_trees_json)
 
   parser.add_argument('--findit-api-url',
                       help='Query findit results from this url.')
   parser.add_argument('--crbug-service-account',
                       help='Path to a service account JSON file to be used to '
                            'search for relevant issues on crbug.com.')
   parser.add_argument('--api-endpoint-prefix',
                       help='Endpoint prefix for posting alerts. Old API '
                            'endpoint will be formed by adding value specified '
                            'in --old-api-path to the prefix, new API endpoints '
                            'will be formed by adding '
                            '/api/v1/alerts/<tree_name>.')
   parser.add_argument('--old-api-path',
                       help='Path to be appended to --api-endpoint-prefix to '
                            'form old API endpoint.')
 
   args = parser.parse_args(args)
   logs.process_argparse_options(args)
   loop_args = outer_loop.process_argparse_options(args)
+  ts_mon.process_argparse_options(args)
 
   # TODO(sergiyb): Remove support for data_url when builder_alerts recipes are
   # updated and using new syntax to call this script.
   if args.data_url:
     if (len(args.data_url) == 1 and args.data_url[0].endswith('alerts') and
         not args.api_endpoint_prefix and not args.old_api_path):
       logging.warn(
           'You are using positional argument to specify URL to post updates '
           'to. Please use --api-endpoint-prefix and --old-api-path instead.')
       slash_index = args.data_url[0].rindex('/')
@@ skipping 20 matching lines @@
         _ConnectionpoolFilter())
 
   def outer_loop_iteration():
     return inner_loop(args)
 
   loop_results = outer_loop.loop(
       task=outer_loop_iteration,
       sleep_timeout=lambda: 5,
       **loop_args)
 
+  ts_mon.flush()
   return 0 if loop_results.success else 1
 
 
 if __name__ == '__main__':
   sys.exit(main(sys.argv[1:]))