From 01916154d9e4cc65717797d51be867186b348ed4 Mon Sep 17 00:00:00 2001 From: stephen mcquay Date: Fri, 13 Mar 2015 11:30:52 -0700 Subject: [PATCH] use os.Getenv --- main.go | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/main.go b/main.go index 011e882..5df810c 100644 --- a/main.go +++ b/main.go @@ -16,15 +16,10 @@ func main() { log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile) binDirs := []string{} srcDirs := []string{} - for _, env := range os.Environ() { - if strings.HasPrefix(env, "GOPATH=") { - i := strings.Index(env, "=") - paths := filepath.SplitList(env[i+1:]) - for _, path := range paths { - binDirs = append(binDirs, filepath.Join(path, "bin")) - srcDirs = append(srcDirs, filepath.Join(path, "src")) - } - } + + for _, path := range filepath.SplitList(os.Getenv("GOPATH")) { + binDirs = append(binDirs, filepath.Join(path, "bin")) + srcDirs = append(srcDirs, filepath.Join(path, "src")) } files := []string{}