带牛仔和websocket的erlang聊天室

Question

我正在尝试用牛仔websocket处理程序创建一个聊天室。我希望那些来自每一个将转发到其他套接字以及，就像一个聊天组。我不知道该如何实施？我不知道如何保存连接到websocket的套接字，以便我们可以向它们发送消息。我有一个牛仔操纵者：

-module(chat_conn).

-export([ init/2
    , websocket_init/1
    , websocket_handle/2
    , websocket_info/2
    , terminate/3
    ]).

-include("chat.hrl").

init(Req, [WsConnIdleTimeout]) ->
    ?LOG_DEBUG("[CHAT-CONN] New HTTP Request on: /api , Pid: ~p", [self()]),
    WsOpts = #{idle_timeout => WsConnIdleTimeout},
    State = #{counter => 0},
    {cowboy_websocket, Req, State, WsOpts}.

websocket_init(State0) ->
    ?LOG_DEBUG("[FOOZI-CONN] HTTP Upgraded to WebSocketm pid: ~p", [self()]),
    NewState = State0,
    {ok, NewState}.

websocket_handle({text, PlainRequest}, #{counter := Counter0} = State) ->
    ?LOG_DEBUG("[HIGGS-CONN] Receive New Message: ~p, Pid: ~p" , [PlainRequest, self()]),
    NewCounter = Counter0 + 1,
    Reply = list_to_binary("Counter is: " ++ integer_to_list(NewCounter)),
    NewState = #{counter => NewCounter},
    {reply, {text, Reply}, NewState};
    %{ok, State};
websocket_handle(Frame, State) ->
    ?LOG_INFO("[HIGGS-CONN] Invalid Frame => ~p", [Frame]),
    {stop, State}.

websocket_info(Message, State) ->
    ?LOG_INFO("[CONN-INFO] Unhandled message! => Message: ~p", [Message]),
    {reply, {text, list_to_binary(Message)}, State}.
    %{ok, State}.

terminate(Reason, _, State) ->
    ?LOG_INFO("[CONN-TERMINATE] Terminated! => Pid: ~p, Reason: ~p, State: ~p", [self(), Reason, State]),
    ok.

Answer 1

我不知道如何保存连接到websocket的套接字，以便向它们发送消息

在erlang中，可以使用gen_servers存储状态，因此您可以创建一个gen_server，websocket处理程序使用它来保存客户机PID。当客户端使用特定路由向您的牛仔服务器发送“开始聊天”消息时，将调用关联的websocket处理程序。在该处理程序中，self()将是该客户端的websocket进程的pid。您可以通过调用gen_server:cast(chat_room，{arrive，self()})，然后在chat_room:handle_cast()中将pid添加到存储在state变量中的pid列表中，将该pid保存在gen_server中。

当牛仔服务器从客户机浏览器接收到聊天消息时，客户机的websocket进程将处理该消息，并且在适当的websocket处理程序中，您可以查询gen_server以获得连接的客户机的PID列表。然后，您可以使用！向每个用户发送消息。反过来，每个客户端的websocket_info()处理程序将处理消息，它可以通过websocket将消息中继回客户端，返回:{reply,{text,text},State}。

对于gen_server，您还需要实现一个leave()函数，适当的websocket处理程序将调用该函数来更新存储在gen_server中的客户端列表，例如gen_server:cast(chat_room,{leave,self()})。