github-wownero/wowlet
1
0
Fork 0
mirror of https://github.com/wownero/wowlet.git synced 2026-10-17 15:21:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

Websocket: fix unnecessary reconnect

Browse Source
This commit is contained in:
tobtoht
2021-10-14 15:41:14 +02:00
parent baf4fdc67d
commit 3ef10e9b53
3 changed files with 18 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/utils/WebsocketClient.cpp
+17 -34
View File
@@ -9,76 +9,59 @@
WebsocketClient::WebsocketClient(QObject *parent)
: QObject(parent)
{
connect(&webSocket, &QWebSocket::binaryMessageReceived, this, &WebsocketClient::onbinaryMessageReceived);
connect(&webSocket, &QWebSocket::connected, this, &WebsocketClient::onConnected);
connect(&webSocket, &QWebSocket::disconnected, this, &WebsocketClient::closed);
connect(&webSocket, &QWebSocket::disconnected, this, &WebsocketClient::onDisconnected);
connect(&webSocket, QOverload<QAbstractSocket::SocketError>::of(&QWebSocket::error), this, &WebsocketClient::onError);
connect(&m_connectionTimer, &QTimer::timeout, this, &WebsocketClient::checkConnection);
connect(&webSocket, &QWebSocket::binaryMessageReceived, this, &WebsocketClient::onbinaryMessageReceived);
// Keep websocket connection alive
connect(&m_pingTimer, &QTimer::timeout, [this]{
if (webSocket.state() == QAbstractSocket::ConnectedState)
if (webSocket.state() == QAbstractSocket::ConnectedState) {
webSocket.ping();
}
});
m_pingTimer.setInterval(30 * 1000);
m_pingTimer.start();
}
void WebsocketClient::sendMsg(const QByteArray &data) {
if (webSocket.state() == QAbstractSocket::ConnectedState)
if (webSocket.state() == QAbstractSocket::ConnectedState) {
webSocket.sendBinaryMessage(data);
}
void WebsocketClient::onToggleConnect(bool connect) {
m_connect = connect;
if (m_connect)
checkConnection();
}
}
void WebsocketClient::start() {
// connect & reconnect on errors/close
#ifdef QT_DEBUG
qDebug() << "WebSocket connect:" << m_url.url();
#endif
if (m_connect)
auto state = webSocket.state();
if (state != QAbstractSocket::ConnectedState && state != QAbstractSocket::ConnectingState) {
webSocket.open(m_url);
if (!m_connectionTimer.isActive()) {
m_connectionTimer.start(2000);
}
}
void WebsocketClient::checkConnection() {
if (!m_connect)
return;
if (webSocket.state() == QAbstractSocket::UnconnectedState) {
#ifdef QT_DEBUG
qDebug() << "WebSocket reconnect";
#endif
this->start();
}
}
void WebsocketClient::onConnected() {
#ifdef QT_DEBUG
qDebug() << "WebSocket connected";
#endif
emit connectionEstablished();
}
void WebsocketClient::onDisconnected() {
qDebug() << "WebSocket disconnected";
QTimer::singleShot(1000, [this]{this->start();});
}
void WebsocketClient::onError(QAbstractSocket::SocketError error) {
qCritical() << "WebSocket error: " << error;
auto state = webSocket.state();
if (state == QAbstractSocket::ConnectedState || state == QAbstractSocket::ConnectingState)
if (state == QAbstractSocket::ConnectedState || state == QAbstractSocket::ConnectingState) {
webSocket.abort();
}
}
void WebsocketClient::onbinaryMessageReceived(const QByteArray &message) {
#ifdef QT_DEBUG
qDebug() << "WebSocket received:" << message;
#endif
if (!Utils::validateJSON(message)) {
qCritical() << "Could not interpret WebSocket message as JSON";
return;