Roll infra_libs to 564aaf7480f24c90687df79d9cef910cc342a54d

client/third_party/infra_libs/ts_mon/config.py

 # 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.
 
 import json
 import logging
 import os
 import socket
 import sys
 import urlparse
@@ skipping 73 matching lines @@
   parser.add_argument(
       '--ts-mon-endpoint',
       help='url (including file://, pubsub://project/topic, https://) to post '
            'monitoring metrics to. If set, overrides the value in '
            '--ts-mon-config-file')
   parser.add_argument(
       '--ts-mon-credentials',
       help='path to a pkcs8 json credential file. If set, overrides the value '
            'in --ts-mon-config-file')
   parser.add_argument(
+      '--ts-mon-ca-certs',
+      help='path to file containing root CA certificates for SSL server '
+           'certificate validation. If not set, a CA cert file bundled with '
+           'httplib2 is used.')
+  parser.add_argument(
       '--ts-mon-flush',
       choices=('manual', 'auto'), default='auto',
       help=('metric push behavior: manual (only send when flush() is called), '
             'or auto (send automatically every --ts-mon-flush-interval-secs '
             'seconds). (default: %(default)s)'))
   parser.add_argument(
       '--ts-mon-flush-interval-secs',
       type=int,
       default=60,
       help=('automatically push metrics on this interval if '
@@ skipping 54 matching lines @@
   parser.add_argument(
       '--ts-mon-task-number', type=int, default=0,
       help='number (e.g. for replication) of this instance of this task '
            '(default: %(default)s)')
 
   parser.add_argument(
       '--ts-mon-metric-name-prefix',
       default='/chrome/infra/',
       help='metric name prefix for all metrics (default: %(default)s)')
 
+  parser.add_argument(
+      '--ts-mon-use-new-proto',
+      default=False, action='store_true',
+      help='use the new proto schema (default: false)')
+
 def process_argparse_options(args):
   """Process command line arguments to initialize the global monitor.
 
   Also initializes the default target.
 
   Starts a background thread to automatically flush monitoring metrics if not
   disabled by command line arguments.
 
   Args:
     args (argparse.Namespace): the result of parsing the command line arguments
@@ skipping 44 matching lines @@
 
   if endpoint.startswith('file://'):
     interface.state.global_monitor = monitors.DebugMonitor(
         endpoint[len('file://'):])
   elif endpoint.startswith('pubsub://'):
     if credentials:
       url = urlparse.urlparse(endpoint)
       project = url.netloc
       topic = url.path.strip('/')
       interface.state.global_monitor = monitors.PubSubMonitor(
-          credentials, project, topic, use_instrumented_http=True)
+          credentials, project, topic, use_instrumented_http=True,
+          ca_certs=args.ts_mon_ca_certs)
     else:
       logging.error('ts_mon monitoring is disabled because credentials are not '
                     'available')
   elif endpoint.startswith('https://'):
-    interface.state.global_monitor = monitors.HttpsMonitor(endpoint,
-                                                           credentials)
+    interface.state.global_monitor = monitors.HttpsMonitor(
+        endpoint, credentials)
   elif endpoint.lower() == 'none':
     logging.info('ts_mon monitoring has been explicitly disabled')
   else:
     logging.error('ts_mon monitoring is disabled because the endpoint provided'
                   ' is invalid or not supported: %s', endpoint)
 
   interface.state.flush_mode = args.ts_mon_flush
+  interface.state.use_new_proto = args.ts_mon_use_new_proto
 
   if args.ts_mon_flush == 'auto':
     interface.state.flush_thread = interface._FlushThread(
         args.ts_mon_flush_interval_secs)
     interface.state.flush_thread.start()
 
   standard_metrics.init()
+