mirror of
https://github.com/golang/go
synced 2024-10-01 13:08:32 -06:00
19e2aca3fd
Extract goimports' logic for walking Go source directories into a separate package, suitable for use in go/packages. No functional changes. Added a convenience feature to fastwalk, allowing the user to say that they're done with a directory and stop receiving callbacks for it. Testing is a little light; I expect goimports' tests to cover most everything we care about. Change-Id: If047ada4414f5f282637d11fd07e8342fadc9c33 Reviewed-on: https://go-review.googlesource.com/c/138877 Run-TryBot: Heschi Kreinick <heschi@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Michael Matloob <matloob@golang.org> Reviewed-by: Brad Fitzpatrick <bradfitz@golang.org>
38 lines
909 B
Go
38 lines
909 B
Go
// Copyright 2016 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.
|
|
|
|
// +build appengine !linux,!darwin,!freebsd,!openbsd,!netbsd
|
|
|
|
package fastwalk
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
)
|
|
|
|
// readDir calls fn for each directory entry in dirName.
|
|
// It does not descend into directories or follow symlinks.
|
|
// If fn returns a non-nil error, readDir returns with that error
|
|
// immediately.
|
|
func readDir(dirName string, fn func(dirName, entName string, typ os.FileMode) error) error {
|
|
fis, err := ioutil.ReadDir(dirName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
skipFiles := false
|
|
for _, fi := range fis {
|
|
if fi.Mode().IsRegular() && skipFiles {
|
|
continue
|
|
}
|
|
if err := fn(dirName, fi.Name(), fi.Mode()&os.ModeType); err != nil {
|
|
if err == SkipFiles {
|
|
skipFiles = true
|
|
continue
|
|
}
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|