// Copyright 2018 The Bazel Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package com.google.devtools.build.lib.runtime;

import com.google.common.collect.ConcurrentHashMultiset;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Multiset;
import com.google.devtools.build.lib.actions.ActionResult;
import com.google.devtools.build.lib.actions.SpawnResult;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.concurrent.atomic.AtomicLong;
import javax.annotation.concurrent.ThreadSafe;

/** Collects results from SpawnResult. */
@ThreadSafe
public class SpawnStats {
  private static final ImmutableList<String> REPORT_FIRST = ImmutableList.of("remote cache hit");

  private final ConcurrentHashMultiset<String> runners = ConcurrentHashMultiset.create();
  private final AtomicLong totalWallTimeMillis = new AtomicLong();

  public void countActionResult(ActionResult actionResult) {
    for (SpawnResult r : actionResult.spawnResults()) {
      countRunnerName(r.getRunnerName());
      totalWallTimeMillis.addAndGet(r.getMetrics().executionWallTime().toMillis());
    }
  }

  public void countRunnerName(String runner) {
    runners.add(runner);
  }

  public long getTotalWallTimeMillis() {
    return totalWallTimeMillis.get();
  }

  /*
   * Returns a human-readable summary of spawns counted.
   */
  public String getSummary() {
    ResultString result = new ResultString();

    // First report cache results.
    for (String s : REPORT_FIRST) {
      int count = runners.setCount(s, 0);
      if (count > 0) {
        result.add(s, count);
      }
    }

    // Sort the rest alphabetically
    ArrayList<Multiset.Entry<String>> list = new ArrayList<>(runners.entrySet());
    Collections.sort(list, Comparator.comparing(e -> e.getElement()));

    for (Multiset.Entry<String> e : list) {
      result.add(e.getElement(), e.getCount());
    }

    int total = result.spawnsCount();
    return total + " process" + (total == 1 ? "" : "es") + result + ".";
  }

  private static class ResultString {
    StringBuilder result = new StringBuilder();
    int spawnsCount = 0;
    int runnersNum = 0;

    public int spawnsCount() {
      return spawnsCount;
    }

    public void add(String name, int count) {
      spawnsCount += count;
      runnersNum += 1;

      if (result.length() > 0) {
        result.append(", ");
      }
      result.append(count);
      result.append(" ");
      result.append(name);
    }

    @Override
    public String toString() {
      if (runnersNum == 0) {
        return "";
      }
      return ": " + result;
    }
  }
}

@@ -1951,24 +1951,198 @@
 Summary() %7B%0A
+    return getSummary(REPORT_FIRST);%0A  %7D%0A%0A  /*%0A   * Returns a human-readable summary of spawns counted.%0A   */%0A  public String getSummary(ImmutableList%3CString%3E reportFirst) %7B%0A
     ResultSt
@@ -2175,16 +2175,16 @@
 ring();%0A
-
 %0A    // 
@@ -2231,28 +2231,27 @@
 ing s : 
-REPORT_FIRST
+reportFirst
 ) %7B%0A    

Allow customizing which runner name is reported first.