| Index: infra/libs/event_mon/router.py
|
| diff --git a/infra/libs/event_mon/router.py b/infra/libs/event_mon/router.py
|
| index 02d5f7911b164c5165d0114070d6380c4c4d6c5d..373ebf50768a7ab9330578a53c975e6af3d91def 100644
|
| --- a/infra/libs/event_mon/router.py
|
| +++ b/infra/libs/event_mon/router.py
|
| @@ -4,6 +4,7 @@
|
|
|
| import Queue
|
| import logging
|
| +import requests
|
| import threading
|
| import time
|
|
|
| @@ -26,10 +27,9 @@ class _Router(object):
|
| ... fill in event ...
|
| router.push_event(event)
|
| """
|
| - def __init__(self, dry_run=True):
|
| - # dry_run is meant for local testing and unit testing. When True, this
|
| - # object should have no side effect.
|
| - self.dry_run = dry_run
|
| + def __init__(self, endpoint=None):
|
| + # endpoint == None means 'dry run'. No data is sent.
|
| + self.endpoint = endpoint
|
|
|
| self.event_queue = Queue.Queue()
|
| self._thread = threading.Thread(target=self._router)
|
| @@ -44,10 +44,11 @@ class _Router(object):
|
|
|
| # Set this time at the very last moment
|
| events.request_time_ms = time_ms()
|
| -
|
| - if not self.dry_run: # pragma: no cover
|
| - # TODO(pgervais): Actually do something
|
| - pass
|
| + if self.endpoint: # pragma: no cover
|
| + # TODO(pgervais): log when something fails.
|
| + requests.post(self.endpoint, data=events.SerializeToString())
|
| + else:
|
| + print('fake post request')
|
|
|
| def close(self, timeout=None):
|
| """
|
| @@ -80,8 +81,9 @@ class _Router(object):
|
| % str(type(event)))
|
| return False
|
|
|
| - # Dumb implementation, can be made more sophisticated (batching)
|
| - request = LogRequestLite()
|
| - request.log_event.extend((event,)) # copies the protobuf
|
| - self.event_queue.put(request)
|
| + # TODO(pgervais): implement batching.
|
| + request_p = LogRequestLite()
|
| + request_p.log_source_name = 'CHROME_INFRA'
|
| + request_p.log_event.extend((event,)) # copies the protobuf
|
| + self.event_queue.put(request_p)
|
| return True
|
|
|