Quick 'n dirty profiling.

Author: nedtwigg

lib-extra/build.gradle

@@ -3,6 +3,9 @@ apply from: rootProject.file('gradle/java-setup.gradle')
 apply from: rootProject.file('gradle/java-publish.gradle')
 
 dependencies {
+	// temporary, for profiling
+	compile 'com.diffplug.durian:durian-debug:1.0.0'
+
 	compile project(':lib')
 	// misc useful utilities
 	compile "com.diffplug.durian:durian-core:${VER_DURIAN}"

lib/build.gradle

@@ -3,6 +3,8 @@ apply from: rootProject.file('gradle/java-setup.gradle')
 apply from: rootProject.file('gradle/java-publish.gradle')
 
 dependencies {
+	// temporary, for profiling
+	compile 'com.diffplug.durian:durian-debug:1.0.0'
 	// zero runtime reqs is a hard requirements for spotless-lib
 	// if you need a dep, put it in lib-extra
 	testCompile "junit:junit:${VER_JUNIT}"

lib/src/main/java/com/diffplug/spotless/FormatterStepImpl.java

@@ -20,6 +20,8 @@
 import java.util.Objects;
 import java.util.Random;
 
+import com.diffplug.common.debug.LapTimer;
+import com.diffplug.common.debug.StepProfiler;
 import com.diffplug.spotless.FormatterStep.Strict;
 
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
@@ -32,7 +34,9 @@
  * from the API.
  */
 @SuppressFBWarnings("SE_TRANSIENT_FIELD_NOT_RESTORED")
-abstract class FormatterStepImpl<Key extends Serializable> extends Strict<Key> {
+public abstract class FormatterStepImpl<Key extends Serializable> extends Strict<Key> {
+	public static final StepProfiler PROFILER = new StepProfiler(LapTimer.createNanoWrap2Sec());
+
 	private static final long serialVersionUID = 1L;
 
 	/** Transient because only the key matters. */
@@ -53,7 +57,12 @@ public String getName() {
 
 	@Override
 	protected Key calculateKey() throws Exception {
-		return keySupplier.get();
+		try {
+			PROFILER.startStep(name + " calculateKey");
+			return keySupplier.get();
+		} finally {
+			PROFILER.finish();
+		}
 	}
 
 	static final class Standard<Key extends Serializable> extends FormatterStepImpl<Key> {
@@ -70,9 +79,16 @@ static final class Standard<Key extends Serializable> extends FormatterStepImpl<
 		@Override
 		protected String format(Key key, String rawUnix, File file) throws Exception {
 			if (formatter == null) {
+				PROFILER.startStep(name + " createFormatter");
 				formatter = keyToFormatter.apply(key());
+				PROFILER.finish();
+			}
+			try {
+				PROFILER.startStep(name + " apply");
+				return formatter.apply(rawUnix);
+			} finally {
+				PROFILER.finish();
 			}
-			return formatter.apply(rawUnix);
 		}
 	}
 
@@ -90,15 +106,24 @@ static class Closeable<Key extends Serializable> extends FormatterStepImpl<Key>
 		@Override
 		protected String format(Key key, String rawUnix, File file) throws Exception {
 			if (formatter == null) {
+				PROFILER.startStep(name + " openFormatter");
 				formatter = keyToFormatter.apply(key());
+				PROFILER.finish();
+			}
+			try {
+				PROFILER.startStep(name + " apply");
+				return formatter.apply(rawUnix);
+			} finally {
+				PROFILER.finish();
 			}
-			return formatter.apply(rawUnix);
 		}
 
 		@Override
 		public void finish() {
 			if (formatter != null) {
+				PROFILER.startStep(name + " closeFormatter");
 				formatter.close();
+				PROFILER.finish();
 				formatter = null;
 			}
 		}

plugin-gradle/build.gradle

@@ -5,7 +5,9 @@ apply from: rootProject.file('gradle/java-publish.gradle')
 apply plugin: 'com.gradle.plugin-publish'
 apply plugin: 'java-gradle-plugin'
 
-dependencies {
+dependencies {	// temporary, for profiling
+	compile 'com.diffplug.durian:durian-debug:1.0.0'
+
 	compile project(':lib')
 	compile project(':lib-extra')
 	compile "com.diffplug.durian:durian-core:${VER_DURIAN}"

plugin-gradle/src/main/java/com/diffplug/gradle/spotless/ApplyFormatTask.java

@@ -20,6 +20,7 @@
 import org.gradle.api.tasks.TaskAction;
 
 import com.diffplug.spotless.Formatter;
+import com.diffplug.spotless.FormatterStepImpl;
 import com.diffplug.spotless.PaddedCellBulk;
 
 public class ApplyFormatTask extends BaseFormatTask {
@@ -35,5 +36,6 @@ public void apply() throws Exception {
 				formatter.applyFormat(file);
 			}
 		}
+		FormatterStepImpl.PROFILER.printResults();
 	}
 }