forked from ebhomengo/niki
1
0
Fork 0
niki/vendor/github.com/go-ozzo/ozzo-validation/each.go

131 lines
1.8 KiB
Go
Raw Normal View History

2024-02-18 10:42:21 +00:00
// Copyright 2016 Qiang Xue. All rights reserved.
2024-02-18 10:42:21 +00:00
// Use of this source code is governed by a MIT-style
2024-02-18 10:42:21 +00:00
// license that can be found in the LICENSE file.
package validation
import (
"errors"
"reflect"
"strconv"
)
// Each returns a validation rule that loops through an iterable (map, slice or array)
2024-02-18 10:42:21 +00:00
// and validates each value inside with the provided rules.
2024-02-18 10:42:21 +00:00
// An empty iterable is considered valid. Use the Required rule to make sure the iterable is not empty.
2024-02-18 10:42:21 +00:00
func Each(rules ...Rule) *EachRule {
2024-02-18 10:42:21 +00:00
return &EachRule{
2024-02-18 10:42:21 +00:00
rules: rules,
}
2024-02-18 10:42:21 +00:00
}
type EachRule struct {
rules []Rule
}
// Loops through the given iterable and calls the Ozzo Validate() method for each value.
2024-02-18 10:42:21 +00:00
func (r *EachRule) Validate(value interface{}) error {
2024-02-18 10:42:21 +00:00
errs := Errors{}
v := reflect.ValueOf(value)
2024-02-18 10:42:21 +00:00
switch v.Kind() {
2024-02-18 10:42:21 +00:00
case reflect.Map:
2024-02-18 10:42:21 +00:00
for _, k := range v.MapKeys() {
2024-02-18 10:42:21 +00:00
val := r.getInterface(v.MapIndex(k))
2024-02-18 10:42:21 +00:00
if err := Validate(val, r.rules...); err != nil {
2024-02-18 10:42:21 +00:00
errs[r.getString(k)] = err
2024-02-18 10:42:21 +00:00
}
2024-02-18 10:42:21 +00:00
}
2024-02-18 10:42:21 +00:00
case reflect.Slice, reflect.Array:
2024-02-18 10:42:21 +00:00
for i := 0; i < v.Len(); i++ {
2024-02-18 10:42:21 +00:00
val := r.getInterface(v.Index(i))
2024-02-18 10:42:21 +00:00
if err := Validate(val, r.rules...); err != nil {
2024-02-18 10:42:21 +00:00
errs[strconv.Itoa(i)] = err
2024-02-18 10:42:21 +00:00
}
2024-02-18 10:42:21 +00:00
}
2024-02-18 10:42:21 +00:00
default:
2024-02-18 10:42:21 +00:00
return errors.New("must be an iterable (map, slice or array)")
2024-02-18 10:42:21 +00:00
}
if len(errs) > 0 {
2024-02-18 10:42:21 +00:00
return errs
2024-02-18 10:42:21 +00:00
}
2024-02-18 10:42:21 +00:00
return nil
2024-02-18 10:42:21 +00:00
}
func (r *EachRule) getInterface(value reflect.Value) interface{} {
2024-02-18 10:42:21 +00:00
switch value.Kind() {
2024-02-18 10:42:21 +00:00
case reflect.Ptr, reflect.Interface:
2024-02-18 10:42:21 +00:00
if value.IsNil() {
2024-02-18 10:42:21 +00:00
return nil
2024-02-18 10:42:21 +00:00
}
2024-02-18 10:42:21 +00:00
return value.Elem().Interface()
2024-02-18 10:42:21 +00:00
default:
2024-02-18 10:42:21 +00:00
return value.Interface()
2024-02-18 10:42:21 +00:00
}
2024-02-18 10:42:21 +00:00
}
func (r *EachRule) getString(value reflect.Value) string {
2024-02-18 10:42:21 +00:00
switch value.Kind() {
2024-02-18 10:42:21 +00:00
case reflect.Ptr, reflect.Interface:
2024-02-18 10:42:21 +00:00
if value.IsNil() {
2024-02-18 10:42:21 +00:00
return ""
2024-02-18 10:42:21 +00:00
}
2024-02-18 10:42:21 +00:00
return value.Elem().String()
2024-02-18 10:42:21 +00:00
default:
2024-02-18 10:42:21 +00:00
return value.String()
2024-02-18 10:42:21 +00:00
}
2024-02-18 10:42:21 +00:00
}