Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 # Copyright 2016 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 os | |
| 6 import logging | |
| 7 import sys | |
| 8 | |
| 9 from devil.android import logcat_monitor | |
| 10 from devil.utils import reraiser_thread | |
| 11 from pylib import constants | |
| 12 | |
| 13 sys.path.insert(0, os.path.abspath(os.path.join( | |
| 14 constants.DIR_SOURCE_ROOT, 'tools', 'swarming_client'))) | |
| 15 from libs.logdog import bootstrap # pylint: disable=import-error | |
| 16 | |
| 17 class LogdogLogcatMonitor(logcat_monitor.LogcatMonitor): | |
| 18 """Logcat monitor that writes logcat to a logdog stream. | |
| 19 The logdog stream client will return a url, where contains the logcat. | |
| 20 """ | |
| 21 def __init__(self, adb, stream_name, clear=True, filter_specs=None): | |
| 22 super(LogdogLogcatMonitor, self).__init__(adb, clear, filter_specs) | |
| 23 self._logcat_url = '' | |
| 24 self._logdog_stream = None | |
| 25 self._stream_client = None | |
| 26 self._stream_name = stream_name | |
| 27 try: | |
| 28 self._stream_client = bootstrap.ButlerBootstrap.probe().stream_client() | |
| 29 self._logdog_stream = self._stream_client.open_text(self._stream_name) | |
| 30 except bootstrap.NotBootstrappedError as e: | |
| 31 logging.exception( | |
| 32 'Error not bootstrapped. Failed to start logdog: %s', e) | |
| 33 except ValueError as e: | |
|
jbudorick
2016/12/01 01:56:18
given the number of KeyErrors that the logdog code
BigBossZhiling
2016/12/01 02:07:41
Done.
| |
| 34 logging.exception('Error cannot create stream_client: %s.', e) | |
| 35 | |
| 36 def GetLogcatURL(self): | |
| 37 """Return logcat url. | |
| 38 | |
| 39 The default logcat url is '', if failed to create stream_client. | |
| 40 """ | |
| 41 return self._logcat_url | |
| 42 | |
| 43 def Stop(self): | |
| 44 """Stops the logcat monitor. | |
| 45 | |
| 46 Close the logdog stream as well. | |
| 47 """ | |
| 48 super(LogdogLogcatMonitor, self)._StopRecording() | |
| 49 if self._logdog_stream: | |
| 50 try: | |
| 51 self._logcat_url = self._stream_client.get_viewer_url(self._stream_name) | |
|
jbudorick
2016/12/01 01:56:18
According to the docs, this can also emit ValueErr
BigBossZhiling
2016/12/01 02:07:41
Done.
| |
| 52 except KeyError as e: | |
| 53 logging.exception('Error cannot get viewer url: %s', e) | |
| 54 self._logdog_stream.close() | |
| 55 | |
| 56 def Start(self): | |
| 57 """Starts the logdog logcat monitor. | |
| 58 | |
| 59 Clears the logcat if |clear| was set in |__init__|. | |
| 60 """ | |
| 61 if self._clear: | |
| 62 self._adb.Logcat(clear=True) | |
| 63 self._StartRecording() | |
| 64 | |
| 65 def _StartRecording(self): | |
| 66 """Starts recording logcat to file. | |
| 67 | |
| 68 Write logcat to stream at the same time. | |
| 69 """ | |
| 70 def record_to_stream(): | |
| 71 if self._logdog_stream: | |
| 72 for data in self._adb.Logcat(filter_specs=self._filter_specs, | |
| 73 logcat_format='threadtime'): | |
| 74 if self._stop_recording_event.isSet(): | |
| 75 return | |
| 76 self._logdog_stream.write(data + '\n') | |
| 77 | |
| 78 self._stop_recording_event.clear() | |
| 79 if not self._record_thread: | |
| 80 self._record_thread = reraiser_thread.ReraiserThread(record_to_stream) | |
| 81 self._record_thread.start() | |
| 82 | |
| 83 def Close(self): | |
| 84 """Override parent's close method.""" | |
| 85 pass | |
| 86 | |
| 87 def __del__(self): | |
| 88 """Override parent's delete method.""" | |
| 89 pass | |
| OLD | NEW |