From: spf13 <steve.francia@gmail.com>
Date: Thu, 3 Oct 2013 13:44:45 +0000 (-0400)
Subject: Enabling Nitro ('--stepAnalysis') again. Fix #58
X-Git-Tag: v0.9~75
X-Git-Url: http://git.maquefel.me/?a=commitdiff_plain;h=dcfcbac5;p=brevno-suite%2Fhugo

Enabling Nitro ('--stepAnalysis') again. Fix #58
---

diff --git a/commands/hugo.go b/commands/hugo.go
index 21f4f0f9..b52e8533 100644
--- a/commands/hugo.go
+++ b/commands/hugo.go
@@ -19,6 +19,7 @@ import (
 	"github.com/mostafah/fsync"
 	"github.com/spf13/cobra"
 	"github.com/spf13/hugo/hugolib"
+	"github.com/spf13/nitro"
 	"log"
 	"os"
 	"path/filepath"
@@ -66,6 +67,7 @@ func init() {
 	HugoCmd.PersistentFlags().BoolVar(&UglyUrls, "uglyurls", false, "if true, use /filename.html instead of /filename/")
 	HugoCmd.PersistentFlags().StringVarP(&BaseUrl, "base-url", "b", "", "hostname (and path) to the root eg. http://spf13.com/")
 	HugoCmd.PersistentFlags().StringVar(&CfgFile, "config", "", "config file (default is path/config.yaml|json|toml)")
+	HugoCmd.PersistentFlags().BoolVar(&nitro.AnalysisOn, "stepAnalysis", false, "display memory and timing of different steps of the program")
 	HugoCmd.Flags().BoolVarP(&BuildWatch, "watch", "w", false, "watch filesystem for changes and recreate as needed")
 }
 
diff --git a/hugolib/site.go b/hugolib/site.go
index 3511b6a1..25ee4d3d 100644
--- a/hugolib/site.go
+++ b/hugolib/site.go
@@ -31,7 +31,7 @@ import (
 	"time"
 )
 
-var DefaultTimer = nitro.Initalize()
+var DefaultTimer *nitro.B
 
 func MakePermalink(base *url.URL, path *url.URL) *url.URL {
 	return base.ResolveReference(path)
@@ -78,6 +78,10 @@ type SiteInfo struct {
 	Config     *Config
 }
 
+func init() {
+	DefaultTimer = nitro.Initalize()
+}
+
 func (s *Site) timerStep(step string) {
 	if s.timer == nil {
 		s.timer = DefaultTimer