From 36a9095c12ab905ec0127a50c6f03679fccc8230 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Leandro=20L=C3=B3pez=20=28inkel=29?= Date: Thu, 27 Jun 2024 10:02:06 -0300 Subject: [PATCH] feat(perf): Precompile getPattern regexps MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit This greatly improves the allocations: $ go test -benchmem -bench=. goos: darwin goarch: arm64 pkg: github.com/hairyhenderson/go-codeowners BenchmarkParseCodeowners-10 70927 16857 ns/op 27211 B/op 242 allocs/op PASS ok github.com/hairyhenderson/go-codeowners 5.297s This reduces the allocations in 62%: $ benchstat main.log precompile.log goos: darwin goarch: arm64 pkg: github.com/hairyhenderson/go-codeowners │ main.log │ precompile.log │ │ sec/op │ sec/op vs base │ ParseCodeowners-10 38.72µ ± 2% 17.63µ ± 6% -54.46% (p=0.000 n=10) │ main.log │ precompile.log │ │ B/op │ B/op vs base │ ParseCodeowners-10 59.95Ki ± 0% 26.56Ki ± 0% -55.70% (p=0.000 n=10) │ main.log │ precompile.log │ │ allocs/op │ allocs/op vs base │ ParseCodeowners-10 641.0 ± 0% 242.0 ± 0% -62.25% (p=0.000 n=10) --- codeowners.go | 28 ++++++++++++++++++++-------- 1 file changed, 20 insertions(+), 8 deletions(-) diff --git a/codeowners.go b/codeowners.go index c0a9f6f..7327f18 100644 --- a/codeowners.go +++ b/codeowners.go @@ -223,21 +223,33 @@ func (c *Codeowners) Owners(path string) []string { return nil } +// precompile all regular expressions +var ( + reCommentIgnore = regexp.MustCompile(`^(\\#|\\!)`) + rePrependSlash = regexp.MustCompile(`([^\/+])/.*\*\.`) + reEscapeDot = regexp.MustCompile(`\.`) + reDoubleStar1 = regexp.MustCompile(`/\*\*/`) + reDoubleStar2 = regexp.MustCompile(`\*\*/`) + reDoubleStar3 = regexp.MustCompile(`/\*\*`) + reEscapeStar1 = regexp.MustCompile(`\\\*`) + reEscapeStar2 = regexp.MustCompile(`\*`) +) + // based on github.com/sabhiram/go-gitignore // but modified so that 'dir/*' only matches files in 'dir/' func getPattern(line string) (*regexp.Regexp, error) { // when # or ! is escaped with a \ - if regexp.MustCompile(`^(\\#|\\!)`).MatchString(line) { + if reCommentIgnore.MatchString(line) { line = line[1:] } // If we encounter a foo/*.blah in a folder, prepend the / char - if regexp.MustCompile(`([^\/+])/.*\*\.`).MatchString(line) && line[0] != '/' { + if rePrependSlash.MatchString(line) && line[0] != '/' { line = "/" + line } // Handle escaping the "." char - line = regexp.MustCompile(`\.`).ReplaceAllString(line, `\.`) + line = reEscapeDot.ReplaceAllString(line, `\.`) magicStar := "#$~" @@ -245,13 +257,13 @@ func getPattern(line string) (*regexp.Regexp, error) { if strings.HasPrefix(line, "/**/") { line = line[1:] } - line = regexp.MustCompile(`/\*\*/`).ReplaceAllString(line, `(/|/.+/)`) - line = regexp.MustCompile(`\*\*/`).ReplaceAllString(line, `(|.`+magicStar+`/)`) - line = regexp.MustCompile(`/\*\*`).ReplaceAllString(line, `(|/.`+magicStar+`)`) + line = reDoubleStar1.ReplaceAllString(line, `(/|/.+/)`) + line = reDoubleStar2.ReplaceAllString(line, `(|.`+magicStar+`/)`) + line = reDoubleStar3.ReplaceAllString(line, `(|/.`+magicStar+`)`) // Handle escaping the "*" char - line = regexp.MustCompile(`\\\*`).ReplaceAllString(line, `\`+magicStar) - line = regexp.MustCompile(`\*`).ReplaceAllString(line, `([^/]*)`) + line = reEscapeStar1.ReplaceAllString(line, `\`+magicStar) + line = reEscapeStar2.ReplaceAllString(line, `([^/]*)`) // Handle escaping the "?" char line = strings.ReplaceAll(line, "?", `\?`)