turbine support worker mode

Author: SupSaiYaJin

src/java_tools/buildjar/java/com/google/devtools/build/java/turbine/BUILD

@@ -7,12 +7,16 @@ package(
 
 java_binary(
     name = "turbine_direct_binary",
-    main_class = "com.google.turbine.main.Main",
+    main_class = "com.google.devtools.build.java.turbine.TurbineWorkerWrapper",
+    srcs = ["TurbineWorkerWrapper.java"],
+    deps = [
+        "//src/main/java/com/google/devtools/build/lib/worker:work_request_handlers",
+        "//third_party:turbine",
+    ],
     runtime_deps = [
         "//src/main/protobuf:deps_java_proto",
         "//third_party:guava",
         "//third_party:jsr305",
-        "//third_party:turbine",
     ],
 )
 

src/java_tools/buildjar/java/com/google/devtools/build/java/turbine/TurbineWorkerWrapper.java

@@ -0,0 +1,51 @@
+package com.google.devtools.build.java.turbine;
+
+import com.google.turbine.main.Main;
+import com.google.devtools.build.lib.worker.ProtoWorkerMessageProcessor;
+import com.google.devtools.build.lib.worker.WorkRequestHandler;
+
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.time.Duration;
+import java.util.List;
+
+/**
+ * A Wrapper for Turbine to support multiplex worker
+ */
+public class TurbineWorkerWrapper {
+
+  public static void main(String[] args) throws IOException {
+    if (args.length == 1 && args[0].equals("--persistent_worker")) {
+      WorkRequestHandler workerHandler =
+          new WorkRequestHandler.WorkRequestHandlerBuilder(
+              TurbineWorkerWrapper::turbine,
+              System.err,
+              new ProtoWorkerMessageProcessor(System.in, System.out))
+              .setCpuUsageBeforeGc(Duration.ofSeconds(10))
+              .build();
+      int exitCode = 1;
+      try {
+        workerHandler.processRequests();
+        exitCode = 0;
+      } catch (IOException e) {
+        System.err.println(e.getMessage());
+      } finally {
+        // Prevent hanging threads from keeping the worker alive.
+        System.exit(exitCode);
+      }
+    } else {
+      Main.main(args);
+    }
+  }
+
+  private static int turbine(List<String> args, PrintWriter pw) {
+      try {
+          Main.compile(args.toArray(new String[0]));
+      } catch (Throwable e) {
+          System.err.println(e.getMessage());
+          return 1;
+      }
+      return 0;
+  }
+
+}

src/main/java/com/google/devtools/build/lib/rules/java/JavaHeaderCompileAction.java

@@ -132,7 +132,7 @@ public static Builder newBuilder(RuleContext ruleContext) {
   public static final class Builder {
 
     private static final ParamFileInfo PARAM_FILE_INFO =
-        ParamFileInfo.builder(UNQUOTED).setCharset(ISO_8859_1).build();
+        ParamFileInfo.builder(UNQUOTED).setCharset(ISO_8859_1).setUseAlways(true).build();
 
     private final RuleContext ruleContext;
 
@@ -457,6 +457,11 @@ public void build(JavaToolchainProvider javaToolchain) {
                 ExecutionRequirements.REMOTE_EXECUTION_INLINE_OUTPUTS,
                 outputDepsProto.getExecPathString());
       }
+      executionInfo = ImmutableMap.<String, String>builder()
+          .putAll(executionInfo)
+          .putAll(ExecutionRequirements.WORKER_MODE_ENABLED)
+          .putAll(ExecutionRequirements.WORKER_MULTIPLEX_MODE_ENABLED)
+          .build();
       if (useDirectClasspath) {
         NestedSet<Artifact> classpath;
         if (!directJars.isEmpty() || classpathEntries.isEmpty()) {
@@ -485,7 +490,7 @@ public void build(JavaToolchainProvider javaToolchain) {
         ruleContext.registerAction(
             new JavaHeaderCompileAction(
                 /* owner= */ ruleContext.getActionOwner(),
-                /* tools= */ NestedSetBuilder.emptySet(Order.STABLE_ORDER),
+                /* tools= */ headerCompiler.tool().getFilesToRun(),
                 /* inputs= */ allInputs,
                 /* outputs= */ outputs.build(),
                 /* resourceSetOrBuilder= */ AbstractAction.DEFAULT_RESOURCE_SET,