qbit/go
1
0
Fork 0
mirror of https://github.com/golang/go synced 2024-10-03 13:31:21 -06:00
Code Releases Activity

gofmt: handle comments correctly in rewrites

Browse Source 
R=rsc
CC=golang-dev
https://golang.org/cl/6294076
This commit is contained in:
Robert Griesemer 2012-06-25 13:58:28 -07:00
parent 51ff2ef409
commit 96a609c2d7
4 changed files with 35 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/cmd/gofmt/gofmt_test.go
View File

@ -66,7 +66,6 @@ func runTest(t *testing.T, in, out, flags string) {
}
}
// TODO(gri) Add more test cases!
var tests = []struct {
in, flags string
}{
@ -78,6 +77,7 @@ var tests = []struct {
{"testdata/rewrite2.input", "-r=int->bool"},
{"testdata/rewrite3.input", "-r=x->x"},
{"testdata/rewrite4.input", "-r=(x)->x"},
{"testdata/rewrite5.input", "-r=x+x->2*x"},
{"testdata/stdin*.input", "-stdin"},
{"testdata/comments.input", ""},
{"testdata/import.input", ""},

5
src/cmd/gofmt/rewrite.go
View File

@ -55,6 +55,7 @@ func dump(msg string, val reflect.Value) {
// rewriteFile applies the rewrite rule 'pattern -> replace' to an entire file.
func rewriteFile(pattern, replace ast.Expr, p *ast.File) *ast.File {
cmap := ast.NewCommentMap(fileSet, p, p.Comments)
m := make(map[string]reflect.Value)
pat := reflect.ValueOf(pattern)
repl := reflect.ValueOf(replace)
@ -73,7 +74,9 @@ func rewriteFile(pattern, replace ast.Expr, p *ast.File) *ast.File {
}
return val
}
return apply(f, reflect.ValueOf(p)).Interface().(*ast.File)
r := apply(f, reflect.ValueOf(p)).Interface().(*ast.File)
r.Comments = cmap.Filter(r).Comments() // recreate comments list
return r
}
// setValue is a wrapper for x.SetValue(y); it protects

15
src/cmd/gofmt/testdata/rewrite5.golden vendored Normal file
View File

@ -0,0 +1,15 @@
// Copyright 2011 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.
// Rewriting of expressions containing nodes with associated comments to
// expressions without those nodes must also eliminate the associated
// comments.
package p
func f(x int) int {
_ = 2 * x // this comment remains in the rewrite
_ = 2 * x
return 2 * x
}

15
src/cmd/gofmt/testdata/rewrite5.input vendored Normal file
View File

@ -0,0 +1,15 @@
// Copyright 2011 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.
// Rewriting of expressions containing nodes with associated comments to
// expressions without those nodes must also eliminate the associated
// comments.
package p
func f(x int) int {
_ = x + x // this comment remains in the rewrite
_ = x /* this comment must not be in the rewrite */ + x
return x /* this comment must not be in the rewrite */ + x
}