add trace profiling

cmd/analyze.go

@@ -25,6 +25,7 @@ import (
 	"github.com/GoogleCloudPlatform/container-diff/differs"
 	pkgutil "github.com/GoogleCloudPlatform/container-diff/pkg/util"
 	"github.com/golang/glog"
+	"github.com/pkg/profile"
 	"github.com/spf13/cobra"
 )
 
@@ -42,6 +43,9 @@ var analyzeCmd = &cobra.Command{
 		return nil
 	},
 	Run: func(cmd *cobra.Command, args []string) {
+		if os.Getenv(containerDiffEnvPrefix) == "1" {
+			defer profile.Start(profile.TraceProfile).Stop()
+		}
 		if err := analyzeImage(args[0], strings.Split(types, ",")); err != nil {
 			glog.Error(err)
 			os.Exit(1)

cmd/diff.go

@@ -26,6 +26,7 @@ import (
 	"github.com/GoogleCloudPlatform/container-diff/differs"
 	pkgutil "github.com/GoogleCloudPlatform/container-diff/pkg/util"
 	"github.com/golang/glog"
+	"github.com/pkg/profile"
 	"github.com/spf13/cobra"
 )
 
@@ -43,6 +44,9 @@ var diffCmd = &cobra.Command{
 		return nil
 	},
 	Run: func(cmd *cobra.Command, args []string) {
+		if os.Getenv(containerDiffEnvPrefix) == "1" {
+			defer profile.Start(profile.TraceProfile).Stop()
+		}
 		if err := diffImages(args[0], args[1], strings.Split(types, ",")); err != nil {
 			glog.Error(err)
 			os.Exit(1)

cmd/root.go

@@ -41,6 +41,8 @@ type validatefxn func(args []string) error
 const (
 	DaemonPrefix = "daemon://"
 	RemotePrefix = "remote://"
+
+	containerDiffEnvPrefix = "CONTAINER_DIFF_ENABLE_PROFILING"
 )
 
 var RootCmd = &cobra.Command{