Fix bugs and add instrumentation

pkg/analyzer_plugin/lib/src/channel/isolate_channel.dart

 // Copyright (c) 2017, 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.
 
 import 'dart:async';
+import 'dart:convert';
 import 'dart:isolate';
 
+import 'package:analyzer/instrumentation/instrumentation.dart';
 import 'package:analyzer_plugin/channel/channel.dart';
 import 'package:analyzer_plugin/protocol/protocol.dart';
 
 /**
  * The object that allows a [ServerPlugin] to receive [Request]s and to return
  * both [Response]s and [Notification]s. It communicates with the analysis
  * server by passing data to the server's main isolate.
  */
 class PluginIsolateChannel implements PluginCommunicationChannel {
   /**
@@ skipping 25 matching lines @@
       _subscription.cancel();
       _subscription = null;
     }
   }
 
   @override
   void listen(void onRequest(Request request),
       {Function onError, void onDone()}) {
     void onData(data) {
       Map<String, Object> requestMap = data;
+//      print('[plugin] Received request: ${JSON.encode(requestMap)}');
       Request request = new Request.fromJson(requestMap);
       if (request != null) {
         onRequest(request);
       }
     }
 
     if (_subscription != null) {
       throw new StateError('Only one listener is allowed per channel');
     }
     _subscription = _receivePort.listen(onData,
         onError: onError, onDone: onDone, cancelOnError: false);
   }
 
   @override
   void sendNotification(Notification notification) {
-    _sendPort.send(notification.toJson());
+    Map<String, Object> json = notification.toJson();
+//    print('[plugin] Send notification: ${JSON.encode(json)}');
+    _sendPort.send(json);
   }
 
   @override
   void sendResponse(Response response) {
-    _sendPort.send(response.toJson());
+    Map<String, Object> json = response.toJson();
+//    print('[plugin] Send response: ${JSON.encode(json)}');
+    _sendPort.send(json);
   }
 }
 
 /**
  * The communication channel that allows an analysis server to send [Request]s
  * to, and to receive both [Response]s and [Notification]s from, a plugin.
  */
 class ServerIsolateChannel implements ServerCommunicationChannel {
   /**
-   * The URI for the plugin that will be run in the isolate that this channel
+   * The URI for the Dart file that will be run in the isolate that this channel
    * communicates with.
    */
-  final Uri uri;
+  final Uri pluginUri;
+
+  /**
+   * The URI for the '.packages' file that will control how 'package:' URIs are
+   * resolved.
+   */
+  final Uri packagesUri;
+
+  /**
+   * The instrumentation service that is being used by the analysis server.
+   */
+  final InstrumentationService instrumentationService;
 
   /**
    * The isolate in which the plugin is running, or `null` if the plugin has
    * not yet been started by invoking [listen].
    */
   Isolate _isolate;
 
   /**
    * The port used to send requests to the plugin, or `null` if the plugin has
    * not yet been started by invoking [listen].
    */
   SendPort _sendPort;
 
+  ReceivePort receivePort;
+
+  ReceivePort errorPort;
+
+  ReceivePort exitPort;
+
   /**
    * Initialize a newly created channel to communicate with an isolate running
    * the code at the given [uri].
    */
-  ServerIsolateChannel(this.uri);
-
+  ServerIsolateChannel(
+      this.pluginUri, this.packagesUri, this.instrumentationService);
   @override
   void close() {
-    // TODO(brianwilkerson) Is there anything useful to do here?
+    receivePort?.close();
+    errorPort?.close();
+    exitPort?.close();
     _isolate = null;
-    _sendPort = null;
+//    _sendPort = null;
+//    receivePort = null;
+//    errorPort = null;
+//    exitPort = null;
   }
 
   @override
   Future<Null> listen(void onResponse(Response response),
       void onNotification(Notification notification),
       {Function onError, void onDone()}) async {
     if (_isolate != null) {
       throw new StateError('Cannot listen to the same channel more than once.');
     }
-    ReceivePort receivePort = new ReceivePort();
-    ReceivePort errorPort;
+    receivePort = new ReceivePort();
     if (onError != null) {
       errorPort = new ReceivePort();
       errorPort.listen((error) {
         onError(error);
       });
     }
-    ReceivePort exitPort;
     if (onDone != null) {
       exitPort = new ReceivePort();
       exitPort.listen((_) {
         onDone();
       });
     }
-    _isolate = await Isolate.spawnUri(uri, <String>[], receivePort.sendPort,
-        automaticPackageResolution: true,
+    _isolate = await Isolate.spawnUri(
+        pluginUri, <String>[], receivePort.sendPort,
         onError: errorPort?.sendPort,
-        onExit: exitPort?.sendPort);
-    _sendPort = await receivePort.first as SendPort;
+        onExit: exitPort?.sendPort,
+        packageConfig: packagesUri);
+    Completer<Null> channelReady = new Completer<Null>();
     receivePort.listen((dynamic input) {
-      if (input is Map) {
+      if (input is SendPort) {
+//        print('[server] Received send port');
+        _sendPort = input;
+        channelReady.complete(null);
+      } else if (input is Map) {
         if (input.containsKey('id') != null) {
+          String encodedInput = JSON.encode(input);
+//          print('[server] Received response: $encodedInput');
+          instrumentationService.logPluginResponse(pluginUri, encodedInput);
           onResponse(new Response.fromJson(input));
         } else if (input.containsKey('event')) {
+          String encodedInput = JSON.encode(input);
+//          print('[server] Received notification: $encodedInput');
+          instrumentationService.logPluginNotification(pluginUri, encodedInput);
           onNotification(new Notification.fromJson(input));
         }
       }
     });
+    return channelReady.future;
   }
 
   @override
   void sendRequest(Request request) {
-    _sendPort.send(request.toJson());
+    Map<String, Object> json = request.toJson();
+    String encodedRequest = JSON.encode(json);
+//    print('[server] Send request: $encodedRequest');
+    instrumentationService.logPluginRequest(pluginUri, encodedRequest);
+    _sendPort.send(json);
   }
 }