Add module support

uudashr · Sep 23, 2018 · 80818a3 · 80818a3
1 parent f5479ab
commit 80818a3
Show file tree

Hide file tree

Showing 3 changed files with 254 additions and 94 deletions.
diff --git a/cmd/gopkgs/main.go b/cmd/gopkgs/main.go
@@ -46,8 +46,8 @@ func main() {
 		flagPerfTrace      *string
 	)
 
-	devMode := os.Getenv("DEV_MODE")
-	if devMode == "1" || devMode == "true" {
+	envDevMode := os.Getenv("DEV_MODE")
+	if envDevMode == "1" || envDevMode == "true" || envDevMode == "on" {
 		flagPerfCPUProfile = flag.String("perf-cpuprofile", "", "Write the CPU profile to a file")
 		flagPerfTrace = flag.String("perf-trace", "", "Write an execution trace to a file")
 	}
@@ -106,7 +106,7 @@ func main() {
 		os.Exit(1)
 	}
 
-	pkgs, err := gopkgs.Packages(gopkgs.Options{
+	pkgs, err := gopkgs.List(gopkgs.Options{
 		WorkDir:  *flagWorkDir,
 		NoVendor: *flagNoVendor,
 	})