fix: invalid map preallocation size (#49)

tmzane · web-flow · commit a22690e6d52b · 2023-05-26T22:03:29.000+03:00
diff --git a/utils.go b/utils.go
@@ -23,9 +23,10 @@ func mainModule() (dir string, packages map[string]struct{}, _ error) {
 		return "", nil, fmt.Errorf("running `go list all`: %w", err)
 	}
 
-	list := strings.TrimSpace(string(out))
-	packages = make(map[string]struct{}, len(list))
-	for _, pkg := range strings.Split(list, "\n") {
+	list := strings.Split(strings.TrimSpace(string(out)), "\n")
+	packages = make(map[string]struct{}, len(list)*2)
+
+	for _, pkg := range list {
 		packages[pkg] = struct{}{}
 		packages[pkg+"_test"] = struct{}{} // `*_test` packages belong to the main module, see issue #24.
 	}

Original file line number	Diff line number	Diff line change
`@@ -23,9 +23,10 @@ func mainModule() (dir string, packages map[string]struct{}, _ error) {`
`23`	`23`	return "", nil, fmt.Errorf("running `go list all`: %w", err)
`24`	`24`	`}`
`25`	`25`
`26`		`- list := strings.TrimSpace(string(out))`
`27`		`- packages = make(map[string]struct{}, len(list))`
`28`		`- for _, pkg := range strings.Split(list, "\n") {`
	`26`	`+ list := strings.Split(strings.TrimSpace(string(out)), "\n")`
	`27`	`+ packages = make(map[string]struct{}, len(list)*2)`
	`28`	`+`
	`29`	`+ for _, pkg := range list {`
`29`	`30`	`packages[pkg] = struct{}{}`
`30`	`31`	packages[pkg+"_test"] = struct{}{} // `*_test` packages belong to the main module, see issue #24.
`31`	`32`	`}`