1. chat_server.erl
-module(chat_server).
-export([start/1]).
-define(TCP_OPTIONS, [list, {packet, 0}, {active, false}, {reuseaddr, true}]).
start(Port) ->
Pid = spawn(fun() -> manage_clients([]) end),
register(client_manager, Pid),
{ok, LSocket} = gen_tcp:listen(Port, ?TCP_OPTIONS),
do_accept(LSocket).
do_accept(LSocket) ->
{ok, Socket} = gen_tcp:accept(LSocket),
spawn(fun() -> handle_client(Socket) end),
client_manager ! {connect, Socket},
do_accept(LSocket).
handle_client(Socket) ->
case gen_tcp:recv(Socket, 0) of
{ok, Data} ->
client_manager ! {data, Data},
handle_client(Socket);
{error, closed} ->
client_manager ! {disconnect, Socket}
end.
manage_clients(Sockets) ->
receive
{connect, Socket} ->
io:fwrite(“Socket connected: ~w~n”, [Socket]),
NewSockets = [Socket | Sockets];
{disconnect, Socket} ->
io:fwrite(“Socket disconnected: ~w~n”, [Socket]),
NewSockets = lists:delete(Socket, Sockets);
{data, Data} ->
send_data(Sockets, Data),
NewSockets = Sockets
end,
manage_clients(NewSockets).
send_data(Sockets, Data) ->
SendData = fun(Socket) ->
gen_tcp:send(Socket, Data)
end,
lists:foreach(SendData, Sockets).
server监听Socket连接,并群发消息到所有的client
2. chat_send_client.erl
-module(chat_send_client).
-export([start/2]).
start(IP, Port) ->
{ok, Socket} = gen_tcp:connect(IP, Port, [binary, {packet, 4}]),
talk(Socket).
talk(Socket) ->
Msg = io:get_line(‘Input you msg:’),
ok = gen_tcp:send(Socket, term_to_binary(Msg)),
talk(Socket).
send_client用来启动一个client发消息的进程
3.chat_recv_client.erl
-module(chat_recv_client).
-export([start/2]).
start(IP, Port) ->
{ok, Socket} = gen_tcp:connect(IP, Port, [binary, {packet, 4}]),
recv_msg(Socket).
recv_msg(Socket) ->
receive
{tcp, Socket, Bin} ->
Msg = binary_to_term(Bin),
io:format(“Received msg: ~p~n”, [Msg]),
recv_msg(Socket)
end.
recv_client用来在client端实时显示所有登录聊天室的client发送的消息
4,启动server
Eshell> chat_server:start(9999).
5,启动send和recv两个client进程
$$ client1
Eshell> chat_send_client:start(“localhost”, 9999).
Eshell> chat_recv_client:start(“localhost”, 9999).
$$ client2
Eshell> chat_send_client:start(“localhost”, 9999).
Eshell> chat_recv_client:start(“localhost”, 9999).
6,交互过程
%% client1/send
Input you msg:我囧
%% client2/send
Input you msg:你囧啥?
%% client1、2/recv
Received msg: “我囧\n”
Received msg: “你囧啥?\n”