lucene/gradle/validation/measure-task-times.gradle

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

70 lines
2.6 KiB
Groovy
Raw Normal View History

/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.
*/
import java.util.concurrent.ConcurrentHashMap
import java.util.stream.Collectors
// Measure aggregate wall time of each task execution time (summed up across
// all projects).
if (propertyOrDefault("task.times", null) != null) {
gradle.taskGraph.whenReady { TaskExecutionGraph graph ->
def taskTimes = new ConcurrentHashMap<String, Long>()
def measured = graph.allTasks.findAll { task -> true }
graph.addTaskExecutionListener(new TaskExecutionListener() {
@Override
void beforeExecute(Task task) {
def start = System.currentTimeMillis()
if (measured.contains(task)) {
taskTimes.put(task.path, start)
}
}
@Override
void afterExecute(Task task, TaskState taskState) {
def stop = System.currentTimeMillis()
if (measured.contains(task)) {
if (taskState.didWork) {
taskTimes.compute(task.path, { k, v -> stop - v })
} else {
taskTimes.remove(task.path)
}
}
}
})
// After the build is finished, check the test count.
gradle.buildFinished {
// Compute aggregate times of all task execution times.
def aggregates = taskTimes.entrySet().stream()
.collect(Collectors.groupingBy({e -> e.key.replaceAll(".*:", "")}))
.entrySet().stream()
.collect(Collectors.toMap(
{e -> e.key },
{e -> e.value.stream().mapToLong({ entry -> entry.value }).sum() }
))
logger.lifecycle("Aggregate task times (possibly running in parallel!):")
aggregates.entrySet().stream()
.sorted({a, b -> Long.compare(b.value, a.value) })
.forEach {v ->
logger.lifecycle(String.format(Locale.ROOT, " %6.2f sec. %s", v.value / 1000.0, v.key))
}
}
}
}