Pulling good patches off bad head of default

1 files changed, 9 insertions, 7 deletions

diff --git a/src/worker_pool_worker.erl b/src/worker_pool_worker.erl
index d3a48119..9ef8c636 100644
--- a/src/worker_pool_worker.erl
+++ b/src/worker_pool_worker.erl
@@ -46,6 +46,8 @@
 -spec(submit/2 :: (pid(), fun (() -> A) | {atom(), atom(), [any()]}) -> A).
 -spec(submit_async/2 ::
       (pid(), fun (() -> any()) | {atom(), atom(), [any()]}) -> 'ok').
+-spec(run/1 :: (fun (() -> A)) -> A;
+               ({atom(), atom(), [any()]}) -> any()).
 
 -endif.
 
@@ -65,6 +67,13 @@ submit(Pid, Fun) ->
 submit_async(Pid, Fun) ->
     gen_server2:cast(Pid, {submit_async, Fun}).
 
+run({M, F, A}) ->
+    apply(M, F, A);
+run(Fun) ->
+    Fun().
+
+%%----------------------------------------------------------------------------
+
 init([WId]) ->
     ok = worker_pool:idle(WId),
     put(worker_pool_worker, true),
@@ -95,10 +104,3 @@ code_change(_OldVsn, State, _Extra) ->
 
 terminate(_Reason, State) ->
     State.
-
-%%----------------------------------------------------------------------------
-
-run({M, F, A}) ->
-    apply(M, F, A);
-run(Fun) ->
-    Fun().