gorm/schema/check.go

33 lines
859 B
Go
Raw Normal View History

2020-02-22 06:15:51 +03:00
package schema
import (
"regexp"
"strings"
)
type Check struct {
Name string
Constraint string // length(phone) >= 10
*Field
}
// ParseCheckConstraints parse schema check constraints
func (schema *Schema) ParseCheckConstraints() map[string]Check {
var checks = map[string]Check{}
for _, field := range schema.FieldsByDBName {
if chk := field.TagSettings["CHECK"]; chk != "" {
names := strings.Split(chk, ",")
2020-02-22 14:41:01 +03:00
if len(names) > 1 && regexp.MustCompile("^[A-Za-z-_]+$").MatchString(names[0]) {
2020-02-22 06:15:51 +03:00
checks[names[0]] = Check{Name: names[0], Constraint: strings.Join(names[1:], ","), Field: field}
} else {
2020-02-22 14:41:01 +03:00
if names[0] == "" {
chk = strings.Join(names[1:], ",")
}
2020-02-22 06:15:51 +03:00
name := schema.namer.CheckerName(schema.Table, field.DBName)
checks[name] = Check{Name: name, Constraint: chk, Field: field}
}
}
}
return checks
}