mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-21 01:01:43 +00:00
57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
// Copyright (c) FIRST and other WPILib contributors.
|
|
// Open Source Software; you can modify and/or share it under the terms of
|
|
// the WPILib BSD license file in the root directory of this project.
|
|
|
|
#ifndef WPIUTIL_WPI_HTTPWEBSOCKETSERVERCONNECTION_INC_
|
|
#define WPIUTIL_WPI_HTTPWEBSOCKETSERVERCONNECTION_INC_
|
|
|
|
#include <memory>
|
|
|
|
#include "wpi/HttpWebSocketServerConnection.h"
|
|
|
|
namespace wpi {
|
|
|
|
template <typename Derived>
|
|
HttpWebSocketServerConnection<Derived>::HttpWebSocketServerConnection(
|
|
std::shared_ptr<uv::Stream> stream, span<const std::string_view> protocols)
|
|
: HttpServerConnection{stream},
|
|
m_helper{m_request},
|
|
m_protocols{protocols.begin(), protocols.end()} {
|
|
// Handle upgrade event
|
|
m_helper.upgrade.connect([this] {
|
|
// Negotiate sub-protocol
|
|
SmallVector<std::string_view, 2> protocols{m_protocols.begin(),
|
|
m_protocols.end()};
|
|
std::string_view protocol = m_helper.MatchProtocol(protocols).second;
|
|
|
|
// Check that the upgrade is valid
|
|
if (!IsValidWsUpgrade(protocol)) {
|
|
return;
|
|
}
|
|
|
|
// Disconnect HttpServerConnection header reader
|
|
m_dataConn.disconnect();
|
|
m_messageCompleteConn.disconnect();
|
|
|
|
// Accepting the stream may destroy this (as it replaces the stream user
|
|
// data), so grab a shared pointer first.
|
|
auto self = this->shared_from_this();
|
|
|
|
// Accept the upgrade
|
|
auto ws = m_helper.Accept(m_stream, protocol);
|
|
|
|
// Set this as the websocket user data to keep it around
|
|
ws->SetData(self);
|
|
|
|
// Store in member
|
|
m_websocket = ws.get();
|
|
|
|
// Call derived class function
|
|
ProcessWsUpgrade();
|
|
});
|
|
}
|
|
|
|
} // namespace wpi
|
|
|
|
#endif // WPIUTIL_WPI_HTTPWEBSOCKETSERVERCONNECTION_INC_
|