-module(websocket_app).-behaviour(application).%% Application callbacks-export([start/2, stop/1]).%% ===================================================================%% Application callbacks%% ===================================================================start(_StartType, _StartArgs) ->    %%启动存储pid的树据 可以采用 ets 表格处理 但是为了方便集群处理 我采用的mnesia    ok = websocket_store:init(),    %% 配置路由    Dispatch = cowboy_router:compile([                      {‘_‘, [                         {"/", cowboy_static, {priv_file, websocket, "index.html"}},                         {"/websocket", websocket_hander, []},                         {"/static/[...]", cowboy_static, {priv_dir,  websocket, "static"}}                        ]}                     ]),    {ok, _} = cowboy:start_http(http, 100, [{port, 8080}],                [{env, [{dispatch, Dispatch}]}]),    %%启动监督树    websocket_sup:start_link().stop(_State) ->    ok.

-module(websocket_hander).-behaviour(cowboy_websocket_handler).%% ------------------------------------------------------------------%% API Function Exports%% ------------------------------------------------------------------%% ------------------------------------------------------------------%% gen_server Function Exports%% -------------------------------------------------------------------export([init/3]).-export([websocket_init/3]).-export([websocket_handle/3]).-export([websocket_info/3]).-export([websocket_terminate/3]).%% ------------------------------------------------------------------%% API Function Definitions%% ------------------------------------------------------------------%% ------------------------------------------------------------------%% gen_server Function Definitions%% ------------------------------------------------------------------%%websocket 握手init({tcp, http}, _Req, _Opts) ->    %%插入数据    websocket_store:dirty_insert(self(),web),    {upgrade, protocol, cowboy_websocket}.%%连接初始websocket_init(_TransportName, Req, _Opts) ->    %%连接一秒后 发送消息    erlang:start_timer(1000, self(), <<"Hello!">>),    {ok, Req, undefined_state}.%%处理客户端发送投递的消息websocket_handle({text, Msg}, Req, State) ->    %% 把消息投递到其余的进程 这里可以 投递到 tcp socket  也可以投递的 websocket 这样就可以实现 服务支持多协议处理     lists:foreach(fun([Pid,SocketType]) ->               case SocketType of                  tcp -> ok;                      web ->                  case Pid == self() of                      true -> ok;                      false -> Pid!{chat,Msg}                  end                                            end          end,websocket_store:dirty_lookall()),    {reply, {text, << "That‘s what she said! ", Msg/binary >>}, Req, State};websocket_handle(_Data, Req, State) ->    {ok, Req, State}.%% 处理erlang 发送的消息 这里是接收 46 行发送的数据 websocket_info({chat,Msg},_Req,State)->    {reply,{text,Msg},_Req,State};%% 处理     erlang:start_timer(1000, self(), <<"Hello!">>) 发送的消息websocket_info({timeout, _Ref, Msg}, Req, State) ->    {reply, {text, Msg}, Req, State};websocket_info(_Info, Req, State) ->    {ok, Req, State}.%%断开socketwebsocket_terminate(_Reason, _Req, _State) ->    websocket_store:dirty_delete(self()),    ok.%% ------------------------------------------------------------------%% Internal Function Definitions%% ------------------------------------------------------------------

%%%-------------------------------------------------------------------%%% @author thinkpad <>%%% @copyright (C) 2014, thinkpad%%% @doc%%%%%% @end%%% Created : 27 Jun 2014 by thinkpad <>%%%--------------------------------------------------------------------module(websocket_store).-include_lib("stdlib/include/qlc.hrl" ).%% API-export([dirty_insert/2,dirty_lookall/0,dirty_lookall_record/0,dirty_delete/1]).-export([init/0,insert/2,delete/1,lookup/1,lookall/0]).-record(socket_to_pid, {pid,socket_type}).-define(WAIT_FOR_TABLES, 10000).%%%===================================================================%%% API%%%===================================================================init()->    dynamic_db_init().    %--------------------------------------------------------------------%% @doc Insert a key and pid.%% @spec insert(Key, Pid) -> void()%% @end%%--------------------------------------------------------------------insert(Pid,SocketType) when is_pid(Pid) ->    Fun = fun() -> mnesia:write(#socket_to_pid{pid = Pid, socket_type = SocketType}) end,    {atomic, _} = mnesia:transaction(Fun).%--------------------------------------------------------------------%% @doc  dirty insert pid and socket%% @spec  dirty_insert(Pid socket)%% @end%%--------------------------------------------------------------------dirty_insert(Pid,SocketType) when is_pid(Pid) ->    mnesia:dirty_write(#socket_to_pid{pid = Pid, socket_type = SocketType}).%--------------------------------------------------------------------%% @doc dirty_read data%% @spec dirty_lookall()%% @end%%--------------------------------------------------------------------dirty_lookall()->    mnesia:dirty_select(socket_to_pid,[{#socket_to_pid{pid=‘$1‘,socket_type = ‘$2‘},[],[‘$$‘]}]).dirty_delete(Pid)->    mnesia:dirty_delete(socket_to_pid,Pid).%--------------------------------------------------------------------%% @doc look all record info%% @spec %% @end%%--------------------------------------------------------------------dirty_lookall_record()->         mnesia:dirty_select(socket_to_pid,[{#socket_to_pid{pid=‘$1‘,socket_type = ‘$2‘},[],[‘$_‘]}]).%%--------------------------------------------------------------------%% @doc Find a pid given a key.%% @spec lookup(Key) -> {ok, Pid} | {error, not_found}%% @end%%--------------------------------------------------------------------lookup(Pid) ->       do(qlc:q([{X#socket_to_pid.pid,X#socket_to_pid.socket_type} || X <- mnesia:table(socket_to_pid),X#socket_to_pid.pid==Pid])).%%--------------------------------------------------------------------%% @doc Find all list%% @spec lookall() -> {List} | {error, not_found}%% @end%%--------------------------------------------------------------------lookall() ->         do(qlc:q([[X#socket_to_pid.pid,X#socket_to_pid.socket_type] || X <- mnesia:table(socket_to_pid)])).%%--------------------------------------------------------------------%% @doc Delete an element by pid from the registrar.%% @spec delete(Pid) -> void()%% @end%%--------------------------------------------------------------------delete(Pid) ->    try        [#socket_to_pid{} = Record] = mnesia:dirty_read(socket_to_pid, Pid, #socket_to_pid.pid),        mnesia:dirty_delete_object(Record)    catch        _C:_E -> ok    end.   %%--------------------------------------------------------------------%% @doc%% @spec%% @end%%--------------------------------------------------------------------%%%===================================================================%%% Internal functions%%%===================================================================dynamic_db_init() ->    delete_schema(),    {atomic, ok} = mnesia:create_table(socket_to_pid, [{attributes, record_info(fields, socket_to_pid)}]),    ok.%% deletes a local schema.delete_schema() ->    mnesia:stop(),    mnesia:delete_schema([node()]),    mnesia:start().do(Query) ->    F = fun() -> qlc:e(Query) end,    {atomic, Value} = mnesia:transaction(F),    Value.