52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package golinters
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
|
|
gocycloAPI "github.com/golangci/gocyclo/pkg/gocyclo"
|
|
"github.com/golangci/golangci-lint/pkg/lint/linter"
|
|
"github.com/golangci/golangci-lint/pkg/result"
|
|
)
|
|
|
|
type Gocyclo struct{}
|
|
|
|
func (Gocyclo) Name() string {
|
|
return "gocyclo"
|
|
}
|
|
|
|
func (Gocyclo) Desc() string {
|
|
return "Computes and checks the cyclomatic complexity of functions"
|
|
}
|
|
|
|
func (g Gocyclo) Run(ctx context.Context, lintCtx *linter.Context) ([]result.Issue, error) {
|
|
var stats []gocycloAPI.Stat
|
|
for _, f := range lintCtx.ASTCache.GetAllValidFiles() {
|
|
stats = gocycloAPI.BuildStats(f.F, f.Fset, stats)
|
|
}
|
|
if len(stats) == 0 {
|
|
return nil, nil
|
|
}
|
|
|
|
sort.Slice(stats, func(i, j int) bool {
|
|
return stats[i].Complexity > stats[j].Complexity
|
|
})
|
|
|
|
res := make([]result.Issue, 0, len(stats))
|
|
for _, s := range stats {
|
|
if s.Complexity <= lintCtx.Settings().Gocyclo.MinComplexity {
|
|
break //Break as the stats is already sorted from greatest to least
|
|
}
|
|
|
|
res = append(res, result.Issue{
|
|
Pos: s.Pos,
|
|
Text: fmt.Sprintf("cyclomatic complexity %d of func %s is high (> %d)",
|
|
s.Complexity, formatCode(s.FuncName, lintCtx.Cfg), lintCtx.Settings().Gocyclo.MinComplexity),
|
|
FromLinter: g.Name(),
|
|
})
|
|
}
|
|
|
|
return res, nil
|
|
}
|