50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
package foreignKeyCheck
|
|
|
|
import (
|
|
"golang.org/x/tools/go/analysis"
|
|
"gormlint/common"
|
|
"strings"
|
|
)
|
|
|
|
// ForeignKeyCheck todo: add URL for foreign key analyzer rules
|
|
var ForeignKeyCheck = &analysis.Analyzer{
|
|
Name: "GormForeignKeyCheck",
|
|
Doc: "Check foreign key in gorm model struct tag",
|
|
Run: run,
|
|
}
|
|
|
|
var models map[string]common.Model
|
|
|
|
func run(pass *analysis.Pass) (any, error) {
|
|
models = make(map[string]common.Model)
|
|
common.ParseModels(pass, &models)
|
|
|
|
for _, model := range models {
|
|
for _, field := range model.Fields {
|
|
for _, param := range field.Params {
|
|
pair := strings.Split(param, ":")
|
|
paramName := pair[0]
|
|
paramValue := pair[1]
|
|
if paramName == "foreignKey" {
|
|
foreignKey, fieldExist := model.Fields[paramValue]
|
|
|
|
if !fieldExist {
|
|
pass.Reportf(field.Pos, "Foreign key \"%s\" mentioned in tag at field \"%s\" doesn't exist in model \"%s\"", paramValue, field.Name, model.Name)
|
|
} else {
|
|
foreignKeyType := common.ResolveBaseType(foreignKey.Type)
|
|
if foreignKeyType == nil {
|
|
pass.Reportf(foreignKey.Pos, "Failed to resolve type of foreign key field \"%s\": %s", field.Name, foreignKey.Type)
|
|
} else {
|
|
// TODO: handle all int types
|
|
if *foreignKeyType != "uint" && *foreignKeyType != "int" {
|
|
pass.Reportf(foreignKey.Pos, "Foreign key should have type like int, not \"%s\"", foreignKey.Type)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|