fix empty biz body unwrapBinary EOF error (#70)

Co-authored-by: xielei.xielei <[email protected]>

thrift/binary.go

@@ -204,10 +204,14 @@ func (p BinaryProtocol) UnwrapBody() (string, TMessageType, int32, FieldID, []by
 		return name, rTyp, seqID, 0, nil, err
 	}
 	// read the success struct
-	_, _, structID, err := p.ReadFieldBegin()
+	_, typeId, structID, err := p.ReadFieldBegin()
 	if err != nil {
 		return name, rTyp, seqID, structID, nil, err
 	}
+	//empty response
+	if typeId == STOP {
+		return name, rTyp, seqID, structID, []byte{}, nil
+	}
 	// there's alway a struct stop by success struct
 	if p.Read > len(p.Buf)-1 {
 		return name, rTyp, seqID, structID, nil, io.EOF

thrift/binary_test.go

@@ -171,3 +171,25 @@ func TestBinaryProtocol_WriteAny_ReadAny(t *testing.T) {
 		})
 	}
 }
+
+func TestBinaryUnwrap(t *testing.T) {
+	b := BinaryProtocol{
+		Buf:  nil,
+		Read: 0,
+	}
+	var i32 int32 = -2147418110
+	var methodName = "DummyNew"
+	b.WriteI32(i32)
+	b.WriteString(methodName)
+	b.WriteI32(1)
+	b.WriteByte(byte(STOP))
+
+	name, rType, _, _, bs, err := UnwrapBinaryMessage(b.Buf)
+
+	require.Equal(t, bs, []byte{})
+	require.Equal(t, name, methodName)
+	require.Equal(t, rType, REPLY)
+	var nilErr error
+	require.Equal(t, err, nilErr)
+
+}