Broadcast messages through websocket.

[komnotmr]

Is there examples of sending broadcast messages through websockets?

[ithewei]

Record all channels, traverse to send.

[ithewei]

#include "WebSocketServer.h" using namespace hv; class ChannelMgr { public: void addChannel(const WebSocketChannelPtr& channel) { std::lock_guard<std::mutex> locker(mutex_); channels[channel->id()] = channel; } void removeChannel(const WebSocketChannelPtr& channel) { std::lock_guard<std::mutex> locker(mutex_); channels.erase(channel->id()); } int foreachChannel(std::function<void(const WebSocketChannelPtr& channel)> fn) { std::lock_guard<std::mutex> locker(mutex_); for (auto& pair : channels) { fn(pair.second); } return channels.size(); } // broadcast thread-safe int broadcast(const void* data, int size) { return foreachChannel([data, size](const WebSocketChannelPtr& channel) { channel->send((const char*)data, size); }); } int broadcast(const std::string& str) { return broadcast(str.data(), str.size()); } private: // id => SocketChannelPtr std::map<int, WebSocketChannelPtr> channels; std::mutex mutex_; }; int main(int argc, char** argv) { if (argc < 2) { printf("Usage: %s port\n", argv[0]); return -10; } int port = atoi(argv[1]); HttpService http; http.GET("/ping", [](const HttpContextPtr& ctx) { return ctx->send("pong"); }); WebSocketService ws; ChannelMgr mgr; ws.onopen = [&mgr](const WebSocketChannelPtr& channel, const HttpRequestPtr& req) { printf("onopen: GET %s\n", req->Path().c_str()); mgr.addChannel(channel); }; ws.onmessage = [&mgr](const WebSocketChannelPtr& channel, const std::string& msg) { printf("onmessage: %d\n", (int)msg.size()); mgr.broadcast(msg); }; ws.onclose = [&mgr](const WebSocketChannelPtr& channel) { printf("onclose\n"); mgr.removeChannel(channel); }; websocket_server_t server; server.port = port; server.service = &http; server.ws = &ws; websocket_server_run(&server); return 0; }

[komnotmr]

Thank you!