65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package golinters
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
|
|
"github.com/alexkohler/prealloc/pkg"
|
|
"golang.org/x/tools/go/analysis"
|
|
|
|
"github.com/golangci/golangci-lint/pkg/config"
|
|
"github.com/golangci/golangci-lint/pkg/golinters/goanalysis"
|
|
"github.com/golangci/golangci-lint/pkg/lint/linter"
|
|
"github.com/golangci/golangci-lint/pkg/result"
|
|
)
|
|
|
|
const preallocName = "prealloc"
|
|
|
|
func NewPreAlloc(settings *config.PreallocSettings) *goanalysis.Linter {
|
|
var mu sync.Mutex
|
|
var resIssues []goanalysis.Issue
|
|
|
|
analyzer := &analysis.Analyzer{
|
|
Name: preallocName,
|
|
Doc: goanalysis.TheOnlyanalyzerDoc,
|
|
Run: func(pass *analysis.Pass) (any, error) {
|
|
issues := runPreAlloc(pass, settings)
|
|
|
|
if len(issues) == 0 {
|
|
return nil, nil
|
|
}
|
|
|
|
mu.Lock()
|
|
resIssues = append(resIssues, issues...)
|
|
mu.Unlock()
|
|
|
|
return nil, nil
|
|
},
|
|
}
|
|
|
|
return goanalysis.NewLinter(
|
|
preallocName,
|
|
"Finds slice declarations that could potentially be pre-allocated",
|
|
[]*analysis.Analyzer{analyzer},
|
|
nil,
|
|
).WithIssuesReporter(func(*linter.Context) []goanalysis.Issue {
|
|
return resIssues
|
|
}).WithLoadMode(goanalysis.LoadModeSyntax)
|
|
}
|
|
|
|
func runPreAlloc(pass *analysis.Pass, settings *config.PreallocSettings) []goanalysis.Issue {
|
|
var issues []goanalysis.Issue
|
|
|
|
hints := pkg.Check(pass.Files, settings.Simple, settings.RangeLoops, settings.ForLoops)
|
|
|
|
for _, hint := range hints {
|
|
issues = append(issues, goanalysis.NewIssue(&result.Issue{
|
|
Pos: pass.Fset.Position(hint.Pos),
|
|
Text: fmt.Sprintf("Consider pre-allocating %s", formatCode(hint.DeclaredSliceName, nil)),
|
|
FromLinter: preallocName,
|
|
}, pass))
|
|
}
|
|
|
|
return issues
|
|
}
|