You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
142 lines
4.5 KiB
142 lines
4.5 KiB
-module(ntPptListener).
|
|
|
|
-include("eNet.hrl").
|
|
-include("ntCom.hrl").
|
|
|
|
-compile(inline).
|
|
-compile({inline_size, 128}).
|
|
|
|
-export([
|
|
start_link/4
|
|
, getOpts/1
|
|
, getListenPort/1
|
|
|
|
, init_it/3
|
|
, system_code_change/4
|
|
, system_continue/3
|
|
, system_get_state/1
|
|
, system_terminate/4
|
|
]).
|
|
|
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% genActor start %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|
-spec(start_link(atom(), atom(), inet:port_number(), [listenOpt()]) -> {ok, pid()} | ignore | {error, term()}).
|
|
start_link(ListenName, AptSupName, Port, ListenOpts) ->
|
|
proc_lib:start_link(?MODULE, init_it, [ListenName, self(), {AptSupName, Port, ListenOpts}], infinity, []).
|
|
|
|
init_it(Name, Parent, Args) ->
|
|
case safeRegister(Name) of
|
|
true ->
|
|
process_flag(trap_exit, true),
|
|
modInit(Parent, Args);
|
|
{false, Pid} ->
|
|
proc_lib:init_ack(Parent, {error, {already_started, Pid}})
|
|
end.
|
|
|
|
-spec system_code_change(term(), module(), undefined | term(), term()) -> {ok, term()}.
|
|
system_code_change(State, _Module, _OldVsn, _Extra) ->
|
|
{ok, State}.
|
|
|
|
-spec system_continue(pid(), [], {module(), atom(), pid(), term()}) -> ok.
|
|
system_continue(_Parent, _Debug, {Parent, State}) ->
|
|
loop(Parent, State).
|
|
|
|
-spec system_get_state(term()) -> {ok, term()}.
|
|
system_get_state(State) ->
|
|
{ok, State}.
|
|
|
|
-spec system_terminate(term(), pid(), [], term()) -> none().
|
|
system_terminate(Reason, _Parent, _Debug, _State) ->
|
|
exit(Reason).
|
|
|
|
safeRegister(Name) ->
|
|
try register(Name, self()) of
|
|
true -> true
|
|
catch
|
|
_:_ -> {false, whereis(Name)}
|
|
end.
|
|
|
|
modInit(Parent, Args) ->
|
|
case init(Args) of
|
|
{ok, State} ->
|
|
proc_lib:init_ack(Parent, {ok, self()}),
|
|
loop(Parent, State);
|
|
{stop, Reason} ->
|
|
proc_lib:init_ack(Parent, {error, Reason}),
|
|
exit(Reason)
|
|
end.
|
|
|
|
loop(Parent, State) ->
|
|
receive
|
|
{system, From, Request} ->
|
|
sys:handle_system_msg(Request, From, Parent, ?MODULE, [], {Parent, State});
|
|
{'EXIT', Parent, Reason} ->
|
|
terminate(Reason, State);
|
|
Msg ->
|
|
case handleMsg(Msg, State) of
|
|
{ok, NewState} ->
|
|
loop(Parent, NewState);
|
|
{stop, Reason} ->
|
|
terminate(Reason, State),
|
|
exit(Reason)
|
|
end
|
|
end.
|
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% genActor end %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|
|
|
-record(state, {
|
|
listenAddr :: inet:ip_address()
|
|
, listenPort :: inet:port_number()
|
|
, lSock :: inet:socket()
|
|
, opts :: [listenOpt()]
|
|
}).
|
|
|
|
-define(DefTcpOpts, [{nodelay, true}, {reuseaddr, true}, {send_timeout, 30000}, {send_timeout_close, true}]).
|
|
|
|
init({AptSupName, Port, ListenOpts}) ->
|
|
TcpOpts = ?getLValue(tcpOpts, ListenOpts, []),
|
|
LastTcpOpts = ntCom:mergeOpts(?DefTcpOpts, TcpOpts),
|
|
%% Don't active the socket...
|
|
case gen_tcp:listen(Port, lists:keystore(active, 1, LastTcpOpts, {active, false})) of
|
|
{ok, LSock} ->
|
|
AptCnt = ?getLValue(aptCnt, ListenOpts, ?AptCnt),
|
|
ConMod = ?getLValue(conMod, ListenOpts, undefined),
|
|
startAcceptor(AptCnt, LSock, AptSupName, ConMod),
|
|
{ok, {LAddr, LPort}} = inet:sockname(LSock),
|
|
% ?ntInfo("success to listen on ~p ~n", [Port]),
|
|
{ok, #state{listenAddr = LAddr, listenPort = LPort, lSock = LSock, opts = [{acceptors, AptCnt}, {tcpOpts, LastTcpOpts}]}};
|
|
{error, Reason} ->
|
|
?ntErr("failed to listen on ~p - ~p (~s) ~n", [Port, Reason, inet:format_error(Reason)]),
|
|
{stop, Reason}
|
|
end.
|
|
|
|
handleMsg({'$gen_call', From, miOpts}, #state{opts = Opts} = State) ->
|
|
gen_server:reply(From, Opts),
|
|
{ok, State};
|
|
|
|
handleMsg({'$gen_call', From, miListenPort}, #state{listenPort = LPort} = State) ->
|
|
gen_server:reply(From, LPort),
|
|
{ok, State};
|
|
|
|
handleMsg(_Msg, State) ->
|
|
?ntErr("~p unexpected info: ~p ~n", [?MODULE, _Msg]),
|
|
{noreply, State}.
|
|
|
|
terminate(_Reason, #state{lSock = LSock, listenAddr = Addr, listenPort = Port}) ->
|
|
?ntInfo("stopped on ~s:~p ~n", [inet:ntoa(Addr), Port]),
|
|
%% 关闭这个监听LSock 监听进程收到tcp_close 然后终止acctptor进程
|
|
catch port_close(LSock),
|
|
ok.
|
|
|
|
startAcceptor(0, _LSock, _AptSupName, _ConMod) ->
|
|
ok;
|
|
startAcceptor(N, LSock, AptSupName, ConMod) ->
|
|
supervisor:start_child(AptSupName, [LSock, ConMod, []]),
|
|
startAcceptor(N - 1, LSock, AptSupName, ConMod).
|
|
|
|
-spec getOpts(pid()) -> [listenOpt()].
|
|
getOpts(Listener) ->
|
|
gen_server:call(Listener, miOpts).
|
|
|
|
-spec getListenPort(pid()) -> inet:port_number().
|
|
getListenPort(Listener) ->
|
|
gen_server:call(Listener, miListenPort).
|
|
|