Merge pull request #111 from nkubala/profile

add trace profiling

Author: nkubala

BUILD

@@ -19,6 +19,7 @@ go_library(
     deps = [
         "//cmd:go_default_library",
         "//vendor/github.com/golang/glog:go_default_library",
+        "//vendor/github.com/pkg/profile:go_default_library",
     ],
 )
 

Godeps/Godeps.json

@@ -23,10 +23,16 @@ import (
 
 	"github.com/GoogleCloudPlatform/container-diff/cmd"
 	"github.com/golang/glog"
+	"github.com/pkg/profile"
 )
 
+const containerDiffEnvPrefix = "CONTAINER_DIFF_ENABLE_PROFILING"
+
 func main() {
 	flag.Parse()
+	if os.Getenv(containerDiffEnvPrefix) == "1" {
+		defer profile.Start(profile.TraceProfile).Stop()
+	}
 	if err := cmd.RootCmd.Execute(); err != nil {
 		fmt.Println(err)
 		os.Exit(1)