aicra/validator/builtin/int.go

59 lines
1.2 KiB
Go

package builtin
import (
"encoding/json"
"math"
"reflect"
"github.com/xdrm-io/aicra/validator"
)
// IntDataType is what its name tells
type IntDataType struct{}
// GoType returns the type of data
func (IntDataType) GoType() reflect.Type {
return reflect.TypeOf(int(0))
}
// Validator returns the validator
func (IntDataType) Validator(typeName string, registry ...validator.Type) validator.ValidateFunc {
// nothing if type not handled
if typeName != "int" {
return nil
}
return func(value interface{}) (interface{}, bool) {
switch cast := value.(type) {
case int:
return cast, true
case uint:
overflows := cast > math.MaxInt64
return int(cast), !overflows
case float64:
intVal := int(cast)
overflows := cast < float64(math.MinInt64) || cast > float64(math.MaxInt64)
return intVal, cast == float64(intVal) && !overflows
// serialized string -> try to convert to float
case string:
num := json.Number(cast)
intVal, err := num.Int64()
return int(intVal), err == nil
// serialized string -> try to convert to float
case []byte:
num := json.Number(cast)
intVal, err := num.Int64()
return int(intVal), err == nil
// unknown type
default:
return 0, false
}
}
}