Delete locks in before send callback

Author: danschultzer

lib/extensions/persistent_session/plug/cookie.ex

@@ -229,11 +229,18 @@ defmodule PowPersistentSession.Plug.Cookie do
   defp filter_invalid!(clauses), do: raise "Invalid get_by clauses stored: #{inspect clauses}"
 
   defp lock_auth_user(conn, token, user, metadata, config) do
+    id    = {[__MODULE__, token], self()}
     nodes = Node.list() ++ [node()]
 
-    case :global.set_lock({[__MODULE__, token], self()}, nodes, 0) do
+    case :global.set_lock(id, nodes, 0) do
       true ->
-        auth_user(conn, user, metadata, config)
+        conn
+        |> register_before_send(fn conn ->
+          :global.del_lock(id, nodes)
+
+          conn
+        end)
+        |> auth_user(user, metadata, config)
 
       false ->
         Plug.assign_current_user(conn, user, config)

lib/pow/plug/session.ex

@@ -246,10 +246,19 @@ defmodule Pow.Plug.Session do
   end
 
   defp lock_create(conn, session_id, user, config) do
+    id    = {[__MODULE__, session_id], self()}
     nodes = Node.list() ++ [node()]
 
-    case :global.set_lock({[__MODULE__, session_id], self()}, nodes, 0) do
-      true  -> create(conn, user, config)
+    case :global.set_lock(id, nodes, 0) do
+      true ->
+        conn
+        |> register_before_send(fn conn ->
+          :global.del_lock(id, nodes)
+
+          conn
+        end)
+        |> create(user, config)
+
       false -> {conn, user}
     end
   end