nicd/mebe-2
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix timer cookie middleware

Browse source
This commit is contained in:
Mikko Ahlroth 2020-04-11 11:00:56 +03:00
parent ac693e1c42
commit 0b2b354363
1 changed files with 3 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
lib/web/middleware/request_time.ex
View file

@ -9,14 +9,14 @@ defmodule Mebe2.Web.Middleware.RequestTime do
@impl true
def process_head(request, state, inner_server) do
put_response_timer()
{parts, inner_server} = Raxx.Server.handle_head(inner_server, request)
{parts, state, inner_server}
{[%Raxx.Response{} = resp], inner_server} = Raxx.Server.handle_head(inner_server, request)
resp = add_timer_cookie(resp)
{[resp], state, inner_server}
end
@impl true
def process_tail(tail, state, inner_server) do
{parts, inner_server} = Raxx.Server.handle_tail(inner_server, tail)
parts = modify_response_parts(parts, state)
{parts, state, inner_server}
end
@ -38,18 +38,6 @@ defmodule Mebe2.Web.Middleware.RequestTime do
:ok
end
defp modify_response_parts(parts, :disengage) do
parts
end
defp modify_response_parts(parts, :engage) do
Enum.flat_map(parts, &do_handle_response_part(&1))
end
defp do_handle_response_part(response = %Raxx.Response{}) do
[add_timer_cookie(response)]
end
@spec add_timer_cookie(Raxx.Response.t()) :: Raxx.Response.t()
defp add_timer_cookie(%Raxx.Response{} = response) do
old_time = Process.get(@timer_key)