feat(tcpsocket): support tune_fun option

Author: keynslug

src/esockd_accept_inet.erl

@@ -94,6 +94,7 @@ tune_socket(Sock, [{tune_buffer, true}|More]) ->
             Error
     end;
 tune_socket(Sock, [{tune_fun, {M, F, A}} | More]) ->
+    %% NOTE: Socket is not part of the argument list, backward compatibility.
     case apply(M, F, A) of
         ok ->
             tune_socket(Sock, More);

src/esockd_accept_socket.erl

@@ -83,7 +83,9 @@ eval_tune_socket_fun(Sock, {Fun, Opts}) ->
 mk_tune_socket_fun(Opts) ->
     TcpOpts = proplists:get_value(tcp_options, Opts, []),
     SockOpts = lists:flatten([sock_opt(O) || O <- merge_sock_defaults(TcpOpts)]),
-    TuneOpts = [{Name, Val} || {Name, Val} <- Opts, Name =:= tune_buffer],
+    TuneOpts = [{Name, Val} || {Name, Val} <- Opts,
+                               Name =:= tune_buffer orelse
+                               Name =:= tune_fun],
     {fun ?MODULE:tune_socket/2, [{setopts, SockOpts} | TuneOpts]}.
 
 tune_socket(Sock, [{setopts, SockOpts} | Rest]) ->
@@ -93,16 +95,24 @@ tune_socket(Sock, [{setopts, SockOpts} | Rest]) ->
         Error ->
             Error
     end;
-tune_socket(Sock, [{tune_buffer, true} | _]) ->
+tune_socket(Sock, [{tune_buffer, true} | Rest]) ->
     try
         BufRecv = ensure(socket:getopt(Sock, {socket, rcvbuf})),
         Buffer = ensure(socket:getopt(Sock, {otp, rcvbuf})),
         Max = max(Buffer, BufRecv),
         ok = ensure(socket:setopt(Sock, {otp, rcvbuf}, Max)),
-        {ok, Sock}
+        tune_socket(Sock, Rest)
     catch
         Error -> Error
     end;
+tune_socket(Sock, [{tune_fun, {M, F, A}} | Rest]) ->
+    %% NOTE: Socket is not part of the argument list, backward compatibility.
+    case apply(M, F, A) of
+        ok ->
+            tune_socket(Sock, Rest);
+        Error ->
+            Error
+    end;
 tune_socket(Sock, _) ->
     {ok, Sock}.