123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152 |
- -module(ranch_server).
- -behaviour(gen_server).
- -export([start_link/0]).
- -export([insert_listener/2]).
- -export([lookup_listener/1]).
- -export([add_acceptor/2]).
- -export([send_to_acceptors/2]).
- -export([add_connection/1]).
- -export([count_connections/1]).
- -export([remove_connection/1]).
- -export([init/1]).
- -export([handle_call/3]).
- -export([handle_cast/2]).
- -export([handle_info/2]).
- -export([terminate/2]).
- -export([code_change/3]).
- -define(TAB, ?MODULE).
- -type key() :: {listener | acceptors, any()}.
- -type monitors() :: [{{reference(), pid()}, key()}].
- -record(state, {
- monitors = [] :: monitors()
- }).
- -spec start_link() -> {ok, pid()}.
- start_link() ->
- gen_server:start_link({local, ?MODULE}, ?MODULE, [], []).
- -spec insert_listener(any(), pid()) -> ok.
- insert_listener(Ref, Pid) ->
- true = ets:insert_new(?TAB, {{listener, Ref}, Pid}),
- gen_server:cast(?MODULE, {insert_listener, Ref, Pid}).
- -spec lookup_listener(any()) -> pid().
- lookup_listener(Ref) ->
- ets:lookup_element(?TAB, {listener, Ref}, 2).
- -spec add_acceptor(any(), pid()) -> ok.
- add_acceptor(Ref, Pid) ->
- gen_server:cast(?MODULE, {add_acceptor, Ref, Pid}).
- -spec send_to_acceptors(any(), any()) -> ok.
- send_to_acceptors(Ref, Msg) ->
- Acceptors = ets:lookup_element(?TAB, {acceptors, Ref}, 2),
- _ = [Pid ! Msg || Pid <- Acceptors],
- ok.
- -spec add_connection(pid()) -> non_neg_integer().
- add_connection(ListenerPid) ->
- ets:update_counter(?TAB, {connections, ListenerPid}, 1).
- -spec count_connections(pid()) -> non_neg_integer().
- count_connections(ListenerPid) ->
- ets:update_counter(?TAB, {connections, ListenerPid}, 0).
- -spec remove_connection(pid()) -> non_neg_integer().
- remove_connection(ListenerPid) ->
- ets:update_counter(?TAB, {connections, ListenerPid}, -1).
- init([]) ->
- {ok, #state{}}.
- handle_call(_Request, _From, State) ->
- {reply, ignore, State}.
- handle_cast({insert_listener, Ref, Pid}, State=#state{monitors=Monitors}) ->
- true = ets:insert_new(?TAB, {{acceptors, Ref}, []}),
- true = ets:insert_new(?TAB, {{connections, Pid}, 0}),
- MonitorRef = erlang:monitor(process, Pid),
- {noreply, State#state{
- monitors=[{{MonitorRef, Pid}, {listener, Ref}}|Monitors]}};
- handle_cast({add_acceptor, Ref, Pid}, State=#state{monitors=Monitors}) ->
- MonitorRef = erlang:monitor(process, Pid),
- Acceptors = ets:lookup_element(?TAB, {acceptors, Ref}, 2),
- true = ets:insert(?TAB, {{acceptors, Ref}, [Pid|Acceptors]}),
- {noreply, State#state{
- monitors=[{{MonitorRef, Pid}, {acceptors, Ref}}|Monitors]}};
- handle_cast({add_connection, Pid}, State) ->
- _ = erlang:monitor(process, Pid),
- {noreply, State};
- handle_cast(_Request, State) ->
- {noreply, State}.
- handle_info({'DOWN', MonitorRef, process, Pid, _},
- State=#state{monitors=Monitors}) ->
- {_, Key} = lists:keyfind({MonitorRef, Pid}, 1, Monitors),
- Monitors2 = remove_process(Key, MonitorRef, Pid, Monitors),
- {noreply, State#state{monitors=Monitors2}};
- handle_info(_Info, State) ->
- {noreply, State}.
- terminate(_Reason, _State) ->
- ok.
- code_change(_OldVsn, State, _Extra) ->
- {ok, State}.
- -spec remove_process(key(), reference(), pid(), Monitors)
- -> Monitors when Monitors::monitors() .
- remove_process(Key = {listener, Ref}, MonitorRef, Pid, Monitors) ->
- true = ets:delete(?TAB, Key),
- true = ets:delete(?TAB, {acceptors, Ref}),
- true = ets:delete(?TAB, {connections, Pid}),
- lists:keydelete({MonitorRef, Pid}, 1, Monitors);
- remove_process(Key = {acceptors, _}, MonitorRef, Pid, Monitors) ->
- Acceptors = ets:lookup_element(?TAB, Key, 2),
- true = ets:insert(?TAB, {Key, lists:delete(Pid, Acceptors)}),
- lists:keydelete({MonitorRef, Pid}, 1, Monitors).
|