Stowage/go
2
0
Fork 0
mirror of https://github.com/golang/go.git synced 2025-12-08 06:10:04 +00:00
Code Issues Projects Releases Packages Wiki Activity

cmd/gofmt: simplify arg handling

Browse source 
First, we can use flag.Args instead of flag.NArg and flag.Arg.

Second, just call filepath.WalkDir directly on each argument. We don't
need to check if each argument is a directory or not, since the function
will still work on regular files as expected.

To continue giving an error in the "gofmt does-not-exist.go" case, we
now need to return and handle errors from filepath.WalkDir, too.
Arguably, that should have always been the case.

While at it, I noticed that the printinf of the "diff" command did not
obey the "out" parameter. Fix that.

Finally, remove the code to ignore IsNotExist errors. It was added in CL
19301, though it didn't include tests and its reasoning is dubious.
Using gofmt on a directory treewhile another program is concurrently
editing or removing files is inherently racy. Hiding errors can hide
valid problems from the user, and such racy usages aren't supported.

Change-Id: I2e74cc04c53eeefb25231d804752b53562b97371
Reviewed-on: https://go-review.googlesource.com/c/go/+/284138
Run-TryBot: Daniel Martí <mvdan@mvdan.cc>
Trust: Daniel Martí <mvdan@mvdan.cc>
Reviewed-by: Bryan C. Mills <bcmills@google.com>
Reviewed-by: Robert Griesemer <gri@golang.org>
TryBot-Result: Go Bot <gobot@golang.org>
This commit is contained in:
Daniel Martí 2021-01-15 23:10:48 +00:00
parent 22a56b629d
commit a54762586f
1 changed files with 8 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

29
src/cmd/gofmt/gofmt.go
View file

@ -151,7 +151,7 @@ func processFile(filename string, in io.Reader, out io.Writer, stdin bool) error
if err != nil { if err != nil {
return fmt.Errorf("computing diff: %s", err) return fmt.Errorf("computing diff: %s", err)
} }
fmt.Printf("diff -u %s %s\n", filepath.ToSlash(filename+".orig"), filepath.ToSlash(filename)) fmt.Fprintf(out, "diff -u %s %s\n", filepath.ToSlash(filename+".orig"), filepath.ToSlash(filename))
out.Write(data) out.Write(data)
} }
} }
@ -164,21 +164,15 @@ func processFile(filename string, in io.Reader, out io.Writer, stdin bool) error
} }
func visitFile(path string, f fs.DirEntry, err error) error { func visitFile(path string, f fs.DirEntry, err error) error {
if err == nil && isGoFile(f) { if err != nil || !isGoFile(f) {
err = processFile(path, nil, os.Stdout, false) return err
} }
// Don't complain if a file was deleted in the meantime (i.e. if err := processFile(path, nil, os.Stdout, false); err != nil {
// the directory changed concurrently while running gofmt).
if err != nil && !os.IsNotExist(err) {
report(err) report(err)
} }
return nil return nil
} }
func walkDir(path string) {
filepath.WalkDir(path, visitFile)
}
func main() { func main() {
// call gofmtMain in a separate function // call gofmtMain in a separate function
// so that it can use defer and have them // so that it can use defer and have them
@ -206,7 +200,8 @@ func gofmtMain() {
initParserMode() initParserMode()
initRewrite() initRewrite()
if flag.NArg() == 0 { args := flag.Args()
if len(args) == 0 {
if *write { if *write {
fmt.Fprintln(os.Stderr, "error: cannot use -w with standard input") fmt.Fprintln(os.Stderr, "error: cannot use -w with standard input")
exitCode = 2 exitCode = 2
@ -218,17 +213,9 @@ func gofmtMain() {
return return
} }
for i := 0; i < flag.NArg(); i++ { for _, arg := range args {
path := flag.Arg(i) if err := filepath.WalkDir(arg, visitFile); err != nil {
switch dir, err := os.Stat(path); {
case err != nil:
report(err) report(err)
case dir.IsDir():
walkDir(path)
default:
if err := processFile(path, nil, os.Stdout, false); err != nil {
report(err)
}
} }
} }
} }