cast json.Number

cast_test.go

@@ -6,6 +6,7 @@
 package cast
 
 import (
+	"encoding/json"
 	"fmt"
 	"html/template"
 	"testing"
@@ -30,6 +31,7 @@ func TestToUintE(t *testing.T) {
 		{uint16(8), 8, false},
 		{uint32(8), 8, false},
 		{uint64(8), 8, false},
+		{json.Number("8"), 8, false},
 		{float32(8.31), 8, false},
 		{float64(8.31), 8, false},
 		{true, 1, false},
@@ -45,6 +47,7 @@ func TestToUintE(t *testing.T) {
 		{float32(-8.31), 0, true},
 		{float64(-8.31), 0, true},
 		{"-8", 0, true},
+		{json.Number("-8"), 0, true},
 		{"test", 0, true},
 		{testing.T{}, 0, true},
 	}
@@ -83,6 +86,7 @@ func TestToUint64E(t *testing.T) {
 		{uint16(8), 8, false},
 		{uint32(8), 8, false},
 		{uint64(8), 8, false},
+		{json.Number("8"), 8, false},
 		{float32(8.31), 8, false},
 		{float64(8.31), 8, false},
 		{true, 1, false},
@@ -98,6 +102,7 @@ func TestToUint64E(t *testing.T) {
 		{float32(-8.31), 0, true},
 		{float64(-8.31), 0, true},
 		{"-8", 0, true},
+		{json.Number("-8"), 0, true},
 		{"test", 0, true},
 		{testing.T{}, 0, true},
 	}
@@ -136,12 +141,14 @@ func TestToUint32E(t *testing.T) {
 		{uint16(8), 8, false},
 		{uint32(8), 8, false},
 		{uint64(8), 8, false},
+		{json.Number("8"), 8, false},
 		{float32(8.31), 8, false},
 		{float64(8.31), 8, false},
 		{true, 1, false},
 		{false, 0, false},
 		{"8", 8, false},
 		{nil, 0, false},
+		// errors
 		{int(-8), 0, true},
 		{int8(-8), 0, true},
 		{int16(-8), 0, true},
@@ -150,7 +157,7 @@ func TestToUint32E(t *testing.T) {
 		{float32(-8.31), 0, true},
 		{float64(-8.31), 0, true},
 		{"-8", 0, true},
-		// errors
+		{json.Number("-8"), 0, true},
 		{"test", 0, true},
 		{testing.T{}, 0, true},
 	}
@@ -189,6 +196,7 @@ func TestToUint16E(t *testing.T) {
 		{uint16(8), 8, false},
 		{uint32(8), 8, false},
 		{uint64(8), 8, false},
+		{json.Number("8"), 8, false},
 		{float32(8.31), 8, false},
 		{float64(8.31), 8, false},
 		{true, 1, false},
@@ -204,6 +212,7 @@ func TestToUint16E(t *testing.T) {
 		{float32(-8.31), 0, true},
 		{float64(-8.31), 0, true},
 		{"-8", 0, true},
+		{json.Number("-8"), 0, true},
 		{"test", 0, true},
 		{testing.T{}, 0, true},
 	}
@@ -242,6 +251,7 @@ func TestToUint8E(t *testing.T) {
 		{uint16(8), 8, false},
 		{uint32(8), 8, false},
 		{uint64(8), 8, false},
+		{json.Number("8"), 8, false},
 		{float32(8.31), 8, false},
 		{float64(8.31), 8, false},
 		{true, 1, false},
@@ -257,6 +267,7 @@ func TestToUint8E(t *testing.T) {
 		{float32(-8.31), 0, true},
 		{float64(-8.31), 0, true},
 		{"-8", 0, true},
+		{json.Number("-8"), 0, true},
 		{"test", 0, true},
 		{testing.T{}, 0, true},
 	}
@@ -300,9 +311,11 @@ func TestToIntE(t *testing.T) {
 		{true, 1, false},
 		{false, 0, false},
 		{"8", 8, false},
+		{json.Number("8"), 8, false},
 		{nil, 0, false},
 		// errors
 		{"test", 0, true},
+		{json.Number("test"), 0, true},
 		{testing.T{}, 0, true},
 	}
 
@@ -345,9 +358,11 @@ func TestToInt64E(t *testing.T) {
 		{true, 1, false},
 		{false, 0, false},
 		{"8", 8, false},
+		{json.Number("8"), 8, false},
 		{nil, 0, false},
 		// errors
 		{"test", 0, true},
+		{json.Number("test"), 0, true},
 		{testing.T{}, 0, true},
 	}
 
@@ -390,9 +405,11 @@ func TestToInt32E(t *testing.T) {
 		{true, 1, false},
 		{false, 0, false},
 		{"8", 8, false},
+		{json.Number("8"), 8, false},
 		{nil, 0, false},
 		// errors
 		{"test", 0, true},
+		{json.Number("test"), 0, true},
 		{testing.T{}, 0, true},
 	}
 
@@ -435,9 +452,11 @@ func TestToInt16E(t *testing.T) {
 		{true, 1, false},
 		{false, 0, false},
 		{"8", 8, false},
+		{json.Number("8"), 8, false},
 		{nil, 0, false},
 		// errors
 		{"test", 0, true},
+		{json.Number("test"), 0, true},
 		{testing.T{}, 0, true},
 	}
 
@@ -480,9 +499,11 @@ func TestToInt8E(t *testing.T) {
 		{true, 1, false},
 		{false, 0, false},
 		{"8", 8, false},
+		{json.Number("8"), 8, false},
 		{nil, 0, false},
 		// errors
 		{"test", 0, true},
+		{json.Number("test"), 0, true},
 		{testing.T{}, 0, true},
 	}
 
@@ -523,10 +544,12 @@ func TestToFloat64E(t *testing.T) {
 		{float32(8), 8, false},
 		{float64(8.31), 8.31, false},
 		{"8", 8, false},
+		{json.Number("8.31"), 8.31, false},
 		{true, 1, false},
 		{false, 0, false},
 		// errors
 		{"test", 0, true},
+		{json.Number("test"), 0, true},
 		{testing.T{}, 0, true},
 	}
 
@@ -567,10 +590,12 @@ func TestToFloat32E(t *testing.T) {
 		{float32(8.31), 8.31, false},
 		{float64(8.31), 8.31, false},
 		{"8", 8, false},
+		{json.Number("8.31"), 8.31, false},
 		{true, 1, false},
 		{false, 0, false},
 		// errors
 		{"test", 0, true},
+		{json.Number("test"), 0, true},
 		{testing.T{}, 0, true},
 	}
 

caste.go

@@ -10,6 +10,7 @@ import (
 	"errors"
 	"fmt"
 	"html/template"
+	"math"
 	"reflect"
 	"strconv"
 	"strings"
@@ -131,6 +132,12 @@ func ToFloat64E(i interface{}) (float64, error) {
 			return 1, nil
 		}
 		return 0, nil
+	case Float64er:
+		v, err := s.Float64()
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to float64", i, i)
+		}
+		return v, nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to float64", i, i)
 	}
@@ -176,6 +183,15 @@ func ToFloat32E(i interface{}) (float32, error) {
 			return 1, nil
 		}
 		return 0, nil
+	case Float64er:
+		v, err := s.Float64()
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to float32", i, i)
+		}
+		if v > math.MaxFloat32 {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to float32", i, i)
+		}
+		return float32(v), nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to float32", i, i)
 	}
@@ -223,6 +239,12 @@ func ToInt64E(i interface{}) (int64, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case Int64er:
+		v, err := s.Int64()
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to int64", i, i)
+		}
+		return v, nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to int64", i, i)
 	}
@@ -270,6 +292,15 @@ func ToInt32E(i interface{}) (int32, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case Int64er:
+		v, err := s.Int64()
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to int32", i, i)
+		}
+		if v > math.MaxInt32 {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to int32", i, i)
+		}
+		return int32(v), nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to int32", i, i)
 	}
@@ -317,6 +348,15 @@ func ToInt16E(i interface{}) (int16, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case Int64er:
+		v, err := s.Int64()
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to int16", i, i)
+		}
+		if v > math.MaxInt16 {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to int16", i, i)
+		}
+		return int16(v), nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to int16", i, i)
 	}
@@ -364,6 +404,15 @@ func ToInt8E(i interface{}) (int8, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case Int64er:
+		v, err := s.Int64()
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to int8", i, i)
+		}
+		if v > math.MaxInt8 {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to int8", i, i)
+		}
+		return int8(v), nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to int8", i, i)
 	}
@@ -411,6 +460,12 @@ func ToIntE(i interface{}) (int, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case Int64er:
+		v, err := s.Int64()
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to int", i, i)
+		}
+		return int(v), nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to int", i, i)
 	}
@@ -479,6 +534,12 @@ func ToUintE(i interface{}) (uint, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case fmt.Stringer:
+		v, err := strconv.ParseUint(s.String(), 10, 64)
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to uint", i, i)
+		}
+		return uint(v), nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to uint", i, i)
 	}
@@ -547,6 +608,12 @@ func ToUint64E(i interface{}) (uint64, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case fmt.Stringer:
+		v, err := strconv.ParseUint(s.String(), 10, 64)
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to uint64", i, i)
+		}
+		return v, nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to uint64", i, i)
 	}
@@ -615,6 +682,12 @@ func ToUint32E(i interface{}) (uint32, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case fmt.Stringer:
+		v, err := strconv.ParseUint(s.String(), 10, 32)
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to uint32", i, i)
+		}
+		return uint32(v), nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to uint32", i, i)
 	}
@@ -683,6 +756,12 @@ func ToUint16E(i interface{}) (uint16, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case fmt.Stringer:
+		v, err := strconv.ParseUint(s.String(), 10, 16)
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to uint16", i, i)
+		}
+		return uint16(v), nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to uint16", i, i)
 	}
@@ -751,6 +830,12 @@ func ToUint8E(i interface{}) (uint8, error) {
 		return 0, nil
 	case nil:
 		return 0, nil
+	case fmt.Stringer:
+		v, err := strconv.ParseUint(s.String(), 10, 8)
+		if err != nil {
+			return 0, fmt.Errorf("unable to cast %#v of type %T to uint8", i, i)
+		}
+		return uint8(v), nil
 	default:
 		return 0, fmt.Errorf("unable to cast %#v of type %T to uint8", i, i)
 	}

types.go

@@ -0,0 +1,11 @@
+package cast
+
+type (
+	Float64er interface {
+		Float64() (float64, error)
+	}
+
+	Int64er interface {
+		Int64() (int64, error)
+	}
+)