update to use bazel_worker package

pkg/analyzer_cli/lib/src/build_mode.dart

 // Copyright (c) 2015, 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 analyzer_cli.src.build_mode;
 
 import 'dart:core' hide Resource;
 import 'dart:io' as io;
 
 import 'package:analyzer/dart/ast/ast.dart' show CompilationUnit;
@@ skipping 10 matching lines @@
 import 'package:analyzer/src/summary/idl.dart';
 import 'package:analyzer/src/summary/link.dart';
 import 'package:analyzer/src/summary/package_bundle_reader.dart';
 import 'package:analyzer/src/summary/summarize_ast.dart';
 import 'package:analyzer/src/summary/summarize_elements.dart';
 import 'package:analyzer/task/dart.dart';
 import 'package:analyzer_cli/src/analyzer_impl.dart';
 import 'package:analyzer_cli/src/driver.dart';
 import 'package:analyzer_cli/src/error_formatter.dart';
 import 'package:analyzer_cli/src/options.dart';
-import 'package:protobuf/protobuf.dart';
-
-import 'message_grouper.dart';
-import 'worker_protocol.pb.dart';
+import 'package:bazel_worker/bazel_worker.dart';
 
 /**
  * Analyzer used when the "--build-mode" option is supplied.
  */
 class BuildMode {
   final CommandLineOptions options;
   final AnalysisStats stats;
 
   final ResourceProvider resourceProvider = PhysicalResourceProvider.INSTANCE;
   SummaryDataStore summaryDataStore;
@@ skipping 223 matching lines @@
       }
       Uri uri = Uri.parse(sourceFile.substring(0, pipeIndex));
       String path = sourceFile.substring(pipeIndex + 1);
       uriToFileMap[uri] = new JavaFile(path);
     }
     return uriToFileMap;
   }
 }
 
 /**
- * Default implementation of [WorkerConnection] that works with stdio.
- */
-class StdWorkerConnection implements WorkerConnection {
-  final MessageGrouper _messageGrouper;
-  final io.Stdout _stdoutStream;
-
-  StdWorkerConnection(io.Stdin stdinStream, this._stdoutStream)
-      : _messageGrouper = new MessageGrouper(stdinStream);
-
-  @override
-  WorkRequest readRequest() {
-    var buffer = _messageGrouper.next;
-    if (buffer == null) return null;
-
-    return new WorkRequest.fromBuffer(buffer);
-  }
-
-  @override
-  void writeResponse(WorkResponse response) {
-    var responseBuffer = response.writeToBuffer();
-
-    var writer = new CodedBufferWriter();
-    writer.writeInt32NoTag(responseBuffer.length);
-    writer.writeRawBytes(responseBuffer);
-
-    _stdoutStream.add(writer.toBuffer());
-  }
-}
-
-/**
- * Connection between a worker and input / output.
- */
-abstract class WorkerConnection {
-  /**
-   * Read a new [WorkRequest]. Returns [null] when there are no more requests.
-   */
-  WorkRequest readRequest();
-
-  /**
-   * Write the given [response] as bytes to the output.
-   */
-  void writeResponse(WorkResponse response);
-}
-
-/**
  * Persistent Bazel worker.
  */
-class WorkerLoop {
-  static const int EXIT_CODE_OK = 0;
-  static const int EXIT_CODE_ERROR = 15;
-
-  final WorkerConnection connection;
-
+class AnalyzerWorkerLoop extends SyncWorkerLoop {
   final StringBuffer errorBuffer = new StringBuffer();
   final StringBuffer outBuffer = new StringBuffer();
 
   final String dartSdkPath;
 
-  WorkerLoop(this.connection, {this.dartSdkPath});
+  AnalyzerWorkerLoop(SyncWorkerConnection connection, {this.dartSdkPath})
+      : super(connection: connection);
 
-  factory WorkerLoop.std(
+  factory AnalyzerWorkerLoop.std(
       {io.Stdin stdinStream, io.Stdout stdoutStream, String dartSdkPath}) {
-    stdinStream ??= io.stdin;
-    stdoutStream ??= io.stdout;
-    WorkerConnection connection =
-        new StdWorkerConnection(stdinStream, stdoutStream);
-    return new WorkerLoop(connection, dartSdkPath: dartSdkPath);
+    SyncWorkerConnection connection = new StdSyncWorkerConnection(
+        stdinStream: stdinStream, stdoutStream: stdoutStream);
+    return new AnalyzerWorkerLoop(connection, dartSdkPath: dartSdkPath);
   }
 
   /**
    * Performs analysis with given [options].
    */
   void analyze(CommandLineOptions options) {
     options.dartSdkPath ??= dartSdkPath;
     new BuildMode(options, new AnalysisStats()).analyze();
   }
 
   /**
-   * Perform a single loop step.  Return `true` if should exit the loop.
+   * Perform a single loop step.
    */
-  bool performSingle() {
+  WorkResponse performRequest(WorkRequest request) {
+    errorBuffer.clear();
+    outBuffer.clear();
     try {
-      WorkRequest request = connection.readRequest();
-      if (request == null) {
-        return true;
-      }
       // Prepare options.
       CommandLineOptions options =
           CommandLineOptions.parse(request.arguments, (String msg) {
         throw new ArgumentError(msg);
       });
       // Analyze and respond.
       analyze(options);
       String msg = _getErrorOutputBuffersText();
-      connection.writeResponse(new WorkResponse()
+      return new WorkResponse()
         ..exitCode = EXIT_CODE_OK
-        ..output = msg);
+        ..output = msg;
     } catch (e, st) {
       String msg = _getErrorOutputBuffersText();
-      msg += '$e \n $st';
-      connection.writeResponse(new WorkResponse()
+      msg += '$e\n$st';
+      return new WorkResponse()
         ..exitCode = EXIT_CODE_ERROR
-        ..output = msg);
+        ..output = msg;
     }
-    return false;
   }
 
   /**
    * Run the worker loop.
    */
+  @override
   void run() {
     errorSink = errorBuffer;
     outSink = outBuffer;
     exitHandler = (int exitCode) {
       return throw new StateError('Exit called: $exitCode');
     };
-    while (true) {
-      errorBuffer.clear();
-      outBuffer.clear();
-      bool shouldExit = performSingle();
-      if (shouldExit) {
-        break;
-      }
-    }
+    super.run();
   }
 
   String _getErrorOutputBuffersText() {
     String msg = '';
     if (errorBuffer.isNotEmpty) {
       msg += errorBuffer.toString() + '\n';
     }
     if (outBuffer.isNotEmpty) {
       msg += outBuffer.toString() + '\n';
     }
     return msg;
   }
 }