Make it stick together just barely

This commit is contained in:
Alex Hultman
2020-06-02 12:40:51 +02:00
parent a8539e7463
commit cb48712b3d
7 changed files with 134 additions and 45 deletions
+57 -4
View File
@@ -25,11 +25,64 @@ int main() {
.idleTimeout = 10,
.maxBackpressure = 1 * 1024 * 1204,
/* Handlers */
.upgrade = [](auto *res, auto *req) {
std::cout << "Upgrade now!" << std::endl;
/* Pass user data from upgrade to open handler here */
//res.upgrade({.something = 15}, req);
//.syncUpgrade()
.upgrade = [](auto *res, auto *req, auto *context) {
/* Immediate path */
res->template upgrade<PerSocketData>(req->getHeader("sec-websocket-key"),
req->getHeader("sec-websocket-protocol"),
req->getHeader("sec-websocket-extensions"),
context);
return;
std::cout << "Upgrade request!" << std::endl;
/* Async path, you have to COPY headers */
std::string secWebSocketKey(req->getHeader("sec-websocket-key"));
std::string secWebSocketProtocol(req->getHeader("sec-websocket-protocol"));
std::string secWebSocketExtensions(req->getHeader("sec-websocket-extensions"));
/* If the client disconnects inbetween, you MUST avoid upgrading */
bool *aborted = new bool(false);
res->onAborted([=]() {
std::cout << "WebSocket upgrade aborted!" << std::endl;
*aborted = true;
});
/* Simulate checking auth for 10 seconds */
struct us_loop_t *loop = (struct us_loop_t *) uWS::Loop::get();
struct us_timer_t *delayTimer = us_create_timer(loop, 0, 0);
us_timer_set(delayTimer, [](struct us_timer_t *t) {
std::cout << "Timer triggered!" << std::endl;
us_timer_close(t);
}, 5000, 0);
/* Simulate doing async work by deferring upgrade to next event loop iteration */
uWS::Loop::get()->defer([=](){
std::cout << "Upgrading now!" << std::endl;
if (!*aborted) {
// this should get some kind of ticket
res->template upgrade<PerSocketData>(secWebSocketKey,
secWebSocketProtocol,
secWebSocketExtensions,
context);
}
delete aborted;
});
/* Immediate path is in all seriousness a simple return bool */
// not exactly, you need to pass UserData - return a moved UserData and bool?
},
.open = [](auto *ws, auto *req) {
/* Open event here, you may access ws->getUserData() which points to a PerSocketData struct */