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
+        |> auth_user(user, metadata, config)
+        |> register_before_send(fn conn ->
+          :global.del_lock(id, nodes)
+
+          conn
+        end)
 
       false ->
         Plug.assign_current_user(conn, user, config)

lib/pow/plug/session.ex

@@ -246,11 +246,23 @@ 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)
-      false -> {conn, user}
+    case :global.set_lock(id, nodes, 0) do
+      true ->
+        {conn, user} = create(conn, user, config)
+
+        conn = register_before_send(conn, fn conn ->
+          :global.del_lock(id, nodes)
+
+          conn
+        end)
+
+        {conn, user}
+
+      false ->
+        {conn, user}
     end
   end