Commit 49bcc88f authored by Robert Griesemer's avatar Robert Griesemer

undo CL 4964067 / 661cb84cc6f0

API change. Needs further reflection.

««« original CL description
path/filepath: Simplify Walk interface

The last argument of filepath.Walk was removed, and the Visitor
interface now contains an Error method that is called on errors.

Fixes #2237.

R=golang-dev, gri, r
CC=golang-dev
https://golang.org/cl/4964067

»»»

R=r
CC=golang-dev
https://golang.org/cl/4974065
parent e5c20dc2
...@@ -200,7 +200,7 @@ func report(err os.Error) { ...@@ -200,7 +200,7 @@ func report(err os.Error) {
func walkDir(path string) { func walkDir(path string) {
v := make(fileVisitor) v := make(fileVisitor)
go func() { go func() {
filepath.Walk(path, v) filepath.Walk(path, v, v)
close(v) close(v)
}() }()
for err := range v { for err := range v {
...@@ -225,10 +225,6 @@ func (v fileVisitor) VisitFile(path string, f *os.FileInfo) { ...@@ -225,10 +225,6 @@ func (v fileVisitor) VisitFile(path string, f *os.FileInfo) {
} }
} }
func (v fileVisitor) Error(path string, err os.Error) {
v <- err
}
func isGoFile(f *os.FileInfo) bool { func isGoFile(f *os.FileInfo) bool {
// ignore non-Go files // ignore non-Go files
return f.IsRegular() && !strings.HasPrefix(f.Name, ".") && strings.HasSuffix(f.Name, ".go") return f.IsRegular() && !strings.HasPrefix(f.Name, ".") && strings.HasSuffix(f.Name, ".go")
......
...@@ -164,14 +164,10 @@ func (v fileVisitor) VisitFile(path string, f *os.FileInfo) { ...@@ -164,14 +164,10 @@ func (v fileVisitor) VisitFile(path string, f *os.FileInfo) {
} }
} }
func (v fileVisitor) Error(path string, err os.Error) {
v <- err
}
func walkDir(path string) { func walkDir(path string) {
v := make(fileVisitor) v := make(fileVisitor)
go func() { go func() {
filepath.Walk(path, v) filepath.Walk(path, v, v)
close(v) close(v)
}() }()
for err := range v { for err := range v {
......
...@@ -104,34 +104,30 @@ func doFile(name string, reader io.Reader) { ...@@ -104,34 +104,30 @@ func doFile(name string, reader io.Reader) {
// Visitor for filepath.Walk - trivial. Just calls doFile on each file. // Visitor for filepath.Walk - trivial. Just calls doFile on each file.
// TODO: if govet becomes richer, might want to process // TODO: if govet becomes richer, might want to process
// a directory (package) at a time. // a directory (package) at a time.
type fileVisitor chan os.Error type V struct{}
func (v fileVisitor) VisitDir(path string, f *os.FileInfo) bool { func (v V) VisitDir(path string, f *os.FileInfo) bool {
return true return true
} }
func (v fileVisitor) VisitFile(path string, f *os.FileInfo) { func (v V) VisitFile(path string, f *os.FileInfo) {
if strings.HasSuffix(path, ".go") { if strings.HasSuffix(path, ".go") {
doFile(path, nil) doFile(path, nil)
} }
} }
func (v fileVisitor) Error(path string, err os.Error) {
v <- err
}
// walkDir recursively walks the tree looking for .go files. // walkDir recursively walks the tree looking for .go files.
func walkDir(root string) { func walkDir(root string) {
v := make(fileVisitor) errors := make(chan os.Error)
done := make(chan bool) done := make(chan bool)
go func() { go func() {
for e := range v { for e := range errors {
errorf("walk error: %s", e) errorf("walk error: %s", e)
} }
done <- true done <- true
}() }()
filepath.Walk(root, v) filepath.Walk(root, V{}, errors)
close(v) close(errors)
<-done <-done
} }
......
...@@ -255,14 +255,14 @@ func Abs(path string) (string, os.Error) { ...@@ -255,14 +255,14 @@ func Abs(path string) (string, os.Error) {
} }
// Visitor methods are invoked for corresponding file tree entries // Visitor methods are invoked for corresponding file tree entries
// visited by Walk. // visited by Walk. The parameter path is the full path of f relative
// to root.
type Visitor interface { type Visitor interface {
VisitDir(path string, f *os.FileInfo) bool VisitDir(path string, f *os.FileInfo) bool
VisitFile(path string, f *os.FileInfo) VisitFile(path string, f *os.FileInfo)
Error(path string, err os.Error)
} }
func walk(path string, f *os.FileInfo, v Visitor) { func walk(path string, f *os.FileInfo, v Visitor, errors chan<- os.Error) {
if !f.IsDirectory() { if !f.IsDirectory() {
v.VisitFile(path, f) v.VisitFile(path, f)
return return
...@@ -274,11 +274,13 @@ func walk(path string, f *os.FileInfo, v Visitor) { ...@@ -274,11 +274,13 @@ func walk(path string, f *os.FileInfo, v Visitor) {
list, err := readDir(path) list, err := readDir(path)
if err != nil { if err != nil {
v.Error(path, err) if errors != nil {
errors <- err
}
} }
for _, e := range list { for _, e := range list {
walk(Join(path, e.Name), e, v) walk(Join(path, e.Name), e, v, errors)
} }
} }
...@@ -314,14 +316,18 @@ func (f fileInfoList) Swap(i, j int) { f[i], f[j] = f[j], f[i] } ...@@ -314,14 +316,18 @@ func (f fileInfoList) Swap(i, j int) { f[i], f[j] = f[j], f[i] }
// v.VisitFile for each directory or file in the tree, including root. // v.VisitFile for each directory or file in the tree, including root.
// If v.VisitDir returns false, Walk skips the directory's entries; // If v.VisitDir returns false, Walk skips the directory's entries;
// otherwise it invokes itself for each directory entry in sorted order. // otherwise it invokes itself for each directory entry in sorted order.
// Walk calls v.Error if an error happens while reading a directory. // An error reading a directory does not abort the Walk.
func Walk(root string, v Visitor) { // If errors != nil, Walk sends each directory read error
// to the channel. Otherwise Walk discards the error.
func Walk(root string, v Visitor, errors chan<- os.Error) {
f, err := os.Lstat(root) f, err := os.Lstat(root)
if err != nil { if err != nil {
v.Error(root, err) if errors != nil {
errors <- err
}
return // can't progress return // can't progress
} }
walk(root, f, v) walk(root, f, v, errors)
} }
// Base returns the last element of path. // Base returns the last element of path.
......
...@@ -299,30 +299,30 @@ func mark(name string) { ...@@ -299,30 +299,30 @@ func mark(name string) {
}) })
} }
type TestVisitor chan os.Error type TestVisitor struct{}
func (v TestVisitor) VisitDir(path string, f *os.FileInfo) bool { func (v *TestVisitor) VisitDir(path string, f *os.FileInfo) bool {
mark(f.Name) mark(f.Name)
return true return true
} }
func (v TestVisitor) VisitFile(path string, f *os.FileInfo) { func (v *TestVisitor) VisitFile(path string, f *os.FileInfo) {
mark(f.Name) mark(f.Name)
} }
func (v TestVisitor) Error(path string, err os.Error) {
v <- err
}
func TestWalk(t *testing.T) { func TestWalk(t *testing.T) {
makeTree(t) makeTree(t)
v := make(TestVisitor, 64) // 1) ignore error handling, expect none
v := &TestVisitor{}
filepath.Walk(tree.name, v, nil)
checkMarks(t)
// 1) no errors expected. // 2) handle errors, expect none
filepath.Walk(tree.name, v) errors := make(chan os.Error, 64)
filepath.Walk(tree.name, v, errors)
select { select {
case err := <-v: case err := <-errors:
t.Errorf("no error expected, found: %s", err) t.Errorf("no error expected, found: %s", err)
default: default:
// ok // ok
...@@ -343,13 +343,14 @@ func TestWalk(t *testing.T) { ...@@ -343,13 +343,14 @@ func TestWalk(t *testing.T) {
tree.entries[1].mark-- tree.entries[1].mark--
tree.entries[3].mark-- tree.entries[3].mark--
// 2) expect two errors // 3) handle errors, expect two
errors = make(chan os.Error, 64)
os.Chmod(filepath.Join(tree.name, tree.entries[1].name), 0) os.Chmod(filepath.Join(tree.name, tree.entries[1].name), 0)
filepath.Walk(tree.name, v) filepath.Walk(tree.name, v, errors)
Loop: Loop:
for i := 1; i <= 2; i++ { for i := 1; i <= 2; i++ {
select { select {
case <-v: case <-errors:
// ok // ok
default: default:
t.Errorf("%d. error expected, none found", i) t.Errorf("%d. error expected, none found", i)
...@@ -357,7 +358,7 @@ func TestWalk(t *testing.T) { ...@@ -357,7 +358,7 @@ func TestWalk(t *testing.T) {
} }
} }
select { select {
case err := <-v: case err := <-errors:
t.Errorf("only two errors expected, found 3rd: %v", err) t.Errorf("only two errors expected, found 3rd: %v", err)
default: default:
// ok // ok
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment