Only profile if asked

nlowe · Dec 19, 2018 · 7e06004 · 7e06004
1 parent b5081c8
commit 7e06004
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/aoc2018.go b/aoc2018.go
@@ -45,10 +45,14 @@ var rootCmd = &cobra.Command{
 		}
 		start = time.Now()
 
-		profiler = profile.Start()
+		if viper.GetBool("profile") {
+			profiler = profile.Start()
+		}
 	},
 	PersistentPostRun: func(_ *cobra.Command, _ []string) {
-		profiler.Stop()
+		if profiler != nil {
+			profiler.Stop()
+		}
 		fmt.Printf("Took %s\n", time.Since(start))
 	},
 }