fix send client

src/client/send_client.erl

@@ -1,5 +1,6 @@
--module(send_client).
 
+-module(send_client).
+-include("../ses.hrl").
 -behaviour(websocket_client_handler).
 
 -export([
@@ -27,27 +28,45 @@ websocket_handle({pong, _}, _ConnState, State) ->
 
 websocket_handle({text, <<"GO">>}, _ConnState, State) ->
     %{reply, {text, Msg}, State}.
-    websocket_client:cast(self(), {text, <<"GO">>}),
-    {reply, {binary, <<"bbbbbbbb">>}, _ConnState, State};
+lager:notice("1111111111111111111"),
+Self = self(),
+    spawn(fun() -> loop(Self) end),
+lager:notice("22222222222222222"),
+    %{reply, {binary, <<"bbbbbbbb">>}, _ConnState, State};
+    {ok, State};
 
 
 websocket_handle({text, Msg}, _ConnState, State) ->
-    io:format("receive from server : ~p~n", [Msg]),
+    lager:notice("receive from server : ~p~n", [Msg]),
     %{reply, {text, Msg}, State}.
     {ok, State}.
 
 
+websocket_info({text, <<"GO">>}, _ConnState, State) ->
+lager:notice("3333333333333333"),
+lager:notice("self() :~p", [self()]),
+    websocket_client:cast(self(), {text, <<"GO">>}),
+    {reply, {binary, <<"bbbbbbbb">>}, _ConnState, State};
+
 
 websocket_info({text, Msg}, _ConnState, State) ->
+lager:notice("4444444444444"),
     {reply, {text, Msg}, _ConnState, State};
 
 websocket_info({timeout, _Ref, _Msg}, _ConnState, State) ->
     %erlang:start_timer(1000, self(), <<"alive2">>),
     %{reply, {text, <<"alive">>}, _ConnState, State};
     {ok, State};
 
+
 websocket_info(start, _ConnState, State) ->
     %{reply, {text, <<"from client, erlang message received">>}, State}.
+    {ok, State};
+
+
+websocket_info(Info, _ConnState, State) ->
+    %{reply, {text, <<"from client, erlang message received">>}, State}.
+lager:notice("555555555555555"),
     {ok, State}.
 
 websocket_terminate(Reason, _ConnState, State) ->
@@ -57,6 +76,13 @@ websocket_terminate(Reason, _ConnState, State) ->
 
 send() ->
     {ok, Pid} = ?MODULE:start_link("127.0.0.1"),
+lager:notice("pid:~p", [Pid]),
     websocket_client:cast(Pid , {text, <<"SEND#ronalfei">>}),
     ok.
 
+loop(Pid) ->
+    websocket_client:cast(Pid ,  {binary, <<"bbbbbbbb">>}),
+    receive
+    after 3000 ->
+        loop(Pid)
+    end.

src/receive_handler.erl

@@ -1,4 +1,5 @@
 -module(receive_handler).
+-include("ses.hrl").
 -behaviour(cowboy_websocket_handler).
 
 -export([init/3]).
@@ -34,13 +35,19 @@ websocket_handle(_Data, Req, State) ->
 
 
 
-websocket_info({timeout, _Ref, {binary, Bin}}, Req, State) ->
+websocket_info({binary, Bin}, Req, State) ->
 	{reply, {binary, Bin}, Req, State};
 
 
+websocket_info({text, Msg}, Req, State) ->
+	{reply, {text, Msg}, Req, State};
+
+
 websocket_info({timeout, _Ref, Msg}, Req, State) ->
 	{reply, {text, Msg}, Req, State};
 
+
+
 websocket_info(Info, Req, State) ->
     lager:warning("receive info : ~p", [Info]),
 	{ok, Req, State}.
@@ -56,7 +63,7 @@ init_ets(TableName) ->
     Options = [set, public, named_table, {read_concurrency, true} ],
     ets:new(Name, Options),
     Leader = group_leader(),
-    lager:debug(" group leader for this connnection is ~p", [Leader]),
+    lager:warning(" group leader for this connnection is ~p", [Leader]),
     ets:give_away(Name, Leader, []).
 
 

src/send_handler.erl

@@ -1,4 +1,5 @@
 -module(send_handler).
+-include("ses.hrl").
 -behaviour(cowboy_websocket_handler).
 
 -export([init/3]).
@@ -12,6 +13,7 @@ init({tcp, http}, _Req, _Opts) ->
 
 websocket_init(_TransportName, Req, _Opts) ->
 	erlang:start_timer(0, self(), <<"HI">>),
+lager:warning("self:~p", [self()]),
 	{ok, Req, {} }.
 
 
@@ -21,18 +23,21 @@ websocket_handle({text, <<"SEND#", ToUserid/binary>>}, Req, State) ->
         {ok, Pid} ->
             NewState = {<<"SEND#">>, Pid},
             insert_ets(pid_map, {NewState, self()}),%% 把发送流的进程插入到ets, 目前没有太大意义
+lager:warning("111111111: ~p", [NewState]),
 	        {reply, {text, <<"GO">>}, Req, NewState};
         {error, Reason} ->
             {reply, {text, Reason}, Req, State}
     end;
 
-websocket_handle({text, Msg}, Req, State) ->
-	%{reply, {text, << "receive", Msg/binary >>}, Req, State};
-	{ok, Req, State};
+websocket_handle({text, Msg}, Req, {<<"SEND#">>, ToPid}) ->
+lager:warning("msg to Pid:~p", [ToPid]),
+    ToPid ! {text, Msg},
+	{ok, Req, {<<"SEND#">>, ToPid}};
 
 
 websocket_handle({binary, Bin}, Req, {<<"SEND#">>, ToPid}) ->
-    erlang:start_timer(0, ToPid, {binary, Bin}),
+    ToPid ! {binary, Bin},
+lager:warning("binary to Pid:~p", [ToPid]),
 	{ok, Req, {<<"SEND#">>, ToPid}};
 
 websocket_handle({binary, Bin}, Req, State) ->
@@ -61,7 +66,7 @@ init_ets(TableName) ->
     Options = [set, public, named_table, {read_concurrency, true} ],
     ets:new(Name, Options),
     Leader = group_leader(),
-    %lager:debug(" group leader for this connnection is ~p", [Leader]),
+    lager:warning(" group leader for this connnection is ~p", [Leader]),
     ets:give_away(Name, Leader, []).
 
 

src/ses.erl

@@ -30,7 +30,7 @@ test() ->
     lager:critical("asdfasdf"),
     lager:alert("asdfasdf"),
     lager:emergency("asdfasdf"),
-    f(10000),
+    %f(10000),
 
     ok.