fix: types in m2m
This commit is contained in:
@@ -1,19 +1,18 @@
|
||||
package common
|
||||
|
||||
import (
|
||||
"github.com/kuzgoga/fogg"
|
||||
"go/ast"
|
||||
"go/token"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Field struct {
|
||||
Name string
|
||||
Type ast.Expr
|
||||
Tags *string
|
||||
Options []string // contains options like "autoCreateTime" or "null"
|
||||
Params []string // contains params like "foreignKey:CustomerId" or "constrain:OnDelete:Cascade"
|
||||
Tag *string
|
||||
Pos token.Pos
|
||||
Comment string
|
||||
Tags fogg.Tag
|
||||
}
|
||||
|
||||
type Model struct {
|
||||
@@ -27,45 +26,3 @@ type Param struct {
|
||||
Name string
|
||||
Value string
|
||||
}
|
||||
|
||||
func (model *Model) GetParam(name string) *Param {
|
||||
for _, field := range model.Fields {
|
||||
for _, param := range field.Params {
|
||||
pair := strings.SplitN(param, ":", 2)
|
||||
if len(pair) != 2 {
|
||||
return nil
|
||||
}
|
||||
if strings.ToLower(pair[0]) == strings.ToLower(name) {
|
||||
return &Param{
|
||||
Name: pair[0],
|
||||
Value: pair[1],
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (model *Model) HasParam(name string) bool {
|
||||
return model.GetParam(name) != nil
|
||||
}
|
||||
|
||||
func (field *Field) HasParam(name string) bool {
|
||||
return field.GetParam(name) != nil
|
||||
}
|
||||
|
||||
func (field *Field) GetParam(name string) *Param {
|
||||
for _, param := range field.Params {
|
||||
pair := strings.SplitN(param, ":", 2)
|
||||
if len(pair) != 2 {
|
||||
return nil
|
||||
}
|
||||
if strings.ToLower(pair[0]) == strings.ToLower(name) {
|
||||
return &Param{
|
||||
Name: pair[0],
|
||||
Value: pair[1],
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user