1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
//
// Copyright (c) 2016-2019 Vinnie Falco (vinnie dot falco at gmail dot com)
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
// Official repository: https://github.com/vinniefalco/CppCon2018
//
#include "shared_state.hpp"
#include "websocket_session.hpp"
shared_state::
shared_state(std::string doc_root)
: doc_root_(std::move(doc_root))
{
}
void
shared_state::
join(websocket_session* session)
{
std::lock_guard<std::mutex> lock(mutex_);
sessions_.insert(session);
}
void
shared_state::
leave(websocket_session* session)
{
std::lock_guard<std::mutex> lock(mutex_);
sessions_.erase(session);
}
// Broadcast a message to all websocket client sessions
void
shared_state::
send(std::string message)
{
// Put the message in a shared pointer so we can re-use it for each client
auto const ss = boost::make_shared<std::string const>(std::move(message));
// Make a local list of all the weak pointers representing
// the sessions, so we can do the actual sending without
// holding the mutex:
std::vector<boost::weak_ptr<websocket_session>> v;
{
std::lock_guard<std::mutex> lock(mutex_);
v.reserve(sessions_.size());
for(auto p : sessions_)
v.emplace_back(p->weak_from_this());
}
// For each session in our local list, try to acquire a strong
// pointer. If successful, then send the message on that session.
for(auto const& wp : v)
if(auto sp = wp.lock())
sp->send(ss);
}
|
