use end_json to end streaming when in streaming mode in addition

to `end_stream`

src/jsx.erl

@@ -107,14 +107,14 @@ is_term(Source) -> is_term(Source, []).
 is_term(Source, Config) -> jsx_verify:is_term(Source, Config).
 
 
--type decoder() :: fun((json_text() | end_stream) -> any()).
+-type decoder() :: fun((json_text() | end_stream | end_json) -> any()).
 
 -spec decoder(Handler::module(), State::any(), Config::list()) -> decoder().
 
 decoder(Handler, State, Config) -> jsx_decoder:decoder(Handler, State, Config).
 
 
--type encoder() :: fun((json_term() | end_stream) -> any()).
+-type encoder() :: fun((json_term() | end_stream | end_json) -> any()).
 
 -spec encoder(Handler::module(), State::any(), Config::list()) -> encoder().
 
@@ -464,5 +464,32 @@ encode_test_() ->
         } || {Title, _, Term, Events} <- Data
     ].
 
+end_stream_test_() ->
+    Tokens = [start_object, end_object, end_json],
+    [
+        {"encoder end_stream", ?_assertEqual(
+            Tokens,
+            begin
+                {incomplete, F} = (jsx:parser(jsx, [], [stream]))([start_object, end_object]),
+                F(end_stream)
+            end
+        )},
+        {"encoder end_json", ?_assertEqual(
+            Tokens,
+            begin
+                {incomplete, F} = (jsx:parser(jsx, [], [stream]))([start_object, end_object]),
+                F(end_json)
+            end
+        )},
+        {"decoder end_stream", ?_assertEqual(
+            Tokens,
+            begin {incomplete, F} = (jsx:decoder(jsx, [], [stream]))(<<"{}">>), F(end_stream) end
+        )},
+        {"decoder end_json", ?_assertEqual(
+            Tokens,
+            begin {incomplete, F} = (jsx:decoder(jsx, [], [stream]))(<<"{}">>), F(end_json) end
+        )}
+    ].
+
 
 -endif.

src/jsx_decoder.erl

@@ -150,7 +150,7 @@ incomplete(State, Rest, Handler, Acc, Stack, Config = #config{stream=false}) ->
 incomplete(State, Rest, Handler, Acc, Stack, Config = #config{incomplete_handler=false}) ->
     {incomplete, fun(Stream) when is_binary(Stream) ->
                 resume(<<Rest/binary, Stream/binary>>, State, Handler, Acc, Stack, Config);
-            (end_stream) ->
+            (End) when End == end_stream; End == end_json ->
                 case resume(<<Rest/binary, ?space/utf8>>, State, Handler, Acc, Stack, Config#config{stream=false}) of
                     {incomplete, _} -> ?error(State, Rest, Handler, Acc, Stack, Config);
                     Else -> Else

src/jsx_parser.erl

@@ -71,7 +71,7 @@ resume(Rest, State, Handler, Stack, Config) ->
 incomplete(State, Handler, Stack, Config=#config{stream=false}) ->
     ?error(State, [], Handler, Stack, Config);
 incomplete(State, Handler, Stack, Config=#config{incomplete_handler=false}) ->
-    {incomplete, fun(end_stream) ->
+    {incomplete, fun(End) when End == end_stream; End == end_json ->
                 case resume([end_json], State, Handler, Stack, Config) of
                     {incomplete, _} -> ?error(State, [], Handler, Stack, Config);
                     Else -> Else