aboutsummaryrefslogtreecommitdiffstats
path: root/src/gun_sse_h.erl
diff options
context:
space:
mode:
Diffstat (limited to 'src/gun_sse_h.erl')
-rw-r--r--src/gun_sse_h.erl18
1 files changed, 11 insertions, 7 deletions
diff --git a/src/gun_sse_h.erl b/src/gun_sse_h.erl
index 11dc443..9d8836e 100644
--- a/src/gun_sse_h.erl
+++ b/src/gun_sse_h.erl
@@ -38,18 +38,22 @@ init(ReplyTo, StreamRef, _, Headers, _) ->
disable
end.
--spec handle(_, binary(), State) -> {done, State} when State::#state{}.
-handle(IsFin, Data, State=#state{reply_to=ReplyTo, stream_ref=StreamRef, sse_state=SSE0}) ->
+-spec handle(_, binary(), State) -> {done, non_neg_integer(), State} when State::#state{}.
+handle(IsFin, Data, State) ->
+ handle(IsFin, Data, State, 0).
+
+handle(IsFin, Data, State=#state{reply_to=ReplyTo, stream_ref=StreamRef, sse_state=SSE0}, Flow) ->
case cow_sse:parse(Data, SSE0) of
{event, Event, SSE} ->
ReplyTo ! {gun_sse, self(), StreamRef, Event},
- handle(IsFin, <<>>, State#state{sse_state=SSE});
+ handle(IsFin, <<>>, State#state{sse_state=SSE}, Flow + 1);
{more, SSE} ->
- _ = case IsFin of
+ Inc = case IsFin of
fin ->
- ReplyTo ! {gun_sse, self(), StreamRef, fin};
+ ReplyTo ! {gun_sse, self(), StreamRef, fin},
+ 1;
_ ->
- ok
+ 0
end,
- {done, State#state{sse_state=SSE}}
+ {done, Flow + Inc, State#state{sse_state=SSE}}
end.