109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
// Copyright 2014 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Package buildutil provides utilities related to the go/build
|
|
// package in the standard library.
|
|
package buildutil
|
|
|
|
import (
|
|
"go/build"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
// AllPackages returns the import path of each Go package in any source
|
|
// directory of the specified build context (e.g. $GOROOT or an element
|
|
// of $GOPATH). Errors are ignored. The results are sorted.
|
|
//
|
|
// The result may include import paths for directories that contain no
|
|
// *.go files, such as "archive" (in $GOROOT/src).
|
|
//
|
|
// All I/O is via the build.Context virtual file system,
|
|
// which must be concurrency-safe.
|
|
//
|
|
func AllPackages(ctxt *build.Context) []string {
|
|
var list []string
|
|
var mu sync.Mutex
|
|
ForEachPackage(ctxt, func(pkg string, _ error) {
|
|
mu.Lock()
|
|
list = append(list, pkg)
|
|
mu.Unlock()
|
|
})
|
|
sort.Strings(list)
|
|
return list
|
|
}
|
|
|
|
// ForEachPackage calls the found function with the import path of
|
|
// each Go package it finds in any source directory of the specified
|
|
// build context (e.g. $GOROOT or an element of $GOPATH).
|
|
//
|
|
// If the package directory exists but could not be read, the second
|
|
// argument to the found function provides the error.
|
|
//
|
|
// The found function and the build.Context virtual file system
|
|
// accessors must be concurrency safe.
|
|
//
|
|
func ForEachPackage(ctxt *build.Context, found func(importPath string, err error)) {
|
|
var wg sync.WaitGroup
|
|
for _, root := range ctxt.SrcDirs() {
|
|
root := root
|
|
wg.Add(1)
|
|
go func() {
|
|
allPackages(ctxt, root, found)
|
|
wg.Done()
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func allPackages(ctxt *build.Context, root string, found func(string, error)) {
|
|
ReadDir := ctxt.ReadDir
|
|
if ReadDir == nil {
|
|
ReadDir = ioutil.ReadDir
|
|
}
|
|
|
|
root = filepath.Clean(root) + string(os.PathSeparator)
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
var walkDir func(dir string)
|
|
walkDir = func(dir string) {
|
|
// Prune search if we encounter any directory with these base names:
|
|
switch filepath.Base(dir) {
|
|
case "testdata", ".hg":
|
|
return
|
|
}
|
|
|
|
pkg := filepath.ToSlash(strings.TrimPrefix(dir, root))
|
|
|
|
// Prune search if we encounter any of these import paths.
|
|
switch pkg {
|
|
case "builtin":
|
|
return
|
|
}
|
|
|
|
files, err := ReadDir(dir)
|
|
if pkg != "" || err != nil {
|
|
found(pkg, err)
|
|
}
|
|
for _, fi := range files {
|
|
fi := fi
|
|
if fi.IsDir() {
|
|
wg.Add(1)
|
|
go func() {
|
|
walkDir(filepath.Join(dir, fi.Name()))
|
|
wg.Done()
|
|
}()
|
|
}
|
|
}
|
|
}
|
|
|
|
walkDir(root)
|
|
wg.Wait()
|
|
}
|