[erlang] gen_tcp传输文件原型(二进制)

-module(tcp).
-export([client/3,server/1]).

client(Port,SiteID,Deploy) ->
    Host = "localhost",
    DeployBin = list_to_binary(Deploy),
    {ok,Sock} = gen_tcp:connect(Host,Port,[binary,{packet,4}]),
    ok = gen_tcp:send(Sock,<<SiteID:32,DeployBin/binary>>),
    receive_data(Sock,[],SiteID,Deploy).

receive_data(Socket,SoFar,SiteID,Deploy) ->
    receive
        {tcp,Socket,<<"error",Msg/binary>>} ->
            io:format("~p~p~p~n",[binary_to_list(Msg),SiteID,Deploy]);
        {tcp,Socket,<<"success",Data/binary>>} ->
            Target = "/tmp/"++ integer_to_list(SiteID) ++ "-" ++ Deploy ++ ".tar",
            case file:write_file(Target,Data) of
                ok             -> ok;
                {error,Reason} -> io:format("~p~n",[Reason])
            end;
        {tcp,Socket,_Bin} ->
            %%receive_data(Socket,[Bin|SoFar],SiteID,Deploy);
            io:format("test");
        {tcp_closed,Socket} ->
%            case lists:reverse(SoFar) of
%                <<"error",Msg/binary>>  ->
%                    {error,binary_to_list(Msg)};
%                <<"success",Data/binary>> ->
%                    Target = "/tmp/"++ SiteID ++ "-" ++ Deploy ++ ".tar",
%                    file:write_file(Target,list_to_binary(Data),
%                    {ok,Target};
%                _                   ->
%                    {error,unknown}
%            end
            io:format("~p~n",[SoFar])
    end.

server(Port) ->
    case gen_tcp:listen(Port,[binary,{packet,4}, %% 每个应用程序消息都从一个4字节长的头部开始
                                       {reuseaddr,true},  %% 
                                       {active,true}]) of
        {ok,Listen} ->
            spawn(fun()-> connect(Listen) end);
        {error,Why} ->
            io:format("~p~n",[Why])
    end.

connect(Listen) ->
    case gen_tcp:accept(Listen) of %% 等待暂停并等待一个连接
        {ok,Socket} ->
            spawn(fun() -> connect(Listen) end),
            loop(Socket);
        {error,Why} ->
            io:format("~p~n",[Why])
    end.

loop(Socket) ->
    receive
        %% 传入正确的二进制格式
        {tcp,Socket,<<SiteID:32,DeployBin/binary>>} ->
            Deploy = binary_to_list(DeployBin),
            Filename = "/tmp/scm/" ++ integer_to_list(SiteID) ++ "-" ++ Deploy ++ ".tar",
            T = case file:read_file(Filename) of
                {ok,Bin1} ->
                    <<"success",Bin1/binary>>;
                {error,Reason} ->
                    RBin = list_to_binary(atom_to_list(Reason)),
                    <<"error",RBin/binary>>
            end,
            gen_tcp:send(Socket,T),
            gen_tcp:close(Socket);
        {tcp,Socket,_Bin}   ->
            Msg = list_to_binary("illegal format"),
            gen_tcp:send(Socket,[<<"error",Msg/binary>>]),
            gen_tcp:close(Socket);
        {tcp_closed,Socket} ->
            io:format("Server socket closed~n")
    end.

http://www.cnblogs.com/bluefrog/archive/2012/09/10/2679040.html 改进版加入了一些二进制处理

改一改还能写个二进制包处理的游戏服务器, 就可以开搞游戏了... 慢慢来咯