Instrument communications

pkg/analysis_server/lib/src/channel/web_socket_channel.dart

 // Copyright (c) 2014, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 library channel.web_socket;
 
 import 'dart:async';
 import 'dart:convert';
 import 'dart:io';
 
 import 'package:analysis_server/src/channel/channel.dart';
 import 'package:analysis_server/src/protocol.dart';
+import 'package:analyzer/instrumentation/instrumentation.dart';
 
 
 /**
  * Instances of the class [WebSocketClientChannel] implement a
  * [ClientCommunicationChannel] that uses a [WebSocket] to communicate with
  * servers.
  */
 class WebSocketClientChannel implements ClientCommunicationChannel {
   /**
    * The socket being wrapped.
@@ skipping 42 matching lines @@
  * [ServerCommunicationChannel] that uses a [WebSocket] to communicate with
  * clients.
  */
 class WebSocketServerChannel implements ServerCommunicationChannel {
   /**
    * The socket being wrapped.
    */
   final WebSocket socket;
 
   /**
+   * The instrumentation server that is to be used by this analysis server.
+   */
+  final InstrumentationServer instrumentationServer;
+
+  /**
    * Initialize a newly create [WebSocket] wrapper to wrap the given [socket].
    */
-  WebSocketServerChannel(this.socket);
+  WebSocketServerChannel(this.socket, this.instrumentationServer);
 
   @override
   void close() {
     socket.close(WebSocketStatus.NORMAL_CLOSURE);
   }
 
   @override
   void listen(void onRequest(Request request), {void onError(), void
       onDone()}) {
     socket.listen(
         (data) => readRequest(data, onRequest),
         onError: onError,
         onDone: onDone);
   }
 
   /**
    * Read a request from the given [data] and use the given function to handle
    * the request.
    */
   void readRequest(Object data, void onRequest(Request request)) {
     if (data is String) {
+      instrumentationServer.log(data);
       // Parse the string as a JSON descriptor and process the resulting
       // structure as a request.
       ServerCommunicationChannel.FromJson.start();
       Request request = new Request.fromString(data);
       ServerCommunicationChannel.FromJson.stop();
       if (request == null) {
         sendResponse(new Response.invalidRequestFormat());
         return;
       }
       onRequest(request);
     } else if (data is List<int>) {
       // TODO(brianwilkerson) Implement a more efficient protocol.
       sendResponse(new Response.invalidRequestFormat());
     } else {
       sendResponse(new Response.invalidRequestFormat());
     }
   }
 
   @override
   void sendNotification(Notification notification) {
     ServerCommunicationChannel.ToJson.start();
     String jsonEncoding = JSON.encode(notification.toJson());
     ServerCommunicationChannel.ToJson.stop();
     socket.add(jsonEncoding);
+    instrumentationServer.log(jsonEncoding);
   }
 
   @override
   void sendResponse(Response response) {
     ServerCommunicationChannel.ToJson.start();
     String jsonEncoding = JSON.encode(response.toJson());
     ServerCommunicationChannel.ToJson.stop();
     socket.add(jsonEncoding);
+    instrumentationServer.log(jsonEncoding);
   }
 }