With some input from Jon I've managed to reduce this back into a macro rather than a fun and a macro calling a fun. He also suggested that on eagain I sleep a small amount of time so as to allow other work to catch up a bit.

This commit is contained in:
Gregory Burd 2013-07-31 15:39:55 -04:00
parent ee904b4769
commit f153509409

View file

@ -21,28 +21,26 @@
%%
%% -------------------------------------------------------------------
-spec async_nif_enqueue(function(), [term()]) -> term() | {error, term()}.
async_nif_enqueue(F, A) ->
R = erlang:make_ref(),
case erlang:apply(F, [R|A]) of
{ok, enqueued} ->
receive
{R, {error, shutdown}=Error} ->
%% Work unit was queued, but not executed.
Error;
{R, {error, _Reason}=Error} ->
%% Work unit returned an error.
Error;
{R, Reply} ->
Reply
end;
{error, eagain} ->
%% Work unit was not queued, try again.
async_nif_enqueue(F, A);
%{error, enomem} ->
%{error, shutdown} ->
Other ->
Other
end.
-define(ASYNC_NIF_CALL(Fun, Args), async_nif_enqueue(Fun, Args)).
-define(ASYNC_NIF_CALL(Fun, Args),
F = fun(F, A, R, T) ->
case erlang:apply(F, [R|A]) of
{ok, enqueued} ->
receive
{R, {error, shutdown}=Error} ->
%% Work unit was queued, but not executed.
Error;
{R, {error, _Reason}=Error} ->
%% Work unit returned an error.
Error;
{R, Reply} ->
Reply
end;
{error, eagain} ->
SleepyTime = min(30, (T+1)*2),
timer:sleep(SleepyTime),
F(F, A, R, SleepyTime);
Other ->
Other
end
end,
F(Fun, Args, erlang:make_ref(), 1)).