Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 # Copyright 2015 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import json | |
| 6 import logging | |
| 7 import os | |
| 8 import socket | |
| 9 import sys | |
| 10 import urlparse | |
| 11 import re | |
| 12 | |
| 13 import monitors | |
| 14 import interface | |
| 15 from common import standard_metrics | |
| 16 from common import targets | |
| 17 | |
| 18 | |
| 19 def initialize(endpoint=None, flush='manual', job_name=None, | |
| 20 service_name=None, instance=0): | |
| 21 """Initialize the global monitor. | |
| 22 | |
| 23 Also initializes the default target if endpoint argument is supplied. | |
| 24 If they aren't, all created metrics will have to supply their own target. | |
| 25 This is generally a bad idea, as many libraries rely on the default target | |
| 26 being set up. | |
| 27 | |
| 28 Args: | |
| 29 endpoint: url (in format pubsub://project/topic) to post ts_mon metrics to | |
| 30 flush: metric push behavior: all (send every metric individually), | |
| 31 or manual (only send when flush() is called) | |
| 32 job_name: name of the job instance of the task | |
| 33 instance: number of instance for this task | |
| 34 service_name: name of service being monitored | |
| 35 | |
| 36 """ | |
| 37 hostname = socket.getfqdn().split('.')[0] | |
| 38 try: | |
| 39 region = fqdn.split('.')[1] | |
| 40 except: | |
| 41 region = '' | |
| 42 | |
| 43 if endpoint.startswith('pubsub://'): | |
|
Sergey Berezin
2015/08/18 00:40:37
nit: I feel this logic should belong in ts_mon its
jshu
2015/08/18 01:15:55
thanks for drawing attention this, going to take o
Sergey Berezin
2015/08/18 17:31:38
Filed http://crbug.com/522135
The danger of dupli
jshu
2015/08/20 00:23:34
Those two files are completely different.
| |
| 44 url = urlparse.urlparse(endpoint) | |
| 45 project = url.netloc | |
| 46 topic = url.path.strip('/') | |
| 47 interface.state.global_monitor = monitors.PubSubMonitor(project, topic) | |
| 48 else: | |
| 49 logging.error('Error: endpoint should begin with pubsub://') | |
| 50 interface.state.global_monitor = monitors.NullMonitor() | |
| 51 | |
| 52 # Reimplement ArgumentParser.error, since we don't have access to the parser | |
| 53 if not service_name: | |
| 54 logging.error('service_name variable is not set for task.') | |
| 55 if not job_name: # pragma: no cover | |
| 56 logging.error('job_name variable is not set for task.') | |
| 57 interface.state.default_target = targets.TaskTarget( | |
| 58 service_name, job_name, region, hostname, instance) | |
| 59 | |
| 60 interface.state.flush_mode = flush | |
| 61 | |
| 62 standard_metrics.init() | |
| OLD | NEW |