return EOF early

Author: FZambia

decode_stream.go

@@ -60,9 +60,9 @@ func (d *JSONStreamCommandDecoder) Decode() (*Command, int, error) {
 	if err != nil {
 		if err == io.EOF && len(cmdBytes) > 0 {
 			var c Command
-			_, err = json.Parse(cmdBytes, &c, 0)
-			if err != nil {
-				return nil, 0, err
+			_, parseErr := json.Parse(cmdBytes, &c, 0)
+			if parseErr != nil {
+				return nil, 0, parseErr
 			}
 			return &c, len(cmdBytes), err
 		}

decode_stream_test.go

@@ -89,13 +89,17 @@ func testDecodingFrame(tb testing.TB, frame []byte, protoType Type) {
 		require.Equal(tb, 10037, size)
 	}
 	_, size, err = dec.Decode()
-	require.NoError(tb, err)
 	if protoType == TypeProtobuf {
 		require.Equal(tb, 10018, size)
 	} else {
 		require.Equal(tb, 10036, size)
 	}
-	_, _, err = dec.Decode()
-	require.ErrorIs(tb, err, io.EOF)
+	if err != nil {
+		require.ErrorIs(tb, err, io.EOF)
+	} else {
+		_, _, err = dec.Decode()
+		require.ErrorIs(tb, err, io.EOF)
+	}
+
 	PutStreamCommandDecoder(protoType, dec)
 }