Merge pull request #28 from ipfs/fix/json-parsing

Fix JSON Parsing

http/response.go

@@ -67,18 +67,15 @@ func (res *Response) RawNext() (interface{}, error) {
 		}
 	}
 
-	a := &cmds.Any{}
-	a.Add(&cmdkit.Error{})
-	a.Add(res.req.Command().Type)
-
-	err := res.dec.Decode(a)
+	m := &cmds.MaybeError{Value: res.req.Command().Type}
+	err := res.dec.Decode(m)
 
 	// last error was sent as value, now we get the same error from the headers. ignore and EOF!
 	if err != nil && res.err != nil && err.Error() == res.err.Error() {
 		err = io.EOF
 	}
 
-	return a.Interface(), err
+	return m.Get(), err
 }
 
 func (res *Response) Next() (interface{}, error) {

any_test.go → maybeerror_test.go

@@ -6,6 +6,8 @@ import (
 	"reflect"
 	"strings"
 	"testing"
+
+	"github.com/ipfs/go-ipfs-cmdkit"
 )
 
 type Foo struct {
@@ -22,29 +24,53 @@ type ValueError struct {
 }
 
 type anyTestCase struct {
-	Types   []interface{}
+	Value   interface{}
 	JSON    string
 	Decoded []ValueError
 }
 
-func TestMaybe(t *testing.T) {
+func TestMaybeError(t *testing.T) {
 	testcases := []anyTestCase{
 		anyTestCase{
-			Types: []interface{}{Foo{}, &Bar{}},
-			JSON:  `{"Bar":2}{"Foo":"abc"}`,
+			Value: &Foo{},
+			JSON:  `{"Bar":23}{"Bar":42}{"Message":"some error", "Type": "error"}`,
+			Decoded: []ValueError{
+				ValueError{Error: nil, Value: &Foo{23}},
+				ValueError{Error: nil, Value: &Foo{42}},
+				ValueError{Error: nil, Value: cmdkit.Error{Message: "some error", Code: 0}},
+			},
+		},
+		anyTestCase{
+			Value: Foo{},
+			JSON:  `{"Bar":23}{"Bar":42}{"Message":"some error", "Type": "error"}`,
 			Decoded: []ValueError{
-				ValueError{Error: nil, Value: &Foo{2}},
-				ValueError{Error: nil, Value: &Bar{"abc"}},
+				ValueError{Error: nil, Value: &Foo{23}},
+				ValueError{Error: nil, Value: &Foo{42}},
+				ValueError{Error: nil, Value: cmdkit.Error{Message: "some error", Code: 0}},
+			},
+		},
+		anyTestCase{
+			Value: &Bar{},
+			JSON:  `{"Foo":""}{"Foo":"Qmabc"}{"Message":"some error", "Type": "error"}`,
+			Decoded: []ValueError{
+				ValueError{Error: nil, Value: &Bar{""}},
+				ValueError{Error: nil, Value: &Bar{"Qmabc"}},
+				ValueError{Error: nil, Value: cmdkit.Error{Message: "some error", Code: 0}},
+			},
+		},
+		anyTestCase{
+			Value: Bar{},
+			JSON:  `{"Foo":""}{"Foo":"Qmabc"}{"Message":"some error", "Type": "error"}`,
+			Decoded: []ValueError{
+				ValueError{Error: nil, Value: &Bar{""}},
+				ValueError{Error: nil, Value: &Bar{"Qmabc"}},
+				ValueError{Error: nil, Value: cmdkit.Error{Message: "some error", Code: 0}},
 			},
 		},
 	}
 
 	for _, tc := range testcases {
-		a := &Any{}
-
-		for _, t := range tc.Types {
-			a.Add(t)
-		}
+		m := &MaybeError{Value: tc.Value}
 
 		r := strings.NewReader(tc.JSON)
 		d := json.NewDecoder(r)
@@ -52,19 +78,19 @@ func TestMaybe(t *testing.T) {
 		var err error
 
 		for _, dec := range tc.Decoded {
-			err = d.Decode(a)
+			err = d.Decode(m)
 			if err != dec.Error {
 				t.Fatalf("error is %v, expected %v", err, dec.Error)
 			}
 
-			rx := a.Interface()
+			rx := m.Get()
 			rxIsPtr := reflect.TypeOf(rx).Kind() == reflect.Ptr
 
 			ex := dec.Value
 			exIsPtr := reflect.TypeOf(ex).Kind() == reflect.Ptr
 
 			if rxIsPtr != exIsPtr {
-				t.Fatalf("value is %#v, expected %#v", a.Interface(), dec.Value)
+				t.Fatalf("value is %#v, expected %#v", m.Get(), dec.Value)
 			}
 
 			if rxIsPtr {
@@ -73,13 +99,13 @@ func TestMaybe(t *testing.T) {
 			}
 
 			if rx != ex {
-				t.Fatalf("value is %#v, expected %#v", a.Interface(), dec.Value)
+				t.Fatalf("value is %#v, expected %#v", m.Get(), dec.Value)
 			}
 		}
 
-		err = d.Decode(a)
+		err = d.Decode(m)
 		if err != io.EOF {
-			t.Fatal("data left in decoder:", a.Interface())
+			t.Fatal("data left in decoder:", m.Get())
 		}
 	}
 }

writer.go

@@ -67,34 +67,24 @@ func (r *readerResponse) Length() uint64 {
 }
 
 func (r *readerResponse) RawNext() (interface{}, error) {
-	a := &Any{}
-	a.Add(cmdkit.Error{})
-	a.Add(r.req.Command().Type)
-
-	err := r.dec.Decode(a)
+	m := &MaybeError{Value: r.req.Command().Type}
+	err := r.dec.Decode(m)
 	if err != nil {
 		return nil, err
 	}
 
 	r.once.Do(func() { close(r.emitted) })
 
-	v := a.Interface()
+	v := m.Get()
 	return v, nil
 }
 
 func (r *readerResponse) Next() (interface{}, error) {
-	a := &Any{}
-	a.Add(cmdkit.Error{})
-	a.Add(r.req.Command().Type)
-
-	err := r.dec.Decode(a)
+	v, err := r.RawNext()
 	if err != nil {
 		return nil, err
 	}
 
-	r.once.Do(func() { close(r.emitted) })
-
-	v := a.Interface()
 	if err, ok := v.(cmdkit.Error); ok {
 		v = &err
 	}
@@ -177,81 +167,32 @@ func (re *WriterResponseEmitter) Emit(v interface{}) error {
 	return re.enc.Encode(v)
 }
 
-type Any struct {
-	types map[reflect.Type]bool
-	order []reflect.Type
+type MaybeError struct {
+	Value interface{} // needs to be a pointer
+	Error cmdkit.Error
 
-	v interface{}
+	isError bool
 }
 
-func (a *Any) UnmarshalJSON(data []byte) error {
-	var (
-		iv  interface{}
-		err error
-	)
-
-	for _, t := range a.order {
-		v := reflect.New(t).Elem().Addr()
-
-		isNil := func(v reflect.Value) (yup, ok bool) {
-			ok = true
-			defer func() {
-				r := recover()
-				if r != nil {
-					ok = false
-				}
-			}()
-			yup = v.IsNil()
-			return
-		}
-
-		isZero := func(v reflect.Value, t reflect.Type) (yup, ok bool) {
-			ok = true
-			defer func() {
-				r := recover()
-				if r != nil {
-					ok = false
-				}
-			}()
-			yup = v.Elem().Interface() == reflect.Zero(t).Interface()
-			return
-		}
-
-		err = json.Unmarshal(data, v.Interface())
-
-		vIsNil, isNilOk := isNil(v)
-		vIsZero, isZeroOk := isZero(v, t)
-
-		nilish := (isNilOk && vIsNil) || (isZeroOk && vIsZero)
-		if err == nil && !nilish {
-			a.v = v.Interface()
-			return nil
-		}
+func (m *MaybeError) Get() interface{} {
+	if m.isError {
+		return m.Error
 	}
-
-	err = json.Unmarshal(data, &iv)
-	a.v = iv
-
-	return err
+	return m.Value
 }
 
-func (a *Any) Add(v interface{}) {
-	if v == nil {
-		return
-	}
-	if a.types == nil {
-		a.types = map[reflect.Type]bool{}
-	}
-	t := reflect.TypeOf(v)
-	isPtr := t.Kind() == reflect.Ptr
-	if isPtr || t.Kind() == reflect.Interface {
-		t = t.Elem()
+func (m *MaybeError) UnmarshalJSON(data []byte) error {
+	err := json.Unmarshal(data, &m.Error)
+	if err == nil {
+		m.isError = true
+		return nil
 	}
 
-	a.types[t] = isPtr
-	a.order = append(a.order, t)
-}
+	// make sure we are working with a pointer here
+	v := reflect.ValueOf(m.Value)
+	if v.Kind() != reflect.Ptr {
+		m.Value = reflect.New(v.Type()).Interface()
+	}
 
-func (a *Any) Interface() interface{} {
-	return a.v
+	return json.Unmarshal(data, m.Value)
 }