Add optional param to run benchmark without presence of flag.

plentico · May 14, 2020 · 68de6d9 · 68de6d9
1 parent b793adc
commit 68de6d9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/cmd/build.go b/cmd/build.go
@@ -48,7 +48,7 @@ you need to deploy for your website.`,
 func Build() {
 
 	build.CheckBenchmarkFlag(BenchmarkFlag)
-	defer build.Benchmark(time.Now(), "Total build")
+	defer build.Benchmark(time.Now(), "Total build", true)
 
 	// Get settings from config file.
 	siteConfig := readers.GetSiteConfig()

diff --git a/cmd/build/benchmark.go b/cmd/build/benchmark.go
@@ -14,7 +14,13 @@ func CheckBenchmarkFlag(flag bool) {
 }
 
 // Benchmark records time for individual build processes.
-func Benchmark(start time.Time, message string) {
+func Benchmark(start time.Time, message string, alwaysRun ...bool) {
+
+	// Check variadic args (used to mimic optional parameters).
+	if len(alwaysRun) == 1 {
+		// Optionally run benchmark even if user didn't pass flag.
+		benchmarkFlag = alwaysRun[0]
+	}
 
 	elapsed := time.Since(start)