AI智能
改变未来

我的erlang TCP服务器

不解释,直接贴代码。

1) socket_options.hrl

-record(socket_options,        {port=8024,         loop,         name,         ip=any,         nodelay=false,         backlog=128,         listen=null,                         active_sockets=0}).

2)  test_server.erl

-module(test_server).-export([start/0, stop/0, looper/1]).-include(\"socket_options.hrl\").%开启服务start()->  test_socket_server:start(#socket_options{loop = {?MODULE, looper}}).%关闭服务stop()->  test_socket_server:stop(). looper(Socket) ->    inet:setopts(Socket, [{active, once}]),    receive    {tcp, Socket, Data} ->        io:format(\"Got packet: ~p~n\", [Data]),        gen_tcp:send(Socket, Data),        looper(Socket);    {tcp_closed, Socket}->        io:format(\"Socket ~p closed~n\", [Socket]);    {tcp_error, Socket, Reason} ->        io:format(\"Error on socket ~p reason: ~p~n\", [Socket, Reason])    end.

3) test_socket_server.erl

-module(test_socket_server).-export([start/0, start/1, stop/0, stop/1]).-export([init/1, handle_call/3, handle_cast/2, terminate/2, code_change/3,         handle_info/2]).-include(\"socket_options.hrl\").-define(RECBUF_SIZE, 8192).start() ->gen_server:start(?MODULE, [], []).start(InitArg = #socket_options{name = Name}) ->    case Name of        undefined ->    gen_server:start({local,?MODULE}, ?MODULE, InitArg, []);        _ ->            gen_server:start({local,Name}, ?MODULE, InitArg, [])end.stop() ->  gen_server:cast(?MODULE, stop).stop(Name) when is_atom(Name) ->    gen_server:cast(Name, stop);stop(Pid) when is_pid(Pid) ->    gen_server:cast(Pid, stop);stop(_State = #socket_options{name = Name}) ->    stop(Name).%回调函数%init(State = #socket_options{ip = Ip, port = Port, backlog = Backlog, nodelay = NoDelay}) ->init(State = #socket_options{port = Port, backlog = Backlog, nodelay = NoDelay}) ->    BaseOpts = [binary,        {reuseaddr, true},        {packet, 0},        {backlog, Backlog},        {recbuf, ?RECBUF_SIZE},        {active, false},        {nodelay, NoDelay}],    listen(Port, BaseOpts, State).    %%监听         handle_call(_Msg, _Caller, State) -> {noreply, State}.handle_info(_Msg, Library) -> {noreply, Library}.terminate(Reason, #socket_options{listen = Listen}) ->    io:format(\"socket close,~p~n\",[Reason]),    gen_tcp:close(Listen).             %%关闭Listencode_change(_OldVsn, State, _Extra) ->    State.handle_cast({accepted, _Pid}, State = #socket_options{active_sockets = ActiveSockets,             listen = Listen, loop = Loop}) ->    NewState = State#socket_options{active_sockets = 1 + ActiveSockets},    test_socket_server_acceptor:start_link(self(), Listen, Loop),    {noreply, NewState}; handle_cast(stop, Name) ->     %io:format(\"关闭服务器!~n\"),    {stop, normal , Name}. listen(Port, Opts, State) ->    case gen_tcp:listen(Port, Opts) of        {ok, Listen} ->test_socket_server_acceptor:start_link(self(), Listen, State#socket_options.loop),NewState = State#socket_options{listen = Listen},{ok, NewState};        {error, Reason} ->            {stop, Reason}    end.

 4) test_socket_server_acceptor.erl

-module(test_socket_server_acceptor).-export([start_link/3, init/3]).start_link(Server, Listen, Loop) ->    proc_lib:spawn_link(?MODULE, init, [Server, Listen, Loop]).init(Server, Listen, Loop) ->    case catch gen_tcp:accept(Listen) of        {ok, Socket} ->            gen_server:cast(Server, {accepted, self()}),            call_loop(Loop, Socket);        {error, closed} ->            exit(normal);        {error, timeout} ->            init(Server, Listen, Loop);        {error, esslaccept} ->            exit(normal);        _ ->            exit({error, accept_failed})    end.call_loop({M, F}, Socket) ->    io:format(\"socket call_loop,~n\"),    M:F(Socket);call_loop({M, F, [A1]}, Socket) ->    M:F(Socket, A1);call_loop({M, F, A}, Socket) ->    erlang:apply(M, F, [Socket | A]);call_loop(Loop, Socket) ->    Loop(Socket).

下一步是想在此基础上写一个聊天室什么的,不过我目前我想先实践完OTP实战这本书。

赞(0) 打赏
未经允许不得转载:爱站程序员基地 » 我的erlang TCP服务器