performance_test.go

Documentation: go/parser

		 1  // Copyright 2012 The Go Authors. All rights reserved.
		 2  // Use of this source code is governed by a BSD-style
		 3  // license that can be found in the LICENSE file.
		 4  
		 5  package parser
		 6  
		 7  import (
		 8  	"go/token"
		 9  	"os"
		10  	"testing"
		11  )
		12  
		13  // TODO(rFindley): use a testdata file or file from another package here, to
		14  //								 avoid a moving target.
		15  var src = readFile("parser.go")
		16  
		17  func readFile(filename string) []byte {
		18  	data, err := os.ReadFile(filename)
		19  	if err != nil {
		20  		panic(err)
		21  	}
		22  	return data
		23  }
		24  
		25  func BenchmarkParse(b *testing.B) {
		26  	b.SetBytes(int64(len(src)))
		27  	for i := 0; i < b.N; i++ {
		28  		if _, err := ParseFile(token.NewFileSet(), "", src, ParseComments); err != nil {
		29  			b.Fatalf("benchmark failed due to parse error: %s", err)
		30  		}
		31  	}
		32  }
		33  
		34  func BenchmarkParseOnly(b *testing.B) {
		35  	b.SetBytes(int64(len(src)))
		36  	for i := 0; i < b.N; i++ {
		37  		if _, err := ParseFile(token.NewFileSet(), "", src, ParseComments|SkipObjectResolution); err != nil {
		38  			b.Fatalf("benchmark failed due to parse error: %s", err)
		39  		}
		40  	}
		41  }
		42  
		43  func BenchmarkResolve(b *testing.B) {
		44  	b.SetBytes(int64(len(src)))
		45  	for i := 0; i < b.N; i++ {
		46  		b.StopTimer()
		47  		fset := token.NewFileSet()
		48  		file, err := ParseFile(fset, "", src, SkipObjectResolution)
		49  		if err != nil {
		50  			b.Fatalf("benchmark failed due to parse error: %s", err)
		51  		}
		52  		b.StartTimer()
		53  		handle := fset.File(file.Package)
		54  		resolveFile(file, handle, nil)
		55  	}
		56  }
		57
View as plain text