Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feat: tag to select child field #1235

Open
wants to merge 5 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions cache.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,12 +21,14 @@ const (
typeKeys
typeEndKeys
typeOmitNil
typeSelect
)

const (
invalidValidation = "Invalid validation tag on field '%s'"
undefinedValidation = "Undefined validation function '%s' on field '%s'"
keysTagNotDefined = "'" + endKeysTag + "' tag encountered without a corresponding '" + keysTag + "' tag"
invalidSelectTag = "'select' tags must have exactly one value"
)

type structCache struct {
Expand Down Expand Up @@ -266,6 +268,22 @@ func (v *Validate) parseFieldTagsRecursive(tag string, fieldName string, alias s
continue

default:
if strings.HasPrefix(t, selectTag) {
vals := strings.SplitN(t, tagKeySeparator, 2)

// Check again for exact match to prevent future conflicts
if vals[0] == selectTag {
if len(vals) == 1 {
panic(invalidSelectTag)
}

current.typeof = typeSelect
current.hasParam = true
current.param = vals[1]
continue
}
}

if t == isdefault {
current.typeof = typeIsDefault
}
Expand Down
14 changes: 14 additions & 0 deletions doc.go
Original file line number Diff line number Diff line change
Expand Up @@ -249,6 +249,20 @@ Example #2
// eq=1|eq=2 will be applied to each array element in the map keys
// required will be applied to map values

# Select

Selects a struct field or map value for which the following tags will be applied.
It is similar to the dive tags for arrays/slices/maps except that it only applies to a single struct field or map value.

Usage: select=FieldName

Example:

// Validates that the field "V" of "MyStruct.Field" is greater than 10 when "Field" is valid
type MyStruct struct {
Field sql.Null[uint] `validate:"omitempty,select=V,gt=10"`
}

# Required

This validates that the value is not the data types default zero value.
Expand Down
54 changes: 53 additions & 1 deletion validator.go
Original file line number Diff line number Diff line change
Expand Up @@ -445,8 +445,60 @@ OUTER:
ct = ct.next
}

default:
case typeSelect:
var name, altName string
var fieldValue reflect.Value
switch kind {
case reflect.Struct:

v.misc = append(v.misc[0:0], cf.name...)
v.misc = append(v.misc, '.')
v.misc = append(v.misc, ct.param...)
name = string(v.misc)

if cf.namesEqual {
altName = name
} else {
v.misc = append(v.misc[0:0], cf.altName...)
v.misc = append(v.misc, '.')
v.misc = append(v.misc, ct.param...)
altName = string(v.misc)
}

fieldValue = current.FieldByName(ct.param)

case reflect.Map:

v.misc = append(v.misc[0:0], cf.name...)
v.misc = append(v.misc, '[')
v.misc = append(v.misc, ct.param...)
v.misc = append(v.misc, ']')
name = string(v.misc)

if cf.namesEqual {
altName = name
} else {
v.misc = append(v.misc[0:0], cf.altName...)
v.misc = append(v.misc, '[')
v.misc = append(v.misc, ct.param...)
v.misc = append(v.misc, ']')
altName = string(v.misc)
}

fieldValue = current.MapIndex(reflect.ValueOf(ct.param))

default:
panic("can't select field on a non struct or map types")
}

v.traverseField(ctx, parent, fieldValue, ns, structNs, &cField{
altName: altName,
name: name,
}, ct.next)

return

default:
// set Field Level fields
v.slflParent = parent
v.flField = current
Expand Down
1 change: 1 addition & 0 deletions validator_instance.go
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ const (
excludedIfTag = "excluded_if"
excludedUnlessTag = "excluded_unless"
skipValidationTag = "-"
selectTag = "select"
diveTag = "dive"
keysTag = "keys"
endKeysTag = "endkeys"
Expand Down
46 changes: 46 additions & 0 deletions validator_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -13794,3 +13794,49 @@ func TestPrivateFieldsStruct(t *testing.T) {
Equal(t, len(errs), tc.errorNum)
}
}

func TestSelectTag(t *testing.T) {
validator := New(WithRequiredStructEnabled())

t.Run("on struct", func(t *testing.T) {
type Test struct {
Int sql.NullInt64 `validate:"required,select=Int64,gt=1"`
}

validCase := Test{sql.NullInt64{Int64: 2}}
zeroCase := Test{}
invalidCase := Test{sql.NullInt64{Int64: 1}}

Equal(t, validator.Struct(validCase), nil)
AssertError(t, validator.Struct(zeroCase), "Test.Int", "Test.Int", "Int", "Int", "required")
AssertError(t, validator.Struct(invalidCase), "Test.Int.Int64", "Test.Int.Int64", "Int.Int64", "Int.Int64", "gt")
})

t.Run("on map", func(t *testing.T) {
type Test struct {
Map map[string]int `validate:"required,select=key,gt=1"`
}

validCase := Test{map[string]int{"key": 2}}
zeroCase := Test{}
invalidCase := Test{map[string]int{"key": 1}}

Equal(t, validator.Struct(validCase), nil)
AssertError(t, validator.Struct(zeroCase), "Test.Map", "Test.Map", "Map", "Map", "required")
AssertError(t, validator.Struct(invalidCase), "Test.Map[key]", "Test.Map[key]", "Map[key]", "Map[key]", "gt")
})

t.Run("missing select value", func(t *testing.T) {
type Test struct {
Int sql.NullInt64 `validate:"required,select"`
}

defer func() {
if r := recover(); r != invalidSelectTag {
t.Errorf("Expected panic %q, got %v", invalidSelectTag, r)
}
}()

_ = validator.Struct(Test{})
})
}