Chromium Code Reviews| Index: tools/telemetry/telemetry/core/backends/chrome_inspector/inspector_websocket.py |
| diff --git a/tools/telemetry/telemetry/core/backends/chrome_inspector/inspector_websocket.py b/tools/telemetry/telemetry/core/backends/chrome_inspector/inspector_websocket.py |
| index 9ee28741dcaf5b7723cef395896c3a819a79c3e0..db12c17cc65ac03122cb310408f0e053cca72040 100644 |
| --- a/tools/telemetry/telemetry/core/backends/chrome_inspector/inspector_websocket.py |
| +++ b/tools/telemetry/telemetry/core/backends/chrome_inspector/inspector_websocket.py |
| @@ -11,10 +11,6 @@ import time |
| from telemetry.core.backends.chrome_inspector import websocket |
| -_DomainHandler = collections.namedtuple( |
| - 'DomainHandler', ['notification_handler', 'will_close_handler']) |
| - |
| - |
| class DispatchNotificationsUntilDoneTimeoutException(Exception): |
| """Exception that can be thrown from DispatchNotificationsUntilDone to |
| indicate timeout exception of the function. |
| @@ -42,8 +38,7 @@ class InspectorWebsocket(object): |
| self._all_data_received = False |
| self._domain_handlers = {} |
| - def RegisterDomain( |
| - self, domain_name, notification_handler, will_close_handler=None): |
| + def RegisterDomain(self, domain_name, notification_handler): |
| """Registers a given domain for handling notification methods. |
| When used as handler for DispatchNotificationsUntilDone, |
| @@ -55,10 +50,7 @@ class InspectorWebsocket(object): |
| if msg['method'] == 'Console.messageAdded': |
| print msg['params']['message'] |
| return True |
| - def OnConsoleClose(self): |
| - pass |
| - inspector_backend.RegisterDomain( |
| - 'Console', OnConsoleNotification, OnConsoleClose) |
| + inspector_backend.RegisterDomain('Console', OnConsoleNotification) |
| Args: |
| domain_name: The devtools domain name. E.g., 'Tracing', 'Memory', 'Page'. |
| @@ -67,16 +59,14 @@ class InspectorWebsocket(object): |
| (via DispatchNotifications and DispatchNotificationsUntilDone). |
| The handler accepts a single paramater: the JSON object representing |
| the notification. |
| - will_close_handler: Handler to be called from Disconnect(). |
| """ |
| assert domain_name not in self._domain_handlers |
| - self._domain_handlers[domain_name] = _DomainHandler( |
| - notification_handler, will_close_handler) |
| + self._domain_handlers[domain_name] = notification_handler |
| def UnregisterDomain(self, domain_name): |
| """Unregisters a previously registered domain.""" |
| assert domain_name in self._domain_handlers |
| - self._domain_handlers.pop(domain_name) |
| + del self._domain_handlers[domain_name] |
|
nednguyen
2015/02/21 06:07:48
What is this change for?
erikchen
2015/02/23 18:27:56
Pop has a return value, whereas del does not. Sinc
|
| def Connect(self, url, timeout=10): |
| assert not self._socket |
| @@ -85,14 +75,7 @@ class InspectorWebsocket(object): |
| self._next_request_id = 0 |
| def Disconnect(self): |
| - """Disconnects the inspector websocket. |
| - |
| - All existing domain handlers will also be unregistered. |
| - """ |
| - for _, handler in self._domain_handlers.items(): |
| - if handler.will_close_handler: |
| - handler.will_close_handler() |
| - |
| + """Disconnects the inspector websocket.""" |
| if self._socket: |
| self._socket.close() |
| self._socket = None |
| @@ -177,7 +160,7 @@ class InspectorWebsocket(object): |
| dot_pos = mname.find('.') |
| domain_name = mname[:dot_pos] |
| if domain_name in self._domain_handlers: |
| - return self._domain_handlers[domain_name].notification_handler(result) |
| + return self._domain_handlers[domain_name](result) |
| logging.warn('Unhandled inspector message: %s', result) |
| return False |